146 lines
4.4 KiB
Python
146 lines
4.4 KiB
Python
import io
|
|
from http import HTTPStatus
|
|
|
|
from fastapi import APIRouter, Depends, HTTPException, Request
|
|
from fastapi.responses import HTMLResponse, StreamingResponse
|
|
from lnbits.core.models import User
|
|
from lnbits.decorators import check_user_exists
|
|
from lnbits.helpers import template_renderer
|
|
|
|
from .crud import chunks, get_withdraw_link
|
|
from .helpers import create_lnurl
|
|
|
|
withdraw_ext_generic = APIRouter()
|
|
|
|
|
|
def withdraw_renderer():
|
|
return template_renderer(["withdraw/templates"])
|
|
|
|
|
|
@withdraw_ext_generic.get("/", response_class=HTMLResponse)
|
|
async def index(request: Request, user: User = Depends(check_user_exists)):
|
|
return withdraw_renderer().TemplateResponse(
|
|
"withdraw/index.html", {"request": request, "user": user.json()}
|
|
)
|
|
|
|
|
|
@withdraw_ext_generic.get("/{link_id}", response_class=HTMLResponse)
|
|
async def display(request: Request, link_id):
|
|
link = await get_withdraw_link(link_id, 0)
|
|
|
|
if not link:
|
|
raise HTTPException(
|
|
status_code=HTTPStatus.NOT_FOUND, detail="Withdraw link does not exist."
|
|
)
|
|
|
|
try:
|
|
lnurl = create_lnurl(link, request)
|
|
except ValueError as exc:
|
|
raise HTTPException(
|
|
status_code=HTTPStatus.INTERNAL_SERVER_ERROR,
|
|
detail=str(exc),
|
|
) from exc
|
|
|
|
return withdraw_renderer().TemplateResponse(
|
|
"withdraw/display.html",
|
|
{
|
|
"request": request,
|
|
"spent": link.is_spent,
|
|
"lnurl_url": str(lnurl.url),
|
|
"enabled": link.enabled,
|
|
},
|
|
)
|
|
|
|
|
|
@withdraw_ext_generic.get("/print/{link_id}", response_class=HTMLResponse)
|
|
async def print_qr(request: Request, link_id):
|
|
link = await get_withdraw_link(link_id)
|
|
if not link:
|
|
raise HTTPException(
|
|
status_code=HTTPStatus.NOT_FOUND, detail="Withdraw link does not exist."
|
|
)
|
|
|
|
if link.uses == 0:
|
|
return withdraw_renderer().TemplateResponse(
|
|
"withdraw/print_qr.html",
|
|
{"request": request, "link": link.json(), "unique": False},
|
|
)
|
|
links = []
|
|
count = 0
|
|
|
|
for _ in link.usescsv.split(","):
|
|
linkk = await get_withdraw_link(link_id, count)
|
|
if not linkk:
|
|
raise HTTPException(
|
|
status_code=HTTPStatus.NOT_FOUND, detail="Withdraw link does not exist."
|
|
)
|
|
try:
|
|
lnurl = create_lnurl(linkk, request)
|
|
except ValueError as exc:
|
|
raise HTTPException(
|
|
status_code=HTTPStatus.INTERNAL_SERVER_ERROR,
|
|
detail=str(exc),
|
|
) from exc
|
|
links.append(str(lnurl.bech32))
|
|
count = count + 1
|
|
page_link = list(chunks(links, 2))
|
|
linked = list(chunks(page_link, 5))
|
|
|
|
if link.custom_url:
|
|
return withdraw_renderer().TemplateResponse(
|
|
"withdraw/print_qr_custom.html",
|
|
{
|
|
"request": request,
|
|
"link": page_link,
|
|
"unique": True,
|
|
"custom_url": link.custom_url,
|
|
"amt": link.max_withdrawable,
|
|
},
|
|
)
|
|
|
|
return withdraw_renderer().TemplateResponse(
|
|
"withdraw/print_qr.html", {"request": request, "link": linked, "unique": True}
|
|
)
|
|
|
|
|
|
@withdraw_ext_generic.get("/csv/{link_id}", response_class=HTMLResponse)
|
|
async def csv(request: Request, link_id):
|
|
link = await get_withdraw_link(link_id)
|
|
if not link:
|
|
raise HTTPException(
|
|
status_code=HTTPStatus.NOT_FOUND, detail="Withdraw link does not exist."
|
|
)
|
|
|
|
if link.uses == 0:
|
|
raise HTTPException(
|
|
status_code=HTTPStatus.BAD_REQUEST, detail="Withdraw is spent."
|
|
)
|
|
|
|
buffer = io.StringIO()
|
|
count = 0
|
|
for _ in link.usescsv.split(","):
|
|
linkk = await get_withdraw_link(link_id, count)
|
|
if not linkk:
|
|
raise HTTPException(
|
|
status_code=HTTPStatus.NOT_FOUND, detail="Withdraw link does not exist."
|
|
)
|
|
try:
|
|
lnurl = create_lnurl(linkk, request)
|
|
except ValueError as exc:
|
|
raise HTTPException(
|
|
status_code=HTTPStatus.INTERNAL_SERVER_ERROR,
|
|
detail=str(exc),
|
|
) from exc
|
|
buffer.write(f"{lnurl.bech32!s}\n")
|
|
count += 1
|
|
|
|
# Move buffer cursor to the beginning
|
|
buffer.seek(0)
|
|
|
|
return StreamingResponse(
|
|
buffer,
|
|
media_type="text/csv",
|
|
headers={
|
|
"Content-Disposition": f"attachment; filename=withdraw-links-{link_id}.csv"
|
|
},
|
|
)
|