from http import HTTPStatus from fastapi.param_functions import Depends from fastapi.templating import Jinja2Templates from starlette.exceptions import HTTPException from starlette.requests import Request from starlette.responses import HTMLResponse from lnbits.core.models import User from lnbits.decorators import check_user_exists from lnbits.extensions.satspay.helpers import public_charge from . import satspay_ext, satspay_renderer from .crud import get_charge templates = Jinja2Templates(directory="templates") @satspay_ext.get("/", response_class=HTMLResponse) async def index(request: Request, user: User = Depends(check_user_exists)): return satspay_renderer().TemplateResponse( "satspay/index.html", {"request": request, "user": user.dict()} ) @satspay_ext.get("/{charge_id}", response_class=HTMLResponse) async def display(request: Request, charge_id: str): charge = await get_charge(charge_id) if not charge: raise HTTPException( status_code=HTTPStatus.NOT_FOUND, detail="Charge link does not exist." ) view_data = { "request": request, "charge_data": public_charge(charge), } if "mempool_endpoint" in charge.config: view_data["mempool_endpoint"] = charge.config["mempool_endpoint"] view_data["network"] = charge.config["network"] return satspay_renderer().TemplateResponse( "satspay/display.html", view_data, )