AI 封面图生成

1
function parseFrontmatter(content: string) {
2
  const match = content.match(/^---\n([\s\S]*?)\n---/);
3
  if (!match) return null;
4

5
  const frontmatter: Record<string, any> = {};
6
  const lines = match[1].split("\n");
7

8
  for (const line of lines) {
9
    const [key, ...valueParts] = line.split(":");
10
    if (key && valueParts.length) {
11
      const value = valueParts.join(":").trim();
12
      frontmatter[key.trim()] = value.replace(/^["']|["']$/g, "");
13
    }
14
  }
15

16
  return frontmatter;
17
}

1
async function generateImage(prompt: string, filename: string): Promise<string> {
2
  console.log(`🎨 正在生成图片：${filename}`);
3

4
  const response = await fetch(`${API_URL}/images/generations`, {
5
    method: "POST",
6
    headers: {
7
      Authorization: `Bearer ${API_KEY}`,
8
      "Content-Type": "application/json",
9
    },
10
    body: JSON.stringify({
11
      model: MODEL,
12
      prompt,
13
      n: 1,
14
      size: "1792x1024",
15
      response_format: "b64_json",
16
    }),
17
  });
18

19
  if (!response.ok) {
20
    const errorText = await response.text();
21
    throw new Error(`API 请求失败 (${response.status}): ${errorText}`);
22
  }
23

24
  const data = await response.json();
25
  const b64Data = data.data?.[0]?.b64_json;
26
  if (!b64Data) throw new Error("生成图片失败");
27

28
  const imageBuffer = Buffer.from(b64Data, "base64");
29
  const outputPath = join(OUTPUT_DIR, filename);
30

31
  await mkdir(OUTPUT_DIR, { recursive: true });
32
  await writeFile(outputPath, imageBuffer);
33
  console.log(`✅ 已保存：${outputPath}`);
34

35
  return `/images/covers/generated/${filename}`;
36
}

1
async function processFile(file: string) {
2
  const filePath = join(BLOG_DIR, file);
3
  const content = await readFile(filePath, "utf-8");
4
  const frontmatter = parseFrontmatter(content);
5

6
  if (!frontmatter) {
7
    console.log(`⚠️  跳过（无 frontmatter）：${file}`);
8
    return;
9
  }
10

11
  // 跳过已有封面的文章（非 api 且非 unsplash）
12
  if (frontmatter.image && frontmatter.image !== "api" && !frontmatter.image.includes("source.unsplash.com")) {
13
    console.log(`⏭️  跳过（已有封面）：${file}`);
14
    return;
15
  }
16

17
  const prompt = generatePrompt(frontmatter);
18
  const filename = `${file.replace(/\.(md|mdx)$/, "")}.png`.replace(/\//g, "-");
19
  const imagePath = await generateImage(prompt, filename);
20

21
  // 更新 frontmatter 的 image 字段
22
  const newContent = content.replace(
23
    /^(---\n[\s\S]*?)image:.*$/m,
24
    `$1image: ${imagePath}`,
25
  );
26

27
  if (newContent === content) {
28
    // 如果没有 image 字段，追加一个
29
    const updatedContent = content.replace(
30
      /^(---\n[\s\S]*?)\n---/,
31
      `$1\nimage: ${imagePath}\n---`,
32
    );
33
    await writeFile(filePath, updatedContent);
34
  } else {
35
    await writeFile(filePath, newContent);
36
  }
37

38
  console.log(`✨ 已更新文章：${file}\n`);
39
}

1
async function main() {
2
  const targetFile = process.argv[2];
3

4
  if (targetFile) {
5
    // 单文章模式
6
    const relativePath = targetFile
7
      .replace(/\\/g, "/")
8
      .replace(/^src\/content\/blog\//, "");
9
    console.log(`📝 处理单个文章：${relativePath}\n`);
10
    try {
11
      await processFile(relativePath);
12
    } catch (error) {
13
      console.error(`❌ 处理失败：${relativePath}`, error);
14
      process.exit(1);
15
    }
16
  } else {
17
    // 批量模式
18
    const files = await readdir(BLOG_DIR, { recursive: true });
19
    const mdFiles = files.filter((f) => f.endsWith(".md") || f.endsWith(".mdx"));
20

21
    console.log(`📝 找到 ${mdFiles.length} 篇文章\n`);
22

23
    for (const file of mdFiles) {
24
      try {
25
        await processFile(file);
26
      } catch (error) {
27
        console.error(`❌ 处理失败：${file}`, error);
28
      }
29
    }
30
  }
31

32
  console.log("\n🎉 全部完成！");
33
}
34

35
main();