mirror of
https://github.com/aljazceru/nutshell.git
synced 2025-12-20 18:44:20 +01:00
144 lines
4.5 KiB
Python
144 lines
4.5 KiB
Python
# type: ignore
|
|
from typing import Optional
|
|
|
|
import httpx
|
|
|
|
from ..core.settings import settings
|
|
from .base import (
|
|
InvoiceResponse,
|
|
PaymentResponse,
|
|
PaymentStatus,
|
|
StatusResponse,
|
|
Wallet,
|
|
)
|
|
|
|
|
|
class LNbitsWallet(Wallet):
|
|
"""https://github.com/lnbits/lnbits"""
|
|
|
|
def __init__(self):
|
|
self.endpoint = settings.mint_lnbits_endpoint
|
|
self.client = httpx.AsyncClient(
|
|
verify=not settings.debug,
|
|
headers={"X-Api-Key": settings.mint_lnbits_key},
|
|
)
|
|
|
|
async def status(self) -> StatusResponse:
|
|
try:
|
|
r = await self.client.get(url=f"{self.endpoint}/api/v1/wallet", timeout=15)
|
|
r.raise_for_status()
|
|
except Exception as exc:
|
|
return StatusResponse(
|
|
error_message=f"Failed to connect to {self.endpoint} due to: {exc}",
|
|
balance_msat=0,
|
|
)
|
|
|
|
try:
|
|
data: dict = r.json()
|
|
except Exception:
|
|
return StatusResponse(
|
|
error_message=(
|
|
f"Failed to connect to {self.endpoint}, got: '{r.text[:200]}...'"
|
|
),
|
|
balance_msat=0,
|
|
)
|
|
if "detail" in data:
|
|
return StatusResponse(
|
|
error_message=f"LNbits error: {data['detail']}", balance_msat=0
|
|
)
|
|
|
|
return StatusResponse(error_message=None, balance_msat=data["balance"])
|
|
|
|
async def create_invoice(
|
|
self,
|
|
amount: int,
|
|
memo: Optional[str] = None,
|
|
description_hash: Optional[bytes] = None,
|
|
unhashed_description: Optional[bytes] = None,
|
|
) -> InvoiceResponse:
|
|
data = {"out": False, "amount": amount}
|
|
if description_hash:
|
|
data["description_hash"] = description_hash.hex()
|
|
if unhashed_description:
|
|
data["unhashed_description"] = unhashed_description.hex()
|
|
|
|
data["memo"] = memo or ""
|
|
try:
|
|
r = await self.client.post(
|
|
url=f"{self.endpoint}/api/v1/payments", json=data
|
|
)
|
|
r.raise_for_status()
|
|
except Exception:
|
|
return InvoiceResponse(
|
|
ok=False,
|
|
error_message=r.json()["detail"],
|
|
)
|
|
|
|
data = r.json()
|
|
checking_id, payment_request = data["checking_id"], data["payment_request"]
|
|
|
|
return InvoiceResponse(
|
|
ok=True,
|
|
checking_id=checking_id,
|
|
payment_request=payment_request,
|
|
)
|
|
|
|
async def pay_invoice(self, bolt11: str, fee_limit_msat: int) -> PaymentResponse:
|
|
try:
|
|
r = await self.client.post(
|
|
url=f"{self.endpoint}/api/v1/payments",
|
|
json={"out": True, "bolt11": bolt11},
|
|
timeout=None,
|
|
)
|
|
r.raise_for_status()
|
|
except Exception:
|
|
return PaymentResponse(error_message=r.json()["detail"])
|
|
if r.status_code > 299:
|
|
return PaymentResponse(error_message=(f"HTTP status: {r.reason_phrase}",))
|
|
if "detail" in r.json():
|
|
return PaymentResponse(error_message=(r.json()["detail"],))
|
|
|
|
data: dict = r.json()
|
|
checking_id = data["payment_hash"]
|
|
|
|
# we do this to get the fee and preimage
|
|
payment: PaymentStatus = await self.get_payment_status(checking_id)
|
|
|
|
return PaymentResponse(
|
|
ok=True,
|
|
checking_id=checking_id,
|
|
fee_msat=payment.fee_msat,
|
|
preimage=payment.preimage,
|
|
)
|
|
|
|
async def get_invoice_status(self, checking_id: str) -> PaymentStatus:
|
|
try:
|
|
r = await self.client.get(
|
|
url=f"{self.endpoint}/api/v1/payments/{checking_id}"
|
|
)
|
|
r.raise_for_status()
|
|
except Exception:
|
|
return PaymentStatus(paid=None)
|
|
data: dict = r.json()
|
|
if data.get("detail"):
|
|
return PaymentStatus(paid=None)
|
|
return PaymentStatus(paid=r.json()["paid"])
|
|
|
|
async def get_payment_status(self, checking_id: str) -> PaymentStatus:
|
|
try:
|
|
r = await self.client.get(
|
|
url=f"{self.endpoint}/api/v1/payments/{checking_id}"
|
|
)
|
|
r.raise_for_status()
|
|
except Exception:
|
|
return PaymentStatus(paid=None)
|
|
data = r.json()
|
|
if "paid" not in data and "details" not in data:
|
|
return PaymentStatus(paid=None)
|
|
|
|
return PaymentStatus(
|
|
paid=data["paid"],
|
|
fee_msat=data["details"]["fee"],
|
|
preimage=data["preimage"],
|
|
)
|