"""
OpenAI 兼容 API 网关
提供统一的 /v1/chat/completions 和 /v1/models 端点
"""

from typing import Any, Dict

from fastapi import APIRouter, HTTPException, Request
from fastapi.responses import JSONResponse

from adapters import get_adapter, get_provider_from_model
from adapters.base import ChatCompletionRequest
from utils.logger import get_logger

logger = get_logger()

router = APIRouter(tags=["OpenAI Compatible API"])


@router.post("/chat/completions")
async def chat_completions(request: Request):
    """
    OpenAI 兼容的聊天补全接口

    根据请求中的 model 字段自动路由到对应的平台适配器：
    - glm-* → 智谱 GLM
    - qwen-* → 阿里云百炼
    - gpt-* / o1-* / o3-* → OpenAI
    - deepseek-* → Deepseek
    """
    try:
        body = await request.json()
    except Exception:
        raise HTTPException(status_code=400, detail="Invalid JSON body")

    # 创建请求对象
    chat_request = ChatCompletionRequest.from_dict(body)
    model = chat_request.model

    # 根据模型名称确定平台
    provider = get_provider_from_model(model)
    logger.info(f"[Gateway] model={model} → provider={provider}")

    # 获取对应平台的适配器
    adapter = get_adapter(provider)
    if adapter is None:
        raise HTTPException(
            status_code=400,
            detail=f"Unsupported model: {model} (provider: {provider})",
        )

    # 检查适配器是否可用
    if not adapter.is_available():
        raise HTTPException(
            status_code=503,
            detail=f"Provider '{provider}' is not available (API key not configured)",
        )

    # 调用适配器处理请求
    return await adapter.chat(chat_request)


@router.get("/models")
async def list_models():
    """
    返回所有可用平台的模型列表

    聚合所有已配置 API Key 的平台模型
    """
    from adapters import get_all_adapters

    all_models = []

    for provider, adapter in get_all_adapters().items():
        if adapter.is_available():
            models = adapter.list_models()
            all_models.extend([m.to_dict() for m in models])

    return {
        "object": "list",
        "data": all_models,
    }


@router.get("/models/{model_id}")
async def get_model(model_id: str):
    """
    获取特定模型信息
    """
    from adapters import get_all_adapters

    for provider, adapter in get_all_adapters().items():
        if adapter.is_available():
            for model in adapter.list_models():
                if model.id == model_id:
                    return {
                        "object": "model",
                        "id": model.id,
                        "owned_by": model.provider,
                        "data": model.to_dict(),
                    }

    raise HTTPException(status_code=404, detail=f"Model not found: {model_id}")


# 初始化时注册适配器
def init_adapters():
    """注册所有适配器"""
    from adapters import register_adapter
    from adapters.dashscope_adapter import DashScopeAdapter
    from adapters.glm_adapter import GLMAdapter
    from adapters.openai_adapter import DeepseekAdapter, OpenAIAdapter

    register_adapter("glm", GLMAdapter)
    register_adapter("dashscope", DashScopeAdapter)
    register_adapter("openai", OpenAIAdapter)
    register_adapter("deepseek", DeepseekAdapter)

    logger.info("[Gateway] Adapters registered: glm, dashscope, openai, deepseek")