"""提示词公共层：与模型家族无关的商品上下文、风格模板、图类型名与文案组件。

各家族模块（alibaba / doubao / gpt / google）只负责"如何对模型说话"，
商品信息提炼与风格体系统一在这里维护，避免多处漂移。
"""
from __future__ import annotations

import re

# ── 风格模板（与插件端 STYLE_SET_OPTIONS 对应；提示词可被用户在插件里改写覆盖）───

STYLE_SETS: dict[int, dict] = {
    1: {
        "name": "北欧极简",
        "tone": "北欧极简风：浅灰或米白背景，柔和漫射光，低饱和色调，画面留白充足，构图克制干净",
        "bg": "",
    },
    2: {
        "name": "清新明亮",
        "tone": "清新明亮风：明亮的白色到浅蓝渐变背景，高调光线，色彩明快通透，整体轻盈干净",
        "bg": "",
    },
    3: {
        "name": "高级感深色",
        "tone": "高级质感风：深灰或炭黑背景，戏剧性侧光打光，突出商品材质与光泽，沉稳高级",
        "bg": "",
    },
    4: {
        "name": "暖调生活",
        "tone": "温暖生活风：暖米色背景，暖色灯光氛围，温馨的家居质感，亲和力强",
        "bg": "",
    },
    5: {
        "name": "纯净棚拍",
        "tone": "标准电商棚拍：纯色浅背景，均匀的正面柔光，无杂物干扰，商品居中突出",
        "bg": "",
    },
}

# ── 图类型中文名（导出文件名用）───────────────────────────────────────────

TYPE_NAMES_ZH: dict[str, str] = {
    "white_bg": "白底主图",
    "key_features": "核心卖点图",
    "selling_pt": "卖点图",
    "material": "材质图",
    "lifestyle": "场景展示图",
    "multi_scene": "多场景拼图",
    "ecommerce_detail": "电商详情图",
    "size_chart": "尺寸标注图",
    "sku_collection": "SKU合集图",
    "custom": "创意图",
}

# ── 图内营销文案渲染规范（各家族共用；语言由平台决定）──────────────────────

TEXT_RENDER = {
    "zh": (
        "Render concise Chinese marketing text inside the image: main headline max 8 Chinese characters, "
        "sub-lines max 12 characters each, font is modern clean sans-serif (Source Han Sans style), "
        "high legibility, tasteful typography layout, colors harmonized with the composition. "
        "No spelling errors, no garbled characters."
    ),
    "en": (
        "Render concise English marketing text inside the image: headline max 5 words, "
        "sub-lines max 8 words each, Helvetica Neue style sans-serif, high legibility, "
        "tasteful typography layout, colors harmonized with the composition. No spelling errors."
    ),
    "ru": (
        "Render concise Russian marketing text inside the image: headline max 4 words, "
        "sub-lines max 6 words each, modern clean sans-serif (Inter / PT Sans style), "
        "proper Cyrillic typography, high legibility, tasteful layout, colors harmonized with the composition. "
        "No spelling errors, no mixed latin/cyrillic gibberish."
    ),
}


def resolve_style(style_set: int, style_prompt: str | None = None) -> dict:
    """用户改写的风格提示词整体覆盖内置模板（tone/bg 整体替换）。"""
    if style_prompt and style_prompt.strip():
        return {"name": "custom", "tone": style_prompt.strip(), "bg": ""}
    return STYLE_SETS.get(style_set, STYLE_SETS[1])


def requirements_block(requirements: str | None) -> str:
    """用户强制要求块：最高优先级、置于提示词最前、覆盖冲突指令（原文保留不翻译）。"""
    if requirements and requirements.strip():
        return (
            "STRICT REQUIREMENTS (highest priority, must be followed exactly, "
            "override any conflicting instruction): "
            + requirements.strip().rstrip(".")
            + "."
        )
    return ""


# ── 商品上下文提炼 ────────────────────────────────────────────────────────

def _shorten(text: str, n: int) -> str:
    text = re.sub(r"\s+", " ", (text or "")).strip()
    return text[:n]

def _clean_title(title: str) -> str:
    """去掉常见堆砌词，让标题更可读。"""
    t = _shorten(title, 60)
    return re.sub(r"[【【】】\\[\\]|/]", " ", t).strip()

def build_context(raw: dict, fallback_name: str = "", fallback_desc: str = "") -> dict:
    """从采集数据提炼生图上下文：标题、描述行、卖点列表、参数行。

    raw: {title, desc, price, params: [{key, value}], sellingPoints}
    """
    title = _clean_title(raw.get("title") or fallback_name or "product")
    desc = _shorten(raw.get("desc") or fallback_desc or "", 200)

    # 卖点：优先显式卖点文本；否则从参数表里挑短而有信息量的键值对
    selling_points: list[dict] = []
    sp_text = raw.get("sellingPoints") or ""
    if sp_text:
        for chunk in re.split(r"[；;\n·]+|(?<!\d)\.(?!\d)", sp_text):
            c = _shorten(chunk, 20)
            if c and len(selling_points) < 5:
                selling_points.append({"zh": c, "en": c})
    if not selling_points:
        for p in (raw.get("params") or [])[:12]:
            k, v = _shorten(p.get("key", ""), 10), _shorten(str(p.get("value", "")), 16)
            if k and v and k.lower() not in {"货号", "sku", "isbn", "上架时间"}:
                selling_points.append({"zh": f"{k} {v}", "en": f"{k} {v}"})
            if len(selling_points) >= 5:
                break

    params_line = "; ".join(
        f"{p.get('key')}: {p.get('value')}" for p in (raw.get("params") or [])[:8]
    )
    return {
        "title": title,
        "title_en": title,  # 采集源多为中文标题，英文场景直接用原词避免乱翻译
        "desc": desc,
        "selling_points": selling_points[:3],
        "params_line": params_line,
        "price": raw.get("price") or "",
    }

def selling_point_lines(ctx: dict, lang: str, max_n: int = 3) -> str:
    """卖点列表 → 单行文案（图内 callout/标题用），无卖点返回空串。"""
    sps = ctx["selling_points"][:max_n]
    if not sps:
        return ""
    key = "zh" if lang == "zh" else "en"
    return "; ".join(s[key] for s in sps if s.get(key))


def type_name(type_id: str) -> str:
    return TYPE_NAMES_ZH.get(type_id, type_id)