import os
import re
from docx import Document
from docx.shared import Pt, Inches, Cm, RGBColor
from docx.enum.text import WD_ALIGN_PARAGRAPH
from docx.oxml.ns import qn


def generate_person_doc(person_name: str, person_info: str, photo_path: str,
                        stories: list[dict], output_path: str):
    """
    为一位讲述者生成最终 DOCX 文档

    person_info: 个人信息文本
    photo_path: 照片路径（可为空）
    stories: [{title, content}]
    """
    doc = Document()

    # 设置默认字体
    style = doc.styles['Normal']
    font = style.font
    font.name = '宋体'
    font.size = Pt(12)
    style.element.rPr.rFonts.set(qn('w:eastAsia'), '宋体')

    # 照片
    if photo_path and os.path.exists(photo_path):
        try:
            doc.add_picture(photo_path, width=Inches(1.5))
            last_paragraph = doc.paragraphs[-1]
            last_paragraph.alignment = WD_ALIGN_PARAGRAPH.CENTER
        except Exception:
            pass

    # 姓名
    name_para = doc.add_paragraph()
    name_para.alignment = WD_ALIGN_PARAGRAPH.CENTER
    name_run = name_para.add_run(person_name)
    name_run.font.size = Pt(18)
    name_run.font.bold = True
    name_run.font.name = '黑体'
    name_run.element.rPr.rFonts.set(qn('w:eastAsia'), '黑体')

    # 个人信息
    if person_info:
        info_para = doc.add_paragraph()
        info_para.alignment = WD_ALIGN_PARAGRAPH.CENTER
        info_run = info_para.add_run(person_info.strip())
        info_run.font.size = Pt(11)
        info_run.font.color.rgb = RGBColor(0x66, 0x66, 0x66)

    # 分隔线
    doc.add_paragraph("─" * 30).alignment = WD_ALIGN_PARAGRAPH.CENTER

    # 故事
    for i, story in enumerate(stories):
        # 故事标题
        title_para = doc.add_paragraph()
        title_run = title_para.add_run(story.get("title", f"故事 {i + 1}"))
        title_run.font.size = Pt(14)
        title_run.font.bold = True
        title_run.font.name = '黑体'
        title_run.element.rPr.rFonts.set(qn('w:eastAsia'), '黑体')

        # 故事梗概
        summary = story.get("summary", "")
        if summary:
            p = doc.add_paragraph()
            run = p.add_run("【故事梗概】")
            run.font.size = Pt(11)
            run.font.bold = True
            run.font.color.rgb = RGBColor(0x33, 0x33, 0x99)
            p = doc.add_paragraph(summary.strip())
            p.paragraph_format.first_line_indent = Cm(0.74)
            p.paragraph_format.line_spacing = 1.5
            for run in p.runs:
                run.font.size = Pt(11)

        # 关键词标签
        tags = story.get("tags", [])
        if tags:
            p = doc.add_paragraph()
            run = p.add_run("【关键词标签】")
            run.font.size = Pt(11)
            run.font.bold = True
            run.font.color.rgb = RGBColor(0x33, 0x33, 0x99)
            tag_text = "、".join(tags)
            p = doc.add_paragraph(tag_text)
            p.paragraph_format.line_spacing = 1.5
            for run in p.runs:
                run.font.size = Pt(10)
                run.font.color.rgb = RGBColor(0x66, 0x66, 0x66)

        # 故事内容（支持 Markdown 格式）
        content = story.get("content", "")
        if content:
            _add_markdown_content(doc, content)

        # 多个故事之间加分隔
        if i < len(stories) - 1:
            doc.add_paragraph("─" * 30).alignment = WD_ALIGN_PARAGRAPH.CENTER

    # 保存
    os.makedirs(os.path.dirname(output_path), exist_ok=True)
    doc.save(output_path)
    return output_path


def _add_markdown_content(doc, content: str):
    """将 Markdown 格式的内容添加到 DOCX，支持标题、引用、列表等"""
    lines = content.split("\n")
    i = 0
    while i < len(lines):
        line = lines[i].strip()

        if not line:
            i += 1
            continue

        # ## 二级标题
        if line.startswith("## "):
            title = line[3:].strip()
            p = doc.add_paragraph()
            run = p.add_run(title)
            run.font.size = Pt(13)
            run.font.bold = True
            run.font.name = '黑体'
            run.element.rPr.rFonts.set(qn('w:eastAsia'), '黑体')
            p.paragraph_format.space_before = Pt(12)

        # ### 三级标题
        elif line.startswith("### "):
            title = line[4:].strip()
            p = doc.add_paragraph()
            run = p.add_run(title)
            run.font.size = Pt(12)
            run.font.bold = True
            run.font.name = '黑体'
            run.element.rPr.rFonts.set(qn('w:eastAsia'), '黑体')
            p.paragraph_format.space_before = Pt(6)

        # > 引用
        elif line.startswith("> "):
            quote_text = line[2:].strip()
            p = doc.add_paragraph()
            p.paragraph_format.left_indent = Cm(1)
            run = p.add_run(quote_text)
            run.font.italic = True
            run.font.color.rgb = RGBColor(0x55, 0x55, 0x55)
            run.font.size = Pt(11)

        # - 列表项
        elif line.startswith("- "):
            item_text = line[2:].strip()
            p = doc.add_paragraph(style='List Bullet')
            run = p.add_run(item_text)
            run.font.size = Pt(12)

        # 普通段落
        else:
            p = doc.add_paragraph(line)
            p.paragraph_format.first_line_indent = Cm(0.74)
            p.paragraph_format.line_spacing = 1.5

        i += 1