"""Pydantic 契约（插件 ↔ 服务端）。"""
from __future__ import annotations

from pydantic import BaseModel, Field

SUPPORTED_TYPES = [
    "white_bg", "key_features", "selling_pt", "material",
    "lifestyle", "multi_scene", "ecommerce_detail",
    "size_chart", "sku_collection", "custom",
]

# 通义（DashScope）生图模型白名单：插件下拉可选的模型
TONGYI_MODELS = ["qwen-image-3.0-pro", "wan2.7-image-pro", "wan2.6-image", "wan2.6-t2i"]

# RightAPI 生图模型白名单（gpt-image 系列 + Google nano-banana 系列，同一中转）
RIGHTAPI_MODELS = [
    "gpt-image-2",
    "gpt-image-2-vip",
    "nano-banana",
    "nano-banana-2",
    "nano-banana-2-lite",
    "nano-banana-pro",
]

# 模型 → provider 推断表：插件只传模型名，服务端据此路由（模型名优先于 provider 字段）
MODEL_PROVIDERS: dict[str, str] = {
    **{m: "tongyi" for m in TONGYI_MODELS},
    **{m: "rightapi" for m in RIGHTAPI_MODELS},
}


def resolve_provider(model: str | None, requested: str | None, default: str) -> str:
    """已知模型名直接定位 provider；未知模型回退到请求指定的 provider 或默认值。"""
    if model and model in MODEL_PROVIDERS:
        return MODEL_PROVIDERS[model]
    return requested or default


# ── 文本素材（规划 / 生成共用）──

class TextMaterial(BaseModel):
    kind: str = Field(..., description="title | params | selling_point | desc | price | brand")
    content: str = ""
    pairs: list[dict] | None = None  # [{key, value}]


# ── 套图生成 ──

# 目标平台 → 文案语言 + 图片比例（平台决定规格，不再单独选语言）
PLATFORM_SPECS: dict[str, dict] = {
    "ozon": {"lang": "ru", "ratio": "3:4", "label": "Ozon"},
    "wb": {"lang": "ru", "ratio": "3:4", "label": "Wildberries"},
    "cn": {"lang": "zh", "ratio": "1:1", "label": "中文（国内平台）"},
}


# ── 无状态套图生成（工具流程：请求自带采集数据）──

class GenerateImageItem(BaseModel):
    url: str = Field(..., description="勾选的图片 URL（源站原图）")
    group_key: str = Field(default="main", description="main | sku | detail")
    variant_name: str | None = Field(default=None, description="SKU 规格名（方案绑定用）")


class PlanItem(BaseModel):
    """出图方案项：一类图 × 数量，可绑定 SKU 规格。"""
    kind: str = Field(default="custom", description="图类型（SUPPORTED_TYPES 之一）")
    title: str = Field(..., description="方案标题，如「主图·粉色」")
    detail: str = Field(default="", description="这张图展示什么（中文）")
    prompt_hint: str = Field(default="", description="构图提示（英文，进生图 prompt）")
    count: int = Field(default=1, ge=0, le=5)
    variant_name: str | None = Field(default=None, description="绑定的 SKU 规格名")


class GenerateRequest(BaseModel):
    texts: list[TextMaterial] = Field(default_factory=list, description="采集的文本素材")
    images: list[GenerateImageItem] = Field(default_factory=list, description="勾选的参考图")
    style_set: int = Field(default=1, ge=1, le=5)
    style_prompt: str | None = Field(default=None, description="用户改写的风格提示词（覆盖 style_set 模板）")
    requirements: str | None = Field(default=None, description="生图要求（最高优先级，强制约束，覆盖其他设定）")
    types: list[str] = Field(default_factory=list, description="旧参数：无方案时按类型生成")
    plan: list[PlanItem] | None = Field(default=None, description="出图方案（优先于 types）")
    platform: str = Field(default="cn", description="目标平台：ozon | wb | cn")
    provider: str | None = Field(default=None, description="覆盖默认 provider（doubao | tongyi）")
    model: str | None = Field(default=None, description="覆盖默认生图模型（tongyi: qwen-image-3.0-pro / wan2.7-image-pro）")


# ── 出图方案规划（DeepSeek）──

class PlanRequest(BaseModel):
    texts: list[TextMaterial] = Field(default_factory=list)
    sku_variants: list[str] = Field(default_factory=list, description="带图的 SKU 规格名")
    image_stats: dict = Field(default_factory=dict, description="分组图片数量统计")
    platform: str = Field(default="cn")
    requirements: str | None = Field(default=None, description="生图要求（最高优先级，规划方案必须遵循）")


class PlanItemOut(BaseModel):
    kind: str
    title: str
    detail: str = ""
    prompt_hint: str = ""
    count: int = 1
    variant_name: str | None = None


class PlanResponse(BaseModel):
    summary: str = ""
    items: list[PlanItemOut]


class SuiteCreateResponse(BaseModel):
    suite_id: str


class SuiteImageOut(BaseModel):
    type_id: str
    name: str
    url: str
    status: str
    error: str | None = None


class SuiteOut(BaseModel):
    id: str
    status: str
    style_set: int
    platform: str
    lang: str
    ratio: str
    provider: str
    model: str | None = None
    total: int = 0  # 计划生成总张数（进度分母；images 是逐张追加，过程中 length < total）
    images: list[SuiteImageOut]
    error: str | None = None