#!/usr/bin/env python3
"""
Word 导出脚本 - word-exporter skill
用法: python3 export_word.py --title "标题" --content "内容" --output "输出路径"
"""
import sys
import os
import argparse

# 确保依赖路径在 sys.path 中
sys.path.insert(0, '/workspace/.local/lib/python3.11/site-packages')

from docx import Document
from docx.shared import Pt, RGBColor
from docx.enum.text import WD_ALIGN_PARAGRAPH


def export_word(title: str, content: str, output_path: str) -> str:
    """将内容导出为 Word 文档"""
    doc = Document()

    # 设置标题
    heading = doc.add_heading(title, level=0)
    heading.alignment = WD_ALIGN_PARAGRAPH.CENTER

    # 按行解析内容（简单 Markdown 支持）
    for line in content.split('\n'):
        line = line.rstrip()
        if line.startswith('## '):
            doc.add_heading(line[3:], level=2)
        elif line.startswith('# '):
            doc.add_heading(line[2:], level=1)
        elif line.startswith('- ') or line.startswith('* '):
            p = doc.add_paragraph(line[2:], style='List Bullet')
        elif line.strip() == '':
            doc.add_paragraph('')
        else:
            doc.add_paragraph(line)

    # 确保输出目录存在
    os.makedirs(os.path.dirname(output_path), exist_ok=True)

    doc.save(output_path)
    return output_path


if __name__ == '__main__':
    parser = argparse.ArgumentParser(description='导出 Word 文档')
    parser.add_argument('--title', required=True, help='文档标题')
    parser.add_argument('--content', required=True, help='文档正文内容')
    parser.add_argument('--output', required=True, help='输出文件路径')
    args = parser.parse_args()

    output = export_word(args.title, args.content, args.output)
    print(f'output_path: {output}')
    print('Word 文档导出成功')