views_lnurl.py

import random
from http import HTTPStatus

from fastapi import APIRouter, Query, Request
from lnbits.core.services import create_invoice

from .crud import get_eightball

eightball_lnurl_router = APIRouter()


@eightball_lnurl_router.get(
    "/api/v1/lnurl/pay/{eightball_id}",
    status_code=HTTPStatus.OK,
    name="eightball.api_lnurl_pay",
)
async def api_lnurl_pay(
    request: Request,
    eightball_id: str,
):
    eightball = await get_eightball(eightball_id)
    if not eightball:
        return {"status": "ERROR", "reason": "No eightball found"}
    if not eightball.lnurlpayamount:
        return {"status": "ERROR", "reason": "Eightball has no lnurlpayamount set"}
    return {
        "callback": str(
            request.url_for(
                "eightball.api_lnurl_pay_callback", eightball_id=eightball_id
            )
        ),
        "maxSendable": eightball.lnurlpayamount * 1000,
        "minSendable": eightball.lnurlpayamount * 1000,
        "metadata": f"""[["text/plain", "{eightball.name}"]]""",
        "tag": "payRequest",
    }


@eightball_lnurl_router.get(
    "/api/v1/lnurl/paycb/{eightball_id}",
    status_code=HTTPStatus.OK,
    name="eightball.api_lnurl_pay_callback",
)
async def api_lnurl_pay_cb(
    request: Request,
    eightball_id: str,
    amount: int = Query(...),
):
    eightball = await get_eightball(eightball_id)
    if not eightball:
        return {"status": "ERROR", "reason": "No eightball found"}
    payment = await create_invoice(
        wallet_id=eightball.wallet,
        amount=int(amount / 1000),
        memo=eightball.name,
        unhashed_description=f'[["text/plain", "{eightball.name}"]]'.encode(),
        extra={
            "tag": "EightBall",
            "eightballId": eightball_id,
            "extra": request.query_params.get("amount"),
        },
    )
    random_word = random.choice(eightball.wordlist.split("\n"))
    return {
        "pr": payment.bolt11,
        "routes": [],
        "successAction": {"tag": "message", "message": random_word},
    }