1
const YML = require('yamljs');
2
const fs = require('fs').promises;
3
const path = require('path');
4

5
// 路径配置
6
const LINK_YML_PATH = path.join('source', '_data', 'link.yml');
7
const OUTPUT_JSON_PATH = path.join('source', 'flink_count.json');
8
const TAKE_COUNT = 2;
9

10
async function processLinks() {
11
  try {
12
    // 读取文件（二进制方式）
13
    const yamlBuffer = await fs.readFile(LINK_YML_PATH);
14
    console.log('文件读取成功，大小:', yamlBuffer.length, '字节');
15

16
    // 处理UTF-8 BOM头（移除BOM）
17
    let cleanBuffer = yamlBuffer;
18
    if (yamlBuffer[0] === 0xef && yamlBuffer[1] === 0xbb && yamlBuffer[2] === 0xbf) {
19
      console.warn('检测到UTF-8 BOM头，已自动移除');
20
      cleanBuffer = yamlBuffer.slice(3); // 移除前3个BOM字节
21
    }
22

23
    // 转换为字符串
24
    const yamlContent = cleanBuffer.toString('utf8');
25

26
    // 显示文件前100个字符用于调试
27
    console.log('文件前100字符:', yamlContent.substring(0, 100).replace(/\n/g, '\\n'));
28

29
    // 更安全的YAML处理 - 只处理确实为空的rss字段
30
    const processedContent = yamlContent
31
      .replace(/^\s*rss:\s*$/gm, 'rss: ""')  // 处理空的rss字段
32
      .replace(/\t/g, '  ');  // 将制表符替换为两个空格（YAML不允许制表符缩进）
33

34
    // 保存处理后的内容用于调试
35
    await fs.writeFile('processed_yaml_debug.yml', processedContent, 'utf8');
36
    console.log('已保存处理后的YAML到 processed_yaml_debug.yml');
37

38
    // 尝试使用更严格的方式定位错误
39
    try {
40
      console.log('开始解析YAML...');
41
      const data = YML.parse(processedContent);
42
      console.log('YAML解析成功，根结构类型:', Array.isArray(data) ? '数组' : typeof data);
43

44
      if (!Array.isArray(data)) {
45
        throw new Error('YAML文件的根结构不是数组');
46
      }
47

48
      const linkList = [];
49
      for (let i = 0; i < Math.min(TAKE_COUNT, data.length); i++) {
50
        const item = data[i];
51
        if (item && Array.isArray(item.link_list)) {
52
          linkList.push(...item.link_list);
53
        } else {
54
          console.warn(`第${i+1}项不包含有效的link_list数组，已跳过`);
55
        }
56
      }
57

58
      const outputData = {
59
        link_list: linkList,
60
        length: linkList.length
61
      };
62

63
      await fs.writeFile(
64
        OUTPUT_JSON_PATH,
65
        JSON.stringify(outputData, null, 2),
66
        'utf8'
67
      );
68

69
      console.log(`成功生成 ${OUTPUT_JSON_PATH}，共包含 ${linkList.length} 条友链`);
70

71
    } catch (parseError) {
72
      console.error('\n=== YAML解析错误 ===');
73
      console.error('错误信息:', parseError.message);
74

75
      // 尝试定位错误行
76
      if (parseError.parsedLine) {
77
        console.error('错误位置: 第', parseError.parsedLine, '行');
78
        // 输出错误行附近的内容
79
        const lines = processedContent.split('\n');
80
        const errorLine = parseError.parsedLine - 1; // 转换为数组索引
81
        // 显示错误行前后3行
82
        for (let i = Math.max(0, errorLine - 3); i <= Math.min(lines.length - 1, errorLine + 3); i++) {
83
          console.log(`${i + 1}: ${lines[i]}`);
84
        }
85
      } else {
86
        console.error('错误位置: 未知（可能是文件开头附近）');
87
      }
88
      console.error('请检查 processed_yaml_debug.yml 文件是否有格式问题');
89
    }
90

91
  } catch (error) {
92
    console.error('\n=== 处理过程出错 ===');
93
    console.error('错误信息:', error.message);
94
  }
95
}
96

97
processLinks();