1import { readFileSync, writeFileSync } from "fs";
2import { unified } from "unified";
3import remarkParse from "remark-parse";
4import remarkCjkFriendly from "remark-cjk-friendly";
5import remarkGfm from "remark-gfm";
6import remarkFrontmatter from "remark-frontmatter";
7import remarkStringify from "remark-stringify";
8import { visit } from "unist-util-visit";
9import YAML from "yaml";
10import { replaceQuotes } from "./quotes";
11import { applyAutocorrect } from "./autocorrect";
12 
13export interface FormatOptions {
14  quotes?: boolean;
15  spacing?: boolean;
16  emphasis?: boolean;
17}
18 
19export interface FormatResult {
20  success: boolean;
21  filePath: string;
22  quotesFixed: boolean;
23  spacingApplied: boolean;
24  emphasisFixed: boolean;
25  error?: string;
26}
27 
28const DEFAULT_OPTIONS: Required<FormatOptions> = {
29  quotes: false,
30  spacing: true,
31  emphasis: true,
32};
33 
34function decodeHtmlEntities(text: string): string {
35  return text.replace(/&#x([0-9A-Fa-f]+);/g, (_, hex) =>
36    String.fromCodePoint(parseInt(hex, 16))
37  );
38}
39 
40function formatFrontmatter(value: string): string | null {
41  try {
42    const doc = YAML.parseDocument(value);
43    return doc.toString({ lineWidth: 0 }).trimEnd();
44  } catch {
45    return null;
46  }
47}
48 
49function formatMarkdownContent(
50  content: string,
51  options: Required<FormatOptions>
52): string {
53  const processor = unified()
54    .use(remarkParse)
55    .use(options.emphasis ? remarkCjkFriendly : [])
56    .use(remarkGfm)
57    .use(remarkFrontmatter, ["yaml"])
58    .use(remarkStringify, {
59      wrap: false,
60    });
61 
62  const tree = processor.parse(content);
63 
64  visit(tree, (node) => {
65    if (node.type === "text" && options.quotes) {
66      const textNode = node as { value: string };
67      textNode.value = replaceQuotes(textNode.value);
68      return;
69    }
70    if (node.type === "yaml") {
71      const yamlNode = node as { value: string };
72      const formatted = formatFrontmatter(yamlNode.value);
73      if (formatted !== null) {
74        yamlNode.value = formatted;
75      }
76      return;
77    }
78  });
79 
80  let result = processor.stringify(tree);
81  if (options.emphasis) {
82    result = decodeHtmlEntities(result);
83  }
84  return result;
85}
86 
87export function formatMarkdown(
88  filePath: string,
89  options?: FormatOptions
90): FormatResult {
91  const opts: Required<FormatOptions> = { ...DEFAULT_OPTIONS, ...options };
92 
93  const result: FormatResult = {
94    success: false,
95    filePath,
96    quotesFixed: false,
97    spacingApplied: false,
98    emphasisFixed: false,
99  };
100 
101  try {
102    const content = readFileSync(filePath, "utf-8");
103    const formattedContent = formatMarkdownContent(content, opts);
104 
105    result.quotesFixed = opts.quotes;
106    result.emphasisFixed = opts.emphasis;
107 
108    writeFileSync(filePath, formattedContent, "utf-8");
109 
110    if (opts.spacing) {
111      result.spacingApplied = applyAutocorrect(filePath);
112    }
113 
114    result.success = true;
115    console.log(`✓ Formatted: ${filePath}`);
116  } catch (error) {
117    result.error = error instanceof Error ? error.message : String(error);
118    console.error(`✗ Format failed: ${result.error}`);
119  }
120 
121  return result;
122}
123 
124function parseArgs(args: string[]): { filePath: string; options: FormatOptions } {
125  const options: FormatOptions = {};
126  let filePath = "";
127 
128  for (let i = 0; i < args.length; i++) {
129    const arg = args[i];
130 
131    if (arg === "--quotes" || arg === "-q") {
132      options.quotes = true;
133    } else if (arg === "--no-quotes") {
134      options.quotes = false;
135    } else if (arg === "--spacing" || arg === "-s") {
136      options.spacing = true;
137    } else if (arg === "--no-spacing") {
138      options.spacing = false;
139    } else if (arg === "--emphasis" || arg === "-e") {
140      options.emphasis = true;
141    } else if (arg === "--no-emphasis") {
142      options.emphasis = false;
143    } else if (arg === "--help" || arg === "-h") {
144      console.log(`Usage: npx -y bun scripts/main.ts <file.md> [options]
145 
146Options:
147  -q, --quotes       Replace ASCII quotes with fullwidth quotes (default: false)
148      --no-quotes    Do not replace quotes
149  -s, --spacing      Add CJK/English spacing via autocorrect (default: true)
150      --no-spacing   Do not add CJK/English spacing
151  -e, --emphasis     Fix CJK emphasis punctuation issues (default: true)
152      --no-emphasis  Do not fix CJK emphasis issues
153  -h, --help         Show this help message`);
154      process.exit(0);
155    } else if (!arg.startsWith("-")) {
156      filePath = arg;
157    }
158  }
159 
160  return { filePath, options };
161}
162 
163if (import.meta.url === `file://${process.argv[1]}`) {
164  const { filePath, options } = parseArgs(process.argv.slice(2));
165 
166  if (!filePath) {
167    console.error("Usage: npx -y bun scripts/main.ts <file.md> [options]");
168    console.error("Use --help for more information.");
169    process.exit(1);
170  }
171 
172  const result = formatMarkdown(filePath, options);
173  if (!result.success) {
174    process.exit(1);
175  }
176}
177