mirror of
https://github.com/lnbits/lnbits-legend.git
synced 2025-02-22 14:22:55 +01:00
d3fc52cd49
a big refactor that:
- fixes some issues that might have happened (or not) with asynchronous
reactions to payments;
- paves the way to https://github.com/lnbits/lnbits/issues/121;
- uses more async/await notation which just looks nice; and
- makes it simple(r?) for one extension to modify stuff from other extensions.
61 lines
1.9 KiB
Python
61 lines
1.9 KiB
Python
import trio # type: ignore
|
|
import json
|
|
import httpx
|
|
|
|
from lnbits.core import db as core_db
|
|
from lnbits.core.models import Payment
|
|
from lnbits.tasks import register_invoice_listener
|
|
|
|
from .crud import get_pay_link
|
|
|
|
|
|
async def register_listeners():
|
|
invoice_paid_chan_send, invoice_paid_chan_recv = trio.open_memory_channel(2)
|
|
register_invoice_listener(invoice_paid_chan_send)
|
|
await wait_for_paid_invoices(invoice_paid_chan_recv)
|
|
|
|
|
|
async def wait_for_paid_invoices(invoice_paid_chan: trio.MemoryReceiveChannel):
|
|
async for payment in invoice_paid_chan:
|
|
await on_invoice_paid(payment)
|
|
|
|
|
|
async def on_invoice_paid(payment: Payment) -> None:
|
|
if "lnurlp" != payment.extra.get("tag"):
|
|
# not an lnurlp invoice
|
|
return
|
|
|
|
if payment.extra.get("wh_status"):
|
|
# this webhook has already been sent
|
|
return
|
|
|
|
pay_link = await get_pay_link(payment.extra.get("link", -1))
|
|
if pay_link and pay_link.webhook_url:
|
|
async with httpx.AsyncClient() as client:
|
|
try:
|
|
r = await client.post(
|
|
pay_link.webhook_url,
|
|
json={
|
|
"payment_hash": payment.payment_hash,
|
|
"payment_request": payment.bolt11,
|
|
"amount": payment.amount,
|
|
"comment": payment.extra.get("comment"),
|
|
"lnurlp": pay_link.id,
|
|
},
|
|
timeout=40,
|
|
)
|
|
await mark_webhook_sent(payment, r.status_code)
|
|
except (httpx.ConnectError, httpx.RequestError):
|
|
await mark_webhook_sent(payment, -1)
|
|
|
|
|
|
async def mark_webhook_sent(payment: Payment, status: int) -> None:
|
|
payment.extra["wh_status"] = status
|
|
|
|
await core_db.execute(
|
|
"""
|
|
UPDATE apipayments SET extra = ?
|
|
WHERE hash = ?
|
|
""",
|
|
(json.dumps(payment.extra), payment.payment_hash),
|
|
)
|