refactor: 重构服务器部分并整理目录和注释

2026-06-16 21:03:59 +08:00 · 2025-12-13 03:37:51 +08:00
parent 063a05a499
commit 5d8dc7c0a5
27 changed files with 367 additions and 378 deletions
@@ -0,0 +1,128 @@
+/**
+ * @fileoverview 配置加载模块
+ * @description 负责读取/解析 `config.yaml`，并提供 API Key 生成能力（供脚本使用）。
+ *
+ * 约定：
+ * - 该模块只负责“读取 + 校验 + 默认值补全”，不负责创建/写入配置文件。
+ * - 初始化/拷贝配置请使用 `config.example.yaml` + `scripts/config-init.js`。
+ */
+
+import fs from 'fs';
+import path from 'path';
+import yaml from 'yaml';
+
+import { logger } from './logger.js';
+
+const CONFIG_PATH = path.join(process.cwd(), 'config.yaml');
+const EXAMPLE_CONFIG_PATH = path.join(process.cwd(), 'config.example.yaml');
+
+// 模块级缓存：确保配置只从磁盘读取一次
+let cachedConfig = null;
+
+/**
+ * 加载并校验配置（只读）
+ * @returns {object} 配置对象
+ */
+export function loadConfig() {
+    // 如果已有缓存，直接返回
+    if (cachedConfig) return cachedConfig;
+
+    if (!fs.existsSync(CONFIG_PATH)) {
+        const hint = fs.existsSync(EXAMPLE_CONFIG_PATH)
+            ? `请复制 ${EXAMPLE_CONFIG_PATH} 为 ${CONFIG_PATH}`
+            : `请创建 ${CONFIG_PATH}（仓库根目录通常提供 config.example.yaml 作为模板）`;
+        throw new Error(`未找到配置文件: ${CONFIG_PATH}。${hint}`);
+    }
+
+    const configFile = fs.readFileSync(CONFIG_PATH, 'utf8');
+    const config = yaml.parse(configFile);
+    if (!config || typeof config !== 'object') {
+        throw new Error(`配置文件解析失败: ${CONFIG_PATH}`);
+    }
+
+    // Docker 路径兼容处理
+    if ((!config.browser?.path || !fs.existsSync(config.browser.path)) &&
+        fs.existsSync('/app/camoufox/camoufox')) {
+        logger.info('配置器', '检测到容器环境，自动修正浏览器路径为 /app/camoufox/camoufox');
+        if (!config.browser) config.browser = {};
+        config.browser.path = '/app/camoufox/camoufox';
+    }
+
+    // 基础配置校验
+    if (!config.server || !config.server.port) {
+        throw new Error('配置文件缺少必需字段: server.port');
+    }
+    if (!config.server.auth) {
+        throw new Error('配置文件缺少必需字段: server.auth');
+    }
+
+    // 设置队列配置默认值
+    if (!config.queue) {
+        config.queue = {
+            maxConcurrent: 1,
+            maxQueueSize: 2,
+            imageLimit: 5
+        };
+    } else {
+        // 强制 maxConcurrent 为 1
+        config.queue.maxConcurrent = 1;
+        if (config.queue.maxQueueSize === undefined) config.queue.maxQueueSize = 2;
+        if (config.queue.imageLimit === undefined) config.queue.imageLimit = 5;
+    }
+
+    // 设置 keepalive 配置默认值（兼容旧字段：keepalive.enable）
+    if (!config.server.keepalive) {
+        config.server.keepalive = { mode: 'comment' };
+    } else {
+        if (config.server.keepalive.mode === undefined) config.server.keepalive.mode = 'comment';
+        // 验证 mode 值
+        if (!['comment', 'content'].includes(config.server.keepalive.mode)) {
+            logger.warn('配置器', `无效的 keepalive.mode: ${config.server.keepalive.mode}，使用默认值 comment`);
+            config.server.keepalive.mode = 'comment';
+        }
+    }
+
+    // 设置 backend 配置默认值
+    if (!config.backend) {
+        config.backend = {
+            type: 'lmarena',
+            geminiBiz: { entryUrl: '' }
+        };
+    }
+
+    // 新增：Merge 配置初始化
+    if (!config.backend.merge) {
+        config.backend.merge = {
+            enable: false,
+            type: ['zai_is', 'lmarena']
+        };
+    }
+
+    // 校验 GeminiBiz 配置
+    if (config.backend.type === 'gemini_biz') {
+        if (!config.backend.geminiBiz || !config.backend.geminiBiz.entryUrl) {
+            throw new Error('backend.type = gemini_biz requires backend.geminiBiz.entryUrl');
+        }
+    }
+
+    logger.debug('配置器', '已加载 config.yaml');
+    logger.debug('配置器', '后端类型:', config.backend.type);
+    logger.debug('配置器', '流式心跳模式:', config.server.keepalive.mode);
+    if (config.backend.type === 'gemini_biz') {
+        logger.debug('配置器', `GeminiBiz 入口: ${config.backend.geminiBiz.entryUrl}`);
+    }
+
+    // 设置日志级别
+    if (config.logLevel) {
+        logger.setLevel(config.logLevel);
+    }
+
+    // 缓存配置
+    cachedConfig = config;
+    return config;
+}
+
+// 默认导出为函数
+export default loadConfig;
+
+