61 lines
2 KiB
Python
61 lines
2 KiB
Python
from http import HTTPStatus
|
|
|
|
from fastapi import APIRouter, Depends, HTTPException, Request
|
|
from fastapi.responses import HTMLResponse
|
|
from lnbits.core.crud import get_wallet
|
|
from lnbits.core.models import User
|
|
from lnbits.decorators import check_user_exists, optional_user_id
|
|
from lnbits.helpers import template_renderer
|
|
|
|
from .crud import get_card_by_external_id, get_hits, get_refunds
|
|
|
|
boltcards_generic_router = APIRouter()
|
|
|
|
|
|
def boltcards_renderer():
|
|
return template_renderer(["boltcards/templates"])
|
|
|
|
|
|
@boltcards_generic_router.get("/", response_class=HTMLResponse)
|
|
async def index(request: Request, user: User = Depends(check_user_exists)):
|
|
return boltcards_renderer().TemplateResponse(
|
|
"boltcards/index.html", {"request": request, "user": user.json()}
|
|
)
|
|
|
|
|
|
@boltcards_generic_router.get("/{card_id}", response_class=HTMLResponse)
|
|
async def display(
|
|
request: Request, card_id: str, user_id: str | None = Depends(optional_user_id)
|
|
):
|
|
if not user_id:
|
|
raise HTTPException(
|
|
status_code=HTTPStatus.UNAUTHORIZED, detail="User not authorized."
|
|
)
|
|
card = await get_card_by_external_id(card_id)
|
|
if not card:
|
|
raise HTTPException(
|
|
status_code=HTTPStatus.NOT_FOUND, detail="Card does not exist."
|
|
)
|
|
|
|
wallet = await get_wallet(card.wallet)
|
|
wallet_balance = 0
|
|
if wallet:
|
|
if wallet.user != user_id:
|
|
raise HTTPException(
|
|
status_code=HTTPStatus.FORBIDDEN, detail="Card does not belong to user."
|
|
)
|
|
wallet_balance = wallet.balance
|
|
hits = await get_hits([card.id])
|
|
hits_json = [hit.json() for hit in hits]
|
|
refunds = [refund.hit_id for refund in await get_refunds([hit.id for hit in hits])]
|
|
card_json = card.json(exclude={"wallet"})
|
|
return boltcards_renderer().TemplateResponse(
|
|
"boltcards/display.html",
|
|
{
|
|
"request": request,
|
|
"card": card_json,
|
|
"hits": hits_json,
|
|
"refunds": refunds,
|
|
"balance": int(wallet_balance),
|
|
},
|
|
)
|