"""DeepSeek-OCR 客户端（子智能体专用）。"""

import base64
import mimetypes
from pathlib import Path
from typing import Dict, List

from openai import OpenAI

from config import OCR_API_BASE_URL, OCR_API_KEY, OCR_MODEL_ID, OCR_MAX_TOKENS
from modules.file_manager import FileManager


class OCRClient:
    """封装 DeepSeek-OCR 调用逻辑。"""

    def __init__(self, project_path: str, file_manager: FileManager):
        self.project_path = Path(project_path).resolve()
        self.file_manager = file_manager

        base_url = (OCR_API_BASE_URL or "").rstrip("/")
        if not base_url.endswith("/v1"):
            base_url = f"{base_url}/v1"

        self.client = OpenAI(
            api_key=OCR_API_KEY,
            base_url=base_url,
        )
        self.model = OCR_MODEL_ID or "deepseek-ai/DeepSeek-OCR"
        self.max_tokens = OCR_MAX_TOKENS or 4096
        self.max_image_size = 10 * 1024 * 1024  # 10MB

    def _validate_image_path(self, path: str):
        valid, error, full_path = self.file_manager._validate_path(path)
        if not valid:
            return False, error, None
        if not full_path.exists():
            return False, "文件不存在", None
        if not full_path.is_file():
            return False, "不是文件", None
        return True, "", full_path

    def ocr_image(self, path: str, prompt: str) -> Dict:
        warnings: List[str] = []

        valid, error, full_path = self._validate_image_path(path)
        if not valid:
            return {"success": False, "error": error, "warnings": warnings}

        if not prompt or not str(prompt).strip():
            return {"success": False, "error": "prompt 不能为空", "warnings": warnings}

        try:
            data = full_path.read_bytes()
        except Exception as exc:
            return {"success": False, "error": f"读取文件失败: {exc}", "warnings": warnings}

        size = len(data)
        if size <= 0:
            return {"success": False, "error": "文件为空，无法识别", "warnings": warnings}

        if size > self.max_image_size:
            return {
                "success": False,
                "error": f"图片过大({size}字节)，上限为{self.max_image_size}字节",
                "warnings": warnings,
            }

        mime_type, _ = mimetypes.guess_type(str(full_path))
        if not mime_type or not mime_type.startswith("image/"):
            warnings.append("无法确定图片类型，已按 JPEG 处理")
            mime_type = "image/jpeg"

        base64_image = base64.b64encode(data).decode("utf-8")
        data_url = f"data:{mime_type};base64,{base64_image}"

        try:
            response = self.client.chat.completions.create(
                model=self.model,
                messages=[
                    {
                        "role": "user",
                        "content": [
                            {"type": "image_url", "image_url": {"url": data_url}},
                            {"type": "text", "text": prompt},
                        ],
                    }
                ],
                max_tokens=self.max_tokens,
                temperature=0,
            )
            content = response.choices[0].message.content if response.choices else ""
            return {"success": True, "content": content or "", "warnings": warnings}
        except Exception as exc:
            return {"success": False, "error": f"OCR 调用失败: {exc}", "warnings": warnings}