#!/usr/bin/env python3
"""
render_card.py — 用模板渲染卡片图，Chrome headless 截图输出 PNG

用法:
  python3 render_card.py \
    --template poster-3-4 \
    --out /tmp/card.png \
    --line1 "OpenClaw" \
    --line2 "有两层" \
    --line3 "model 配置" \
    --highlight "#22a854" \
    --bg "#e6f5ef" \
    --footer "公众号 · 早早集市"

模板位于 assets/templates/<template>.html（相对于本脚本所在的 skills/z-card-image/）
"""

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

SKILL_DIR = Path(__file__).parent.parent
TEMPLATES_DIR = SKILL_DIR / "assets" / "templates"
ICONS_DIR = SKILL_DIR / "assets" / "icons"
WECHAT_SPLIT_DEFAULT_ICON = SKILL_DIR / "assets" / "icons" / "openclaw-logo.svg"

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

WECHAT_SPLIT_WINDOW_EXTRA_HEIGHT = 87

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

def main():
    ap = argparse.ArgumentParser()
    ap.add_argument("--template", default="poster-3-4")
    ap.add_argument("--out", required=True)
    ap.add_argument("--line1", default="")
    ap.add_argument("--line2", default="")
    ap.add_argument("--line3", default="")
    ap.add_argument("--hl1", action="store_true", help="第一行高亮")
    ap.add_argument("--hl2", action="store_true", help="第二行高亮")
    ap.add_argument("--hl3", action="store_true", help="第三行高亮")
    ap.add_argument("--highlight", default="#22a854")
    ap.add_argument("--bg", default="#e6f5ef")
    ap.add_argument("--footer", default="公众号")
    ap.add_argument("--icon", default=None, help="顶部图标路径，不传则自动判断")
    ap.add_argument("--highlight-words", default="", help="要高亮的词，逗号分隔，如 '测试,openclaw'")
    args = ap.parse_args()

    if args.template == "wechat-cover-split":
        lines = [line for line in [args.line1, args.line2, args.line3] if line]
        args.line1 = lines[0] if lines else ""
        args.line2 = "".join(lines[1:]) if len(lines) > 1 else ""
        args.line3 = ""
        args.hl2 = args.hl2 or args.hl3
        args.hl3 = False

    # 自动选图标
    if args.icon:
        icon_path = args.icon
    elif args.template == "wechat-cover-split":
        icon_path = WECHAT_SPLIT_DEFAULT_ICON
    else:
        texts = " ".join([args.line1, args.line2, args.line3]).lower()
        if "openclaw" in texts:
            icon_path = str(ICONS_DIR / "openclaw-logo.svg")
        else:
            icon_path = str(ICONS_DIR / "zzclub-logo-gray.svg")

    tpl_path = TEMPLATES_DIR / f"{args.template}.html"
    if not tpl_path.exists():
        sys.exit(f"Template not found: {tpl_path}")

    html = tpl_path.read_text(encoding="utf-8")
    replacements = {
        "{{MAIN_TEXT_LINE1}}": escape(args.line1),
        "{{MAIN_TEXT_LINE2}}": escape(args.line2),
        "{{MAIN_TEXT_LINE3}}": escape(args.line3),
        "{{LINE1_CLASS}}": "highlight" if args.hl1 else "",
        "{{LINE2_CLASS}}": "highlight" if args.hl2 else "",
        "{{LINE3_CLASS}}": "highlight" if args.hl3 else "",
        "{{HIGHLIGHT_COLOR}}": args.highlight,
        "{{BG_COLOR}}": args.bg,
        "{{FOOTER_TEXT}}": escape(args.footer),
        "{{ICON_PATH}}": icon_path,
        "{{FONT_PATH}}": str(SKILL_DIR / "assets" / "fonts" / "AlimamaShuHeiTi-Bold.ttf"),
        "{{AVATAR_PATH}}": str(SKILL_DIR / "assets" / "icons" / "avatar_jinx_cartoon.jpg"),
    }
    for k, v in replacements.items():
        html = html.replace(k, v)

    # 词级高亮：把指定词用 <span class="highlight"> 包起来
    # 注意：line 内容已经被 html.escape，所以匹配时用转义后的词
    if args.highlight_words:
        import re
        words = [w.strip() for w in args.highlight_words.split(",") if w.strip()]
        for word in words:
            escaped_word = escape(word)
            html = re.sub(
                re.escape(escaped_word),
                f'<span class="highlight">{escaped_word}</span>',
                html
            )

    # 判断尺寸
    size_map = {
        "poster-3-4": (900, 1200),
        "wechat-cover-split": (1340, 400),
    }
    w, h = size_map.get(args.template, (900, 1200))

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

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

    # 输出路径统一用 workspace/tmp/，不要用 /tmp/（飞书无法上传系统临时目录）
    out = Path(args.out)
    screenshot_path = out
    window_h = h
    if args.template == "wechat-cover-split":
        window_h = h + WECHAT_SPLIT_WINDOW_EXTRA_HEIGHT
        screenshot_path = out.with_name(f"{out.stem}.raw{out.suffix}")

    cmd = [
        chrome,
        "--headless",
        "--disable-gpu",
        "--no-sandbox",
        "--disable-web-security=false",
        f"--screenshot={screenshot_path}",
        f"--window-size={w},{window_h}",
        f"file://{tmp_html}",
    ]
    result = subprocess.run(cmd, capture_output=True)
    if result.returncode != 0:
        sys.exit(f"Chrome failed:\n{result.stderr.decode()}")

    if args.template == "wechat-cover-split":
        ffmpeg = shutil.which("ffmpeg")
        if not ffmpeg:
            sys.exit("wechat-cover-split 需要 ffmpeg 做顶部裁切")
        crop_cmd = [
            ffmpeg,
            "-y",
            "-loglevel",
            "error",
            "-i",
            str(screenshot_path),
            "-vf",
            f"crop={w}:{h}:0:0",
            "-frames:v",
            "1",
            str(out),
        ]
        crop_result = subprocess.run(crop_cmd, capture_output=True)
        screenshot_path.unlink(missing_ok=True)
        if crop_result.returncode != 0:
            sys.exit(f"ffmpeg crop failed:\n{crop_result.stderr.decode()}")

    Path(tmp_html).unlink(missing_ok=True)
    print(f"✅ Saved to {out}")

if __name__ == "__main__":
    main()