livestream extension.

(except docs)
2021-04-11 00:44:55 -03:00
parent 10671e5732
commit 76199c66a1
16 changed files with 1086 additions and 4 deletions
@@ -0,0 +1,99 @@
+from quart import g, jsonify
+from http import HTTPStatus
+from lnurl.exceptions import InvalidUrl as LnurlInvalidUrl  # type: ignore
+
+from lnbits.decorators import api_check_wallet_key, api_validate_post_request
+
+from . import livestream_ext
+from .crud import (
+    get_or_create_livestream_by_wallet,
+    add_track,
+    get_tracks,
+    get_producers,
+    update_livestream_fee,
+    update_current_track,
+    delete_track_from_livestream,
+)
+
+
+@livestream_ext.route("/api/v1/livestream", methods=["GET"])
+@api_check_wallet_key("invoice")
+async def api_livestream_from_wallet():
+    ls = await get_or_create_livestream_by_wallet(g.wallet.id)
+    tracks = await get_tracks(ls.id)
+    producers = await get_producers(ls.id)
+
+    try:
+        return (
+            jsonify(
+                {
+                    **ls._asdict(),
+                    **{
+                        "lnurl": ls.lnurl,
+                        "tracks": [track._asdict() for track in tracks],
+                        "producers": [producer._asdict() for producer in producers],
+                    },
+                }
+            ),
+            HTTPStatus.OK,
+        )
+    except LnurlInvalidUrl:
+        return (
+            jsonify({"message": "LNURLs need to be delivered over a publically accessible `https` domain or Tor."}),
+            HTTPStatus.UPGRADE_REQUIRED,
+        )
+
+
+@livestream_ext.route("/api/v1/livestream/track/<track_id>", methods=["PUT"])
+@api_check_wallet_key("invoice")
+async def api_update_track(track_id):
+    try:
+        id = int(track_id)
+    except ValueError:
+        id = 0
+    if id <= 0:
+        id = None
+
+    ls = await get_or_create_livestream_by_wallet(g.wallet.id)
+    await update_current_track(ls.id, id)
+    return "", HTTPStatus.NO_CONTENT
+
+
+@livestream_ext.route("/api/v1/livestream/fee/<fee_pct>", methods=["PUT"])
+@api_check_wallet_key("invoice")
+async def api_update_fee(fee_pct):
+    ls = await get_or_create_livestream_by_wallet(g.wallet.id)
+    await update_livestream_fee(ls.id, int(fee_pct))
+    return "", HTTPStatus.NO_CONTENT
+
+
+@livestream_ext.route("/api/v1/livestream/tracks", methods=["POST"])
+@api_check_wallet_key("invoice")
+@api_validate_post_request(
+    schema={
+        "name": {"type": "string", "empty": False, "required": True},
+        "download_url": {"type": "string", "empty": False, "required": False},
+        "price_msat": {"type": "number", "min": 1, "required": True},
+        "producer_id": {"type": "number", "required": True, "excludes": "producer_name"},
+        "producer_name": {"type": "string", "required": True, "excludes": "producer_id"},
+    }
+)
+async def api_add_track():
+    ls = await get_or_create_livestream_by_wallet(g.wallet.id)
+    await add_track(
+        ls.id,
+        g.data["name"],
+        g.data.get("download_url"),
+        g.data["price_msat"],
+        g.data.get("producer_name"),
+        g.data.get("producer_id"),
+    )
+    return "", HTTPStatus.CREATED
+
+
+@livestream_ext.route("/api/v1/livestream/tracks/<track_id>", methods=["DELETE"])
+@api_check_wallet_key("invoice")
+async def api_delete_track(track_id):
+    ls = await get_or_create_livestream_by_wallet(g.wallet.id)
+    await delete_track_from_livestream(ls.id, track_id)
+    return "", HTTPStatus.NO_CONTENT