mirror of
https://github.com/lnbits/lnbits-legend.git
synced 2025-03-12 19:03:11 +01:00
87 lines
3.2 KiB
Python
87 lines
3.2 KiB
Python
import asyncio
|
|
from http import HTTPStatus
|
|
|
|
import httpx
|
|
from lnurl import decode as lnurl_decode
|
|
from loguru import logger
|
|
from starlette.exceptions import HTTPException
|
|
|
|
from lnbits.core.crud import get_wallet
|
|
from lnbits.core.models import Payment
|
|
from lnbits.core.services import pay_invoice
|
|
from lnbits.helpers import get_current_extension_name
|
|
from lnbits.tasks import register_invoice_listener
|
|
|
|
from .crud import get_lnurlpayout_from_wallet
|
|
|
|
|
|
async def wait_for_paid_invoices():
|
|
invoice_queue = asyncio.Queue()
|
|
register_invoice_listener(invoice_queue, get_current_extension_name())
|
|
|
|
while True:
|
|
payment = await invoice_queue.get()
|
|
await on_invoice_paid(payment)
|
|
|
|
|
|
async def on_invoice_paid(payment: Payment) -> None:
|
|
try:
|
|
# Check its got a payout associated with it
|
|
lnurlpayout_link = await get_lnurlpayout_from_wallet(payment.wallet_id)
|
|
logger.debug("LNURLpayout", lnurlpayout_link)
|
|
if lnurlpayout_link:
|
|
|
|
# Check the wallet balance is more than the threshold
|
|
|
|
wallet = await get_wallet(lnurlpayout_link.wallet)
|
|
assert wallet
|
|
threshold = lnurlpayout_link.threshold + (lnurlpayout_link.threshold * 0.02)
|
|
|
|
if wallet.balance < threshold:
|
|
return
|
|
# Get the invoice from the LNURL to pay
|
|
async with httpx.AsyncClient() as client:
|
|
try:
|
|
url = lnurl_decode(lnurlpayout_link.lnurlpay)
|
|
|
|
try:
|
|
r = await client.get(str(url), timeout=40)
|
|
res = r.json()
|
|
try:
|
|
r = await client.get(
|
|
res["callback"]
|
|
+ "?amount="
|
|
+ str(
|
|
int((wallet.balance - wallet.balance * 0.02) * 1000)
|
|
),
|
|
timeout=40,
|
|
)
|
|
res = r.json()
|
|
|
|
if hasattr(res, "status") and res["status"] == "ERROR":
|
|
raise HTTPException(
|
|
status_code=HTTPStatus.FORBIDDEN,
|
|
detail=res["reason"],
|
|
)
|
|
try:
|
|
await pay_invoice(
|
|
wallet_id=payment.wallet_id,
|
|
payment_request=res["pr"],
|
|
extra={"tag": "lnurlpayout"},
|
|
)
|
|
return
|
|
except:
|
|
pass
|
|
|
|
except Exception as e:
|
|
print("ERROR", str(e))
|
|
return
|
|
except (httpx.ConnectError, httpx.RequestError):
|
|
return
|
|
except Exception:
|
|
raise HTTPException(
|
|
status_code=HTTPStatus.FORBIDDEN,
|
|
detail="Failed to save LNURLPayout",
|
|
)
|
|
except:
|
|
return
|