@zzclub/pipeline 0.2.0 → 0.4.0

package/src/imgx/render-ascii-portrait.ts

@@ -0,0 +1,53 @@
+#!/usr/bin/env bun
+import { getArg, getIntArg, parseArgs, requireArg } from "./cli";
+import { ensureParentDir, findChrome, printSaved, resolveInputPath } from "./runtime";
+import { renderAsciiPortraitPng } from "./ascii-portrait";
+
+export function runRenderAsciiPortraitCli(argv: string[]): void {
+  const parsed = parseArgs(argv);
+  const outPath = requireArg(parsed, "out");
+  const avatarPath = resolveInputPath(requireArg(parsed, "avatar"));
+  const template = getArg(parsed, "template", "ascii-portrait-3-4");
+  const defaultChars =
+    template === "ascii-portrait-tile"
+      ? "@#X$%WM*+=-"
+      : "@#X$%WM*+=-";
+  const title = getArg(parsed, "title");
+  const footer = getArg(parsed, "footer", "公众号 · 早早集市");
+  const iconPath = resolveInputPath(getArg(parsed, "icon"));
+  const bg = getArg(parsed, "bg", template === "ascii-portrait-tile" ? "#f5f0eb" : "auto");
+  const chars = getArg(parsed, "chars", defaultChars);
+  const columns = getIntArg(parsed, "columns", 0);
+  const fontSize = getIntArg(parsed, "font-size", 0);
+  const lineHeight = getIntArg(parsed, "line-height", 0);
+  const width = getIntArg(parsed, "width", 900);
+  const height = getIntArg(parsed, "height", 1200);
+
+  const chromePath = findChrome();
+  if (chromePath === null) {
+    throw new Error("Chrome/Chromium not found");
+  }
+
+  ensureParentDir(outPath);
+  renderAsciiPortraitPng({
+    chromePath,
+    outPath,
+    avatarPath,
+    bg,
+    chars,
+    columns,
+    fontSize,
+    lineHeight,
+    width,
+    height,
+    templateName: template === "ascii-portrait-tile" ? "ascii-portrait-tile" : "ascii-portrait-3-4",
+    title: title.length > 0 ? title : undefined,
+    footer: footer.length > 0 ? footer : undefined,
+    iconPath: iconPath.length > 0 ? iconPath : undefined,
+  });
+  printSaved(outPath);
+}
+
+if (import.meta.main) {
+  runRenderAsciiPortraitCli(process.argv.slice(2));
+}

package/src/imgx/render-card.ts

@@ -0,0 +1,232 @@
+#!/usr/bin/env bun
+import { rmSync, writeFileSync } from "node:fs";
+import { join } from "node:path";
+import { pathToFileURL } from "node:url";
+import { getArg, getArgs, hasFlag, parseArgs, requireArg } from "./cli";
+import { buildPosterConfig, serializePosterConfig, type TipItem } from "./poster-recipe";
+import {
+  PACKAGE_ROOT,
+  cropTop,
+  ensureParentDir,
+  escapeHtml,
+  findChrome,
+  FONTS_DIR,
+  ICONS_DIR,
+  printSaved,
+  readUtf8,
+  renderTemplate,
+  resolveInputPath,
+  screenshotHtml,
+  TEMPLATES_DIR,
+} from "./runtime";
+
+const SIZE_MAP: Record<string, { width: number; height: number }> = {
+  "poster-3-4": { width: 900, height: 1200 },
+  "wechat-cover-split": { width: 1340, height: 400 },
+  "tips-3-4": { width: 900, height: 1200 },
+};
+const DEFAULT_SIZE = { width: 900, height: 1200 } as const;
+
+const WECHAT_SPLIT_WINDOW_EXTRA_HEIGHT = 87;
+const PRETEXT_SCREENSHOT_VIRTUAL_TIME_BUDGET_MS = 6_000;
+
+function detectIcon(template: string, _text: string, _line1: string, _line2: string, _line3: string, fallbackIcon: string): string {
+  if (template === "wechat-cover-split") {
+    return join(ICONS_DIR, "logo.png");
+  }
+
+  if (fallbackIcon.length > 0) return fallbackIcon;
+  return join(ICONS_DIR, "logo.png");
+}
+
+function splitWechatTitle(text: string): { line1: string; line2: string } {
+  const normalized = text.replace(/\s+/g, " ").trim();
+  if (normalized.length === 0) return { line1: "", line2: "" };
+
+  const chars = Array.from(normalized);
+  if (chars.length <= 16) return { line1: normalized, line2: "" };
+
+  // Find all word boundaries: any position where adjacent chars cross
+  // script boundaries (CJK↔Latin) or have a space.
+  type Candidate = { index: number; score: number };
+  const candidates: Candidate[] = [];
+  const minIndex = Math.max(2, Math.min(8, chars.length - 2));
+  const maxIndex = Math.min(chars.length - 2, 20);
+
+  const CJK_RE = /[一-鿿]/;
+  const splitHints = new Set([" ", "，", "。", "、", "：", "；", "-", "｜", "|", "·"]);
+
+  for (let idx = minIndex; idx <= maxIndex; idx++) {
+    const left = chars.slice(0, idx).join("").trim();
+    const right = chars.slice(idx).join("").trim();
+    if (left.length === 0 || right.length === 0) continue;
+
+    let score = 0;
+
+    // 1. Balance penalty (weight reduced — visual balance matters more than
+    //    character-count balance for mixed text)
+    score += Math.abs(left.length - right.length) * 2;
+
+    // 2. Punctuation / word boundary at split point = strong signal
+    const prevChar = chars[idx - 1] ?? "";
+    const nextChar = chars[idx] ?? "";
+
+    if (splitHints.has(prevChar)) score -= 25;  // Strong signal: split after
+    if (splitHints.has(nextChar)) score -= 20;  // Moderate signal: split before
+
+    // 3. Script boundary bonus: splitting between CJK and Latin is natural
+    const prevIsCJK = CJK_RE.test(prevChar);
+    const nextIsCJK = CJK_RE.test(nextChar);
+    if (prevIsCJK !== nextIsCJK) score -= 10;
+
+    // 4. Word-internal penalty: don't split inside an English word
+    const prevIsAlpha = /[A-Za-z]/.test(prevChar);
+    const nextIsAlpha = /[A-Za-z]/.test(nextChar);
+    if (prevIsAlpha && nextIsAlpha) score += 15;
+
+    // 5. Line length caps (visual estimate: CJK ≈ 2 units, Latin ≈ 1 unit)
+    const leftVisual = [...left].reduce((sum, ch) => sum + (CJK_RE.test(ch) ? 2 : 1), 0);
+    const rightVisual = [...right].reduce((sum, ch) => sum + (CJK_RE.test(ch) ? 2 : 1), 0);
+    if (leftVisual > 28 || rightVisual > 28) score += 15;
+
+    // 6. Slight midpoint preference
+    score += Math.abs(idx - chars.length / 2) * 0.5;
+
+    candidates.push({ index: idx, score });
+  }
+
+  candidates.sort((a, b) => a.score - b.score);
+  const best = candidates[0] ?? { index: Math.ceil(chars.length / 2) };
+
+  return {
+    line1: chars.slice(0, best.index).join("").trim(),
+    line2: chars.slice(best.index).join("").trim(),
+  };
+}
+
+export function runRenderCardCli(argv: string[]): void {
+  const parsed = parseArgs(argv);
+  const template = getArg(parsed, "template", "poster-3-4");
+  const outPath = requireArg(parsed, "out");
+  let line1 = getArg(parsed, "line1");
+  let line2 = getArg(parsed, "line2");
+  let line3 = getArg(parsed, "line3");
+  let hl2 = hasFlag(parsed, "hl2");
+  let hl3 = hasFlag(parsed, "hl3");
+  const text = getArg(parsed, "text").replace(/\\n/g, "\n");
+  const highlight = getArg(parsed, "highlight", "#22a854");
+  const bg = getArg(parsed, "bg", "#e6f5ef");
+  const footer = getArg(parsed, "footer", "公众号 · 早早集市");
+  const fallbackIcon = getArg(parsed, "fallback-icon");
+  const highlightWords = getArg(parsed, "highlight-words");
+  const tips: TipItem[] = getArgs(parsed, "tip").map(raw => {
+    const sep = raw.indexOf("::");
+    if (sep === -1) return { title: raw.trim(), description: "" };
+    return { title: raw.slice(0, sep).trim(), description: raw.slice(sep + 2).trim() };
+  });
+
+  if (template === "wechat-cover-split") {
+    if (!line1 && !line2 && !line3 && text.trim().length > 0) {
+      const split = splitWechatTitle(text);
+      line1 = split.line1;
+      line2 = split.line2;
+    }
+    const lines = [line1, line2, line3].filter(Boolean);
+    line1 = lines[0] ?? "";
+    line2 = lines.slice(1).join("");
+    line3 = "";
+    hl2 = hl2 || hl3;
+    hl3 = false;
+  }
+
+  const chromePath = findChrome();
+  if (chromePath === null) {
+    throw new Error("Chrome/Chromium not found");
+  }
+  const usesPretext = template === "poster-3-4" || template === "wechat-cover-split" || template === "tips-3-4";
+
+  const iconPath = resolveInputPath(getArg(parsed, "icon") || detectIcon(template, text, line1, line2, line3, fallbackIcon));
+  const templatePath = join(TEMPLATES_DIR, `${template}.html`);
+  const templateHtml = readUtf8(templatePath);
+
+  const debugArgsPath = process.env.TEST_RENDER_CARD_ARGS_PATH;
+  if (debugArgsPath) {
+    ensureParentDir(debugArgsPath);
+    writeFileSync(debugArgsPath, JSON.stringify(argv), "utf-8");
+  }
+  if (process.env.TEST_RENDER_CARD_STUB === "1") {
+    ensureParentDir(outPath);
+    writeFileSync(outPath, "stub", "utf-8");
+    return;
+  }
+
+  const html = renderTemplate(templateHtml, {
+    "{{MAIN_TEXT_LINE1}}": escapeHtml(line1),
+    "{{MAIN_TEXT_LINE2}}": escapeHtml(line2),
+    "{{MAIN_TEXT_LINE3}}": escapeHtml(line3),
+    "{{LINE1_CLASS}}": hasFlag(parsed, "hl1") ? "highlight" : "",
+    "{{LINE2_CLASS}}": hl2 ? "highlight" : "",
+    "{{LINE3_CLASS}}": hl3 ? "highlight" : "",
+    "{{HIGHLIGHT_COLOR}}": highlight,
+    "{{BG_COLOR}}": bg,
+    "{{FOOTER_TEXT}}": escapeHtml(footer),
+    "{{ICON_PATH}}": iconPath,
+    "{{FONT_PATH}}": join(FONTS_DIR, "AlimamaShuHeiTi-Bold.ttf"),
+    "{{AVATAR_PATH}}": join(ICONS_DIR, "logo.png"),
+    "{{POSTER_CONFIG_JSON}}": serializePosterConfig(
+      buildPosterConfig({
+        text,
+        line1,
+        line2,
+        line3,
+        hl1: hasFlag(parsed, "hl1"),
+        hl2,
+        hl3,
+        highlightWords,
+        highlightColor: highlight,
+        tips,
+      }),
+    ),
+    "{{PRETEXT_MODULE_URL}}": pathToFileURL(
+      join(PACKAGE_ROOT, "node_modules/@chenglou/pretext/dist/layout.js"),
+    ).href,
+  });
+
+  const size = SIZE_MAP[template] ?? DEFAULT_SIZE;
+  ensureParentDir(outPath);
+
+  if (template === "wechat-cover-split") {
+    const rawPath = outPath.replace(/(\.[^.]+)?$/, ".raw$1");
+    screenshotHtml({
+      chromePath,
+      html,
+      outPath: rawPath,
+      width: size.width,
+      height: size.height + WECHAT_SPLIT_WINDOW_EXTRA_HEIGHT,
+      virtualTimeBudgetMs: usesPretext ? PRETEXT_SCREENSHOT_VIRTUAL_TIME_BUDGET_MS : undefined,
+    });
+    cropTop({
+      inputPath: rawPath,
+      outPath,
+      width: size.width,
+      height: size.height,
+    });
+    rmSync(rawPath, { force: true });
+    printSaved(outPath);
+    return;
+  }
+
+  screenshotHtml({
+    chromePath,
+    html,
+    outPath,
+    width: size.width,
+    height: size.height,
+    virtualTimeBudgetMs: usesPretext ? PRETEXT_SCREENSHOT_VIRTUAL_TIME_BUDGET_MS : undefined,
+  });
+  printSaved(outPath);
+}
+
+if (import.meta.main) {
+  runRenderCardCli(process.argv.slice(2));
+}

package/src/imgx/render-x-like-posts.ts

@@ -0,0 +1,235 @@
+#!/usr/bin/env bun
+import { readFileSync } from "node:fs";
+import { join } from "node:path";
+import { getArg, getArgs, parseArgs, requireArg } from "./cli";
+import {
+  escapeHtml,
+  findChrome,
+  FONTS_DIR,
+  ICONS_DIR,
+  printSaved,
+  readUtf8,
+  renderTemplate,
+  resolveInputPath,
+  screenshotHtml,
+  TEMPLATES_DIR,
+} from "./runtime";
+
+const WIDTH = 900;
+const MIN_HEIGHT = 1200;
+const MAX_HEIGHT = 16000;
+const TEXT_WIDTH_CHARS = 26;
+
+type PostRecord = {
+  text: string;
+  created_at?: string;
+  url?: string;
+  favorite_count?: number;
+  retweet_count?: number;
+};
+
+function weightedLength(text: string): number {
+  let total = 0;
+  for (const ch of text) {
+    if (ch === "\n") total += 8;
+    else if (ch.charCodeAt(0) < 128) total += 0.55;
+    else total += 1;
+  }
+  return total;
+}
+
+function estimatePostHeight(text: string): number {
+  const paragraphs = text.split(/\r?\n/).filter(block => block.trim().length > 0);
+  const blocks = paragraphs.length > 0 ? paragraphs : [text];
+  let wrappedLines = 0;
+  for (const block of blocks) {
+    wrappedLines += Math.max(1, Math.ceil(weightedLength(block) / TEXT_WIDTH_CHARS));
+  }
+  wrappedLines += Math.max(0, blocks.length - 1);
+  return 120 + wrappedLines * 48;
+}
+
+function estimateCanvasHeight(posts: PostRecord[]): number {
+  let total = 210;
+  for (const post of posts) {
+    total += estimatePostHeight(post.text) + 20;
+  }
+  return Math.max(MIN_HEIGHT, Math.min(MAX_HEIGHT, total + 120));
+}
+
+function loadPosts(parsed: ReturnType<typeof parseArgs>): PostRecord[] {
+  const posts: PostRecord[] = [];
+
+  for (const text of [...getArgs(parsed, "post"), ...getArgs(parsed, "tweet")]) {
+    if (text.trim().length > 0) posts.push({ text: text.trim() });
+  }
+
+  const inputFile = getArg(parsed, "posts-file") || getArg(parsed, "tweets-file");
+  if (inputFile.length > 0) {
+    const raw = readFileSync(inputFile, "utf8");
+    const parsedJson = JSON.parse(raw);
+    if (!Array.isArray(parsedJson)) {
+      throw new Error("posts file must be a JSON array");
+    }
+
+    for (const item of parsedJson) {
+      if (typeof item === "string" && item.trim().length > 0) {
+        posts.push({ text: item.trim() });
+        continue;
+      }
+      if (typeof item === "object" && item !== null) {
+        const text = String((item as Record<string, unknown>).text ?? "").trim();
+        if (text.length === 0) continue;
+        posts.push({
+          text,
+          created_at: String((item as Record<string, unknown>).created_at ?? "").trim() || undefined,
+          url: String((item as Record<string, unknown>).url ?? "").trim() || undefined,
+          favorite_count: toInt((item as Record<string, unknown>).favorite_count),
+          retweet_count: toInt((item as Record<string, unknown>).retweet_count),
+        });
+      }
+    }
+  }
+
+  if (posts.length === 0) {
+    throw new Error("需要至少一条帖子：传 --post/--tweet 或 --posts-file/--tweets-file");
+  }
+  return posts;
+}
+
+function toInt(value: unknown): number {
+  const num = Number.parseInt(String(value ?? "0"), 10);
+  return Number.isFinite(num) ? num : 0;
+}
+
+function textToHtml(text: string): string {
+  const paragraphs = text
+    .split(/\r?\n/)
+    .map(block => block.trim())
+    .filter(Boolean)
+    .map(block => `<p>${escapeHtml(block)}</p>`);
+  return paragraphs.length > 0 ? paragraphs.join("") : `<p>${escapeHtml(text)}</p>`;
+}
+
+function parseCreatedAt(value: string | undefined): Date | null {
+  if (value === undefined || value.length === 0) return null;
+  const date = new Date(value);
+  return Number.isNaN(date.getTime()) ? null : date;
+}
+
+function formatCreatedAt(value: string | undefined): string {
+  const date = parseCreatedAt(value);
+  if (date === null) return value ?? "";
+  const formatter = new Intl.DateTimeFormat("zh-CN", {
+    timeZone: "Asia/Shanghai",
+    year: "numeric",
+    month: "2-digit",
+    day: "2-digit",
+    hour: "2-digit",
+    minute: "2-digit",
+    hour12: false,
+  });
+  return `${formatter.format(date).replace(/\//g, "-")} UTC+8`;
+}
+
+function buildDateLabel(posts: PostRecord[]): string {
+  const dates = posts
+    .map(post => parseCreatedAt(post.created_at))
+    .filter((date): date is Date => date !== null)
+    .sort((a, b) => b.getTime() - a.getTime());
+  if (dates.length === 0) return "日期未知";
+  const formatter = new Intl.DateTimeFormat("en-CA", {
+    timeZone: "Asia/Shanghai",
+    year: "numeric",
+    month: "2-digit",
+    day: "2-digit",
+  });
+  return formatter.format(dates[0]);
+}
+
+function buildStats(post: PostRecord): string {
+  const parts: string[] = [];
+  const createdAt = formatCreatedAt(post.created_at);
+  if (createdAt.length > 0) {
+    parts.push(`<span class="stat">${escapeHtml(createdAt)}</span>`);
+  }
+
+  if ((post.retweet_count ?? 0) > 0) {
+    parts.push(`<span class="stat"><strong>${post.retweet_count}</strong> RT</span>`);
+  }
+  if ((post.favorite_count ?? 0) > 0) {
+    parts.push(`<span class="stat"><strong>${post.favorite_count}</strong> Likes</span>`);
+  }
+  if ((post.url ?? "").length > 0) {
+    parts.push('<span class="stat">x.com</span>');
+  }
+  return parts.length > 0 ? parts.join("") : '<span class="stat">Forwarded from X</span>';
+}
+
+function buildPostItems(posts: PostRecord[], author: string, handle: string, avatarPath: string): string {
+  return posts
+    .map((post, index) => {
+      return `
+<article class="tweet">
+  <div class="avatar-wrap">
+    <img src="${avatarPath}" alt="avatar">
+  </div>
+  <div class="tweet-main">
+    <div class="tweet-top">
+      <span class="name">${escapeHtml(author)}</span>
+      <span class="handle">${escapeHtml(handle)}</span>
+      <span class="meta">· Post</span>
+    </div>
+    <div class="tweet-text">${textToHtml(post.text)}</div>
+    <div class="tweet-footer">
+      <span class="tweet-index">${index + 1} / ${posts.length}</span>
+      <span class="tweet-stats">${buildStats(post)}</span>
+    </div>
+  </div>
+</article>`.trim();
+    })
+    .join("\n");
+}
+
+export function runRenderXLikePostsCli(argv: string[]): void {
+  const parsed = parseArgs(argv);
+  const chromePath = findChrome();
+  if (chromePath === null) throw new Error("Chrome/Chromium not found");
+
+  const posts = loadPosts(parsed);
+  const author = getArg(parsed, "author", "Unknown Author");
+  const handle = getArg(parsed, "handle", "@twitter");
+  const avatarPath = resolveInputPath(getArg(parsed, "avatar") || join(ICONS_DIR, "logo.png"));
+  const template = readUtf8(join(TEMPLATES_DIR, "x-like-posts.html"));
+  const html = renderTemplate(template, {
+    "{{BG_COLOR}}": getArg(parsed, "bg", "#f5f8fa"),
+    "{{CARD_BG}}": getArg(parsed, "card-bg", "#ffffff"),
+    "{{TEXT_COLOR}}": getArg(parsed, "text", "#0f1419"),
+    "{{MUTED_COLOR}}": getArg(parsed, "muted", "#536471"),
+    "{{BORDER_COLOR}}": getArg(parsed, "border", "#e6ecf0"),
+    "{{ACCENT_COLOR}}": getArg(parsed, "accent", "#1d9bf0"),
+    "{{AUTHOR_NAME}}": escapeHtml(author),
+    "{{AUTHOR_HANDLE}}": escapeHtml(handle),
+    "{{AUTHOR_AVATAR}}": avatarPath,
+    "{{HEADER_LABEL}}": escapeHtml(getArg(parsed, "header-label", "X-like 帖子分享图")),
+    "{{TIME_RANGE_LABEL}}": escapeHtml(buildDateLabel(posts)),
+    "{{FOOTER_TEXT}}": escapeHtml(getArg(parsed, "footer", "整理转发 · via zzhub-media-imgx")),
+    "{{FONT_PATH}}": join(FONTS_DIR, "AlimamaShuHeiTi-Bold.ttf"),
+    "{{TWEET_ITEMS}}": buildPostItems(posts, author, handle, avatarPath),
+  });
+
+  const outPath = requireArg(parsed, "out");
+  screenshotHtml({
+    chromePath,
+    html,
+    outPath,
+    width: WIDTH,
+    height: estimateCanvasHeight(posts),
+    hideScrollbars: true,
+  });
+  printSaved(outPath);
+}
+
+if (import.meta.main) {
+  runRenderXLikePostsCli(process.argv.slice(2));
+}