views.py

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 lnbits.settings import settings

from . import tpos_ext, tpos_renderer
from .crud import get_clean_tpos, get_tpos

templates = Jinja2Templates(directory="templates")


@tpos_ext.get("/", response_class=HTMLResponse)
async def index(request: Request, user: User = Depends(check_user_exists)):
    return tpos_renderer().TemplateResponse(
        "tpos/index.html", {"request": request, "user": user.dict()}
    )


@tpos_ext.get("/{tpos_id}")
async def tpos(request: Request, tpos_id):
    tpos = await get_clean_tpos(tpos_id)
    if not tpos:
        raise HTTPException(
            status_code=HTTPStatus.NOT_FOUND, detail="TPoS does not exist."
        )
    withdrawpinopen = 0
    if tpos.withdrawpindisabled:
        tposDirty = await get_tpos(tpos_id)
        if not tposDirty:
            raise HTTPException(
                status_code=HTTPStatus.NOT_FOUND, detail="TPoS does not exist."
            )
        withdrawpinopen = tposDirty.withdrawpin
    return tpos_renderer().TemplateResponse(
        "tpos/tpos.html",
        {
            "request": request,
            "tpos": tpos,
            "withdrawpinopen": withdrawpinopen,
            "withdrawamtposs": tpos.withdrawamtposs,
            "web_manifest": f"/tpos/manifest/{tpos_id}.webmanifest",
        },
    )


@tpos_ext.get("/manifest/{tpos_id}.webmanifest")
async def manifest(tpos_id: str):
    tpos = await get_clean_tpos(tpos_id)
    if not tpos:
        raise HTTPException(
            status_code=HTTPStatus.NOT_FOUND, detail="TPoS does not exist."
        )

    return {
        "short_name": settings.lnbits_site_title,
        "name": tpos.name + " - " + settings.lnbits_site_title,
        "icons": [
            {
                "src": settings.lnbits_custom_logo
                if settings.lnbits_custom_logo
                else "https://cdn.jsdelivr.net/gh/lnbits/lnbits@0.3.0/docs/logos/lnbits.png",
                "type": "image/png",
                "sizes": "900x900",
            }
        ],
        "start_url": "/tpos/" + tpos_id,
        "background_color": "#1F2234",
        "description": "Bitcoin Lightning tPOS",
        "display": "standalone",
        "scope": "/tpos/" + tpos_id,
        "theme_color": "#1F2234",
        "shortcuts": [
            {
                "name": tpos.name + " - " + settings.lnbits_site_title,
                "short_name": tpos.name,
                "description": tpos.name + " - " + settings.lnbits_site_title,
                "url": "/tpos/" + tpos_id,
            }
        ],
    }