lnbits-legend/lnbits/extensions/livestream/lnurl.py

import hashlib
import math
from quart import jsonify, url_for, request
from lnurl import LnurlPayResponse, LnurlPayActionResponse, LnurlErrorResponse  # type: ignore

from lnbits.core.services import create_invoice

from . import livestream_ext
from .crud import get_livestream, get_livestream_by_track, get_track


@livestream_ext.route("/lnurl/<ls_id>", methods=["GET"])
async def lnurl_livestream(ls_id):
    ls = await get_livestream(ls_id)
    if not ls:
        return jsonify({"status": "ERROR", "reason": "Livestream not found."})

    track = await get_track(ls.current_track)
    if not track:
        return jsonify({"status": "ERROR", "reason": "This livestream is offline."})

    resp = LnurlPayResponse(
        callback=url_for(
            "livestream.lnurl_callback", track_id=track.id, _external=True
        ),
        min_sendable=track.min_sendable,
        max_sendable=track.max_sendable,
        metadata=await track.lnurlpay_metadata(),
    )

    params = resp.dict()
    params["commentAllowed"] = 300

    return jsonify(params)


@livestream_ext.route("/lnurl/t/<track_id>", methods=["GET"])
async def lnurl_track(track_id):
    track = await get_track(track_id)
    if not track:
        return jsonify({"status": "ERROR", "reason": "Track not found."})

    resp = LnurlPayResponse(
        callback=url_for(
            "livestream.lnurl_callback", track_id=track.id, _external=True
        ),
        min_sendable=track.min_sendable,
        max_sendable=track.max_sendable,
        metadata=await track.lnurlpay_metadata(),
    )

    params = resp.dict()
    params["commentAllowed"] = 300

    return jsonify(params)


@livestream_ext.route("/lnurl/cb/<track_id>", methods=["GET"])
async def lnurl_callback(track_id):
    track = await get_track(track_id)
    if not track:
        return jsonify({"status": "ERROR", "reason": "Couldn't find track."})

    amount_received = int(request.args.get("amount"))

    if amount_received < track.min_sendable:
        return (
            jsonify(
                LnurlErrorResponse(
                    reason=f"Amount {round(amount_received / 1000)} is smaller than minimum {math.floor(track.min_sendable)}."
                ).dict()
            ),
        )
    elif track.max_sendable < amount_received:
        return (
            jsonify(
                LnurlErrorResponse(
                    reason=f"Amount {round(amount_received / 1000)} is greater than maximum {math.floor(track.max_sendable)}."
                ).dict()
            ),
        )

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

    ls = await get_livestream_by_track(track_id)

    payment_hash, payment_request = await create_invoice(
        wallet_id=ls.wallet,
        amount=int(amount_received / 1000),
        memo=await track.fullname(),
        description_hash=hashlib.sha256(
            (await track.lnurlpay_metadata()).encode("utf-8")
        ).digest(),
        extra={"tag": "livestream", "track": track.id, "comment": comment},
    )

    if amount_received < track.price_msat:
        success_action = None
    else:
        success_action = track.success_action(payment_hash)

    resp = LnurlPayActionResponse(
        pr=payment_request,
        success_action=success_action,
        routes=[],
    )

    return jsonify(resp.dict())