From 8a7652c06fb70994782d69665c0fa58b3d1645b5 Mon Sep 17 00:00:00 2001
From: Kristjan <kristjan.grm1@gmail.com>
Date: Sat, 2 Jan 2021 20:49:39 +0100
Subject: [PATCH] started working on payment listeners

---
 lnbits/extensions/subdomains/__init__.py |  5 +++
 lnbits/extensions/subdomains/tasks.py    | 47 ++++++++++++++++++++++++
 2 files changed, 52 insertions(+)
 create mode 100644 lnbits/extensions/subdomains/tasks.py

diff --git a/lnbits/extensions/subdomains/__init__.py b/lnbits/extensions/subdomains/__init__.py
index 51a82117..2e8e48361 100644
--- a/lnbits/extensions/subdomains/__init__.py
+++ b/lnbits/extensions/subdomains/__init__.py
@@ -8,3 +8,8 @@ subdomains_ext: Blueprint = Blueprint("subdomains", __name__, static_folder="sta
 
 from .views_api import *  # noqa
 from .views import *  # noqa
+
+from .tasks import register_listeners
+from lnbits.tasks import record_async
+
+subdomains_ext.record(record_async(register_listeners))
\ No newline at end of file
diff --git a/lnbits/extensions/subdomains/tasks.py b/lnbits/extensions/subdomains/tasks.py
new file mode 100644
index 00000000..004093f3
--- /dev/null
+++ b/lnbits/extensions/subdomains/tasks.py
@@ -0,0 +1,47 @@
+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)
\ No newline at end of file