"""OCR 插件入口。"""
import logging

from app.utils.ocr_engine import get_ocr_model_info, get_ocr_engine

logger = logging.getLogger(__name__)


class OCRPlugin:
    def __init__(self):
        self.name = "ocr"
        self.version = "2.0.0"
        self.enabled = False

    def on_enable(self):
        self.enabled = True
        logger.info(f"OCR 插件 {self.version} 已启用")

    def on_disable(self):
        self.enabled = False
        logger.info(f"OCR 插件 {self.version} 已禁用")

    def get_status(self) -> dict:
        """三态状态：enabled + engine_ready + model_id，让前端区分「未启用/未加载/就绪」。"""
        engine_ready = False
        model_id = None
        try:
            # 触发一次引擎实例化以判断"是否已加载"——失败不算禁用，仅未就绪
            get_ocr_engine()
            engine_ready = True
            model_id = get_ocr_model_info().get("det_model")
        except Exception:
            # 依赖缺失或首次加载失败，由前端提示用户
            engine_ready = False
        return {
            "name": self.name,
            "version": self.version,
            "enabled": self.enabled,
            "engine_ready": engine_ready,
            "model_id": model_id,
        }


plugin = OCRPlugin()