Add files via upload

tasks.py

@@ -0,0 +1,79 @@
+import asyncio
+import json
+
+import httpx
+from loguru import logger
+
+from lnbits.core.crud import update_payment_extra
+from lnbits.core.models import Payment
+from lnbits.helpers import get_current_extension_name
+from lnbits.tasks import register_invoice_listener
+
+from .crud import get_pay_link
+
+
+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):
+    if payment.extra.get("tag") != "lnurlp":
+        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: httpx.Response = 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,
+                        "body": json.loads(pay_link.webhook_body)
+                        if pay_link.webhook_body
+                        else "",
+                    },
+                    headers=json.loads(pay_link.webhook_headers)
+                    if pay_link.webhook_headers
+                    else None,
+                    timeout=40,
+                )
+                await mark_webhook_sent(
+                    payment.payment_hash,
+                    r.status_code,
+                    r.is_success,
+                    r.reason_phrase,
+                    r.text,
+                )
+            except Exception as ex:
+                logger.error(ex)
+                await mark_webhook_sent(
+                    payment.payment_hash, -1, False, "Unexpected Error", str(ex)
+                )
+
+
+async def mark_webhook_sent(
+    payment_hash: str, status: int, is_success: bool, reason_phrase="", text=""
+) -> None:
+
+    await update_payment_extra(
+        payment_hash,
+        {
+            "wh_status": status,  # keep for backwards compability
+            "wh_success": is_success,
+            "wh_message": reason_phrase,
+            "wh_response": text,
+        },
+    )