mirror of
https://github.com/lnbits/lnbits-legend.git
synced 2025-02-24 22:58:46 +01:00
128 lines
4 KiB
Python
128 lines
4 KiB
Python
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
|