diff --git a/lnbits/extensions/paywall/__init__.py b/lnbits/extensions/paywall/__init__.py index cf9570a15..bdc05620d 100644 --- a/lnbits/extensions/paywall/__init__.py +++ b/lnbits/extensions/paywall/__init__.py @@ -1,12 +1,18 @@ -from quart import Blueprint +from fastapi import APIRouter + from lnbits.db import Database +from lnbits.helpers import template_renderer db = Database("ext_paywall") -paywall_ext: Blueprint = Blueprint( - "paywall", __name__, static_folder="static", template_folder="templates" +paywall_ext: APIRouter = APIRouter( + prefix="/paywall", + tags=["Paywall"] ) +def paywall_renderer(): + return template_renderer(["lnbits/extensions/paywall/templates"]) + -from .views_api import * # noqa from .views import * # noqa +from .views_api import * # noqa diff --git a/lnbits/extensions/paywall/crud.py b/lnbits/extensions/paywall/crud.py index c13aba439..1f888167f 100644 --- a/lnbits/extensions/paywall/crud.py +++ b/lnbits/extensions/paywall/crud.py @@ -3,17 +3,12 @@ from typing import List, Optional, Union from lnbits.helpers import urlsafe_short_hash from . import db -from .models import Paywall +from .models import CreatePaywall, Paywall async def create_paywall( - *, wallet_id: str, - url: str, - memo: str, - description: Optional[str] = None, - amount: int = 0, - remembers: bool = True, + data: CreatePaywall ) -> Paywall: paywall_id = urlsafe_short_hash() await db.execute( @@ -21,7 +16,7 @@ async def create_paywall( INSERT INTO paywall.paywalls (id, wallet, url, memo, description, amount, remembers) VALUES (?, ?, ?, ?, ?, ?, ?) """, - (paywall_id, wallet_id, url, memo, description, amount, int(remembers)), + (paywall_id, wallet_id, data.url, data.memo, data.description, data.amount, int(data.remembers)), ) paywall = await get_paywall(paywall_id) @@ -33,7 +28,6 @@ async def get_paywall(paywall_id: str) -> Optional[Paywall]: row = await db.fetchone( "SELECT * FROM paywall.paywalls WHERE id = ?", (paywall_id,) ) - return Paywall.from_row(row) if row else None diff --git a/lnbits/extensions/paywall/models.py b/lnbits/extensions/paywall/models.py index d7f2451df..c150372db 100644 --- a/lnbits/extensions/paywall/models.py +++ b/lnbits/extensions/paywall/models.py @@ -1,15 +1,30 @@ import json - from sqlite3 import Row -from typing import NamedTuple, Optional +from typing import Optional + +from fastapi import Query +from pydantic import BaseModel -class Paywall(NamedTuple): +class CreatePaywall(BaseModel): + url: str = Query(...) + memo: str = Query(...) + description: str = Query(None) + amount: int = Query(..., ge=0) + remembers: bool = Query(...) + +class CreatePaywallInvoice(BaseModel): + amount: int = Query(..., ge=1) + +class CheckPaywallInvoice(BaseModel): + payment_hash: str = Query(...) + +class Paywall(BaseModel): id: str wallet: str url: str memo: str - description: str + description: Optional[str] amount: int time: int remembers: bool diff --git a/lnbits/extensions/paywall/templates/paywall/_api_docs.html b/lnbits/extensions/paywall/templates/paywall/_api_docs.html index 1157fa467..ceadf2f08 100644 --- a/lnbits/extensions/paywall/templates/paywall/_api_docs.html +++ b/lnbits/extensions/paywall/templates/paywall/_api_docs.html @@ -18,7 +18,7 @@
Curl example
curl -X GET {{ request.url_root }}api/v1/paywalls -H "X-Api-Key: {{ - g.user.wallets[0].inkey }}" + user.wallets[0].inkey }}" @@ -52,7 +52,7 @@ <string>, "memo": <string>, "description": <string>, "amount": <integer>, "remembers": <boolean>}' -H "Content-type: application/json" -H "X-Api-Key: {{ - g.user.wallets[0].adminkey }}" + user.wallets[0].adminkey }}" @@ -139,7 +139,7 @@ curl -X DELETE {{ request.url_root }}api/v1/paywalls/<paywall_id> -H "X-Api-Key: {{ - g.user.wallets[0].adminkey }}" + user.wallets[0].adminkey }}" diff --git a/lnbits/extensions/paywall/templates/paywall/display.html b/lnbits/extensions/paywall/templates/paywall/display.html index 7bc7d9b88..394578b3a 100644 --- a/lnbits/extensions/paywall/templates/paywall/display.html +++ b/lnbits/extensions/paywall/templates/paywall/display.html @@ -102,11 +102,11 @@ }, createInvoice: function () { var self = this - + console.log(this.amount) axios .post( '/paywall/api/v1/paywalls/{{ paywall.id }}/invoice', - {amount: this.amount} + {amount: self.amount} ) .then(function (response) { self.paymentReq = response.data.payment_request.toUpperCase() diff --git a/lnbits/extensions/paywall/templates/paywall/index.html b/lnbits/extensions/paywall/templates/paywall/index.html index 8be3b2fab..482d1465e 100644 --- a/lnbits/extensions/paywall/templates/paywall/index.html +++ b/lnbits/extensions/paywall/templates/paywall/index.html @@ -237,7 +237,7 @@ LNbits.api .request( 'GET', - '/paywall/api/v1/paywalls?all_wallets', + '/paywall/api/v1/paywalls?all_wallets=true', this.g.user.wallets[0].inkey ) .then(function (response) { diff --git a/lnbits/extensions/paywall/views.py b/lnbits/extensions/paywall/views.py index 0dcbad2f5..e07596c4f 100644 --- a/lnbits/extensions/paywall/views.py +++ b/lnbits/extensions/paywall/views.py @@ -1,22 +1,26 @@ -from quart import g, abort, render_template from http import HTTPStatus -from lnbits.decorators import check_user_exists, validate_uuids +from fastapi import Depends +from starlette.exceptions import HTTPException +from starlette.requests import Request -from . import paywall_ext +from lnbits.core.models import User +from lnbits.decorators import check_user_exists + +from . import paywall_ext, paywall_renderer from .crud import get_paywall -@paywall_ext.route("/") -@validate_uuids(["usr"], required=True) -@check_user_exists() -async def index(): - return await render_template("paywall/index.html", user=g.user) +@paywall_ext.get("/") +async def index(request: Request, user: User = Depends(check_user_exists)): + return paywall_renderer().TemplateResponse("paywall/index.html", {"request": request, "user": user.dict()}) -@paywall_ext.route("/") -async def display(paywall_id): - paywall = await get_paywall(paywall_id) or abort( - HTTPStatus.NOT_FOUND, "Paywall does not exist." - ) - return await render_template("paywall/display.html", paywall=paywall) +@paywall_ext.get("/{paywall_id}") +async def display(request: Request, paywall_id): + paywall = await get_paywall(paywall_id) + if not paywall: + raise HTTPException( + status_code=HTTPStatus.NOT_FOUND, detail="Paywall does not exist." + ) + return paywall_renderer().TemplateResponse("paywall/display.html", {"request": request, "paywall": paywall}) diff --git a/lnbits/extensions/paywall/views_api.py b/lnbits/extensions/paywall/views_api.py index 45c80af4b..38f8d0b6a 100644 --- a/lnbits/extensions/paywall/views_api.py +++ b/lnbits/extensions/paywall/views_api.py @@ -1,81 +1,68 @@ -from quart import g, jsonify, request from http import HTTPStatus +from fastapi import Depends, Query +from starlette.exceptions import HTTPException + from lnbits.core.crud import get_user, get_wallet -from lnbits.core.services import create_invoice, check_invoice_status -from lnbits.decorators import api_check_wallet_key, api_validate_post_request +from lnbits.core.services import check_invoice_status, create_invoice +from lnbits.decorators import WalletTypeInfo, get_key_type from . import paywall_ext -from .crud import create_paywall, get_paywall, get_paywalls, delete_paywall +from .crud import create_paywall, delete_paywall, get_paywall, get_paywalls +from .models import CheckPaywallInvoice, CreatePaywall, CreatePaywallInvoice -@paywall_ext.route("/api/v1/paywalls", methods=["GET"]) -@api_check_wallet_key("invoice") -async def api_paywalls(): - wallet_ids = [g.wallet.id] +@paywall_ext.get("/api/v1/paywalls") +async def api_paywalls(wallet: WalletTypeInfo = Depends(get_key_type), all_wallets: bool = Query(False)): + wallet_ids = [wallet.wallet.id] - if "all_wallets" in request.args: - wallet_ids = (await get_user(g.wallet.user)).wallet_ids + if all_wallets: + wallet_ids = (await get_user(wallet.wallet.user)).wallet_ids - return ( - jsonify([paywall._asdict() for paywall in await get_paywalls(wallet_ids)]), - HTTPStatus.OK, - ) + return [paywall.dict() for paywall in await get_paywalls(wallet_ids)] -@paywall_ext.route("/api/v1/paywalls", methods=["POST"]) -@api_check_wallet_key("invoice") -@api_validate_post_request( - schema={ - "url": {"type": "string", "empty": False, "required": True}, - "memo": {"type": "string", "empty": False, "required": True}, - "description": { - "type": "string", - "empty": True, - "nullable": True, - "required": False, - }, - "amount": {"type": "integer", "min": 0, "required": True}, - "remembers": {"type": "boolean", "required": True}, - } -) -async def api_paywall_create(): - paywall = await create_paywall(wallet_id=g.wallet.id, **g.data) - return jsonify(paywall._asdict()), HTTPStatus.CREATED +@paywall_ext.post("/api/v1/paywalls") +async def api_paywall_create(data: CreatePaywall, wallet: WalletTypeInfo = Depends(get_key_type)): + paywall = await create_paywall(wallet_id=wallet.wallet.id, data=data) + return paywall.dict() -@paywall_ext.route("/api/v1/paywalls/", methods=["DELETE"]) -@api_check_wallet_key("invoice") -async def api_paywall_delete(paywall_id): +@paywall_ext.delete("/api/v1/paywalls/{paywall_id}") +async def api_paywall_delete(paywall_id, wallet: WalletTypeInfo = Depends(get_key_type)): paywall = await get_paywall(paywall_id) if not paywall: - return jsonify({"message": "Paywall does not exist."}), HTTPStatus.NOT_FOUND + raise HTTPException( + status_code=HTTPStatus.NOT_FOUND, + detail="Paywall does not exist." + ) - if paywall.wallet != g.wallet.id: - return jsonify({"message": "Not your paywall."}), HTTPStatus.FORBIDDEN + if paywall.wallet != wallet.wallet.id: + raise HTTPException( + status_code=HTTPStatus.FORBIDDEN, + detail="Not your paywall." + ) await delete_paywall(paywall_id) - - return "", HTTPStatus.NO_CONTENT + raise HTTPException(status_code=HTTPStatus.NO_CONTENT) -@paywall_ext.route("/api/v1/paywalls//invoice", methods=["POST"]) -@api_validate_post_request( - schema={"amount": {"type": "integer", "min": 1, "required": True}} -) -async def api_paywall_create_invoice(paywall_id): +@paywall_ext.post("/api/v1/paywalls/{paywall_id}/invoice") +async def api_paywall_create_invoice(paywall_id, data: CreatePaywallInvoice, wallet: WalletTypeInfo = Depends(get_key_type)): paywall = await get_paywall(paywall_id) - - if g.data["amount"] < paywall.amount: - return ( - jsonify({"message": f"Minimum amount is {paywall.amount} sat."}), - HTTPStatus.BAD_REQUEST, - ) + print("PAYW", paywall) + print("DATA", data) + + if data.amount < paywall.amount: + raise HTTPException( + status_code=HTTPStatus.BAD_REQUEST, + detail=f"Minimum amount is {paywall.amount} sat." + ) try: amount = ( - g.data["amount"] if g.data["amount"] > paywall.amount else paywall.amount + data.amount if data.amount > paywall.amount else paywall.amount ) payment_hash, payment_request = await create_invoice( wallet_id=paywall.wallet, @@ -84,38 +71,35 @@ async def api_paywall_create_invoice(paywall_id): extra={"tag": "paywall"}, ) except Exception as e: - return jsonify({"message": str(e)}), HTTPStatus.INTERNAL_SERVER_ERROR + raise HTTPException( + status_code=HTTPStatus.INTERNAL_SERVER_ERROR, + detail=str(e) + ) - return ( - jsonify({"payment_hash": payment_hash, "payment_request": payment_request}), - HTTPStatus.CREATED, - ) + return {"payment_hash": payment_hash, "payment_request": payment_request} -@paywall_ext.route("/api/v1/paywalls//check_invoice", methods=["POST"]) -@api_validate_post_request( - schema={"payment_hash": {"type": "string", "empty": False, "required": True}} -) -async def api_paywal_check_invoice(paywall_id): +@paywall_ext.post("/api/v1/paywalls/{paywall_id}/check_invoice") +async def api_paywal_check_invoice(data: CheckPaywallInvoice, paywall_id): paywall = await get_paywall(paywall_id) - + payment_hash = data.payment_hash if not paywall: - return jsonify({"message": "Paywall does not exist."}), HTTPStatus.NOT_FOUND + raise HTTPException( + status_code=HTTPStatus.NOT_FOUND, + detail="Paywall does not exist." + ) try: - status = await check_invoice_status(paywall.wallet, g.data["payment_hash"]) + status = await check_invoice_status(paywall.wallet, payment_hash) is_paid = not status.pending except Exception: - return jsonify({"paid": False}), HTTPStatus.OK + return {"paid": False} if is_paid: wallet = await get_wallet(paywall.wallet) - payment = await wallet.get_payment(g.data["payment_hash"]) + payment = await wallet.get_payment(payment_hash) await payment.set_pending(False) - return ( - jsonify({"paid": True, "url": paywall.url, "remembers": paywall.remembers}), - HTTPStatus.OK, - ) + return {"paid": True, "url": paywall.url, "remembers": paywall.remembers} - return jsonify({"paid": False}), HTTPStatus.OK + return {"paid": False}