add listener to invoice

lnbits/extensions/satspay/__init__.py

@@ -1,7 +1,10 @@
+import asyncio
+
 from fastapi import APIRouter
 
 from lnbits.db import Database
 from lnbits.helpers import template_renderer
+from lnbits.tasks import catch_everything_and_restart
 
 db = Database("ext_satspay")
 
@@ -12,6 +15,11 @@ satspay_ext: APIRouter = APIRouter(prefix="/satspay", tags=["satspay"])
 def satspay_renderer():
     return template_renderer(["lnbits/extensions/satspay/templates"])
 
-
+from .tasks import wait_for_paid_invoices
 from .views import *  # noqa
 from .views_api import *  # noqa
+
+
+def satspay_start():
+    loop = asyncio.get_event_loop()
+    loop.create_task(catch_everything_and_restart(wait_for_paid_invoices))

lnbits/extensions/satspay/crud.py

@@ -25,7 +25,10 @@ async def create_charge(user: str, data: CreateCharge) -> Charges:
         onchainaddress = None
     if data.lnbitswallet:
         payment_hash, payment_request = await create_invoice(
-            wallet_id=data.lnbitswallet, amount=data.amount, memo=charge_id
+            wallet_id=data.lnbitswallet,
+            amount=data.amount,
+            memo=charge_id,
+            extra={"tag": "charge"},
         )
     else:
         payment_hash = None
@@ -112,7 +115,7 @@ async def check_address_balance(charge_id: str) -> List[Charges]:
                 pass
         if charge.lnbitswallet:
             invoice_status = await api_payment(charge.payment_hash)
-
+            
             if invoice_status["paid"]:
                 return await update_charge(charge_id=charge_id, balance=charge.amount)
     row = await db.fetchone("SELECT * FROM satspay.charges WHERE id = ?", (charge_id,))

lnbits/extensions/satspay/tasks.py

@@ -0,0 +1,31 @@
+import asyncio
+
+from lnbits.core.models import Payment
+from lnbits.extensions.satspay.crud import check_address_balance, get_charge
+from lnbits.tasks import register_invoice_listener
+
+# from .crud import get_ticket, set_ticket_paid
+
+
+async def wait_for_paid_invoices():
+    invoice_queue = asyncio.Queue()
+    register_invoice_listener(invoice_queue)
+
+    while True:
+        payment = await invoice_queue.get()
+        await on_invoice_paid(payment)
+
+
+async def on_invoice_paid(payment: Payment) -> None:
+    if "charge" != payment.extra.get("tag"):
+        # not a charge invoice
+        return
+
+    charge = await get_charge(payment.memo)
+    if not charge:
+        print("this should never happen", payment)
+        return
+
+    await payment.set_pending(False)
+    await check_address_balance(charge_id=charge.id)
+    

lnbits/extensions/satspay/templates/satspay/display.html

@@ -235,10 +235,24 @@
         charge_time_elapsed: '{{charge.time_elapsed}}',
         charge_amount: '{{charge.amount}}',
         charge_balance: '{{charge.balance}}',
-        charge_paid: '{{charge.paid}}'
+        charge_paid: '{{charge.paid}}',
+        wallet: {
+          inkey: ''
+        },
+        cancelListener: () => {}
       }
     },
     methods: {
+      startPaymentNotifier(){
+        this.cancelListener()
+
+        this.cancelListener = LNbits.event.onInvoicePaid(
+          this.wallet,
+          payment => {
+            this.checkBalance()
+          }
+        )
+      },
       checkBalance: function () {
         var self = this
         LNbits.api
@@ -307,12 +321,14 @@
         this.lnbtc = false
         this.onbtc = true
       }
+      this.wallet.inkey = '{{ wallet_inkey }}'
       this.getTheTime()
       this.getThePercentage()
       var timerCount = this.timerCount
       if ('{{ charge.paid }}' == 'False') {
         timerCount()
       }
+      this.startPaymentNotifier()
     }
   })
 </script>

lnbits/extensions/satspay/views.py

@@ -6,6 +6,7 @@ from starlette.exceptions import HTTPException
 from starlette.requests import Request
 from starlette.responses import HTMLResponse
 
+from lnbits.core.crud import get_wallet
 from lnbits.core.models import User
 from lnbits.decorators import check_user_exists
 
@@ -29,6 +30,7 @@ async def display(request: Request, charge_id):
         raise HTTPException(
             status_code=HTTPStatus.NOT_FOUND, detail="Charge link does not exist."
         )
+    wallet = await get_wallet(charge.lnbitswallet)
     return satspay_renderer().TemplateResponse(
-        "satspay/display.html", {"request": request, "charge": charge}
+        "satspay/display.html", {"request": request, "charge": charge, "wallet_key": wallet.inkey}
     )