#!/usr/bin/env python3
"""
fast_pptx_renderer.py — 极速 PPTX 渲染器
接收 DeckContract JSON，生成专业 PPTX

使用方法:
    python fast_pptx_renderer.py --deck-json <path> --output <path> --theme <theme>

主题:
    deepblue  - 深海蓝（默认）
    caramel   - 焦糖色
    chinese   - 中国红

输出:
    JSON 格式: { "ok": true, "path": "...", "size": 12345, "slides": 7 }
"""
from __future__ import annotations

import argparse
import json
import os
import sys
import time
from pathlib import Path
from typing import Any

from pptx import Presentation
from pptx.dml.color import RGBColor
from pptx.enum.shapes import MSO_AUTO_SHAPE_TYPE
from pptx.enum.text import MSO_ANCHOR, PP_ALIGN
from pptx.util import Inches, Pt


# ─── 主题配色 ────────────────────────────────────────────────

THEMES = {
    "deepblue": {
        "bg": RGBColor(10, 28, 50),
        "bg_gradient_end": RGBColor(15, 40, 70),
        "panel": RGBColor(255, 255, 255),
        "primary": RGBColor(255, 255, 255),
        "text_dark": RGBColor(30, 30, 40),
        "accent": RGBColor(24, 137, 141),
        "accent_light": RGBColor(200, 230, 230),
        "gold": RGBColor(201, 160, 39),
        "muted": RGBColor(150, 160, 175),
        "line": RGBColor(220, 225, 235),
        "good": RGBColor(24, 168, 100),
        "alert": RGBColor(200, 60, 60),
    },
    "caramel": {
        "bg": RGBColor(248, 245, 239),
        "bg_gradient_end": RGBColor(240, 235, 225),
        "panel": RGBColor(255, 255, 255),
        "primary": RGBColor(38, 38, 38),
        "text_dark": RGBColor(38, 38, 38),
        "accent": RGBColor(206, 138, 44),
        "accent_light": RGBColor(240, 230, 208),
        "gold": RGBColor(206, 138, 44),
        "muted": RGBColor(96, 96, 96),
        "line": RGBColor(223, 214, 195),
        "good": RGBColor(24, 108, 79),
        "alert": RGBColor(173, 68, 52),
    },
    "chinese": {
        "bg": RGBColor(180, 30, 30),
        "bg_gradient_end": RGBColor(140, 20, 20),
        "panel": RGBColor(255, 255, 255),
        "primary": RGBColor(255, 255, 255),
        "text_dark": RGBColor(40, 40, 40),
        "accent": RGBColor(220, 180, 60),
        "accent_light": RGBColor(255, 240, 200),
        "gold": RGBColor(220, 180, 60),
        "muted": RGBColor(180, 160, 160),
        "line": RGBColor(200, 180, 180),
        "good": RGBColor(100, 180, 100),
        "alert": RGBColor(255, 200, 100),
    },
}

FONT_CN = "Microsoft YaHei"


# ─── 工具函数 ────────────────────────────────────────────────

def money(value: float | int) -> str:
    """格式化金额。"""
    return f"{round(float(value)):,}"


def pct(value: float) -> str:
    """格式化百分比。"""
    return f"{value:.2f}%"


def add_bg(slide, colors: dict):
    """设置幻灯片背景色。"""
    fill = slide.background.fill
    fill.solid()
    fill.fore_color.rgb = colors["bg"]


def add_textbox(slide, left, top, width, height, text, size=18, bold=False,
                color=None, align=PP_ALIGN.LEFT, line_spacing=1.5, colors=None):
    """添加文本框。"""
    if color is None:
        color = colors["primary"] if colors else RGBColor(255, 255, 255)
    box = slide.shapes.add_textbox(left, top, width, height)
    tf = box.text_frame
    tf.clear()
    tf.word_wrap = True
    tf.vertical_anchor = MSO_ANCHOR.TOP
    p = tf.paragraphs[0]
    p.alignment = align
    p.line_spacing = line_spacing
    run = p.add_run()
    run.text = str(text)
    run.font.name = FONT_CN
    run.font.size = Pt(size)
    run.font.bold = bold
    run.font.color.rgb = color
    return box


def add_paragraphs(slide, left, top, width, height, lines, size=15, color=None,
                   bullet=False, line_spacing=1.5, colors=None):
    """添加多行文本。"""
    if color is None:
        color = colors["primary"] if colors else RGBColor(255, 255, 255)
    box = slide.shapes.add_textbox(left, top, width, height)
    tf = box.text_frame
    tf.clear()
    tf.word_wrap = True
    tf.vertical_anchor = MSO_ANCHOR.TOP
    for idx, line in enumerate(lines):
        p = tf.paragraphs[0] if idx == 0 else tf.add_paragraph()
        p.alignment = PP_ALIGN.LEFT
        p.line_spacing = line_spacing
        p.text = f"• {line}" if bullet else str(line)
        if p.runs:
            run = p.runs[0]
            run.font.name = FONT_CN
            run.font.size = Pt(size)
            run.font.color.rgb = color
    return box


def add_card(slide, left, top, width, height, title, value, subtitle="",
             accent=False, colors=None, title_size=12, value_size=20, subtitle_size=10.5):
    """添加卡片。"""
    if colors is None:
        colors = THEMES["deepblue"]
    shape = slide.shapes.add_shape(MSO_AUTO_SHAPE_TYPE.ROUNDED_RECTANGLE, left, top, width, height)
    shape.fill.solid()
    shape.fill.fore_color.rgb = colors["accent_light"] if accent else colors["panel"]
    shape.line.color.rgb = colors["line"]
    add_textbox(slide, left + Inches(0.14), top + Inches(0.12), width - Inches(0.28),
                Inches(0.22), title, size=title_size, bold=True, color=colors["muted"],
                line_spacing=1.2, colors=colors)
    add_textbox(slide, left + Inches(0.14), top + Inches(0.38), width - Inches(0.28),
                Inches(0.4), value, size=value_size, bold=True,
                color=colors["accent"] if accent else colors["text_dark"],
                line_spacing=1.2, colors=colors)
    if subtitle:
        add_textbox(slide, left + Inches(0.14), top + Inches(0.82), width - Inches(0.28),
                    Inches(0.22), subtitle, size=subtitle_size, color=colors["muted"],
                    line_spacing=1.2, colors=colors)


def add_footer(slide, text, colors=None):
    """添加页脚。"""
    if colors is None:
        colors = THEMES["deepblue"]
    add_textbox(slide, Inches(0.65), Inches(7.0), Inches(12.0), Inches(0.2),
                text, size=10, color=colors["muted"], line_spacing=1.2, colors=colors)


# ─── 幻灯片生成 ──────────────────────────────────────────────

def add_cover_slide(prs: Presentation, product: dict, customer: dict, company: dict, colors: dict):
    """封面页。"""
    slide = prs.slides.add_slide(prs.slide_layouts[6])  # 空白布局
    add_bg(slide, colors)

    product_name = product.get("productName", "保险计划书")
    customer_name = customer.get("name", "客户")
    company_name = company.get("displayName", "")

    # 客户姓名
    add_textbox(slide, Inches(0.8), Inches(1.5), Inches(11.7), Inches(1.0),
                customer_name, font_size=42, bold=True, colors=colors)

    # 副标题
    add_textbox(slide, Inches(0.8), Inches(2.5), Inches(11.7), Inches(0.6),
                "财富增值与传承方案", size=24, color=colors["muted"], colors=colors)

    # 产品名称
    add_textbox(slide, Inches(0.8), Inches(3.5), Inches(11.7), Inches(0.5),
                product_name, size=16, color=colors["muted"], colors=colors)

    # 公司名称
    if company_name:
        add_textbox(slide, Inches(0.8), Inches(4.2), Inches(11.7), Inches(0.4),
                    company_name, size=14, color=colors["accent"], colors=colors)

    # 关键指标卡片
    policy = product.get("policy", {})
    annual_premium = policy.get("annualPremium", 0)
    pay_years = policy.get("payYears", 0)
    benefit_rows = product.get("benefitRows", [])

    total_premium = annual_premium * pay_years
    multiple = "-"
    if benefit_rows:
        last_row = benefit_rows[-1]
        final_value = last_row.get("totalSurrenderValue", 0)
        if total_premium > 0:
            multiple = f"{final_value / total_premium:.1f}x"

    metrics = [
        ("年缴保费", f"${money(annual_premium)}"),
        ("缴费年期", f"{pay_years}年"),
        ("总投入", f"${money(total_premium)}"),
        ("期末倍数", multiple),
    ]
    for i, (label, value) in enumerate(metrics):
        x = Inches(0.8 + i * 3.0)
        add_card(slide, x, Inches(5.0), Inches(2.5), Inches(1.2),
                 label, value, colors=colors, accent=(i == 3))


def add_overview_slide(prs: Presentation, product: dict, company: dict, colors: dict):
    """数据总览页。"""
    slide = prs.slides.add_slide(prs.slide_layouts[6])
    add_bg(slide, colors)

    # 标题
    add_textbox(slide, Inches(0.5), Inches(0.3), Inches(12.3), Inches(0.6),
                "保单数据总览", size=28, bold=True, colors=colors)

    # 基本信息
    insured = product.get("insured", {})
    policy = product.get("policy", {})

    info_items = [
        ("产品名称", product.get("productName", "")),
        ("受保人", f"{insured.get('name', '')} ({insured.get('age', '')}岁)"),
        ("保单货币", policy.get("currency", "USD")),
        ("年缴保费", f"${money(policy.get('annualPremium', 0))}"),
        ("缴费年期", f"{policy.get('payYears', 0)}年"),
        ("保障期间", policy.get("coveragePeriod", "")),
    ]

    for i, (label, value) in enumerate(info_items):
        y = Inches(1.2 + i * 0.6)
        # 标签
        add_textbox(slide, Inches(1.0), y, Inches(3.0), Inches(0.4),
                    label, size=14, color=colors["muted"], colors=colors)
        # 值
        add_textbox(slide, Inches(4.0), y, Inches(8.0), Inches(0.4),
                    str(value), size=14, bold=True, color=colors["text_dark"], colors=colors)

    # 销售洞察
    sales_insights = product.get("salesInsights", {})
    if sales_insights:
        add_textbox(slide, Inches(0.5), Inches(5.0), Inches(12.3), Inches(0.5),
                    "销售洞察", size=20, bold=True, colors=colors)

        key_points = sales_insights.get("keySellingPoints", [])
        if key_points:
            add_paragraphs(slide, Inches(1.0), Inches(5.5), Inches(11.0), Inches(1.5),
                           key_points[:4], size=13, bullet=True, colors=colors)


def add_benefit_table_slide(prs: Presentation, product: dict, colors: dict):
    """利益演示页。"""
    benefit_rows = product.get("benefitRows", [])
    if not benefit_rows:
        return

    slide = prs.slides.add_slide(prs.slide_layouts[6])
    add_bg(slide, colors)

    # 标题
    add_textbox(slide, Inches(0.5), Inches(0.3), Inches(12.3), Inches(0.6),
                "逐年利益演示", size=28, bold=True, colors=colors)

    # 表头
    headers = ["保单年度", "年龄", "已缴保费", "保证现金价值", "归原红利", "终期分红", "退保发还总额"]
    col_widths = [1.5, 1.2, 1.8, 2.0, 1.8, 1.8, 2.2]
    col_x = [0.5]
    for w in col_widths[:-1]:
        col_x.append(col_x[-1] + w)

    for header, x in zip(headers, col_x):
        add_textbox(slide, Inches(x), Inches(1.1), Inches(col_widths[headers.index(header)]),
                    Inches(0.4), header, size=11, bold=True, color=colors["accent"], colors=colors)

    # 数据行（最多20行）
    display_rows = benefit_rows[:20]
    for row_idx, row in enumerate(display_rows):
        y = Inches(1.5 + row_idx * 0.35)
        values = [
            str(row.get("policyYear", "")),
            str(row.get("age", "")),
            f"${money(row.get('totalPremiumPaid', 0))}",
            f"${money(row.get('guaranteedCashValue', 0))}",
            f"${money(row.get('reversionaryBonus', 0))}",
            f"${money(row.get('terminalDividend', 0))}",
            f"${money(row.get('totalSurrenderValue', 0))}",
        ]
        for value, x in zip(values, col_x):
            add_textbox(slide, Inches(x), y, Inches(1.5), Inches(0.3),
                        value, size=10, color=colors["text_dark"], colors=colors)

    add_footer(slide, f"共 {len(benefit_rows)} 年数据，显示前 20 年", colors=colors)


def add_withdrawal_slide(prs: Presentation, product: dict, colors: dict):
    """退保演示页。"""
    withdrawal_rows = product.get("withdrawalRows", [])
    if not withdrawal_rows:
        return

    slide = prs.slides.add_slide(prs.slide_layouts[6])
    add_bg(slide, colors)

    # 标题
    add_textbox(slide, Inches(0.5), Inches(0.3), Inches(12.3), Inches(0.6),
                "退保/提取演示", size=28, bold=True, colors=colors)

    # 表头
    headers = ["保单年度", "年龄", "当年提取", "累计提取", "提取后退保价值"]
    col_widths = [1.8, 1.5, 2.0, 2.0, 2.5]
    col_x = [0.8]
    for w in col_widths[:-1]:
        col_x.append(col_x[-1] + w)

    for header, x in zip(headers, col_x):
        add_textbox(slide, Inches(x), Inches(1.1), Inches(col_widths[headers.index(header)]),
                    Inches(0.4), header, size=12, bold=True, color=colors["accent"], colors=colors)

    # 数据行
    display_rows = withdrawal_rows[:15]
    for row_idx, row in enumerate(display_rows):
        y = Inches(1.6 + row_idx * 0.38)
        values = [
            str(row.get("policyYear", "")),
            str(row.get("age", "")),
            f"${money(row.get('annualWithdrawal', 0))}",
            f"${money(row.get('cumulativeWithdrawal', 0))}",
            f"${money(row.get('surrenderValueAfter', 0))}",
        ]
        for value, x in zip(values, col_x):
            add_textbox(slide, Inches(x), y, Inches(2.0), Inches(0.3),
                        value, size=11, color=colors["text_dark"], colors=colors)


def add_summary_slide(prs: Presentation, product: dict, colors: dict):
    """总结页。"""
    slide = prs.slides.add_slide(prs.slide_layouts[6])
    add_bg(slide, colors)

    policy = product.get("policy", {})
    benefit_rows = product.get("benefitRows", [])

    annual_premium = policy.get("annualPremium", 0)
    pay_years = policy.get("payYears", 0)
    total_premium = annual_premium * pay_years

    final_value = 0
    multiple = "-"
    if benefit_rows:
        last_row = benefit_rows[-1]
        final_value = last_row.get("totalSurrenderValue", 0)
        if total_premium > 0:
            multiple = f"{final_value / total_premium:.1f}x"

    # 标题
    add_textbox(slide, Inches(0.8), Inches(2.0), Inches(11.7), Inches(1.0),
                "方案总结", size=36, bold=True, align=PP_ALIGN.CENTER, colors=colors)

    # 核心数据
    add_textbox(slide, Inches(0.8), Inches(3.2), Inches(11.7), Inches(0.6),
                f"总投入 ${money(total_premium)} → 期末价值 ${money(final_value)}",
                size=24, color=colors["muted"], align=PP_ALIGN.CENTER, colors=colors)

    add_textbox(slide, Inches(0.8), Inches(4.0), Inches(11.7), Inches(0.5),
                f"倍数增长 {multiple}", size=28, bold=True, color=colors["gold"],
                align=PP_ALIGN.CENTER, colors=colors)

    # 销售洞察
    sales_insights = product.get("salesInsights", {})
    if sales_insights:
        narrative = sales_insights.get("suggestedNarrative", "")
        if narrative:
            add_textbox(slide, Inches(1.5), Inches(5.0), Inches(10.3), Inches(1.0),
                        narrative, size=16, color=colors["muted"],
                        align=PP_ALIGN.CENTER, colors=colors)


# ─── 主渲染函数 ──────────────────────────────────────────────

def render_deck(deck: dict, output_path: str, theme: str = "deepblue") -> dict:
    """渲染 DeckContract 为 PPTX。"""
    colors = THEMES.get(theme, THEMES["deepblue"])

    prs = Presentation()
    prs.slide_width = Inches(13.33)
    prs.slide_height = Inches(7.5)

    products = deck.get("products", [])
    customer = deck.get("customer", {})
    company = deck.get("company", {})

    for product in products:
        # 封面页
        add_cover_slide(prs, product, customer, company, colors)
        # 数据总览
        add_overview_slide(prs, product, company, colors)
        # 利益演示
        add_benefit_table_slide(prs, product, colors)
        # 退保演示（如果有）
        add_withdrawal_slide(prs, product, colors)
        # 总结页
        add_summary_slide(prs, product, colors)

    # 保存
    prs.save(output_path)

    # 统计
    file_size = os.path.getsize(output_path)
    slide_count = len(prs.slides)

    return {
        "ok": True,
        "path": output_path,
        "size": file_size,
        "slides": slide_count,
    }


# ─── CLI 入口 ────────────────────────────────────────────────

def main():
    parser = argparse.ArgumentParser(description="Fast PPTX Renderer")
    parser.add_argument("--deck-json", required=True, help="DeckContract JSON 文件路径")
    parser.add_argument("--output", required=True, help="输出 PPTX 文件路径")
    parser.add_argument("--theme", default="deepblue", choices=["deepblue", "caramel", "chinese"],
                        help="主题配色")
    args = parser.parse_args()

    try:
        # 读取 DeckContract JSON
        with open(args.deck_json, "r", encoding="utf-8") as f:
            deck = json.load(f)

        # 渲染
        result = render_deck(deck, args.output, args.theme)

        # 输出结果到 stdout
        print(json.dumps(result, ensure_ascii=False))

    except Exception as e:
        print(json.dumps({"ok": False, "error": str(e)}, ensure_ascii=False))
        sys.exit(1)


if __name__ == "__main__":
    main()