"""Provider 工厂，按设置中的 type 字段实例化。"""
from __future__ import annotations

from typing import Optional

from app.schemas.common import ProviderConfig

from .base import OCRProvider, VLMProvider
from .ocr_http import HttpOCR
from .ocr_paddle import PaddleOCRProvider
from .ocr_tesseract import TesseractOCR
from .ocr_vision import VisionOCR
from .vlm_openai import OpenAICompatVLM

# OCR Provider 类型常量
OCR_TYPES = ("tesseract", "paddleocr", "http", "vision", "none")
VLM_TYPES = ("openai_compat", "none")
RECOGNITION_MODES = ("ocr", "vision", "hybrid")


def build_ocr_provider(
    cfg: ProviderConfig | None,
    *,
    allow_upload: bool = True,
) -> Optional[OCRProvider]:
    """根据配置构造传统 OCR / 视觉 OCR Provider。"""
    if cfg is None or cfg.type in ("", "none", "disabled"):
        return None
    if cfg.type == "tesseract":
        return TesseractOCR(
            lang=cfg.extra.get("lang", "chi_sim+eng"),
            cmd=cfg.extra.get("cmd"),
        )
    if cfg.type == "paddleocr":
        return PaddleOCRProvider(lang=cfg.extra.get("lang", "ch"))
    if cfg.type == "http":
        if not cfg.base_url:
            raise ValueError("HTTP OCR 需要配置 base_url")
        return HttpOCR(
            base_url=cfg.base_url,
            api_key=cfg.api_key or "",
            text_path=str(cfg.extra.get("text_path", "text")),
            headers=cfg.extra.get("headers") if isinstance(cfg.extra.get("headers"), dict) else None,
            timeout=float(cfg.extra.get("timeout", 30)),
        )
    if cfg.type == "vision":
        return build_vision_ocr(cfg, allow_upload=allow_upload)
    raise ValueError(f"未知 OCR Provider 类型: {cfg.type}")


def build_vision_ocr(
    cfg: ProviderConfig | None,
    *,
    allow_upload: bool = True,
) -> Optional[VisionOCR]:
    """从 ProviderConfig 构造视觉 OCR（可与 VLM 共用同一套接口配置）。"""
    if cfg is None or cfg.type in ("", "none", "disabled"):
        return None
    base_url = cfg.base_url or "http://localhost:11434/v1"
    model = cfg.model or "qwen2.5vl:7b"
    return VisionOCR(
        base_url=base_url,
        api_key=cfg.api_key or "",
        model=model,
        timeout=float(cfg.extra.get("timeout", 60)),
        allow_upload=allow_upload,
    )


def build_vlm_provider(cfg: ProviderConfig | None) -> Optional[VLMProvider]:
    """根据配置构造 VLM Provider（AI 分类/摘要/标签）。"""
    if cfg is None or cfg.type in ("", "none", "disabled"):
        return None
    if cfg.type in ("openai_compat", "openai", "ollama", "glm", "minimax", "moonshot", "vision"):
        return OpenAICompatVLM(
            base_url=cfg.base_url or "http://localhost:11434/v1",
            api_key=cfg.api_key or "",
            model=cfg.model or "gpt-4o-mini",
            timeout=float(cfg.extra.get("timeout", 60)),
        )
    raise ValueError(f"未知 VLM Provider 类型: {cfg.type}")