from http import HTTPStatus from fastapi import Depends, Request from fastapi.templating import Jinja2Templates from starlette.exceptions import HTTPException from starlette.responses import HTMLResponse from lnbits.core.models import User from lnbits.decorators import check_user_exists from . import subdomains_ext, subdomains_renderer from .crud import get_domain templates = Jinja2Templates(directory="templates") @subdomains_ext.get("/", response_class=HTMLResponse) async def index( request: Request, user: User = Depends(check_user_exists) # type:ignore ): return subdomains_renderer().TemplateResponse( "subdomains/index.html", {"request": request, "user": user.dict()} ) @subdomains_ext.get("/{domain_id}") async def display(request: Request, domain_id): domain = await get_domain(domain_id) if not domain: raise HTTPException( status_code=HTTPStatus.NOT_FOUND, detail="Domain does not exist." ) allowed_records = ( domain.allowed_record_types.replace('"', "").replace(" ", "").split(",") ) return subdomains_renderer().TemplateResponse( "subdomains/display.html", { "request": request, "domain_id": domain.id, "domain_domain": domain.domain, "domain_desc": domain.description, "domain_cost": domain.cost, "domain_allowed_record_types": allowed_records, }, )