Add files via upload

mediaflow_proxy/extractors/doodstream.py

@@ -0,0 +1,39 @@
+import re
+import time
+from typing import Dict
+
+from mediaflow_proxy.extractors.base import BaseExtractor, ExtractorError
+
+
+class DoodStreamExtractor(BaseExtractor):
+    """DoodStream URL extractor."""
+
+    def __init__(self, request_headers: dict):
+        super().__init__(request_headers)
+        self.base_url = "https://d000d.com"
+
+    async def extract(self, url: str, **kwargs) -> Dict[str, str]:
+        """Extract DoodStream URL."""
+        response = await self._make_request(url)
+
+        # Extract URL pattern
+        pattern = r"(\/pass_md5\/.*?)'.*(\?token=.*?expiry=)"
+        match = re.search(pattern, response.text, re.DOTALL)
+        if not match:
+            raise ExtractorError("Failed to extract URL pattern")
+
+        # Build final URL
+        pass_url = f"{self.base_url}{match[1]}"
+        referer = f"{self.base_url}/"
+        headers = {"range": "bytes=0-", "referer": referer}
+
+        response = await self._make_request(pass_url, headers=headers)
+        timestamp = str(int(time.time()))
+        final_url = f"{response.text}123456789{match[2]}{timestamp}"
+
+        self.base_headers["referer"] = referer
+        return {
+            "destination_url": final_url,
+            "request_headers": self.base_headers,
+            "mediaflow_endpoint": self.mediaflow_endpoint,
+        }