lnurlp fastAPI port

lnbits/extensions/lnurlp/views_api.py

@@ -1,9 +1,13 @@
 from typing import Optional
+from fastapi.params import Depends
 from fastapi.param_functions import Query
 from pydantic.main import BaseModel
-from quart import g, jsonify, request
+
 from http import HTTPStatus
 from lnurl.exceptions import InvalidUrl as LnurlInvalidUrl  # type: ignore
+from starlette.exceptions import HTTPException
+from starlette.requests import Request
+from starlette.responses import HTMLResponse, JSONResponse  # type: ignore
 
 from lnbits.core.crud import get_user
 from lnbits.decorators import api_check_wallet_key, api_validate_post_request
@@ -20,16 +24,16 @@ from .crud import (
 
 @lnurlp_ext.get("/api/v1/currencies")
 async def api_list_currencies_available():
-    return jsonify(list(currencies.keys()))
+    return list(currencies.keys())
 
 
-@lnurlp_ext.get("/api/v1/links")
-@api_check_wallet_key("invoice")
-async def api_links():
-    wallet_ids = [g.wallet.id]
+@lnurlp_ext.get("/api/v1/links", status_code=HTTPStatus.OK)
+# @api_check_wallet_key("invoice")
+async def api_links(wallet: WalletTypeInfo = Depends(get_key_type)):
+    wallet_ids = [wallet.wallet.id]
 
-    if "all_wallets" in request.args:
-        wallet_ids = (await get_user(g.wallet.user)).wallet_ids
+    if "all_wallets" in Request.path_parameters:
+        wallet_ids = (await get_user(wallet.wallet.user)).wallet_ids
 
     try:
         return (
@@ -37,29 +41,40 @@ async def api_links():
                     {**link._asdict(), **{"lnurl": link.lnurl}}
                     for link in await get_pay_links(wallet_ids)
                 ],
-            HTTPStatus.OK,
         )
     except LnurlInvalidUrl:
-        return (
-                {
-                    "message": "LNURLs need to be delivered over a publically accessible `https` domain or Tor."
-                },
-            HTTPStatus.UPGRADE_REQUIRED,
+        raise HTTPException(
+            status_code=HTTPStatus.UPGRADE_REQUIRED,
+            detail="LNURLs need to be delivered over a publically accessible `https` domain or Tor.",
         )
+        # return (
+        #         {
+        #             "message": "LNURLs need to be delivered over a publically accessible `https` domain or Tor."
+        #         },
+        #     HTTPStatus.UPGRADE_REQUIRED,
+        # )
 
 
-@lnurlp_ext.get("/api/v1/links/{link_id}")
-@api_check_wallet_key("invoice")
-async def api_link_retrieve(link_id):
+@lnurlp_ext.get("/api/v1/links/{link_id}", status_code=HTTPStatus.OK)
+# @api_check_wallet_key("invoice")
+async def api_link_retrieve(link_id, wallet: WalletTypeInfo = Depends(get_key_type)):
     link = await get_pay_link(link_id)
 
     if not link:
-        return {"message": "Pay link does not exist."}, HTTPStatus.NOT_FOUND
+        raise HTTPException(
+            detail="Pay link does not exist.",
+            status_code=HTTPStatus.NOT_FOUND
+        )
+        # return {"message": "Pay link does not exist."}, HTTPStatus.NOT_FOUND
 
-    if link.wallet != g.wallet.id:
-        return {"message": "Not your pay link."}, HTTPStatus.FORBIDDEN
+    if link.wallet != wallet.wallet.id:
+        raise HTTPException(
+            detail="Not your pay link.",
+            status_code=HTTPStatus.FORBIDDEN
+        )
+        # return {"message": "Not your pay link."}, HTTPStatus.FORBIDDEN
 
-    return {**link._asdict(), **{"lnurl": link.lnurl}}, HTTPStatus.OK
+    return {**link._asdict(), **{"lnurl": link.lnurl}}
 
 class CreateData(BaseModel):
     description:  str
@@ -71,67 +86,92 @@ class CreateData(BaseModel):
     success_text:  Optional[str]
     success_url:  Optional[str]
 
-@lnurlp_ext.post("/api/v1/links")
-@lnurlp_ext.put("/api/v1/links/{link_id}")
-@api_check_wallet_key("invoice")
-async def api_link_create_or_update(data: CreateData, link_id=None):
+@lnurlp_ext.post("/api/v1/links", status_code=HTTPStatus.CREATED)
+@lnurlp_ext.put("/api/v1/links/{link_id}", status_code=HTTPStatus.OK)
+# @api_check_wallet_key("invoice")
+async def api_link_create_or_update(data: CreateData, link_id=None, wallet: WalletTypeInfo = Depends(get_key_type)):
     if data.min > data.max:
-        return {"message": "Min is greater than max."}, HTTPStatus.BAD_REQUEST
+        raise HTTPException(
+            detail="Min is greater than max.",
+            status_code=HTTPStatus.BAD_REQUEST
+        )
+        # return {"message": "Min is greater than max."}, HTTPStatus.BAD_REQUEST
 
     if data.currency == None and (
         round(data.min) != data.min or round(data.max) != data.max
     ):
-        return {"message": "Must use full satoshis."}, HTTPStatus.BAD_REQUEST
+        raise HTTPException(
+            detail="Must use full satoshis.",
+            status_code=HTTPStatus.BAD_REQUEST
+        )
+        # return {"message": "Must use full satoshis."}, HTTPStatus.BAD_REQUEST
 
     if "success_url" in data and data.success_url[:8] != "https://":
-        return (
-            {"message": "Success URL must be secure https://..."},
-            HTTPStatus.BAD_REQUEST,
+        raise HTTPException(
+            detail="Success URL must be secure https://...",
+            status_code=HTTPStatus.BAD_REQUEST
         )
+        # return (
+        #     {"message": "Success URL must be secure https://..."},
+        #     HTTPStatus.BAD_REQUEST,
+        # )
 
     if link_id:
         link = await get_pay_link(link_id)
 
         if not link:
-            return (
-                {"message": "Pay link does not exist."},
-                HTTPStatus.NOT_FOUND,
+            raise HTTPException(
+                detail="Pay link does not exist.",
+                status_code=HTTPStatus.NOT_FOUND
             )
+            # return (
+            #     {"message": "Pay link does not exist."},
+            #     HTTPStatus.NOT_FOUND,
+            # )
 
         if link.wallet != g.wallet.id:
-            return {"message": "Not your pay link."}, HTTPStatus.FORBIDDEN
+            raise HTTPException(
+                detail="Not your pay link.",
+                status_code=HTTPStatus.FORBIDDEN
+            )
+            # return {"message": "Not your pay link."}, HTTPStatus.FORBIDDEN
 
         link = await update_pay_link(link_id, **data)
     else:
-        link = await create_pay_link(wallet_id=g.wallet.id, **data)
+        link = await create_pay_link(wallet_id=wallet.wallet.id, **data)
 
-    return (
-        {**link._asdict(), **{"lnurl": link.lnurl}},
-        HTTPStatus.OK if link_id else HTTPStatus.CREATED,
-    )
+    return {**link._asdict(), **{"lnurl": link.lnurl}}
 
 
 @lnurlp_ext.delete("/api/v1/links/{link_id}")
-@api_check_wallet_key("invoice")
-async def api_link_delete(link_id):
+# @api_check_wallet_key("invoice")
+async def api_link_delete(link_id, , wallet: WalletTypeInfo = Depends(get_key_type)):
     link = await get_pay_link(link_id)
 
     if not link:
-        return {"message": "Pay link does not exist."}, HTTPStatus.NOT_FOUND
+        raise HTTPException(
+            detail="Pay link does not exist.",
+            status_code=HTTPStatus.NOT_FOUND
+        )
+        # return {"message": "Pay link does not exist."}, HTTPStatus.NOT_FOUND
 
-    if link.wallet != g.wallet.id:
-        return {"message": "Not your pay link."}, HTTPStatus.FORBIDDEN
+    if link.wallet != wallet.wallet.id:
+        raise HTTPException(
+            detail="Not your pay link.",
+            status_code=HTTPStatus.FORBIDDEN
+        )
+        # return {"message": "Not your pay link."}, HTTPStatus.FORBIDDEN
 
     await delete_pay_link(link_id)
-
-    return "", HTTPStatus.NO_CONTENT
+    raise HTTPException(status_code=HTTPStatus.NO_CONTENT)
+    # return "", HTTPStatus.NO_CONTENT
 
 
-@lnurlp_ext.get("/api/v1/rate/{currency}")
+@lnurlp_ext.get("/api/v1/rate/{currency}", status_code=HTTPStatus.OK)
 async def api_check_fiat_rate(currency):
     try:
         rate = await get_fiat_rate_satoshis(currency)
     except AssertionError:
         rate = None
 
-    return {"rate": rate}, HTTPStatus.OK
+    return {"rate": rate}