lnbits-legend/lnbits/extensions/gerty/crud.py

from typing import List, Optional, Union

import httpx
from lnbits.helpers import urlsafe_short_hash
import time
import json

from loguru import logger

from . import db
from .models import Gerty, Mempool, MempoolEndpoint


async def create_gerty(wallet_id: str, data: Gerty) -> Gerty:
    gerty_id = urlsafe_short_hash()
    await db.execute(
        """
        INSERT INTO gerty.gertys (
        id,
        name,
        utc_offset,
        type,
        wallet,
        lnbits_wallets,
        mempool_endpoint,
        exchange,
        display_preferences,
        refresh_time,
        urls
        )
        VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
        """,
        (
            gerty_id,
            data.name,
            data.utc_offset,
            data.type,
            wallet_id,
            data.lnbits_wallets,
            data.mempool_endpoint,
            data.exchange,
            data.display_preferences,
            data.refresh_time,
            data.urls,
        ),
    )

    gerty = await get_gerty(gerty_id)
    assert gerty, "Newly created gerty couldn't be retrieved"
    return gerty


async def update_gerty(gerty_id: str, **kwargs) -> Gerty:
    q = ", ".join([f"{field[0]} = ?" for field in kwargs.items()])
    await db.execute(
        f"UPDATE gerty.gertys SET {q} WHERE id = ?", (*kwargs.values(), gerty_id)
    )
    return await get_gerty(gerty_id)


async def get_gerty(gerty_id: str) -> Optional[Gerty]:
    row = await db.fetchone("SELECT * FROM gerty.gertys WHERE id = ?", (gerty_id,))
    return Gerty(**row) if row else None


async def get_gertys(wallet_ids: Union[str, List[str]]) -> List[Gerty]:
    if isinstance(wallet_ids, str):
        wallet_ids = [wallet_ids]

    q = ",".join(["?"] * len(wallet_ids))
    rows = await db.fetchall(
        f"SELECT * FROM gerty.gertys WHERE wallet IN ({q})", (*wallet_ids,)
    )

    return [Gerty(**row) for row in rows]


async def delete_gerty(gerty_id: str) -> None:
    await db.execute("DELETE FROM gerty.gertys WHERE id = ?", (gerty_id,))


#############MEMPOOL###########


async def get_mempool_info(endPoint: str, gerty) -> Optional[Mempool]:
    logger.debug(endPoint)
    endpoints = MempoolEndpoint()
    url = ""
    for endpoint in endpoints:
        if endPoint == endpoint[0]:
            url = endpoint[1]
    row = await db.fetchone(
        "SELECT * FROM gerty.mempool WHERE endpoint = ? AND mempool_endpoint = ?",
        (
            endPoint,
            gerty.mempool_endpoint,
        ),
    )
    if not row:
        async with httpx.AsyncClient() as client:
            response = await client.get(gerty.mempool_endpoint + url)
            logger.debug(gerty.mempool_endpoint + url)
            mempool_id = urlsafe_short_hash()
            await db.execute(
                """
                INSERT INTO gerty.mempool (
                    id,
                    data,
                    endpoint,
                    time,
                    mempool_endpoint
                )
                VALUES (?, ?, ?, ?, ?)
                """,
                (
                    mempool_id,
                    json.dumps(response.json()),
                    endPoint,
                    int(time.time()),
                    gerty.mempool_endpoint,
                ),
            )
            return response.json()
    if int(time.time()) - row.time > 20:
        async with httpx.AsyncClient() as client:
            response = await client.get(gerty.mempool_endpoint + url)
            await db.execute(
                "UPDATE gerty.mempool SET data = ?, time = ? WHERE endpoint = ? AND mempool_endpoint = ?",
                (
                    json.dumps(response.json()),
                    int(time.time()),
                    endPoint,
                    gerty.mempool_endpoint,
                ),
            )
            return response.json()
    return json.loads(row.data)