lnaddress_ext proposal untested

lnbits/extensions/lnaddress/lnurl.py

@@ -0,0 +1,92 @@
+import hashlib
+from datetime import datetime, timedelta
+
+import httpx
+from fastapi.params import Query
+from lnurl import (  # type: ignore
+    LnurlErrorResponse,
+    LnurlPayActionResponse,
+    LnurlPayResponse,
+)
+from starlette.requests import Request
+
+from . import lnaddress_ext
+from .crud import get_address, get_address_by_username, get_domain
+
+
+async def lnurl_response(username: str, domain: str, request: Request):
+    address = await get_address_by_username(username, domain)
+
+    if not address:
+        return {"status": "ERROR", "reason": "Address not found."}
+
+    ## CHECK IF USER IS STILL VALID/PAYING
+    now = datetime.now().timestamp()
+    start = datetime.fromtimestamp(address.time)
+    expiration = (start  + timedelta(days = address.duration)).timestamp()
+
+    if now > expiration:
+        return LnurlErrorResponse(reason="Address has expired.").dict()
+
+    resp = LnurlPayResponse(
+        callback=request.url_for("lnaddress.lnurl_callback", address_id=address.id, _external=True),
+        min_sendable=1000,
+        max_sendable=1000000000,
+        metadata=await address.lnurlpay_metadata(),
+    )
+
+    return resp.dict()
+
+@lnaddress_ext.get("/lnurl/cb/{address_id}", name="lnaddress.lnurl_callback")
+async def lnurl_callback(address_id, amount: int = Query(...)):
+    address = await get_address(address_id)
+
+    if not address:
+        return LnurlErrorResponse(
+                reason=f"Address not found"
+            ).dict()
+
+    amount_received = amount
+    # min = 1000
+    # max = 1000000000
+
+    # if amount_received < min:
+    #     return LnurlErrorResponse(
+    #             reason=f"Amount {amount_received} is smaller than minimum."
+    #         ).dict()
+
+    # elif amount_received > max:
+    #     return jsonify(
+    #         LnurlErrorResponse(
+    #             reason=f"Amount {amount_received} is greater than maximum."
+    #         ).dict()
+    #     )
+
+    domain = await get_domain(address.domain)
+
+    base_url = address.wallet_endpoint[:-1] if address.wallet_endpoint.endswith('/') else address.wallet_endpoint
+
+    async with httpx.AsyncClient() as client:
+        try:
+            call = await client.post(
+                base_url + "/api/v1/payments",
+                headers={"X-Api-Key": address.wallet_key, "Content-Type": "application/json"},
+                json={
+                    "out": False,
+                    "amount": int(amount_received / 1000),
+                    "description_hash": hashlib.sha256((await address.lnurlpay_metadata()).encode("utf-8")).hexdigest(),
+                    "extra": {"tag": f"Payment to {address.username}@{domain.domain}"},
+                },
+                timeout=40,
+            )
+
+            r = call.json()
+        except AssertionError as e:
+            return LnurlErrorResponse(reason="ERROR")
+
+    resp = LnurlPayActionResponse(
+        pr=r["payment_request"],
+        routes=[],
+    )
+
+    return resp.dict()