from pydantic import BaseModel
from enum import Enum
from typing import Optional
from datetime import datetime


class DownloadStatus(str, Enum):
    PENDING = "pending"
    DOWNLOADING = "downloading"
    PAUSED = "paused"
    COMPLETED = "completed"
    FAILED = "failed"
    CANCELLED = "cancelled"


class HostType(str, Enum):
    RAPIDFILE = "rapidfile"
    UNFICHIER = "1fichier"
    DOODSTREAM = "doodstream"
    OTHER = "other"


class DownloadTask(BaseModel):
    id: str
    url: str
    filename: str
    host: HostType
    status: DownloadStatus
    progress: float = 0.0
    downloaded_bytes: int = 0
    total_bytes: Optional[int] = None
    speed: float = 0.0
    error: Optional[str] = None
    created_at: datetime
    started_at: Optional[datetime] = None
    completed_at: Optional[datetime] = None
    file_path: Optional[str] = None


class DownloadRequest(BaseModel):
    url: str
    filename: Optional[str] = None


class AnimeMetadata(BaseModel):
    """Metadata for anime series"""
    synopsis: Optional[str] = None
    genres: list[str] = []
    rating: Optional[str] = None  # Could be "PG-13", "R", etc., or numeric like "8.5/10"
    release_year: Optional[int] = None
    studio: Optional[str] = None
    poster_image: Optional[str] = None
    banner_image: Optional[str] = None
    total_episodes: Optional[int] = None
    status: Optional[str] = None  # "Ongoing", "Completed", etc.
    alternative_titles: list[str] = []


class AnimeSearchResult(BaseModel):
    """Enhanced search result with metadata"""
    title: str
    url: str
    cover_image: Optional[str] = None
    type: str  # "search_result" or "direct"
    metadata: Optional[AnimeMetadata] = None