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/", 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/", 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/", 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())