"""健康检查端点。"""

from __future__ import annotations

from fastapi import APIRouter

from src import __version__
from src.engine.rapidocr import RapidOCREngine
from src.engine.paddleocr_vl import PaddleOCRVLEngine
from src.schemas.ocr import EngineStatus, HealthResponse

router = APIRouter()

# 全局引擎实例（由 main.py startup 注入）
rapidocr: RapidOCREngine | None = None
paddleocr_vl: PaddleOCRVLEngine | None = None


def init(
    rapidocr_engine: RapidOCREngine,
    paddleocr_vl_engine: PaddleOCRVLEngine,
) -> None:
    """注入引擎实例。"""
    global rapidocr, paddleocr_vl
    rapidocr = rapidocr_engine
    paddleocr_vl = paddleocr_vl_engine


@router.get("/healthz", response_model=HealthResponse)
async def health() -> HealthResponse:
    engines_status: dict = {}

    for name, engine in [("ppocrv6", rapidocr), ("paddleocr-vl", paddleocr_vl)]:
        if engine is None:
            engines_status[name] = EngineStatus(status="unavailable")
        else:
            h = await engine.health()
            engines_status[name] = EngineStatus(status=h.get("status", "error"))

    # 整体状态
    ok_count = sum(1 for e in engines_status.values() if e.status == "ok")
    unavailable_count = sum(1 for e in engines_status.values() if e.status == "unavailable")
    total = len(engines_status)
    if ok_count == total:
        overall = "ok"
    elif unavailable_count == total:
        overall = "error"
    else:
        overall = "degraded"

    return HealthResponse(
        status=overall,
        engines=engines_status,
        version=__version__,
    )