"""
模型配置中心
用于集中管理所有支持的 Gemini 模型

使用方式:
    1. 添加新模型: 在对应的模型列表中添加新的模型字典
    2. 临时关闭模型: 将模型的 enabled 字段设为 False
    3. 重新启用模型: 将模型的 enabled 字段改回 True

模型类型:
    - GEMINI_MODELS: Nano Banana Pro 图像生成模型
    - GEMINI_FLASH_MODELS: Google Gemini Flash 文本生成模型

示例:
    添加新模型:
    {
        "id": "gemini-新模型名称",
        "description": "模型说明和特点",
        "enabled": True,
        "endpoint_type": "standard"  # 端点类型: "dynamic", "standard", "flatfee"
    }
    
    临时关闭模型:
    将对应模型的 "enabled": True 改为 "enabled": False
"""

from typing import List, Dict, Optional


# ============================================================
# 模型配置列表
# ============================================================

# ============================================================
# Nano Banana Pro 图像生成模型
# ============================================================

GEMINI_MODELS = [
    {
        "id": "nano-banana-pro",
        "description": "Nano Banana Pro,根据分辨率自动选择端点 (1K/2K/4K),高性能图像生成模型",
        "enabled": True,
        "endpoint_type": "dynamic",
        "endpoint": None  # 动态端点，由代码根据分辨率选择
    },
    {
        "id": "gemini-3-pro-image-preview-url",
        "description": "URL 模式,根据分辨率自动选择端点 (1K/2K/4K),推荐用于需要不同分辨率的场景",
        "enabled": False,
        "endpoint_type": "dynamic",
        "endpoint": None  # 动态端点，由代码根据分辨率选择
    },
    {
        "id": "gemini-3-pro-image-preview",
        "description": "标准模式,固定端点,适用于常规图像生成",
        "enabled": True,
        "endpoint_type": "standard",
        "endpoint": "/v1beta/models/gemini-3-pro-image-preview:generateContent"
    },
    {
        "id": "gemini-3-pro-image-preview-flatfee",
        "description": "固定费用模式,固定端点,按固定价格计费 (暂时不可用-504错误)",
        "enabled": False,  # 暂时禁用：端点返回 504 错误
        "endpoint_type": "flatfee",
        "endpoint": "/v1beta/models/gemini-3-pro-image-preview-flatfee:generateContent"
    },
    {
        "id": "nano-banana-2",
        "description": "Nano Banana 2 模型,固定端点,适用于高质量图像生成",
        "enabled": False,
        "endpoint_type": "standard",
        "endpoint": "/v1beta/models/nano-banana-2:generateContent"
    }
]


# ============================================================
# Google Gemini Flash 文本生成模型
# ============================================================

GEMINI_FLASH_MODELS = [
    {
        "id": "gemini-3-flash-preview",
        "description": "Gemini 3 Flash,快速多模态文本生成,支持图片和视频输入",
        "enabled": True,
        "endpoints": {
            "不思考": "/v1beta/models/gemini-3-flash-preview-nothinking:generateContent",
            "高": "/v1beta/models/gemini-3-flash-preview-high:generateContent"
        }
    }
]


# ============================================================
# 工具函数
# ============================================================

def get_enabled_models() -> List[str]:
    """
    获取所有启用的模型 ID 列表
    
    Returns:
        启用的模型 ID 列表
    
    Example:
        >>> get_enabled_models()
        ['gemini-3-pro-image-preview-url', 'gemini-3-pro-image-preview', ...]
    """
    return [model["id"] for model in GEMINI_MODELS if model.get("enabled", False)]


def get_all_models() -> List[str]:
    """
    获取所有模型 ID 列表（包括已禁用的）
    
    Returns:
        所有模型 ID 列表
    
    Example:
        >>> get_all_models()
        ['gemini-3-pro-image-preview-url', 'gemini-3-pro-image-preview', ...]
    """
    return [model["id"] for model in GEMINI_MODELS]


def get_model_config(model_id: str) -> Optional[Dict]:
    """
    根据模型 ID 获取完整的模型配置
    
    Args:
        model_id: 模型 ID
    
    Returns:
        模型配置字典，如果未找到则返回 None
    
    Example:
        >>> config = get_model_config("gemini-3-pro-image-preview-url")
        >>> print(config["description"])
        URL 模式,根据分辨率自动选择端点 (1K/2K/4K)
    """
    for model in GEMINI_MODELS:
        if model["id"] == model_id:
            return model
    return None


def is_model_enabled(model_id: str) -> bool:
    """
    检查指定模型是否启用
    
    Args:
        model_id: 模型 ID
    
    Returns:
        True 如果模型启用，False 如果禁用或不存在
    
    Example:
        >>> is_model_enabled("gemini-3-pro-image-preview-url")
        True
    """
    config = get_model_config(model_id)
    if config is None:
        return False
    return config.get("enabled", False)


def get_model_description(model_id: str) -> str:
    """
    获取模型的描述信息
    
    Args:
        model_id: 模型 ID
    
    Returns:
        模型描述，如果未找到则返回空字符串
    
    Example:
        >>> get_model_description("gemini-3-pro-image-preview")
        '标准模式,固定端点,适用于常规图像生成'
    """
    config = get_model_config(model_id)
    if config is None:
        return ""
    return config.get("description", "")


def get_endpoint_type(model_id: str) -> Optional[str]:
    """
    获取模型的端点类型
    
    Args:
        model_id: 模型 ID
    
    Returns:
        端点类型 ("dynamic", "standard", "flatfee")，如果未找到则返回 None
    
    Example:
        >>> get_endpoint_type("gemini-3-pro-image-preview-url")
        'dynamic'
    """
    config = get_model_config(model_id)
    if config is None:
        return None
    return config.get("endpoint_type")


def get_model_endpoint(model_id: str) -> Optional[str]:
    """
    获取模型的 API 端点
    
    Args:
        model_id: 模型 ID
    
    Returns:
        API 端点路径，如果未找到或为动态端点则返回 None
    
    Example:
        >>> get_model_endpoint("gemini-3-pro-image-preview")
        '/v1beta/models/gemini-3-pro-image-preview:generateContent'
        >>> get_model_endpoint("gemini-3-pro-image-preview-url")
        None  # 动态端点
    """
    config = get_model_config(model_id)
    if config is None:
        return None
    return config.get("endpoint")


# ============================================================
# Gemini Flash 模型工具函数
# ============================================================

def get_enabled_flash_models() -> List[str]:
    """
    获取所有启用的 Flash 模型 ID 列表
    
    Returns:
        启用的 Flash 模型 ID 列表
    
    Example:
        >>> get_enabled_flash_models()
        ['gemini-3-flash-preview']
    """
    return [model["id"] for model in GEMINI_FLASH_MODELS if model.get("enabled", False)]


def get_all_flash_models() -> List[str]:
    """
    获取所有 Flash 模型 ID 列表（包括已禁用的）
    
    Returns:
        所有 Flash 模型 ID 列表
    """
    return [model["id"] for model in GEMINI_FLASH_MODELS]


def get_flash_model_config(model_id: str) -> Optional[Dict]:
    """
    根据模型 ID 获取 Flash 模型的完整配置
    
    Args:
        model_id: 模型 ID
    
    Returns:
        模型配置字典，如果未找到则返回 None
    
    Example:
        >>> config = get_flash_model_config("gemini-3-flash-preview")
        >>> print(config["description"])
        'Gemini 3 Flash,快速多模态文本生成,支持图片和视频输入'
    """
    for model in GEMINI_FLASH_MODELS:
        if model["id"] == model_id:
            return model
    return None


def is_flash_model_enabled(model_id: str) -> bool:
    """
    检查指定 Flash 模型是否启用
    
    Args:
        model_id: 模型 ID
    
    Returns:
        True 如果模型启用，False 如果禁用或不存在
    """
    config = get_flash_model_config(model_id)
    if config is None:
        return False
    return config.get("enabled", False)


def get_flash_model_endpoint(model_id: str, thinking_depth: str = "不思考") -> Optional[str]:
    """
    获取 Flash 模型的 API 端点
    
    Args:
        model_id: 模型 ID
        thinking_depth: 思考深度 ("不思考" 或 "高")
    
    Returns:
        API 端点路径，如果未找到则返回 None
    
    Example:
        >>> get_flash_model_endpoint("gemini-3-flash-preview", "不思考")
        '/v1beta/models/gemini-3-flash-preview-nothinking:generateContent'
        >>> get_flash_model_endpoint("gemini-3-flash-preview", "高")
        '/v1beta/models/gemini-3-flash-preview-high:generateContent'
    """
    config = get_flash_model_config(model_id)
    if config is None:
        return None
    
    endpoints = config.get("endpoints", {})
    return endpoints.get(thinking_depth)


def get_flash_model_description(model_id: str) -> str:
    """
    获取 Flash 模型的描述信息
    
    Args:
        model_id: 模型 ID
    
    Returns:
        模型描述，如果未找到则返回空字符串
    """
    config = get_flash_model_config(model_id)
    if config is None:
        return ""
    return config.get("description", "")


# ============================================================
# 向后兼容性检查
# ============================================================

def validate_models_config() -> None:
    """
    验证模型配置的完整性
    
    检查:
    - 每个模型必须有 id, description, enabled, endpoint_type, endpoint 字段
    - 非动态端点模型必须配置有效的 endpoint
    - 至少有一个模型是启用的
    
    Raises:
        ValueError: 如果配置不合法
    """
    if not GEMINI_MODELS:
        raise ValueError("GEMINI_MODELS 列表不能为空")
    
    required_fields = ["id", "description", "enabled", "endpoint_type", "endpoint"]
    valid_endpoint_types = ["dynamic", "standard", "flatfee"]
    
    for i, model in enumerate(GEMINI_MODELS):
        # 检查必需字段
        for field in required_fields:
            if field not in model:
                raise ValueError(f"模型 #{i} 缺少必需字段: {field}")
        
        # 检查 endpoint_type 是否合法
        if model["endpoint_type"] not in valid_endpoint_types:
            raise ValueError(
                f"模型 {model['id']} 的 endpoint_type '{model['endpoint_type']}' 不合法。"
                f"必须是: {', '.join(valid_endpoint_types)}"
            )
        
        # 检查非动态端点模型必须有有效的 endpoint
        if model["endpoint_type"] != "dynamic" and not model.get("endpoint"):
            raise ValueError(
                f"模型 {model['id']} 的 endpoint_type 为 '{model['endpoint_type']}'，"
                f"但未配置有效的 endpoint 字段"
            )
        
        # 检查端点格式（如果配置了）
        endpoint = model.get("endpoint")
        if endpoint and not endpoint.startswith("/v1beta/models/"):
            raise ValueError(
                f"模型 {model['id']} 的 endpoint '{endpoint}' 格式不正确。"
                f"应以 '/v1beta/models/' 开头"
            )
    
    # 检查至少有一个启用的模型
    if not get_enabled_models():
        raise ValueError("至少需要启用一个模型")


def validate_flash_models_config() -> None:
    """
    验证 Flash 模型配置的完整性
    
    检查:
    - 每个模型必须有 id, description, enabled, endpoints 字段
    - endpoints 必须包含所有思考深度选项
    - 至少有一个模型是启用的
    
    Raises:
        ValueError: 如果配置不合法
    """
    if not GEMINI_FLASH_MODELS:
        raise ValueError("GEMINI_FLASH_MODELS 列表不能为空")
    
    required_fields = ["id", "description", "enabled", "endpoints"]
    required_thinking_depths = ["不思考", "高"]
    
    for i, model in enumerate(GEMINI_FLASH_MODELS):
        # 检查必需字段
        for field in required_fields:
            if field not in model:
                raise ValueError(f"Flash 模型 #{i} 缺少必需字段: {field}")
        
        # 检查 endpoints 字典
        endpoints = model.get("endpoints", {})
        if not isinstance(endpoints, dict):
            raise ValueError(f"Flash 模型 {model['id']} 的 endpoints 必须是字典")
        
        # 检查所有思考深度选项都有对应端点
        for depth in required_thinking_depths:
            if depth not in endpoints:
                raise ValueError(
                    f"Flash 模型 {model['id']} 的 endpoints 缺少 '{depth}' 思考深度"
                )
            
            endpoint = endpoints[depth]
            if not endpoint or not endpoint.startswith("/v1beta/models/"):
                raise ValueError(
                    f"Flash 模型 {model['id']} 的端点 '{endpoint}' 格式不正确。"
                    f"应以 '/v1beta/models/' 开头"
                )
    
    # 检查至少有一个启用的模型
    if not get_enabled_flash_models():
        raise ValueError("至少需要启用一个 Flash 模型")


# 在模块加载时验证配置
try:
    validate_models_config()
except ValueError as e:
    print(f"⚠️ 图像模型配置验证失败: {str(e)}")
    print(f"⚠️ 请检查 models_config.py 文件")

try:
    validate_flash_models_config()
except ValueError as e:
    print(f"⚠️ Flash 模型配置验证失败: {str(e)}")
    print(f"⚠️ 请检查 models_config.py 文件")