"""
Download models
"""

from dataclasses import dataclass
from typing import Optional, Dict, Any
from enum import Enum

from .base import BaseRequest, BaseResponse, OperationStatus


class PodcastPlatform(str, Enum):
    """Supported podcast platforms"""
    APPLE_PODCAST = "apple_podcast"
    XIAOYUZHOU = "xiaoyuzhou"


@dataclass
class DownloadRequest(BaseRequest):
    """Request model for podcast download"""
    url: str
    platform: PodcastPlatform
    output_directory: Optional[str] = None
    auto_transcribe: bool = False
    enable_speaker_diarization: bool = False


@dataclass
class DownloadResponse(BaseResponse):
    """Response model for podcast download"""
    original_url: str = ""
    audio_file_path: Optional[str] = None
    file_size_mb: Optional[float] = None
    duration_seconds: Optional[float] = None
    
    @classmethod
    def success(
        cls,
        original_url: str,
        audio_file_path: str,
        file_size_mb: Optional[float] = None,
        duration_seconds: Optional[float] = None,
        message: str = "下载成功"
    ) -> "DownloadResponse":
        """Create successful response"""
        return cls(
            status=OperationStatus.SUCCESS,
            message=message,
            original_url=original_url,
            audio_file_path=audio_file_path,
            file_size_mb=file_size_mb,
            duration_seconds=duration_seconds
        )
    
    @classmethod
    def failed(
        cls,
        original_url: str,
        error_message: str,
        error_code: str = "DOWNLOAD_ERROR",
        error_details: Optional[Dict[str, Any]] = None
    ) -> "DownloadResponse":
        """Create failed response"""
        return cls(
            status=OperationStatus.FAILED,
            message=error_message,
            error_code=error_code,
            error_details=error_details,
            original_url=original_url
        )