"""Shared infrastructure for qianwen-vision scripts.

Thin wrapper around qianwen_lib that re-exports common symbols and adds
vision-specific helpers (streaming with reasoning content, base64 encoding for
smaller files, shared multimodal content builder). Stdlib only -- no pip install
required.
"""
from __future__ import annotations

import sys

if sys.version_info < (3, 9):
    print(f"Error: Python 3.9+ required (found {sys.version}). "
          "Install: https://www.python.org/downloads/", file=sys.stderr)
    sys.exit(1)

from pathlib import Path
from typing import Any

sys.path.insert(0, str(Path(__file__).resolve().parent))

# Re-export shared infrastructure so existing scripts keep working with
# ``from vision_lib import ...``.
from qianwen_lib import (  # noqa: E402,F401
    chat_url,
    extract_text,
    has_oss_url,
    http_post,
    load_request,
    native_base_url,
    require_api_key as _require_api_key_base,
    resolve_file,
    run_update_signal,
    save_result,
    stream_sse,
    try_parse_json,
    upload_local_file,
)

# ---------------------------------------------------------------------------
# Vision-specific credential wrapper
# ---------------------------------------------------------------------------

def require_api_key() -> str:
    """Load API key with vision-specific validation.

    Token Plan keys (``sk-sp-...``) are rejected outright for vision models
    rather than just warned about.
    """
    key = _require_api_key_base(script_file=__file__, domain="Vision")
    if key.startswith("sk-sp-"):
        print(
            "Error: Token Plan key detected (sk-sp-...). "
            "Vision models are not available on Token Plan.\n"
            "Docs: https://platform.qianwenai.com/docs/token-plan/overview",
            file=sys.stderr,
        )
        sys.exit(1)
    return key

# ---------------------------------------------------------------------------
# Update-check signal (convenience)
# ---------------------------------------------------------------------------

def prompt_update_check_install() -> None:
    """Emit structured signals about update-check skill status to stderr."""
    run_update_signal(caller=__file__)

# ---------------------------------------------------------------------------
# Shared multimodal content builder
# ---------------------------------------------------------------------------

def build_content(
    req: dict[str, Any],
    detail: str = "auto",
    *,
    upload_key: str | None = None,
    upload_model: str | None = None,
) -> list[dict[str, Any]]:
    """Build multimodal content parts from image/video/video_frames input.

    Shared by analyze.py and reason.py. The ``detail`` parameter is only
    applied to image_url items and is ignored for video content.

    Supports:
      - ``video_frames``: list of frame image paths/URLs → video content part
      - ``video``: single video URL/path → video_url content part
      - ``images``: list of image paths/URLs → multiple image_url parts
      - ``image``: single image path/URL → one image_url part
    """
    parts: list[dict[str, Any]] = []

    if req.get("video_frames"):
        frames = [
            resolve_file(str(frame), api_key=upload_key, model=upload_model)
            for frame in req["video_frames"]
        ]
        video_obj: dict[str, Any] = {"type": "video", "video": frames}
        if req.get("fps"):
            video_obj["fps"] = float(req["fps"])
        parts.append(video_obj)
    elif req.get("video"):
        video_url = resolve_file(str(req["video"]), api_key=upload_key, model=upload_model)
        video_obj = {
            "type": "video_url",
            "video_url": {"url": video_url},
        }
        if req.get("fps"):
            video_obj["fps"] = float(req["fps"])
        parts.append(video_obj)
    elif req.get("images"):
        for img in req["images"]:
            resolved_url = resolve_file(str(img), api_key=upload_key, model=upload_model)
            parts.append({"type": "image_url", "image_url": {"url": resolved_url, "detail": detail}})
    elif req.get("image"):
        resolved_url = resolve_file(str(req["image"]), api_key=upload_key, model=upload_model)
        parts.append({"type": "image_url", "image_url": {"url": resolved_url, "detail": detail}})
    else:
        raise ValueError("Provide 'image', 'images', 'video', or 'video_frames'")

    return parts