mirror of
https://github.com/lnbits/lnbits-legend.git
synced 2024-11-20 10:39:59 +01:00
148 lines
4.8 KiB
Python
148 lines
4.8 KiB
Python
import asyncio
|
|
import json
|
|
import httpx
|
|
from os import getenv
|
|
from typing import Optional, Dict, AsyncGenerator
|
|
|
|
from .base import (
|
|
StatusResponse,
|
|
InvoiceResponse,
|
|
PaymentResponse,
|
|
PaymentStatus,
|
|
Wallet,
|
|
)
|
|
|
|
|
|
class LntxbotWallet(Wallet):
|
|
"""https://github.com/fiatjaf/lntxbot/blob/master/api.go"""
|
|
|
|
def __init__(self):
|
|
endpoint = getenv("LNTXBOT_API_ENDPOINT")
|
|
self.endpoint = endpoint[:-1] if endpoint.endswith("/") else endpoint
|
|
|
|
key = (
|
|
getenv("LNTXBOT_KEY")
|
|
or getenv("LNTXBOT_ADMIN_KEY")
|
|
or getenv("LNTXBOT_INVOICE_KEY")
|
|
)
|
|
self.auth = {"Authorization": f"Basic {key}"}
|
|
|
|
async def status(self) -> StatusResponse:
|
|
async with httpx.AsyncClient() as client:
|
|
r = await client.get(
|
|
f"{self.endpoint}/balance", headers=self.auth, timeout=40
|
|
)
|
|
try:
|
|
data = r.json()
|
|
except:
|
|
return StatusResponse(
|
|
f"Failed to connect to {self.endpoint}, got: '{r.text[:200]}...'", 0
|
|
)
|
|
|
|
if data.get("error"):
|
|
return StatusResponse(data["message"], 0)
|
|
|
|
return StatusResponse(None, data["BTC"]["AvailableBalance"] * 1000)
|
|
|
|
async def create_invoice(
|
|
self,
|
|
amount: int,
|
|
memo: Optional[str] = None,
|
|
description_hash: Optional[bytes] = None,
|
|
) -> InvoiceResponse:
|
|
data: Dict = {"amt": str(amount)}
|
|
if description_hash:
|
|
data["description_hash"] = description_hash.hex()
|
|
else:
|
|
data["memo"] = memo or ""
|
|
|
|
async with httpx.AsyncClient() as client:
|
|
r = await client.post(
|
|
f"{self.endpoint}/addinvoice", headers=self.auth, json=data, timeout=40
|
|
)
|
|
|
|
if r.is_error:
|
|
try:
|
|
data = r.json()
|
|
error_message = data["message"]
|
|
except:
|
|
error_message = r.text
|
|
pass
|
|
|
|
return InvoiceResponse(False, None, None, error_message)
|
|
|
|
data = r.json()
|
|
return InvoiceResponse(True, data["payment_hash"], data["pay_req"], None)
|
|
|
|
async def pay_invoice(self, bolt11: str) -> PaymentResponse:
|
|
async with httpx.AsyncClient() as client:
|
|
r = await client.post(
|
|
f"{self.endpoint}/payinvoice",
|
|
headers=self.auth,
|
|
json={"invoice": bolt11},
|
|
timeout=100,
|
|
)
|
|
|
|
if "error" in r.json():
|
|
try:
|
|
data = r.json()
|
|
error_message = data["message"]
|
|
except:
|
|
error_message = r.text
|
|
pass
|
|
|
|
return PaymentResponse(False, None, 0, None, error_message)
|
|
|
|
data = r.json()
|
|
checking_id = data["payment_hash"]
|
|
fee_msat = -data["fee_msat"]
|
|
preimage = data["payment_preimage"]
|
|
return PaymentResponse(True, checking_id, fee_msat, preimage, None)
|
|
|
|
async def get_invoice_status(self, checking_id: str) -> PaymentStatus:
|
|
async with httpx.AsyncClient() as client:
|
|
r = await client.post(
|
|
f"{self.endpoint}/invoicestatus/{checking_id}?wait=false",
|
|
headers=self.auth,
|
|
)
|
|
|
|
data = r.json()
|
|
if r.is_error or "error" in data:
|
|
return PaymentStatus(None)
|
|
|
|
if "preimage" not in data:
|
|
return PaymentStatus(False)
|
|
|
|
return PaymentStatus(True)
|
|
|
|
async def get_payment_status(self, checking_id: str) -> PaymentStatus:
|
|
async with httpx.AsyncClient() as client:
|
|
r = await client.post(
|
|
url=f"{self.endpoint}/paymentstatus/{checking_id}", headers=self.auth
|
|
)
|
|
|
|
data = r.json()
|
|
if r.is_error or "error" in data:
|
|
return PaymentStatus(None)
|
|
|
|
statuses = {"complete": True, "failed": False, "pending": None, "unknown": None}
|
|
return PaymentStatus(statuses[data.get("status", "unknown")])
|
|
|
|
async def paid_invoices_stream(self) -> AsyncGenerator[str, None]:
|
|
url = f"{self.endpoint}/payments/stream"
|
|
|
|
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, httpx.ReadTimeout, httpx.ConnectError):
|
|
pass
|
|
|
|
print("lost connection to lntxbot /payments/stream, retrying in 5 seconds")
|
|
await asyncio.sleep(5)
|