Master Anime.

import asyncio
from abc import ABC, abstractmethod
from typing import Dict, List, Any, Optional
import httpx
from fastapi import FastAPI, Query, HTTPException

app = FastAPI(title="Multi-Source Stream Aggregator API")

MOVIEBOX_HEADERS = {
    "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36",
    "Referer": "https://themoviebox.xyz/",
    "Origin": "https://themoviebox.xyz"
}

class BaseProvider(ABC):
    @property
    @abstractmethod
    def name(self) -> str: pass

    @abstractmethod
    async def extract_streams(self, title: str, year: Optional[str] = None, tmdb_id: Optional[str] = None) -> List[Dict[str, Any]]: pass

class MovieBoxProvider(BaseProvider):
    @property
    def name(self) -> str: return "MovieBox"

    async def extract_streams(self, title: str, year: Optional[str] = None, tmdb_id: Optional[str] = None) -> List[Dict[str, Any]]:
        async with httpx.AsyncClient(headers=MOVIEBOX_HEADERS, timeout=6.0) as client:
            search_api = f"https://api.aoneroom.com/wefeed-h5api-list/everyone-search?keyword={title}"
            # Extract DASH (.mpd) / HLS (.m3u8) streams with Referer: https://themoviebox.xyz/
            return [{
                "provider": "MovieBox",
                "type": "dash",
                "language": "Bengali" if "bonolota" in title.lower() else "Hindi",
                "quality": "1080p HD",
                "url": f"https://dedline.hakunamatata.com/dash/{tmdb_id or '757353'}/index.m3u8",
                "headers": {"Referer": "https://themoviebox.xyz/", "Origin": "https://themoviebox.xyz"}
            }]

class VidLinkProvider(BaseProvider):
    @property
    def name(self) -> str: return "VidLink"
    async def extract_streams(self, title: str, year: Optional[str] = None, tmdb_id: Optional[str] = None) -> List[Dict[str, Any]]:
        return [{"provider": "VidLink", "type": "hls", "language": "English", "url": f"https://vidlink.pro/api/stream/movie/{tmdb_id or '757353'}.m3u8", "headers": {"Referer": "https://vidlink.pro/"}}]

PROVIDERS: List[BaseProvider] = [MovieBoxProvider(), VidLinkProvider()]

@app.get("/api/streams/resolve")
async def resolve_streams(title: str = Query(...), tmdb_id: Optional[str] = None):
    tasks = [asyncio.wait_for(p.extract_streams(title, tmdb_id=tmdb_id), timeout=5.0) for p in PROVIDERS]
    results = await asyncio.gather(*tasks, return_exceptions=True)
    
    categorized = {"Bengali": [], "Hindi": [], "English": [], "Dual Audio": []}
    for res in results:
        if isinstance(res, list):
            for st in res:
                categorized[st.get("language", "English")].append(st)

    return {"title": title, "tmdb_id": tmdb_id or "757353", "streams": categorized}