New Extractor!

mediaflow_proxy/extractors/base.py

@@ -31,6 +31,7 @@ async def _make_request(
             async with create_httpx_client() as client:
                 request_headers = self.base_headers
                 request_headers.update(headers or {})
+                print(request_headers)
                 response = await client.request(
                     method,
                     url,

mediaflow_proxy/extractors/factory.py

@@ -7,6 +7,7 @@
 from mediaflow_proxy.extractors.uqload import UqloadExtractor
 from mediaflow_proxy.extractors.streamtape import StreamtapeExtractor
 from mediaflow_proxy.extractors.supervideo import SupervideoExtractor
+from mediaflow_proxy.extractors.vixcloud import VixCloudExtractor
 
 
 
@@ -20,6 +21,7 @@ class ExtractorFactory:
         "Mixdrop": MixdropExtractor,
         "Streamtape": StreamtapeExtractor,
         "Supervideo": SupervideoExtractor,
+        "VixCloud": VixCloudExtractor,
         "LiveTV": LiveTVExtractor,
     }
 

mediaflow_proxy/extractors/vixcloud.py

@@ -0,0 +1,65 @@
+import re
+from typing import Dict, Any
+from bs4 import BeautifulSoup, SoupStrainer
+from mediaflow_proxy.extractors.base import BaseExtractor, ExtractorError
+import json
+from urllib.parse import urlparse, parse_qs
+
+
+class VixCloudExtractor(BaseExtractor):
+    """VixCloud URL extractor."""
+
+    async def version(self, domain: str) -> str:
+        """Get version of VixCloud Parent Site."""
+        DOMAIN = domain
+        base_url = f"https://streamingcommunity.{DOMAIN}/richiedi-un-titolo"
+        response = await self._make_request(
+            base_url,
+            headers={
+                "Referer": f"https://streamingcommunity.{DOMAIN}/",
+                "Origin": f"https://streamingcommunity.{DOMAIN}",
+            },
+        )
+        if response.status_code != 200:
+            raise ExtractorError("Outdated Domain")
+        # Soup the response
+        soup = BeautifulSoup(response.text, "lxml", parse_only=SoupStrainer("div", {"id": "app"}))
+        if soup:
+            # Extract version
+            version = json.loads(soup.find("div", {"id": "app"}).get("data-page"))["version"]
+            return version
+
+    async def extract(self, url: str, **kwargs) -> Dict[str, Any]:
+        """Extract Vixcloud URL."""
+        domain = url.split("://")[1].split("/")[0].split(".")[1]
+        version = await self.version(domain)
+        response = await self._make_request(url, headers={"x-inertia": "true", "x-inertia-version": version})
+        soup = BeautifulSoup(response.text, "lxml", parse_only=SoupStrainer("iframe"))
+        iframe = soup.find("iframe").get("src")
+        parsed_url = urlparse(iframe)
+        query_params = parse_qs(parsed_url.query)
+        response = await self._make_request(iframe, headers={"x-inertia": "true", "x-inertia-version": version})
+
+        if response.status_code != 200:
+            raise ExtractorError("Failed to extract URL components, Invalid Request")
+        soup = BeautifulSoup(response.text, "lxml", parse_only=SoupStrainer("body"))
+        if soup:
+            script = soup.find("body").find("script").text
+            token = re.search(r"'token':\s*'(\w+)'", script).group(1)
+            expires = re.search(r"'expires':\s*'(\d+)'", script).group(1)
+            quality = re.search(r'"quality":(\d+)', script).group(1)
+            vixid = iframe.split("/embed/")[1].split("?")[0]
+            base_url = iframe.split("://")[1].split("/")[0]
+            final_url = f"https://{base_url}/playlist/{vixid}.m3u8?token={token}&expires={expires}"
+            if "canPlayFHD" in query_params:
+                canPlayFHD = "h=1"
+                final_url += "&h=1"
+            if "b" in query_params:
+                b = "b=1"
+                final_url += "&b=1"
+            self.base_headers["referer"] = url
+            return {
+                "destination_url": final_url,
+                "request_headers": self.base_headers,
+                "mediaflow_endpoint": self.mediaflow_endpoint,
+            }

mediaflow_proxy/schemas.py

@@ -63,7 +63,7 @@ class MPDSegmentParams(GenericParams):
 
 
 class ExtractorURLParams(GenericParams):
-    host: Literal["Doodstream", "Mixdrop", "Uqload", "Streamtape", "Supervideo", "LiveTV"] = Field(
+    host: Literal["Doodstream", "Mixdrop", "Uqload", "Streamtape", "Supervideo", "VixCloud", "LiveTV"] = Field(
         ..., description="The host to extract the URL from."
     )
     destination: str = Field(..., description="The URL of the stream.", alias="d")