import re from quart import g, jsonify, request from http import HTTPStatus from lnbits.core.crud import get_user, get_wallet from lnbits.core.services import create_invoice, check_invoice_status from lnbits.decorators import api_check_wallet_key, api_validate_post_request from lnbits.extensions.lnticket import lnticket_ext from .crud import ( create_ticket, update_ticket, get_ticket, get_tickets, delete_ticket, create_form, update_form, get_form, get_forms, delete_form, ) #########FORMS########## @lnticket_ext.route("/api/v1/forms", methods=["GET"]) @api_check_wallet_key("invoice") async def api_forms(): wallet_ids = [g.wallet.id] if "all_wallets" in request.args: wallet_ids = get_user(g.wallet.user).wallet_ids return jsonify([form._asdict() for form in get_forms(wallet_ids)]), HTTPStatus.OK @lnticket_ext.route("/api/v1/forms", methods=["POST"]) @lnticket_ext.route("/api/v1/forms/", methods=["PUT"]) @api_check_wallet_key("invoice") @api_validate_post_request( schema={ "wallet": {"type": "string", "empty": False, "required": True}, "name": {"type": "string", "empty": False, "required": True}, "description": {"type": "string", "min": 0, "required": True}, "costpword": {"type": "integer", "min": 0, "required": True}, } ) async def api_form_create(form_id=None): if form_id: form = get_form(form_id) if not form: return jsonify({"message": "Form does not exist."}), HTTPStatus.NOT_FOUND if form.wallet != g.wallet.id: return jsonify({"message": "Not your form."}), HTTPStatus.FORBIDDEN form = update_form(form_id, **g.data) else: form = create_form(**g.data) return jsonify(form._asdict()), HTTPStatus.CREATED @lnticket_ext.route("/api/v1/forms/", methods=["DELETE"]) @api_check_wallet_key("invoice") async def api_form_delete(form_id): form = get_form(form_id) if not form: return jsonify({"message": "Form does not exist."}), HTTPStatus.NOT_FOUND if form.wallet != g.wallet.id: return jsonify({"message": "Not your form."}), HTTPStatus.FORBIDDEN delete_form(form_id) return "", HTTPStatus.NO_CONTENT #########tickets########## @lnticket_ext.route("/api/v1/tickets", methods=["GET"]) @api_check_wallet_key("invoice") async def api_tickets(): wallet_ids = [g.wallet.id] if "all_wallets" in request.args: wallet_ids = get_user(g.wallet.user).wallet_ids return jsonify([form._asdict() for form in get_tickets(wallet_ids)]), HTTPStatus.OK @lnticket_ext.route("/api/v1/tickets/", methods=["POST"]) @api_validate_post_request( schema={ "form": {"type": "string", "empty": False, "required": True}, "name": {"type": "string", "empty": False, "required": True}, "email": {"type": "string", "empty": True, "required": True}, "ltext": {"type": "string", "empty": False, "required": True}, } ) async def api_ticket_make_ticket(form_id): form = get_form(form_id) if not form: return jsonify({"message": "LNTicket does not exist."}), HTTPStatus.NOT_FOUND try: nwords = len(re.split(r"\s+", g.data["ltext"])) sats = nwords * form.costpword payment_hash, payment_request = create_invoice( wallet_id=form.wallet, amount=sats, memo=f"ticket with {nwords} words on {form_id}", extra={"tag": "lnticket"}, ) except Exception as e: return jsonify({"message": str(e)}), HTTPStatus.INTERNAL_SERVER_ERROR ticket = create_ticket(payment_hash=payment_hash, wallet=form.wallet, sats=sats, **g.data) if not ticket: return jsonify({"message": "LNTicket could not be fetched."}), HTTPStatus.NOT_FOUND return jsonify({"payment_hash": payment_hash, "payment_request": payment_request}), HTTPStatus.OK @lnticket_ext.route("/api/v1/tickets/", methods=["GET"]) async def api_ticket_send_ticket(payment_hash): ticket = get_ticket(payment_hash) try: is_paid = not check_invoice_status(ticket.wallet, payment_hash).pending except Exception: return jsonify({"message": "Not paid."}), HTTPStatus.NOT_FOUND if is_paid: wallet = get_wallet(ticket.wallet) payment = wallet.get_payment(payment_hash) payment.set_pending(False) ticket = update_ticket(paid=True, payment_hash=payment_hash) return jsonify({"paid": True, "ticket_id": ticket.id}), HTTPStatus.OK return jsonify({"paid": False}), HTTPStatus.OK @lnticket_ext.route("/api/v1/tickets/", methods=["DELETE"]) @api_check_wallet_key("invoice") async def api_ticket_delete(ticket_id): ticket = get_ticket(ticket_id) if not ticket: return jsonify({"message": "Paywall does not exist."}), HTTPStatus.NOT_FOUND if ticket.wallet != g.wallet.id: return jsonify({"message": "Not your ticket."}), HTTPStatus.FORBIDDEN delete_ticket(ticket_id) return "", HTTPStatus.NO_CONTENT