livestream extension.

(except docs)

lnbits/extensions/livestream/models.py

@@ -0,0 +1,57 @@
+import json
+from quart import url_for
+from typing import NamedTuple, Optional
+from lnurl import Lnurl, encode as lnurl_encode  # type: ignore
+from lnurl.types import LnurlPayMetadata  # type: ignore
+from lnurl.models import LnurlPaySuccessAction, UrlAction  # type: ignore
+
+
+class Livestream(NamedTuple):
+    id: int
+    wallet: str
+    fee_pct: int
+    current_track: Optional[int]
+
+    @property
+    def lnurl(self) -> Lnurl:
+        url = url_for("livestream.lnurl_response", ls_id=self.id, _external=True)
+        return lnurl_encode(url)
+
+
+class Track(NamedTuple):
+    id: int
+    download_url: str
+    price_msat: int
+    name: str
+    producer: int
+
+    async def description(self) -> str:
+        from .crud import get_producer
+
+        producer = await get_producer(self.producer)
+        if producer:
+            producer_name = producer.name
+        else:
+            producer_name = "unknown author"
+
+        return f"Track '{self.name}', from {producer_name}."
+
+    @property
+    async def lnurlpay_metadata(self) -> LnurlPayMetadata:
+        return LnurlPayMetadata(json.dumps([["text/plain", await self.description()]]))
+
+    def success_action(self, payment_hash: str) -> Optional[LnurlPaySuccessAction]:
+        if not self.download_url:
+            return None
+
+        return UrlAction(
+            url=url_for("livestream.track_redirect_download", track_id=self.id, p=payment_hash, _external=True),
+            description=f"Download the track {self.name}!",
+        )
+
+
+class Producer(NamedTuple):
+    id: int
+    user: str
+    wallet: str
+    name: str