"""无状态套图生成：请求自带采集数据，任务存进程内注册表。"""
from __future__ import annotations

from fastapi import APIRouter, BackgroundTasks, HTTPException

from config import get_settings
from schemas import (
    GenerateRequest, PLATFORM_SPECS, SUPPORTED_TYPES, TONGYI_MODELS, RIGHTAPI_MODELS,
    SuiteCreateResponse, TextMaterial, resolve_provider,
    PlanRequest, PlanResponse, PlanItemOut,
)
from services.generator import run_suite
from services.planner import generate_plan
from services.prompt import type_name
from services.tasks import create_task

router = APIRouter(prefix="/api", tags=["generate"])


def texts_to_raw(texts: list[TextMaterial]) -> dict:
    """插件上传的文本素材 → prompt 上下文用的 raw dict（后写的覆盖先写的）。"""
    raw: dict = {}
    for t in texts:
        if t.kind == "title" and t.content:
            raw["title"] = t.content
        elif t.kind == "price" and t.content:
            raw["price"] = t.content
        elif t.kind == "brand" and t.content:
            raw["brand"] = t.content
        elif t.kind == "params" and t.pairs:
            merged = {p["key"]: p["value"] for p in (raw.get("params") or [])}
            for p in t.pairs:
                merged.setdefault(p["key"], p["value"])
            raw["params"] = [{"key": k, "value": v} for k, v in merged.items()]
        elif t.kind == "selling_point" and t.content:
            raw["sellingPoints"] = t.content
        elif t.kind == "desc" and t.content:
            raw["desc"] = t.content
        elif t.kind == "sales" and t.content:
            raw["sales"] = t.content
        elif t.kind == "shop" and t.content:
            raw["shop"] = t.content
    return raw


@router.post("/generate", response_model=SuiteCreateResponse)
async def generate_suite(
    req: GenerateRequest,
    background: BackgroundTasks,
) -> SuiteCreateResponse:
    if not req.images:
        raise HTTPException(status_code=400, detail="未勾选任何图片，无法生成")

    # 生成任务列表：方案优先；无方案时按 types（空则默认四种）
    if req.plan:
        jobs: list[dict] = []
        for item in req.plan:
            if item.count <= 0:
                continue
            if item.kind not in SUPPORTED_TYPES:
                raise HTTPException(status_code=400, detail=f"方案项「{item.title}」的图类型不支持: {item.kind}")
            # 同一项多张 → 展开为多任务，第二张起在标题上加序号
            for n in range(item.count):
                jobs.append({
                    "kind": item.kind,
                    "title": item.title if item.count == 1 else f"{item.title}{n + 1}",
                    "detail": item.detail,
                    "prompt_hint": item.prompt_hint,
                    "variant_name": item.variant_name,
                })
        if not jobs:
            raise HTTPException(status_code=400, detail="方案中所有项的数量都是 0")
    else:
        types = req.types or ["white_bg", "key_features", "lifestyle", "multi_scene"]
        bad = [t for t in types if t not in SUPPORTED_TYPES]
        if bad:
            raise HTTPException(status_code=400, detail=f"不支持的图类型: {bad}")
        jobs = [{"kind": t, "title": type_name(t), "detail": "", "prompt_hint": "", "variant_name": None} for t in types]

    if req.platform not in PLATFORM_SPECS:
        raise HTTPException(status_code=400, detail=f"不支持的目标平台: {req.platform}（ozon | wb | cn）")
    spec = PLATFORM_SPECS[req.platform]

    settings = get_settings()
    # 插件只传模型名：已知模型直接路由到对应 provider（gpt-image-2 → rightapi）
    provider_name = resolve_provider(req.model, req.provider, settings.image_provider)
    model = req.model
    if provider_name == "tongyi" and model and model not in TONGYI_MODELS:
        raise HTTPException(status_code=400, detail=f"不支持的模型: {model}（tongyi 支持: {TONGYI_MODELS}）")
    if provider_name == "rightapi" and model and model not in RIGHTAPI_MODELS:
        raise HTTPException(status_code=400, detail=f"不支持的模型: {model}（rightapi 支持: {RIGHTAPI_MODELS}）")

    task = create_task(
        status="pending",
        platform=req.platform,
        lang=spec["lang"],
        ratio=spec["ratio"],
        style_set=req.style_set,
        style_prompt=req.style_prompt,
        requirements=req.requirements,
        provider=provider_name,
        model=model,
        total=len(jobs),
        context=texts_to_raw(req.texts),
        plan=jobs,
        # 参考图池：main 组优先，其余组按序补充（variant 绑定靠 variant_name 匹配）
        ref_images=[
            {
                "url": i.url,
                "group_key": i.group_key,
                "variant_name": i.variant_name,
            }
            for i in sorted(req.images, key=lambda x: 0 if x.group_key == "main" else 1)
        ],
    )

    background.add_task(run_suite, task)
    return SuiteCreateResponse(suite_id=task.id)


@router.post("/plan", response_model=PlanResponse)
async def plan_suite(req: PlanRequest) -> PlanResponse:
    """DeepSeek 根据商品信息生成出图方案。"""
    product_info = texts_to_raw(req.texts)
    if not product_info.get("title"):
        raise HTTPException(status_code=400, detail="缺少商品标题，无法规划")
    try:
        result = await generate_plan(product_info, req.sku_variants, req.image_stats, req.platform, requirements=req.requirements)
    except RuntimeError as exc:
        raise HTTPException(status_code=502, detail=str(exc)) from exc
    except Exception as exc:  # noqa: BLE001
        raise HTTPException(status_code=502, detail=f"规划失败: {exc}") from exc
    return PlanResponse(
        summary=result["summary"],
        items=[PlanItemOut(**i) for i in result["items"]],
    )