basic invoice listeners.

lnbits/wallets/lnpay.py

@@ -1,6 +1,9 @@
+import asyncio
+import aiohttp
 from os import getenv
-from typing import Optional, Dict
+from typing import Optional, Dict, AsyncGenerator
 from requests import get, post
+from quart import request
 
 from .base import InvoiceResponse, PaymentResponse, PaymentStatus, Wallet
 
@@ -15,9 +18,13 @@ class LNPayWallet(Wallet):
         self.auth_invoice = getenv("LNPAY_INVOICE_KEY")
         self.auth_read = getenv("LNPAY_READ_KEY")
         self.auth_api = {"X-Api-Key": getenv("LNPAY_API_KEY")}
+        self.queue = asyncio.Queue()
 
     def create_invoice(
-        self, amount: int, memo: Optional[str] = None, description_hash: Optional[bytes] = None
+        self,
+        amount: int,
+        memo: Optional[str] = None,
+        description_hash: Optional[bytes] = None,
     ) -> InvoiceResponse:
         data: Dict = {"num_satoshis": f"{amount}"}
         if description_hash:
@@ -30,7 +37,12 @@ class LNPayWallet(Wallet):
             headers=self.auth_api,
             json=data,
         )
-        ok, checking_id, payment_request, error_message = r.status_code == 201, None, None, r.text
+        ok, checking_id, payment_request, error_message = (
+            r.status_code == 201,
+            None,
+            None,
+            r.text,
+        )
 
         if ok:
             data = r.json()
@@ -55,10 +67,30 @@ class LNPayWallet(Wallet):
         return self.get_payment_status(checking_id)
 
     def get_payment_status(self, checking_id: str) -> PaymentStatus:
-        r = get(url=f"{self.endpoint}/user/lntx/{checking_id}", headers=self.auth_api)
+        r = get(
+            url=f"{self.endpoint}/user/lntx/{checking_id}?fields=settled",
+            headers=self.auth_api,
+        )
 
         if not r.ok:
             return PaymentStatus(None)
 
         statuses = {0: None, 1: True, -1: False}
         return PaymentStatus(statuses[r.json()["settled"]])
+
+    async def paid_invoices_stream(self) -> AsyncGenerator[str, None]:
+        while True:
+            yield await self.queue.get()
+            self.queue.task_done()
+
+    async def webhook_listener(self):
+        data = await request.get_json()
+        if "event" not in data or data["event"].get("name") != "wallet_receive":
+            return ""
+
+        lntx_id = data["data"]["wtx"]["lnTx"]["id"]
+        async with aiohttp.ClientSession() as session:
+            async with session.get(f"{self.endpoint}/user/lntx/{lntx_id}?fields=settled") as resp:
+                data = await resp.json()
+                if data["settled"]:
+                    self.queue.put_nowait(lntx_id)

lnbits/wallets/spark.py

@@ -1,7 +1,9 @@
 import random
 import requests
+import json
+from aiohttp_sse_client import client as sse_client
 from os import getenv
-from typing import Optional
+from typing import Optional, AsyncGenerator
 
 from .base import InvoiceResponse, PaymentResponse, PaymentStatus, Wallet
 
@@ -16,7 +18,7 @@ class UnknownError(Exception):
 
 class SparkWallet(Wallet):
     def __init__(self):
-        self.url = getenv("SPARK_URL")
+        self.url = getenv("SPARK_URL").replace("/rpc", "")
         self.token = getenv("SPARK_TOKEN")
 
     def __getattr__(self, key):
@@ -28,7 +30,9 @@ class SparkWallet(Wallet):
             elif kwargs:
                 params = kwargs
 
-            r = requests.post(self.url, headers={"X-Access": self.token}, json={"method": key, "params": params})
+            r = requests.post(
+                self.url + "/rpc", headers={"X-Access": self.token}, json={"method": key, "params": params}
+            )
             try:
                 data = r.json()
             except:
@@ -91,3 +95,15 @@ class SparkWallet(Wallet):
                 return PaymentStatus(False)
             return PaymentStatus(None)
         raise KeyError("supplied an invalid checking_id")
+
+    async def paid_invoices_stream(self) -> AsyncGenerator[str, None]:
+        url = self.url + "/stream?access-key=" + self.token
+        conn = sse_client.EventSource(url)
+        async with conn as es:
+            async for event in es:
+                try:
+                    if event.type == "inv-paid":
+                        data = json.loads(event.data)
+                        yield data["label"]
+                except ConnectionError:
+                    pass