# -*- coding: utf-8 -*-
"""Shared bootstrap helpers for process startup."""

from __future__ import annotations

import sys


def ensure_models_downloaded(interactive: bool) -> bool:
    """Ensure declared deployment models exist locally, downloading if needed."""
    try:
        from app.utils.download_models import check_all_models, download_models

        missing = check_all_models()
        if not missing:
            return True

        print(f"\n⚠️  检测到 {len(missing)} 个模型未下载")
        for model_id, *_ in missing:
            print(f"  - {model_id}")

        print("\n将自动下载缺失模型后继续启动。")
        if download_models(auto_mode=True):
            return True

        print("\n模型自动下载失败。")
        if interactive:
            print("可手动运行以下命令排查：")
            print("  uv run python -m app.utils.download_models")
            print("  ./scripts/prepare-models.sh")
        else:
            print("非交互式终端下请确认网络可用，或预先准备模型缓存。")
        return False
    except Exception as exc:
        print(f"⚠️  模型检查失败: {exc}")
        return False


def run_cli_preflight() -> bool:
    """Preflight checks for the CLI entrypoint."""
    try:
        from app.core.accelerator import get_accelerator_info, validate_accelerator_runtime

        ok, message = validate_accelerator_runtime()
        info = get_accelerator_info()
        print(
            "Accelerator | "
            f"vendor={info.vendor} runtime={info.runtime} device={info.device} "
            f"count={info.device_count}"
        )
        if not ok:
            print(f"加速器运行栈检查失败: {message}")
            return False
    except Exception as exc:
        print(f"加速器检查失败: {exc}")
        return False

    return ensure_models_downloaded(interactive=sys.stdin.isatty())