lnurl.py

from http import HTTPStatus

from fastapi import Request, Query
from lnurl import LnurlErrorResponse, LnurlPayActionResponse, LnurlPayResponse
from starlette.exceptions import HTTPException

from lnbits.core.services import create_invoice
from lnbits.utils.exchange_rates import get_fiat_rate_satoshis

from . import lnurlp_ext
from .crud import increment_pay_link, get_pay_link, get_address_data
from loguru import logger
from urllib.parse import urlparse
import json
from . import nostr_publickey


@lnurlp_ext.get(
    "/api/v1/lnurl/cb/lnaddr/{link_id}",
    status_code=HTTPStatus.OK,
    name="lnurlp.api_lnurl_lnaddr_callback",
)
async def api_lnurl_lnaddr_callback(
    request: Request, link_id, amount: int = Query(...)
):
    return await api_lnurl_callback(request, link_id, amount, lnaddress=True)


@lnurlp_ext.get(
    "/api/v1/lnurl/cb/{link_id}",
    status_code=HTTPStatus.OK,
    name="lnurlp.api_lnurl_callback",
)
async def api_lnurl_callback(
    request: Request, link_id, amount: int = Query(...), lnaddress=False
):
    link = await increment_pay_link(link_id, served_pr=1)
    if not link:
        raise HTTPException(
            status_code=HTTPStatus.NOT_FOUND, detail="Pay link does not exist."
        )
    min, max = link.min, link.max
    rate = await get_fiat_rate_satoshis(link.currency) if link.currency else 1
    if link.currency:
        # allow some fluctuation (as the fiat price may have changed between the calls)
        min = rate * 995 * link.min
        max = rate * 1010 * link.max
    else:
        min = link.min * 1000
        max = link.max * 1000

    amount = amount
    if amount < min:
        return LnurlErrorResponse(
            reason=f"Amount {amount} is smaller than minimum {min}."
        ).dict()

    elif amount > max:
        return LnurlErrorResponse(
            reason=f"Amount {amount} is greater than maximum {max}."
        ).dict()

    comment = request.query_params.get("comment")
    if len(comment or "") > link.comment_chars:
        return LnurlErrorResponse(
            reason=f"Got a comment with {len(comment)} characters, but can only accept {link.comment_chars}"
        ).dict()

    if lnaddress:
        # for lnaddress, we have to set this otherwise the metadata won't have the identifier
        link.domain = urlparse(str(request.url)).netloc

    extra = {
        "tag": "lnurlp",
        "link": link.id,
        "extra": request.query_params.get("amount"),
    }

    if comment:
        extra["comment"] = (comment,)
        #extra["comment"] = comment

    # nip 57
    nostr = request.query_params.get("nostr")
    if nostr:
        extra["nostr"] = nostr  # put it here for later publishing in tasks.py

    if lnaddress and link.username and link.domain:
        extra["lnaddress"] = f"{link.username}@{link.domain}"

    payment_hash, payment_request = await create_invoice(
        wallet_id=link.wallet,
        amount=int(amount / 1000),
        memo=link.description,
        unhashed_description=nostr.encode()
        if nostr  # we take the zap request as the description instead of the LNURL metadata if present
        else link.lnurlpay_metadata.encode(),
        extra=extra,
    )

    success_action = link.success_action(payment_hash)
    if success_action:
        resp = LnurlPayActionResponse(
            pr=payment_request, success_action=success_action, routes=[]  # type: ignore
        )
    else:
        resp = LnurlPayActionResponse(pr=payment_request, routes=[])  # type: ignore

    logger.info(f"api_lnurl_callback: resp({resp.dict()})")

    return resp.dict()


@lnurlp_ext.get(
    "/api/v1/lnurl/{link_id}",  # Backwards compatibility for old LNURLs / QR codes (with long URL)
    status_code=HTTPStatus.OK,
    name="lnurlp.api_lnurl_response.deprecated",
)
@lnurlp_ext.get(
    "/{link_id}",
    status_code=HTTPStatus.OK,
    name="lnurlp.api_lnurl_response",
)
async def api_lnurl_response(request: Request, link_id, lnaddress=False):
    link = await increment_pay_link(link_id, served_meta=1)
    if not link:
        raise HTTPException(
            status_code=HTTPStatus.NOT_FOUND, detail="Pay link does not exist."
        )

    rate = await get_fiat_rate_satoshis(link.currency) if link.currency else 1

    if lnaddress:
        # for lnaddress, we have to set this otherwise the metadata won't have the identifier
        link.domain = urlparse(str(request.url)).netloc
        callback = str(request.url_for("lnurlp.api_lnurl_lnaddr_callback", link_id=link.id))
    else:
        callback = str(request.url_for("lnurlp.api_lnurl_callback", link_id=link.id))

    resp = LnurlPayResponse(
        callback=callback,
        min_sendable=round(link.min * rate) * 1000,  # type: ignore
        max_sendable=round(link.max * rate) * 1000,  # type: ignore
        metadata=link.lnurlpay_metadata,
    )
    params = resp.dict()

    if link.comment_chars > 0:
        params["commentAllowed"] = link.comment_chars

    if link.zaps:
        params["allowsNostr"] = True
        params["nostrPubkey"] = nostr_publickey.hex()
    return params