#!/usr/bin/env python3
"""
render_article.py — 将文本渲染成单张 article-3-4 卡片图

LLM 负责分页逻辑，每页单独调用本脚本。

用法（单页模式，推荐）:
  python3 render_article.py \
    --title "文章标题" \
    --text "该页正文（段落间空行分隔）" \
    --page-num 1 \
    --page-total 3 \
    --out /path/to/workspace/tmp/card_01.png \
    [--highlight "#22a854"] \
    [--bg "#e6f5ef"] \
    [--footer "公众号 · 早早集市"]

用法（批量模式，兼容保留）:
  python3 render_article.py \
    --title "文章标题" \
    --text "全文..." \
    --out-dir /path/to/output \
    [--chars-per-page 280]
"""

import argparse, shutil, subprocess, sys, tempfile, re
from html import escape
from pathlib import Path

SKILL_DIR = Path(__file__).parent.parent
TEMPLATE_PATH = SKILL_DIR / "assets" / "templates" / "article-3-4.html"
MD_CSS_PATH = SKILL_DIR / "assets" / "styles" / "md.css"
ICONS_DIR = SKILL_DIR / "assets" / "icons"
FONTS_DIR = SKILL_DIR / "assets" / "fonts"

CHROME_PATHS = [
    "/Applications/Google Chrome.app/Contents/MacOS/Google Chrome",
    "/Applications/Chromium.app/Contents/MacOS/Chromium",
    "google-chrome",
    "chromium",
]

W, H = 900, 1200
CHARS_PER_PAGE = 280


def find_chrome():
    for p in CHROME_PATHS:
        if Path(p).exists() or shutil.which(p):
            return p
    return None


def split_at_sentence_boundary(text: str, limit: int) -> tuple:
    """
    在 limit 字符以内，找最后一个句子结束符（。！？…）处截断。
    宁少勿多：找不到就在最后一个逗号/分号处截，再找不到就硬截到 limit*0.85。
    返回 (taken, rest)
    """
    if len(text) <= limit:
        return text, ''

    # 在 limit 范围内从后往前找强句末
    strong_ends = set('。！？…\n')
    weak_ends = set('，；,.;')

    candidate = -1
    for i in range(min(limit, len(text)) - 1, max(limit // 2, 0) - 1, -1):
        if text[i] in strong_ends:
            candidate = i + 1
            break

    if candidate == -1:
        # 找弱分隔符
        for i in range(min(limit, len(text)) - 1, max(limit // 2, 0) - 1, -1):
            if text[i] in weak_ends:
                candidate = i + 1
                break

    if candidate == -1:
        # 实在没有，保守截到 85%
        candidate = int(limit * 0.85)

    return text[:candidate].strip(), text[candidate:].strip()


def split_text_into_pages(text: str, chars_per_page: int) -> list:
    """
    按段落优先、句子边界兜底的分页逻辑。
    宁少勿多：每页预留 10% buffer，不塞满。
    """
    safe_limit = int(chars_per_page * 0.9)  # 保守上限
    paragraphs = [p.strip() for p in re.split(r'\n{2,}', text.strip()) if p.strip()]

    pages = []
    current_chunks = []
    current_len = 0

    for para in paragraphs:
        # 超长段落先按句子边界切碎
        while len(para) > safe_limit:
            taken, para = split_at_sentence_boundary(para, safe_limit)
            if current_chunks:
                pages.append(current_chunks)
                current_chunks = []
                current_len = 0
            pages.append([taken])

        if not para:
            continue

        # 加入当前页会不会超限
        if current_len + len(para) > safe_limit and current_chunks:
            pages.append(current_chunks)
            current_chunks = []
            current_len = 0

        current_chunks.append(para)
        current_len += len(para)

    if current_chunks:
        pages.append(current_chunks)

    return pages


def text_to_html(text: str) -> str:
    """把文本整体交给 markdown 渲染，支持完整 MD 语法"""
    try:
        import markdown as md_lib
    except ImportError:
        sys.exit('需要安装 markdown 库：pip install markdown')
    return md_lib.markdown(text, extensions=['fenced_code', 'tables', 'nl2br'])


def md_to_html(text: str) -> str:
    """把 Markdown 转成 HTML 片段，需要 pip install markdown"""
    try:
        import markdown
        return markdown.markdown(text, extensions=['fenced_code', 'tables', 'nl2br'])
    except ImportError:
        sys.exit('需要安装 markdown 库：pip install markdown')


def render_page(chrome, tpl, out_path, title, content_html, page_label, bottom_tip,
                highlight, bg, footer, icon_path, avatar_path, font_path, md_css_path=''):
    html = tpl
    replacements = {
        '{{MD_CSS_PATH}}': str(md_css_path) if md_css_path else '',
        '{{TITLE}}': escape(title),
        '{{CONTENT_HTML}}': content_html,
        '{{PAGE_LABEL}}': escape(page_label),
        '{{BOTTOM_TIP}}': escape(bottom_tip),
        '{{HIGHLIGHT_COLOR}}': highlight,
        '{{BG_COLOR}}': bg,
        '{{FOOTER_TEXT}}': escape(footer),
        '{{ICON_PATH}}': icon_path,
        '{{AVATAR_PATH}}': avatar_path,
        '{{FONT_PATH}}': font_path,
    }
    for k, v in replacements.items():
        html = html.replace(k, v)

    with tempfile.NamedTemporaryFile(suffix='.html', delete=False, mode='w', encoding='utf-8') as f:
        f.write(html)
        tmp_html = f.name

    cmd = [
        chrome, '--headless', '--disable-gpu', '--no-sandbox',
        f'--screenshot={out_path}',
        f'--window-size={W},{H}',
        f'file://{tmp_html}',
    ]
    result = subprocess.run(cmd, capture_output=True)
    Path(tmp_html).unlink(missing_ok=True)
    if result.returncode != 0:
        sys.exit(f'Chrome failed:\n{result.stderr.decode()}')
    print(f'✅ {out_path}')


def main():
    ap = argparse.ArgumentParser()
    ap.add_argument('--title', required=True)
    ap.add_argument('--text', default='')
    ap.add_argument('--text-file', default='')
    # 单页模式
    ap.add_argument('--page-num', type=int, default=0)
    ap.add_argument('--page-total', type=int, default=0)
    ap.add_argument('--out', default='')
    # 批量模式
    ap.add_argument('--out-dir', default='')
    ap.add_argument('--chars-per-page', type=int, default=CHARS_PER_PAGE)
    ap.add_argument('--md', action='store_true', help='输入为 Markdown，自动转 HTML 渲染')
    # 样式
    ap.add_argument('--highlight', default='#3d6b4f')
    ap.add_argument('--bg', default='#f9fcfa')
    ap.add_argument('--footer', default='公众号 · 早早集市')
    ap.add_argument('--icon', default='')
    args = ap.parse_args()

    if args.text_file:
        text = Path(args.text_file).read_text(encoding='utf-8')
    elif args.text:
        text = args.text
    else:
        sys.exit('需要 --text 或 --text-file')

    chrome = find_chrome()
    if not chrome:
        sys.exit('Chrome/Chromium not found')

    # MD 和纯文本共用同一模板，都走 markdown 渲染
    tpl = TEMPLATE_PATH.read_text(encoding='utf-8')
    icon_path = args.icon or str(ICONS_DIR / 'zzclub-logo-gray.svg')
    avatar_path = str(ICONS_DIR / 'avatar_jinx_cartoon.jpg')
    font_path = str(FONTS_DIR / 'AlimamaShuHeiTi-Bold.ttf')
    md_css_path = str(MD_CSS_PATH)

    def to_content_html(t: str) -> str:
        return text_to_html(t)  # text_to_html 已内置 markdown 渲染

    # 单页模式
    if args.page_num > 0 and args.out:
        page_total = args.page_total if args.page_total > 0 else args.page_num
        page_label = f'{args.page_num} / {page_total}'
        bottom_tip = '· 全文完' if args.page_num == page_total else '← 滑动查看更多'
        render_page(
            chrome=chrome, tpl=tpl, out_path=Path(args.out),
            title=args.title, content_html=to_content_html(text),
            page_label=page_label, bottom_tip=bottom_tip,
            highlight=args.highlight, bg=args.bg, footer=args.footer,
            icon_path=icon_path, avatar_path=avatar_path, font_path=font_path,
            md_css_path=md_css_path,
        )
        return

    # 批量模式
    if not args.out_dir:
        sys.exit('需要 --out (单页模式) 或 --out-dir (批量模式)')

    pages = split_text_into_pages(text, args.chars_per_page)
    total = len(pages)
    print(f'共 {total} 页')
    out_dir = Path(args.out_dir)
    out_dir.mkdir(parents=True, exist_ok=True)
    for i, chunks in enumerate(pages, 1):
        parts = []
        for chunk in chunks:
            c = escape(chunk)
            c = re.sub(r'\*\*(.+?)\*\*', r'<strong>\1</strong>', c)
            parts.append(f'<p>{c}</p>')
        content_html = '\n'.join(parts)
        page_label = f'{i} / {total}'
        bottom_tip = '· 全文完' if i == total else '← 滑动查看更多'
        render_page(
            chrome=chrome, tpl=tpl, out_path=out_dir / f'card_{i:02d}.png',
            title=args.title, content_html=content_html,
            page_label=page_label, bottom_tip=bottom_tip,
            highlight=args.highlight, bg=args.bg, footer=args.footer,
            icon_path=icon_path, avatar_path=avatar_path, font_path=font_path,
        )
    print(f'\n🎉 完成，共输出 {total} 张图到 {out_dir}')


if __name__ == '__main__':
    main()