duojie-helper 0.2.21 → 0.2.22

package/src/tools/droid.js

@@ -1,261 +1,262 @@
-import fs from 'fs-extra';
-import path from 'path';
-import os from 'os';
-import { API_CONFIG, getModels, getApiBaseForProtocol } from '../index.js';
-
-function normalizeBaseUrl(url) {
-  return `${url || ''}`.replace(/\/+$/, '');
-}
-
-/**
- * 获取 Droid (Factory) 配置路径
- */
-function getDroidConfigPaths() {
-  const home = os.homedir();
-  return {
-    configDir: path.join(home, '.factory'),
-    configFile: path.join(home, '.factory', 'config.json'),
-    settingsFile: path.join(home, '.factory', 'settings.json'),
-  };
-}
-
-/**
- * 生成友好的模型显示名称
- * @param {string} modelId - 模型 ID
- * @returns {string} 显示名称
- */
-function generateDisplayName(modelId) {
-  // 将 model id 转换为友好名称
-  // claude-opus-4-6-kiro -> Opus 4.6 Kiro
-  // gpt-5.2-codex -> GPT 5.2 Codex
-  // gemini-3-pro-preview -> Gemini 3 Pro Preview
-  
-  let name = modelId;
-  
-  // 移除常见前缀
-  if (name.startsWith('claude-')) {
-    name = name.substring(7);
-  } else if (name.startsWith('gpt-')) {
-    name = 'GPT ' + name.substring(4);
-  } else if (name.startsWith('gemini-')) {
-    name = 'Gemini ' + name.substring(7);
-  } else if (name.startsWith('glm-')) {
-    name = 'GLM-' + name.substring(4);
-  }
-  
-  // 将连字符转换为空格，并首字母大写
-  name = name
-    .split('-')
-    .map(part => {
-      // 处理版本号（如 4-5 -> 4.5）
-      if (/^\d+$/.test(part)) {
-        return part;
-      }
-      // 首字母大写
-      return part.charAt(0).toUpperCase() + part.slice(1);
-    })
-    .join(' ');
-  
-  // 处理版本号格式（4 5 -> 4.5）
-  name = name.replace(/(\d+)\s+(\d+)(?=\s|$)/g, '$1.$2');
-  
-  // 添加后缀标识
-  return `${name} [duojie.games]`;
-}
-
-/**
- * 根据模型 ID 确定 provider
- * @param {string} modelId - 模型 ID
- * @param {string[]} endpoints - 支持的 endpoint 类型
- * @returns {string} provider 名称
- */
-function determineProvider(modelId, endpoints = []) {
-  if (modelId.startsWith('gpt')) return 'openai';
-  return 'anthropic';
-}
-
-/**
- * 配置 Droid (Factory)
- * 自动配置所有可用模型到 custom_models
- */
-export async function configureDroid(apiKey) {
-  const paths = getDroidConfigPaths();
-  
-  try {
-    // 确保目录存在
-    await fs.ensureDir(paths.configDir);
-
-    // 1. 读取现有配置（如果存在）
-    let existingConfig = {};
-    if (await fs.pathExists(paths.configFile)) {
-      try {
-        existingConfig = await fs.readJson(paths.configFile);
-      } catch {
-        existingConfig = {};
-      }
-    }
-
-    // 2. 获取模型列表
-    const models = getModels();
-    const rawModels = API_CONFIG.rawModels || [];
-    
-    // 创建模型 ID 到 endpoints 的映射
-    const endpointsMap = {};
-    for (const m of rawModels) {
-      endpointsMap[m.id] = m.supported_endpoint_types || [];
-    }
-
-    // 3. 构建 Duojie 的 custom_models
-    const duojieModels = [];
-    
-    // 处理所有模型类别
-    const allModels = [
-      ...(models.claude || []),
-      ...(models.gpt || []),
-      ...(models.gemini || []),
-      ...(models.other || []),
-    ];
-
-    // 默认模型优先放在列表前（便于用户在 UI 中选择）
-    const preferredId = 'gpt-5.2';
-    const preferred = allModels.find(m => m.id === preferredId) || null;
-    const orderedModels = preferred
-      ? [preferred, ...allModels.filter(m => m.id !== preferredId)]
-      : allModels;
-    
-    for (const m of orderedModels) {
-      const endpoints = endpointsMap[m.id] || [];
-      const provider = determineProvider(m.id, endpoints);
-      
-      duojieModels.push({
-        model_display_name: generateDisplayName(m.id),
-        model: m.id,
-        base_url: normalizeBaseUrl(getApiBaseForProtocol(provider)),
-        api_key: apiKey,
-        provider: provider,
-        supports_vision: true,
-        max_tokens: 16384,
-      });
-    }
-
-    if (duojieModels.length === 0) {
-      return {
-        success: false,
-        message: '没有可用的模型',
-      };
-    }
-
-    // 4. 合并配置（保留用户其他自定义模型）
-    let existingCustomModels = existingConfig.custom_models || [];
-    
-    // 过滤掉已有的 Duojie 模型（通过 base_url 或 display_name 判断）
-    existingCustomModels = existingCustomModels.filter(m => {
-      const isDuojie = 
-        (m.base_url && m.base_url.includes('duojie')) ||
-        (m.model_display_name && m.model_display_name.includes('[duojie.games]'));
-      return !isDuojie;
-    });
-
-    // 5. 合并新旧模型
-    const newConfig = {
-      ...existingConfig,
-      custom_models: [
-        ...existingCustomModels,
-        ...duojieModels,
-      ],
-    };
-
-    // 6. 写入配置文件
-    await fs.writeJson(paths.configFile, newConfig, { spaces: 2 });
-
-    // 7. 清空 settings.json 中的 customModels（如文件和字段存在）
-    if (await fs.pathExists(paths.settingsFile)) {
-      try {
-        const settings = await fs.readJson(paths.settingsFile);
-        if (Array.isArray(settings.customModels) && settings.customModels.length > 0) {
-          settings.customModels = [];
-          await fs.writeJson(paths.settingsFile, settings, { spaces: 2 });
-        }
-      } catch {
-        // 忽略，不影响主流程
-      }
-    }
-
-    return {
-      success: true,
-      message: `已配置 ${duojieModels.length} 个模型 → ${paths.configFile}`,
-      configPath: paths.configFile,
-      hint: '重启 Factory 使配置生效',
-    };
-  } catch (error) {
-    return {
-      success: false,
-      message: `配置失败: ${error.message}`,
-    };
-  }
-}
-
-/**
- * 检查 Droid 配置状态
- */
-configureDroid.checkStatus = async function() {
-  const paths = getDroidConfigPaths();
-  
-  if (await fs.pathExists(paths.configFile)) {
-    try {
-      const config = await fs.readJson(paths.configFile);
-      
-      // 检查是否有 Duojie 的自定义模型
-      const customModels = config.custom_models || [];
-      const duojieModels = customModels.filter(m => 
-        (m.base_url && m.base_url.includes('duojie')) ||
-        (m.model_display_name && m.model_display_name.includes('[duojie.games]'))
-      );
-      
-      if (duojieModels.length > 0) {
-        return {
-          configured: true,
-          message: `已配置 ${duojieModels.length} 个 Duojie 模型`,
-        };
-      } else if (customModels.length > 0) {
-        return {
-          configured: true,
-          message: '已配置（非 Duojie）',
-        };
-      }
-    } catch {
-      // ignore
-    }
-  }
-  
-  return { configured: false };
-};
-
-/**
- * 撤销 Droid 配置
- */
-configureDroid.revoke = async function() {
-  const paths = getDroidConfigPaths();
-  
-  if (await fs.pathExists(paths.configFile)) {
-    try {
-      const config = await fs.readJson(paths.configFile);
-      
-      // 只删除 Duojie 的自定义模型
-      if (config.custom_models) {
-        config.custom_models = config.custom_models.filter(m => {
-          const isDuojie = 
-            (m.base_url && m.base_url.includes('duojie')) ||
-            (m.model_display_name && m.model_display_name.includes('[duojie.games]'));
-          return !isDuojie;
-        });
-      }
-      
-      await fs.writeJson(paths.configFile, config, { spaces: 2 });
-    } catch {
-      // ignore
-    }
-  }
-};
-
-export default configureDroid;
+import fs from 'fs-extra';
+import path from 'path';
+import os from 'os';
+import { API_CONFIG, getModels, getApiBaseForProtocol } from '../index.js';
+
+function normalizeBaseUrl(url) {
+  return `${url || ''}`.replace(/\/+$/, '');
+}
+
+/**
+ * 获取 Droid (Factory) 配置路径
+ */
+function getDroidConfigPaths() {
+  const home = os.homedir();
+  return {
+    configDir: path.join(home, '.factory'),
+    configFile: path.join(home, '.factory', 'config.json'),
+    settingsFile: path.join(home, '.factory', 'settings.json'),
+  };
+}
+
+/**
+ * 生成友好的模型显示名称
+ * @param {string} modelId - 模型 ID
+ * @returns {string} 显示名称
+ */
+function generateDisplayName(modelId) {
+  // 将 model id 转换为友好名称
+  // claude-opus-4-6-kiro -> Opus 4.6 Kiro
+  // gpt-5.2-codex -> GPT 5.2 Codex
+  // gemini-3-pro-preview -> Gemini 3 Pro Preview
+  
+  let name = modelId;
+  
+  // 移除常见前缀
+  if (name.startsWith('claude-')) {
+    name = name.substring(7);
+  } else if (name.startsWith('gpt-')) {
+    name = 'GPT ' + name.substring(4);
+  } else if (name.startsWith('gemini-')) {
+    name = 'Gemini ' + name.substring(7);
+  } else if (name.startsWith('glm-')) {
+    name = 'GLM-' + name.substring(4);
+  }
+  
+  // 将连字符转换为空格，并首字母大写
+  name = name
+    .split('-')
+    .map(part => {
+      // 处理版本号（如 4-5 -> 4.5）
+      if (/^\d+$/.test(part)) {
+        return part;
+      }
+      // 首字母大写
+      return part.charAt(0).toUpperCase() + part.slice(1);
+    })
+    .join(' ');
+  
+  // 处理版本号格式（4 5 -> 4.5）
+  name = name.replace(/(\d+)\s+(\d+)(?=\s|$)/g, '$1.$2');
+  
+  // 添加后缀标识
+  return `${name} [duojie.games]`;
+}
+
+/**
+ * 根据模型 ID 确定 provider
+ * @param {string} modelId - 模型 ID
+ * @param {string[]} endpoints - 支持的 endpoint 类型
+ * @returns {string} provider 名称
+ */
+function determineProvider(modelId, endpoints = []) {
+  if (modelId.startsWith('gpt')) return 'openai';
+  if (modelId.startsWith('glm')) return 'generic-chat-completion-api';
+  return 'anthropic';
+}
+
+/**
+ * 配置 Droid (Factory)
+ * 自动配置所有可用模型到 custom_models
+ */
+export async function configureDroid(apiKey) {
+  const paths = getDroidConfigPaths();
+  
+  try {
+    // 确保目录存在
+    await fs.ensureDir(paths.configDir);
+
+    // 1. 读取现有配置（如果存在）
+    let existingConfig = {};
+    if (await fs.pathExists(paths.configFile)) {
+      try {
+        existingConfig = await fs.readJson(paths.configFile);
+      } catch {
+        existingConfig = {};
+      }
+    }
+
+    // 2. 获取模型列表
+    const models = getModels();
+    const rawModels = API_CONFIG.rawModels || [];
+    
+    // 创建模型 ID 到 endpoints 的映射
+    const endpointsMap = {};
+    for (const m of rawModels) {
+      endpointsMap[m.id] = m.supported_endpoint_types || [];
+    }
+
+    // 3. 构建 Duojie 的 custom_models
+    const duojieModels = [];
+    
+    // 处理所有模型类别
+    const allModels = [
+      ...(models.claude || []),
+      ...(models.gpt || []),
+      ...(models.gemini || []),
+      ...(models.other || []),
+    ];
+
+    // 默认模型优先放在列表前（便于用户在 UI 中选择）
+    const preferredId = 'claude-opus-4-6-gemini';
+    const preferred = allModels.find(m => m.id === preferredId) || null;
+    const orderedModels = preferred
+      ? [preferred, ...allModels.filter(m => m.id !== preferredId)]
+      : allModels;
+    
+    for (const m of orderedModels) {
+      const endpoints = endpointsMap[m.id] || [];
+      const provider = determineProvider(m.id, endpoints);
+      
+      duojieModels.push({
+        model_display_name: generateDisplayName(m.id),
+        model: m.id,
+        base_url: normalizeBaseUrl(getApiBaseForProtocol(provider)),
+        api_key: apiKey,
+        provider: provider,
+        supports_vision: true,
+        max_tokens: 16384,
+      });
+    }
+
+    if (duojieModels.length === 0) {
+      return {
+        success: false,
+        message: '没有可用的模型',
+      };
+    }
+
+    // 4. 合并配置（保留用户其他自定义模型）
+    let existingCustomModels = existingConfig.custom_models || [];
+    
+    // 过滤掉已有的 Duojie 模型（通过 base_url 或 display_name 判断）
+    existingCustomModels = existingCustomModels.filter(m => {
+      const isDuojie = 
+        (m.base_url && m.base_url.includes('duojie')) ||
+        (m.model_display_name && m.model_display_name.includes('[duojie.games]'));
+      return !isDuojie;
+    });
+
+    // 5. 合并新旧模型
+    const newConfig = {
+      ...existingConfig,
+      custom_models: [
+        ...existingCustomModels,
+        ...duojieModels,
+      ],
+    };
+
+    // 6. 写入配置文件
+    await fs.writeJson(paths.configFile, newConfig, { spaces: 2 });
+
+    // 7. 清空 settings.json 中的 customModels（如文件和字段存在）
+    if (await fs.pathExists(paths.settingsFile)) {
+      try {
+        const settings = await fs.readJson(paths.settingsFile);
+        if (Array.isArray(settings.customModels) && settings.customModels.length > 0) {
+          settings.customModels = [];
+          await fs.writeJson(paths.settingsFile, settings, { spaces: 2 });
+        }
+      } catch {
+        // 忽略，不影响主流程
+      }
+    }
+
+    return {
+      success: true,
+      message: `已配置 ${duojieModels.length} 个模型 → ${paths.configFile}`,
+      configPath: paths.configFile,
+      hint: '重启 Factory 使配置生效',
+    };
+  } catch (error) {
+    return {
+      success: false,
+      message: `配置失败: ${error.message}`,
+    };
+  }
+}
+
+/**
+ * 检查 Droid 配置状态
+ */
+configureDroid.checkStatus = async function() {
+  const paths = getDroidConfigPaths();
+  
+  if (await fs.pathExists(paths.configFile)) {
+    try {
+      const config = await fs.readJson(paths.configFile);
+      
+      // 检查是否有 Duojie 的自定义模型
+      const customModels = config.custom_models || [];
+      const duojieModels = customModels.filter(m => 
+        (m.base_url && m.base_url.includes('duojie')) ||
+        (m.model_display_name && m.model_display_name.includes('[duojie.games]'))
+      );
+      
+      if (duojieModels.length > 0) {
+        return {
+          configured: true,
+          message: `已配置 ${duojieModels.length} 个 Duojie 模型`,
+        };
+      } else if (customModels.length > 0) {
+        return {
+          configured: true,
+          message: '已配置（非 Duojie）',
+        };
+      }
+    } catch {
+      // ignore
+    }
+  }
+  
+  return { configured: false };
+};
+
+/**
+ * 撤销 Droid 配置
+ */
+configureDroid.revoke = async function() {
+  const paths = getDroidConfigPaths();
+  
+  if (await fs.pathExists(paths.configFile)) {
+    try {
+      const config = await fs.readJson(paths.configFile);
+      
+      // 只删除 Duojie 的自定义模型
+      if (config.custom_models) {
+        config.custom_models = config.custom_models.filter(m => {
+          const isDuojie = 
+            (m.base_url && m.base_url.includes('duojie')) ||
+            (m.model_display_name && m.model_display_name.includes('[duojie.games]'));
+          return !isDuojie;
+        });
+      }
+      
+      await fs.writeJson(paths.configFile, config, { spaces: 2 });
+    } catch {
+      // ignore
+    }
+  }
+};
+
+export default configureDroid;