refactor: share markdown renderer via baoyu-md

skills/baoyu-post-to-wechat/scripts/md-to-wechat.ts

@@ -1,13 +1,19 @@
-import fs from 'node:fs';
-import path from 'node:path';
-import { mkdir, writeFile } from 'node:fs/promises';
-import os from 'node:os';
-import { createHash } from 'node:crypto';
-import { fileURLToPath } from 'node:url';
-import https from 'node:https';
-import http from 'node:http';
-import { spawnSync } from 'node:child_process';
-import process from 'node:process';
+import fs from "node:fs";
+import os from "node:os";
+import path from "node:path";
+import process from "node:process";
+
+import {
+  extractSummaryFromBody,
+  extractTitleFromMarkdown,
+  parseFrontmatter,
+  renderMarkdownDocument,
+  replaceMarkdownImagesWithPlaceholders,
+  resolveColorToken,
+  resolveContentImages,
+  serializeFrontmatter,
+  stripWrappingQuotes,
+} from "baoyu-md";
 
 interface ImageInfo {
   placeholder: string;
@@ -23,200 +29,53 @@ interface ParsedResult {
   contentImages: ImageInfo[];
 }
 
-function downloadFile(url: string, destPath: string): Promise<void> {
-  return new Promise((resolve, reject) => {
-    const protocol = url.startsWith('https') ? https : http;
-    const file = fs.createWriteStream(destPath);
-
-    const request = protocol.get(url, { headers: { 'User-Agent': 'Mozilla/5.0' } }, (response) => {
-      if (response.statusCode === 301 || response.statusCode === 302) {
-        const redirectUrl = response.headers.location;
-        if (redirectUrl) {
-          file.close();
-          fs.unlinkSync(destPath);
-          downloadFile(redirectUrl, destPath).then(resolve).catch(reject);
-          return;
-        }
-      }
-
-      if (response.statusCode !== 200) {
-        file.close();
-        fs.unlinkSync(destPath);
-        reject(new Error(`Failed to download: ${response.statusCode}`));
-        return;
-      }
-
-      response.pipe(file);
-      file.on('finish', () => {
-        file.close();
-        resolve();
-      });
-    });
-
-    request.on('error', (err) => {
-      file.close();
-      fs.unlink(destPath, () => {});
-      reject(err);
-    });
-
-    request.setTimeout(30000, () => {
-      request.destroy();
-      reject(new Error('Download timeout'));
-    });
-  });
-}
-
-function getImageExtension(urlOrPath: string): string {
-  const match = urlOrPath.match(/\.(jpg|jpeg|png|gif|webp)(\?|$)/i);
-  return match ? match[1]!.toLowerCase() : 'png';
-}
-
-async function resolveImagePath(imagePath: string, baseDir: string, tempDir: string): Promise<string> {
-  if (imagePath.startsWith('http://') || imagePath.startsWith('https://')) {
-    const hash = createHash('md5').update(imagePath).digest('hex').slice(0, 8);
-    const ext = getImageExtension(imagePath);
-    const localPath = path.join(tempDir, `remote_${hash}.${ext}`);
-
-    if (!fs.existsSync(localPath)) {
-      console.error(`[md-to-wechat] Downloading: ${imagePath}`);
-      await downloadFile(imagePath, localPath);
-    }
-    return localPath;
-  }
-
-  if (path.isAbsolute(imagePath)) {
-    return imagePath;
-  }
-
-  return path.resolve(baseDir, imagePath);
-}
-
-function parseFrontmatter(content: string): { frontmatter: Record<string, string>; body: string } {
-  const match = content.match(/^---\r?\n([\s\S]*?)\r?\n---\r?\n([\s\S]*)$/);
-  if (!match) return { frontmatter: {}, body: content };
-
-  const frontmatter: Record<string, string> = {};
-  const lines = match[1]!.split('\n');
-  for (const line of lines) {
-    const colonIdx = line.indexOf(':');
-    if (colonIdx > 0) {
-      const key = line.slice(0, colonIdx).trim();
-      let value = line.slice(colonIdx + 1).trim();
-      if ((value.startsWith('"') && value.endsWith('"')) || (value.startsWith("'") && value.endsWith("'"))) {
-        value = value.slice(1, -1);
-      }
-      frontmatter[key] = value;
-    }
-  }
-
-  return { frontmatter, body: match[2]! };
-}
-
 export async function convertMarkdown(
   markdownPath: string,
-  options?: { title?: string; theme?: string; color?: string; citeStatus?: boolean }
+  options?: { title?: string; theme?: string; color?: string; citeStatus?: boolean },
 ): Promise<ParsedResult> {
   const baseDir = path.dirname(markdownPath);
-  const content = fs.readFileSync(markdownPath, 'utf-8');
-  const theme = options?.theme ?? 'default';
+  const content = fs.readFileSync(markdownPath, "utf-8");
   const citeStatus = options?.citeStatus ?? true;
 
   const { frontmatter, body } = parseFrontmatter(content);
 
-  let title = options?.title ?? frontmatter.title ?? '';
-  let bodyWithoutTitle = body;
+  let title = stripWrappingQuotes(options?.title ?? "")
+    || stripWrappingQuotes(frontmatter.title ?? "")
+    || extractTitleFromMarkdown(body);
   if (!title) {
-    const lines = body.split('\n');
-    for (const line of lines) {
-      const trimmed = line.trim();
-      if (!trimmed) continue;
-      const headingMatch = trimmed.match(/^#{1,2}\s+(.+)$/);
-      if (headingMatch) {
-        title = headingMatch[1]!;
-        bodyWithoutTitle = body.replace(/^#{1,2}\s+.+\r?\n?/, '');
-      }
-      break;
-    }
-  } else {
-    bodyWithoutTitle = body.replace(/^#{1,2}\s+.+\r?\n?/, '');
+    title = path.basename(markdownPath, path.extname(markdownPath));
   }
-  if (!title) title = path.basename(markdownPath, path.extname(markdownPath));
-  const author = frontmatter.author || '';
-  let summary = frontmatter.description || frontmatter.summary || '';
 
+  const author = stripWrappingQuotes(frontmatter.author ?? "");
+  let summary = stripWrappingQuotes(frontmatter.description ?? "")
+    || stripWrappingQuotes(frontmatter.summary ?? "");
   if (!summary) {
-    const lines = bodyWithoutTitle.split('\n');
-    for (const line of lines) {
-      const trimmed = line.trim();
-      if (!trimmed) continue;
-      if (trimmed.startsWith('#')) continue;
-      if (trimmed.startsWith('![')) continue;
-      if (trimmed.startsWith('>')) continue;
-      if (trimmed.startsWith('-') || trimmed.startsWith('*')) continue;
-      if (/^\d+\./.test(trimmed)) continue;
-
-      const cleanText = trimmed
-        .replace(/\*\*(.+?)\*\*/g, '$1')
-        .replace(/\*(.+?)\*/g, '$1')
-        .replace(/\[([^\]]+)\]\([^)]+\)/g, '$1')
-        .replace(/`([^`]+)`/g, '$1');
-
-      if (cleanText.length > 20) {
-        summary = cleanText.length > 120 ? cleanText.slice(0, 117) + '...' : cleanText;
-        break;
-      }
-    }
+    summary = extractSummaryFromBody(body, 120);
   }
 
-  const images: Array<{ src: string; placeholder: string }> = [];
-  let imageCounter = 0;
+  const { images, markdown: rewrittenBody } = replaceMarkdownImagesWithPlaceholders(
+    body,
+    "WECHATIMGPH_",
+  );
+  const rewrittenMarkdown = `${serializeFrontmatter(frontmatter)}${rewrittenBody}`;
 
-  const modifiedBody = bodyWithoutTitle.replace(/!\[([^\]]*)\]\(([^)]+)\)/g, (match, alt, src) => {
-    const placeholder = `WECHATIMGPH_${++imageCounter}`;
-    images.push({ src, placeholder });
-    return placeholder;
+  const tempDir = fs.mkdtempSync(path.join(os.tmpdir(), "wechat-article-images-"));
+  const htmlPath = path.join(tempDir, "temp-article.html");
+
+  console.error(
+    `[md-to-wechat] Rendering markdown with theme: ${options?.theme ?? "default"}${options?.color ? `, color: ${options.color}` : ""}, citeStatus: ${citeStatus}`,
+  );
+
+  const { html } = await renderMarkdownDocument(rewrittenMarkdown, {
+    citeStatus,
+    defaultTitle: title,
+    keepTitle: false,
+    primaryColor: resolveColorToken(options?.color),
+    theme: options?.theme,
   });
+  fs.writeFileSync(htmlPath, html, "utf-8");
 
-  const modifiedMarkdown = `---\n${Object.entries(frontmatter).map(([k, v]) => `${k}: ${v}`).join('\n')}\n---\n${modifiedBody}`;
-
-  const tempDir = fs.mkdtempSync(path.join(os.tmpdir(), 'wechat-article-images-'));
-  const tempMdPath = path.join(tempDir, 'temp-article.md');
-  await writeFile(tempMdPath, modifiedMarkdown, 'utf-8');
-
-  const __filename = fileURLToPath(import.meta.url);
-  const __dirname = path.dirname(__filename);
-  const renderScript = path.join(__dirname, 'md', 'render.ts');
-
-  const renderArgs = ['-y', 'bun', renderScript, tempMdPath, '--theme', theme];
-  if (options?.color) renderArgs.push('--color', options.color);
-  if (citeStatus) renderArgs.push('--cite');
-
-  console.error(`[md-to-wechat] Rendering markdown with theme: ${theme}${options?.color ? `, color: ${options.color}` : ''}, citeStatus: ${citeStatus}`);
-
-  const result = spawnSync('npx', renderArgs, {
-    stdio: ['inherit', 'pipe', 'pipe'],
-    cwd: baseDir,
-  });
-
-  if (result.status !== 0) {
-    const stderr = result.stderr?.toString() || '';
-    throw new Error(`Render failed: ${stderr}`);
-  }
-
-  const htmlPath = tempMdPath.replace(/\.md$/i, '.html');
-  if (!fs.existsSync(htmlPath)) {
-    throw new Error(`HTML file not generated: ${htmlPath}`);
-  }
-
-  const contentImages: ImageInfo[] = [];
-  for (const img of images) {
-    const localPath = await resolveImagePath(img.src, baseDir, tempDir);
-    contentImages.push({
-      placeholder: img.placeholder,
-      localPath,
-      originalPath: img.src,
-    });
-  }
+  const contentImages = await resolveContentImages(images, baseDir, tempDir, "md-to-wechat");
 
   return {
     title,
@@ -264,7 +123,7 @@ Example:
 
 async function main(): Promise<void> {
   const args = process.argv.slice(2);
-  if (args.length === 0 || args.includes('--help') || args.includes('-h')) {
+  if (args.length === 0 || args.includes("--help") || args.includes("-h")) {
     printUsage();
   }
 
@@ -276,23 +135,23 @@ async function main(): Promise<void> {
 
   for (let i = 0; i < args.length; i++) {
     const arg = args[i]!;
-    if (arg === '--title' && args[i + 1]) {
+    if (arg === "--title" && args[i + 1]) {
       title = args[++i];
-    } else if (arg === '--theme' && args[i + 1]) {
+    } else if (arg === "--theme" && args[i + 1]) {
       theme = args[++i];
-    } else if (arg === '--color' && args[i + 1]) {
+    } else if (arg === "--color" && args[i + 1]) {
       color = args[++i];
-    } else if (arg === '--cite') {
+    } else if (arg === "--cite") {
       citeStatus = true;
-    } else if (arg === '--no-cite') {
+    } else if (arg === "--no-cite") {
       citeStatus = false;
-    } else if (!arg.startsWith('-')) {
+    } else if (!arg.startsWith("-")) {
       markdownPath = arg;
     }
   }
 
   if (!markdownPath) {
-    console.error('Error: Markdown file path is required');
+    console.error("Error: Markdown file path is required");
     process.exit(1);
   }
 
@@ -305,7 +164,7 @@ async function main(): Promise<void> {
   console.log(JSON.stringify(result, null, 2));
 }
 
-await main().catch((err) => {
-  console.error(`Error: ${err instanceof Error ? err.message : String(err)}`);
+await main().catch((error) => {
+  console.error(`Error: ${error instanceof Error ? error.message : String(error)}`);
   process.exit(1);
 });