1import fs from "node:fs";
2import os from "node:os";
3import path from "node:path";
4import process from "node:process";
5 
6import {
7  cleanSummaryText,
8  extractSummaryFromBody,
9  extractTitleFromMarkdown,
10  parseFrontmatter,
11  renderMarkdownDocument,
12  replaceMarkdownImagesWithPlaceholders,
13  resolveColorToken,
14  resolveContentImages,
15  serializeFrontmatter,
16  stripWrappingQuotes,
17} from "baoyu-md";
18 
19interface ImageInfo {
20  placeholder: string;
21  localPath: string;
22  originalPath: string;
23}
24 
25interface ParsedResult {
26  title: string;
27  author: string;
28  summary: string;
29  htmlPath: string;
30  contentImages: ImageInfo[];
31}
32 
33export async function convertMarkdown(
34  markdownPath: string,
35  options?: { title?: string; theme?: string; color?: string; citeStatus?: boolean },
36): Promise<ParsedResult> {
37  const baseDir = path.dirname(markdownPath);
38  const content = fs.readFileSync(markdownPath, "utf-8");
39  const citeStatus = options?.citeStatus ?? true;
40 
41  const { frontmatter, body } = parseFrontmatter(content);
42 
43  let title = stripWrappingQuotes(options?.title ?? "")
44    || stripWrappingQuotes(frontmatter.title ?? "")
45    || extractTitleFromMarkdown(body);
46  if (!title) {
47    title = path.basename(markdownPath, path.extname(markdownPath));
48  }
49 
50  const author = stripWrappingQuotes(frontmatter.author ?? "");
51  const frontmatterSummary = stripWrappingQuotes(frontmatter.description ?? "")
52    || stripWrappingQuotes(frontmatter.summary ?? "");
53  let summary = cleanSummaryText(frontmatterSummary);
54  if (!summary) {
55    summary = extractSummaryFromBody(body, 120);
56  }
57 
58  const { images, markdown: rewrittenBody } = replaceMarkdownImagesWithPlaceholders(
59    body,
60    "WECHATIMGPH_",
61  );
62  const rewrittenMarkdown = `${serializeFrontmatter(frontmatter)}${rewrittenBody}`;
63 
64  const tempDir = fs.mkdtempSync(path.join(os.tmpdir(), "wechat-article-images-"));
65  const htmlPath = path.join(tempDir, "temp-article.html");
66 
67  console.error(
68    `[md-to-wechat] Rendering markdown with theme: ${options?.theme ?? "default"}${options?.color ? `, color: ${options.color}` : ""}, citeStatus: ${citeStatus}`,
69  );
70 
71  const { html } = await renderMarkdownDocument(rewrittenMarkdown, {
72    citeStatus,
73    defaultTitle: title,
74    keepTitle: false,
75    primaryColor: resolveColorToken(options?.color),
76    theme: options?.theme,
77  });
78  fs.writeFileSync(htmlPath, html, "utf-8");
79 
80  const contentImages = await resolveContentImages(images, baseDir, tempDir, "md-to-wechat");
81 
82  return {
83    title,
84    author,
85    summary,
86    htmlPath,
87    contentImages,
88  };
89}
90 
91function printUsage(): never {
92  console.log(`Convert Markdown to WeChat-ready HTML with image placeholders
93 
94Usage:
95  npx -y bun md-to-wechat.ts <markdown_file> [options]
96 
97Options:
98  --title <title>     Override title
99  --theme <name>      Theme name (default, grace, simple, modern)
100  --color <name|hex>  Primary color (blue, green, vermilion, etc. or hex)
101  --no-cite           Disable bottom citations for ordinary external links
102  --help              Show this help
103 
104Output JSON format:
105{
106  "title": "Article Title",
107  "htmlPath": "/tmp/wechat-article-images/temp-article.html",
108  "contentImages": [
109    {
110      "placeholder": "WECHATIMGPH_1",
111      "localPath": "/tmp/wechat-image/img.png",
112      "originalPath": "imgs/image.png"
113    }
114  ]
115}
116 
117Example:
118  npx -y bun md-to-wechat.ts article.md
119  npx -y bun md-to-wechat.ts article.md --theme grace
120  npx -y bun md-to-wechat.ts article.md --theme modern --color blue
121  npx -y bun md-to-wechat.ts article.md --no-cite
122`);
123  process.exit(0);
124}
125 
126async function main(): Promise<void> {
127  const args = process.argv.slice(2);
128  if (args.length === 0 || args.includes("--help") || args.includes("-h")) {
129    printUsage();
130  }
131 
132  let markdownPath: string | undefined;
133  let title: string | undefined;
134  let theme: string | undefined;
135  let color: string | undefined;
136  let citeStatus = true;
137 
138  for (let i = 0; i < args.length; i++) {
139    const arg = args[i]!;
140    if (arg === "--title" && args[i + 1]) {
141      title = args[++i];
142    } else if (arg === "--theme" && args[i + 1]) {
143      theme = args[++i];
144    } else if (arg === "--color" && args[i + 1]) {
145      color = args[++i];
146    } else if (arg === "--cite") {
147      citeStatus = true;
148    } else if (arg === "--no-cite") {
149      citeStatus = false;
150    } else if (!arg.startsWith("-")) {
151      markdownPath = arg;
152    }
153  }
154 
155  if (!markdownPath) {
156    console.error("Error: Markdown file path is required");
157    process.exit(1);
158  }
159 
160  if (!fs.existsSync(markdownPath)) {
161    console.error(`Error: File not found: ${markdownPath}`);
162    process.exit(1);
163  }
164 
165  const result = await convertMarkdown(markdownPath, { title, theme, color, citeStatus });
166  console.log(JSON.stringify(result, null, 2));
167}
168 
169await main().catch((error) => {
170  console.error(`Error: ${error instanceof Error ? error.message : String(error)}`);
171  process.exit(1);
172});
173