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

from typing import List, Optional, Union

# from lnbits.db import open_ext_db
from . import db
from .models import Charges

from lnbits.helpers import urlsafe_short_hash

from quart import jsonify
import httpx
from lnbits.core.services import create_invoice, check_invoice_status
from ..watchonly.crud import get_watch_wallet, get_fresh_address, get_mempool


###############CHARGES##########################


async def create_charge(
    user: str,
    description: str = None,
    onchainwallet: Optional[str] = None,
    lnbitswallet: Optional[str] = None,
    webhook: Optional[str] = None,
    completelink: Optional[str] = None,
    completelinktext: Optional[str] = "Back to Merchant",
    time: Optional[int] = None,
    amount: Optional[int] = None,
) -> Charges:
    charge_id = urlsafe_short_hash()
    if onchainwallet:
        wallet = await get_watch_wallet(onchainwallet)
        onchain = await get_fresh_address(onchainwallet)
        onchainaddress = onchain.address
    else:
        onchainaddress = None
    if lnbitswallet:
        payment_hash, payment_request = await create_invoice(
            wallet_id=lnbitswallet, amount=amount, memo=charge_id
        )
    else:
        payment_hash = None
        payment_request = None
    await db.execute(
        """
        INSERT INTO charges (
            id,
            user,
            description,
            onchainwallet,
            onchainaddress,
            lnbitswallet,
            payment_request,
            payment_hash,
            webhook,
            completelink,
            completelinktext,
            time,
            amount,
            balance
        )
        VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
        """,
        (
            charge_id,
            user,
            description,
            onchainwallet,
            onchainaddress,
            lnbitswallet,
            payment_request,
            payment_hash,
            webhook,
            completelink,
            completelinktext,
            time,
            amount,
            0,
        ),
    )
    return await get_charge(charge_id)


async def update_charge(charge_id: str, **kwargs) -> Optional[Charges]:
    q = ", ".join([f"{field[0]} = ?" for field in kwargs.items()])
    await db.execute(
        f"UPDATE charges SET {q} WHERE id = ?", (*kwargs.values(), charge_id)
    )
    row = await db.fetchone("SELECT * FROM charges WHERE id = ?", (charge_id,))
    return Charges.from_row(row) if row else None


async def get_charge(charge_id: str) -> Charges:
    row = await db.fetchone("SELECT * FROM charges WHERE id = ?", (charge_id,))
    return Charges.from_row(row) if row else None


async def get_charges(user: str) -> List[Charges]:
    rows = await db.fetchall("SELECT * FROM charges WHERE user = ?", (user,))
    return [Charges.from_row(row) for row in rows]


async def delete_charge(charge_id: str) -> None:
    await db.execute("DELETE FROM charges WHERE id = ?", (charge_id,))


async def check_address_balance(charge_id: str) -> List[Charges]:
    charge = await get_charge(charge_id)
    if not charge.paid:
        if charge.onchainaddress:
            mempool = await get_mempool(charge.user)
            try:
                async with httpx.AsyncClient() as client:
                    r = await client.get(
                        mempool.endpoint + "/api/address/" + charge.onchainaddress
                    )
                    respAmount = r.json()["chain_stats"]["funded_txo_sum"]
                    if respAmount >= charge.balance:
                        await update_charge(charge_id=charge_id, balance=respAmount)
            except Exception:
                pass
        if charge.lnbitswallet:
            invoice_status = await check_invoice_status(
                charge.lnbitswallet, charge.payment_hash
            )
            if invoice_status.paid:
                return await update_charge(charge_id=charge_id, balance=charge.amount)
    row = await db.fetchone("SELECT * FROM charges WHERE id = ?", (charge_id,))
    return Charges.from_row(row) if row else None