1import fs from "node:fs";
2import os from "node:os";
3import path from "node:path";
4import process from "node:process";
5 
6import {
7  extractSummaryFromBody,
8  extractTitleFromMarkdown,
9  parseFrontmatter,
10  pickFirstString,
11  renderMarkdownDocument,
12  replaceMarkdownImagesWithPlaceholders,
13  resolveColorToken,
14  resolveContentImages,
15  resolveImagePath,
16  serializeFrontmatter,
17  stripWrappingQuotes,
18} from "baoyu-md";
19 
20interface ImageInfo {
21  placeholder: string;
22  localPath: string;
23  originalPath: string;
24  alt?: string;
25}
26 
27interface ParsedMarkdown {
28  title: string;
29  summary: string;
30  shortSummary: string;
31  coverImage: string | null;
32  contentImages: ImageInfo[];
33  html: string;
34}
35 
36export async function parseMarkdown(
37  markdownPath: string,
38  options?: {
39    coverImage?: string;
40    title?: string;
41    tempDir?: string;
42    theme?: string;
43    color?: string;
44    citeStatus?: boolean;
45  },
46): Promise<ParsedMarkdown> {
47  const content = fs.readFileSync(markdownPath, "utf-8");
48  const baseDir = path.dirname(markdownPath);
49  const tempDir = options?.tempDir ?? fs.mkdtempSync(path.join(os.tmpdir(), "weibo-article-images-"));
50 
51  const { frontmatter, body } = parseFrontmatter(content);
52 
53  let title = stripWrappingQuotes(options?.title ?? "")
54    || stripWrappingQuotes(frontmatter.title ?? "")
55    || extractTitleFromMarkdown(body);
56  if (!title) {
57    title = path.basename(markdownPath, path.extname(markdownPath));
58  }
59 
60  let summary = stripWrappingQuotes(frontmatter.summary ?? "")
61    || stripWrappingQuotes(frontmatter.description ?? "")
62    || stripWrappingQuotes(frontmatter.excerpt ?? "");
63  if (!summary) {
64    summary = extractSummaryFromBody(body, 44);
65  }
66  const shortSummary = extractSummaryFromBody(body, 44);
67 
68  const coverImagePath = stripWrappingQuotes(options?.coverImage ?? "")
69    || pickFirstString(frontmatter, ["featureImage", "cover_image", "coverImage", "cover", "image"])
70    || null;
71 
72  const { images, markdown: rewrittenBody } = replaceMarkdownImagesWithPlaceholders(
73    body,
74    "WBIMGPH_",
75  );
76  const rewrittenMarkdown = `${serializeFrontmatter(frontmatter)}${rewrittenBody}`;
77 
78  const { html } = await renderMarkdownDocument(rewrittenMarkdown, {
79    citeStatus: options?.citeStatus ?? false,
80    defaultTitle: title,
81    keepTitle: false,
82    primaryColor: resolveColorToken(options?.color),
83    theme: options?.theme,
84  });
85 
86  const contentImages = await resolveContentImages(images, baseDir, tempDir, "md-to-html");
87 
88  let resolvedCoverImage: string | null = null;
89  if (coverImagePath) {
90    resolvedCoverImage = await resolveImagePath(coverImagePath, baseDir, tempDir, "md-to-html");
91  }
92 
93  return {
94    title,
95    summary,
96    shortSummary,
97    coverImage: resolvedCoverImage,
98    contentImages,
99    html,
100  };
101}
102 
103async function main(): Promise<void> {
104  const args = process.argv.slice(2);
105  if (args.length === 0 || args.includes("--help") || args.includes("-h")) {
106    console.log(`Convert Markdown to HTML for Weibo article publishing
107 
108Usage:
109  npx -y bun md-to-html.ts <markdown_file> [options]
110 
111Options:
112  --title <title>       Override title
113  --cover <image>       Override cover image
114  --output <json|html>  Output format (default: json)
115  --html-only           Output only the HTML content
116  --save-html <path>    Save HTML to file
117  --help                Show this help
118`);
119    process.exit(0);
120  }
121 
122  let markdownPath: string | undefined;
123  let title: string | undefined;
124  let coverImage: string | undefined;
125  let outputFormat: "json" | "html" = "json";
126  let htmlOnly = false;
127  let saveHtmlPath: string | undefined;
128 
129  for (let i = 0; i < args.length; i++) {
130    const arg = args[i]!;
131    if (arg === "--title" && args[i + 1]) {
132      title = args[++i];
133    } else if (arg === "--cover" && args[i + 1]) {
134      coverImage = args[++i];
135    } else if (arg === "--output" && args[i + 1]) {
136      outputFormat = args[++i] as "json" | "html";
137    } else if (arg === "--html-only") {
138      htmlOnly = true;
139    } else if (arg === "--save-html" && args[i + 1]) {
140      saveHtmlPath = args[++i];
141    } else if (!arg.startsWith("-")) {
142      markdownPath = arg;
143    }
144  }
145 
146  if (!markdownPath || !fs.existsSync(markdownPath)) {
147    console.error("Error: Valid markdown file path required");
148    process.exit(1);
149  }
150 
151  const result = await parseMarkdown(markdownPath, { title, coverImage });
152 
153  if (saveHtmlPath) {
154    fs.writeFileSync(saveHtmlPath, result.html, "utf-8");
155    console.error(`[md-to-html] HTML saved to: ${saveHtmlPath}`);
156  }
157 
158  if (htmlOnly || outputFormat === "html") {
159    console.log(result.html);
160  } else {
161    console.log(JSON.stringify(result, null, 2));
162  }
163}
164 
165if (import.meta.main ?? (process.argv[1] && path.resolve(process.argv[1]) === path.resolve(import.meta.filename ?? ""))) {
166  await main().catch((error) => {
167    console.error(`Error: ${error instanceof Error ? error.message : String(error)}`);
168    process.exit(1);
169  });
170}
171