UnHided/mediaflow_proxy/extractors/doodstream.py

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,
        }