highperfocused/lnbits
1
0
Fork 0
mirror of https://github.com/lnbits/lnbits.git synced 2025-10-09 20:12:34 +02:00
Code Issues Packages Projects Releases Wiki Activity

add lntxbot invoices listener.

Browse Source
This commit is contained in:
fiatjaf
2020-10-14 23:58:57 -03:00
parent d528d4e09e
commit f02ec67f35
1 changed files with 27 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
lnbits/wallets/lntxbot.py
View File

@@ -1,4 +1,5 @@
import trio # type: ignore import trio # type: ignore
import json
import httpx import httpx
from os import getenv from os import getenv
from typing import Optional, Dict, AsyncGenerator from typing import Optional, Dict, AsyncGenerator
@@ -42,7 +43,7 @@ class LntxbotWallet(Wallet):
data["memo"] = memo or "" data["memo"] = memo or ""
r = httpx.post( r = httpx.post(
url=f"{self.endpoint}/addinvoice", f"{self.endpoint}/addinvoice",
headers=self.auth, headers=self.auth,
json=data, json=data,
timeout=40, timeout=40,
@@ -63,7 +64,7 @@ class LntxbotWallet(Wallet):
def pay_invoice(self, bolt11: str) -> PaymentResponse: def pay_invoice(self, bolt11: str) -> PaymentResponse:
r = httpx.post( r = httpx.post(
url=f"{self.endpoint}/payinvoice", f"{self.endpoint}/payinvoice",
headers=self.auth, headers=self.auth,
json={"invoice": bolt11}, json={"invoice": bolt11},
timeout=40, timeout=40,
@@ -86,7 +87,10 @@ class LntxbotWallet(Wallet):
return PaymentResponse(True, checking_id, fee_msat, preimage, None) return PaymentResponse(True, checking_id, fee_msat, preimage, None)
def get_invoice_status(self, checking_id: str) -> PaymentStatus: def get_invoice_status(self, checking_id: str) -> PaymentStatus:
r = httpx.post(url=f"{self.endpoint}/invoicestatus/{checking_id}?wait=false", headers=self.auth) r = httpx.post(
f"{self.endpoint}/invoicestatus/{checking_id}?wait=false",
headers=self.auth,
)
data = r.json() data = r.json()
if r.is_error or "error" in data: if r.is_error or "error" in data:
@@ -98,7 +102,10 @@ class LntxbotWallet(Wallet):
return PaymentStatus(True) return PaymentStatus(True)
def get_payment_status(self, checking_id: str) -> PaymentStatus: def get_payment_status(self, checking_id: str) -> PaymentStatus:
r = httpx.post(url=f"{self.endpoint}/paymentstatus/{checking_id}", headers=self.auth) r = httpx.post(
url=f"{self.endpoint}/paymentstatus/{checking_id}",
headers=self.auth,
)
data = r.json() data = r.json()
if r.is_error or "error" in data: if r.is_error or "error" in data:
@@ -108,6 +115,19 @@ class LntxbotWallet(Wallet):
return PaymentStatus(statuses[data.get("status", "unknown")]) return PaymentStatus(statuses[data.get("status", "unknown")])
async def paid_invoices_stream(self) -> AsyncGenerator[str, None]: async def paid_invoices_stream(self) -> AsyncGenerator[str, None]:
print("lntxbot does not support paid invoices stream yet") url = f"{self.endpoint}/payments/stream"
await trio.sleep(5)
yield "" while True:
try:
async with httpx.AsyncClient(timeout=None, headers=self.auth) as client:
async with client.stream("GET", url) as r:
async for line in r.aiter_lines():
if line.startswith("data:"):
data = json.loads(line[5:])
if "payment_hash" in data and data.get("msatoshi") > 0:
yield data["payment_hash"]
except (OSError, httpx.ReadError):
pass
print("lost connection to lntxbot /payments/stream, retrying in 5 seconds")
await trio.sleep(5)