Add files via upload

lnbits/wallets/lndrest.py

@@ -0,0 +1,78 @@
+from os import getenv
+import os
+from requests import get, post
+from .base import InvoiceResponse, PaymentResponse, PaymentStatus, Wallet
+
+
+class LndRestWallet(Wallet):
+    """https://api.lightning.community/rest/index.html#lnd-rest-api-reference"""
+
+
+    def __init__(self):
+
+        endpoint = getenv("LND_REST_ENDPOINT") + ":" + getenv("LND_REST_PORT")
+        self.endpoint = endpoint[:-1] if endpoint.endswith("/") else endpoint
+        print(self.endpoint)
+        self.auth_admin = getenv("LND_REST_ADMIN_MACAROON")
+        self.auth_invoice = getenv("LND_REST_INVOICE_MACAROON")
+        self.auth_read = getenv("LND_REST_READ_MACAROON")
+        self.auth_cert = getenv("LND_REST_CERT")
+
+
+    def create_invoice(self, amount: int, memo: str = "") -> InvoiceResponse:
+
+        r = post(
+            url=f"{self.endpoint}/v1/invoices",
+            headers=self.auth_admin, verify=False,
+            json={"value": amount, "memo": memo, "private": True},
+        )
+        print(r)
+        
+
+        ok, checking_id, payment_request, error_message = r.ok, None, None, None
+
+        if r.ok:
+            data = r.json()
+            payment_request = data["payment_request"]
+
+        r = get(url=f"{self.endpoint}/v1/payreq/{payment_request}", headers=self.auth_read, verify=False,)
+
+        if r.ok:
+            dataa = r.json()
+            checking_id = dataa["payment_hash"]
+            error_message = None
+            ok = True
+
+        return InvoiceResponse(ok, checking_id, payment_request, error_message)
+
+
+    def pay_invoice(self, bolt11: str) -> PaymentResponse:
+        r = post(url=f"{self.endpoint}/v2/withdrawals", headers=self.auth_admin, json={"type": "ln", "address": bolt11})
+        ok, checking_id, fee_msat, error_message = r.ok, None, 0, None
+
+        if r.ok:
+            data = r.json()["data"]
+            checking_id, fee_msat = data["id"], data["fee"] * 1000
+        else:
+            error_message = r.json()["message"]
+
+        return PaymentResponse(ok, checking_id, fee_msat, error_message)
+
+
+    def get_invoice_status(self, checking_id: str) -> PaymentStatus:
+        r = get(url=f"{self.endpoint}/v1/charge/{checking_id}", headers=self.auth_invoice)
+
+        if not r.ok:
+            return PaymentStatus(None)
+
+        statuses = {0: None, 1: True, -1: False}
+        return PaymentStatus(statuses[r.json()["data"]["status"]])
+
+    def get_payment_status(self, checking_id: str) -> PaymentStatus:
+        r = get(url=f"{self.endpoint}/v1/withdrawal/{checking_id}", headers=self.auth_admin)
+
+        if not r.ok:
+            return PaymentStatus(None)
+
+        statuses = {0: None, 1: True, -1: False}
+        return PaymentStatus(statuses[r.json()["data"]["status"]])