add download support

backend/config.py

@@ -45,11 +45,13 @@ proxy_server = os.environ.get("PROXY_SERVER")
 @dataclasses.dataclass
 class LinkWithType:
     upstream: str
+    download: bool = False
     ctype: None | str = None
 
 
 class ProxyCreateLink(pydantic.BaseModel):
     upstream: None | pydantic.HttpUrl
+    download: None | pydantic.StrictBool
     ctype: None | pydantic.StrictStr
     region: None | pydantic.StrictStr
 
@@ -91,6 +93,7 @@ class ProxyElem:
                         {
                             "upstream": url.upstream,
                             "ctype": url.ctype,
+                            "download": url.download,
                             "region": self.region,
                         }
                     )

backend/stream.py

@@ -2,13 +2,14 @@
 import json
 import sys
 import base64
+import dataclasses
+from typing import Optional, cast, Any
 import logging
 import asyncio
 import tornado.web
 import tornado.routing
 import config
 import stream_providers
-from typing import Optional, cast, Any
 
 logging.basicConfig(
     format="[%(filename)s:%(lineno)d] %(message)s",
@@ -85,6 +86,14 @@ class UpstreamHandler:
                             future.cancel()
 
 
+@dataclasses.dataclass
+class VideoInfo:
+    upstream: None | str = None
+    download: None | str = None
+    poster: None | str = None
+    ctype: None | str = None
+
+
 class NoDataError(Exception):
     pass
 
@@ -127,10 +136,11 @@ class MainHandler(tornado.web.RequestHandler):
         if provider_data is None:
             raise NoDataError()
 
-        video_info = {}
+        video_info = VideoInfo()
+        video_info.ctype = provider_data.ctype()
         if handler.direct:
-            video_info["upstream"] = provider_data.upstream()
-            video_info["poster"] = provider_data.thumbnail()
+            video_info.upstream = provider_data.upstream()
+            video_info.poster = provider_data.thumbnail()
         else:
             proxied = await handler.proxy.proxy_url(
                 [
@@ -138,16 +148,30 @@ class MainHandler(tornado.web.RequestHandler):
                         upstream=provider_data.upstream(),
                         ctype=provider_data.proxy_ctype(),
                     ),
+                    config.LinkWithType(
+                        upstream=provider_data.upstream(),
+                        ctype=provider_data.proxy_ctype(),
+                        download=True,
+                    ),
                     config.LinkWithType(
                         upstream=provider_data.thumbnail(),
                     ),
                 ]
             )
-            video_info["upstream"] = proxied[0]
-            video_info["poster"] = proxied[1]
+            video_info.upstream = proxied[0]
+            video_info.download = proxied[1]
+            if (
+                isinstance(video_info.ctype, str)
+                and "mpegurl" in video_info.ctype.lower()
+            ):
+                video_info.download = None
+            video_info.poster = proxied[2]
 
-        video_info["ctype"] = provider_data.ctype()
-        return (video_info, provider_data.meta(), provider_data.title())
+        return (
+            dataclasses.asdict(video_info),
+            provider_data.meta(),
+            provider_data.title(),
+        )
 
     async def handle_raw(self, handler):
         try: