duojie-helper 0.2.21 → 0.2.22

package/src/tools/opencode.js

@@ -1,216 +1,216 @@
-import fs from 'fs-extra';
-import path from 'path';
-import os from 'os';
-import { API_CONFIG, getModels } from '../index.js';
-
-/**
- * 获取 OpenCode 配置路径
- */
-function getOpenCodeConfigPaths() {
-  const home = os.homedir();
-  
-  // OpenCode 统一使用 ~/.config/opencode 目录
-  const configDir = path.join(home, '.config', 'opencode');
-  
-  return {
-    configDir,
-    configFile: path.join(configDir, 'opencode.json'),
-  };
-}
-
-/**
- * 配置 OpenCode
- * 自动配置所有可用模型
- */
-export async function configureOpenCode(apiKey) {
-  const paths = getOpenCodeConfigPaths();
-  
-  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();
-    
-    // 3. 构建 Duojie provider 的模型配置
-    const duojieModels = {};
-    
-    // Claude 模型
-    for (const m of (models.claude || [])) {
-      duojieModels[m.id] = {
-        limit: {
-          context: 200000,
-          output: 16384,
-        },
-        modalities: {
-          input: ['text', 'image'],
-          output: ['text'],
-        },
-      };
-    }
-    
-    // GPT 模型
-    for (const m of (models.gpt || [])) {
-      duojieModels[m.id] = {
-        limit: {
-          context: 128000,
-          output: 16384,
-        },
-        modalities: {
-          input: ['text', 'image'],
-          output: ['text'],
-        },
-      };
-    }
-    
-    // Gemini 模型
-    for (const m of (models.gemini || [])) {
-      duojieModels[m.id] = {
-        limit: {
-          context: 1000000,
-          output: 16384,
-        },
-        modalities: {
-          input: ['text', 'image'],
-          output: ['text'],
-        },
-      };
-    }
-    
-    // 其他模型
-    for (const m of (models.other || [])) {
-      duojieModels[m.id] = {
-        limit: {
-          context: 128000,
-          output: 4096,
-        },
-        modalities: {
-          input: ['text', 'image'],
-          output: ['text'],
-        },
-      };
-    }
-
-    const totalModels = Object.keys(duojieModels).length;
-    
-    if (totalModels === 0) {
-      return {
-        success: false,
-        message: '没有可用的模型',
-      };
-    }
-
-    // 4. 构建 Duojie provider 配置
-    const duojieProvider = {
-      npm: '@ai-sdk/anthropic',
-      name: 'Duojie API',
-      options: {
-        baseURL: API_CONFIG.baseUrl,
-      },
-      models: duojieModels,
-    };
-
-    // 5. 合并配置（保留用户其他配置）
-    // 删除旧的 duojie provider
-    if (existingConfig.provider?.duojie) {
-      delete existingConfig.provider.duojie;
-    }
-
-    const preferredModel = duojieModels['gpt-5.2'] ? 'duojie/gpt-5.2' : null;
-
-    const newConfig = {
-      ...existingConfig,
-      $schema: existingConfig.$schema || 'https://opencode.ai/config.json',
-      provider: {
-        ...existingConfig.provider,
-        duojie: duojieProvider,
-      },
-      ...(preferredModel ? { model: preferredModel } : {}),
-    };
-
-    // 6. 写入配置文件
-    await fs.writeJson(paths.configFile, newConfig, { spaces: 2 });
-
-    // 7. 提示用户设置环境变量
-    return {
-      success: true,
-      message: `已配置 ${totalModels} 个模型 → ${paths.configFile}`,
-      configPath: paths.configFile,
-      hint: '请设置环境变量 ANTHROPIC_API_KEY 或在 OpenCode 中配置 API Key',
-    };
-  } catch (error) {
-    return {
-      success: false,
-      message: `配置失败: ${error.message}`,
-    };
-  }
-}
-
-/**
- * 检查 OpenCode 配置状态
- */
-configureOpenCode.checkStatus = async function() {
-  const paths = getOpenCodeConfigPaths();
-  
-  if (await fs.pathExists(paths.configFile)) {
-    try {
-      const config = await fs.readJson(paths.configFile);
-      
-      if (config.provider?.duojie) {
-        const modelCount = Object.keys(config.provider.duojie.models || {}).length;
-        return {
-          configured: true,
-          message: `已配置 ${modelCount} 个 Duojie 模型`,
-        };
-      } else if (config.provider && Object.keys(config.provider).length > 0) {
-        return {
-          configured: true,
-          message: '已配置（非 Duojie）',
-        };
-      }
-    } catch {
-      // ignore
-    }
-  }
-  
-  return { configured: false };
-};
-
-/**
- * 撤销 OpenCode 配置
- */
-configureOpenCode.revoke = async function() {
-  const paths = getOpenCodeConfigPaths();
-  
-  // 清理配置文件中的 duojie provider
-  if (await fs.pathExists(paths.configFile)) {
-    try {
-      const config = await fs.readJson(paths.configFile);
-      
-      if (config.provider?.duojie) {
-        delete config.provider.duojie;
-      }
-      
-      // 如果默认模型是 duojie 的，清除它
-      if (config.model?.startsWith('duojie/')) {
-        delete config.model;
-      }
-      
-      await fs.writeJson(paths.configFile, config, { spaces: 2 });
-    } catch {
-      // ignore
-    }
-  }
-};
-
-export default configureOpenCode;
+import fs from 'fs-extra';
+import path from 'path';
+import os from 'os';
+import { API_CONFIG, getModels } from '../index.js';
+
+/**
+ * 获取 OpenCode 配置路径
+ */
+function getOpenCodeConfigPaths() {
+  const home = os.homedir();
+  
+  // OpenCode 统一使用 ~/.config/opencode 目录
+  const configDir = path.join(home, '.config', 'opencode');
+  
+  return {
+    configDir,
+    configFile: path.join(configDir, 'opencode.json'),
+  };
+}
+
+/**
+ * 配置 OpenCode
+ * 自动配置所有可用模型
+ */
+export async function configureOpenCode(apiKey) {
+  const paths = getOpenCodeConfigPaths();
+  
+  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();
+    
+    // 3. 构建 Duojie provider 的模型配置
+    const duojieModels = {};
+    
+    // Claude 模型
+    for (const m of (models.claude || [])) {
+      duojieModels[m.id] = {
+        limit: {
+          context: 200000,
+          output: 16384,
+        },
+        modalities: {
+          input: ['text', 'image'],
+          output: ['text'],
+        },
+      };
+    }
+    
+    // GPT 模型
+    for (const m of (models.gpt || [])) {
+      duojieModels[m.id] = {
+        limit: {
+          context: 128000,
+          output: 16384,
+        },
+        modalities: {
+          input: ['text', 'image'],
+          output: ['text'],
+        },
+      };
+    }
+    
+    // Gemini 模型
+    for (const m of (models.gemini || [])) {
+      duojieModels[m.id] = {
+        limit: {
+          context: 1000000,
+          output: 16384,
+        },
+        modalities: {
+          input: ['text', 'image'],
+          output: ['text'],
+        },
+      };
+    }
+    
+    // 其他模型
+    for (const m of (models.other || [])) {
+      duojieModels[m.id] = {
+        limit: {
+          context: 128000,
+          output: 4096,
+        },
+        modalities: {
+          input: ['text', 'image'],
+          output: ['text'],
+        },
+      };
+    }
+
+    const totalModels = Object.keys(duojieModels).length;
+    
+    if (totalModels === 0) {
+      return {
+        success: false,
+        message: '没有可用的模型',
+      };
+    }
+
+    // 4. 构建 Duojie provider 配置
+    const duojieProvider = {
+      npm: '@ai-sdk/anthropic',
+      name: 'Duojie API',
+      options: {
+        baseURL: API_CONFIG.baseUrl,
+      },
+      models: duojieModels,
+    };
+
+    // 5. 合并配置（保留用户其他配置）
+    // 删除旧的 duojie provider
+    if (existingConfig.provider?.duojie) {
+      delete existingConfig.provider.duojie;
+    }
+
+    const preferredModel = duojieModels['gpt-5.2'] ? 'duojie/gpt-5.2' : null;
+
+    const newConfig = {
+      ...existingConfig,
+      $schema: existingConfig.$schema || 'https://opencode.ai/config.json',
+      provider: {
+        ...existingConfig.provider,
+        duojie: duojieProvider,
+      },
+      ...(preferredModel ? { model: preferredModel } : {}),
+    };
+
+    // 6. 写入配置文件
+    await fs.writeJson(paths.configFile, newConfig, { spaces: 2 });
+
+    // 7. 提示用户设置环境变量
+    return {
+      success: true,
+      message: `已配置 ${totalModels} 个模型 → ${paths.configFile}`,
+      configPath: paths.configFile,
+      hint: '请设置环境变量 ANTHROPIC_API_KEY 或在 OpenCode 中配置 API Key',
+    };
+  } catch (error) {
+    return {
+      success: false,
+      message: `配置失败: ${error.message}`,
+    };
+  }
+}
+
+/**
+ * 检查 OpenCode 配置状态
+ */
+configureOpenCode.checkStatus = async function() {
+  const paths = getOpenCodeConfigPaths();
+  
+  if (await fs.pathExists(paths.configFile)) {
+    try {
+      const config = await fs.readJson(paths.configFile);
+      
+      if (config.provider?.duojie) {
+        const modelCount = Object.keys(config.provider.duojie.models || {}).length;
+        return {
+          configured: true,
+          message: `已配置 ${modelCount} 个 Duojie 模型`,
+        };
+      } else if (config.provider && Object.keys(config.provider).length > 0) {
+        return {
+          configured: true,
+          message: '已配置（非 Duojie）',
+        };
+      }
+    } catch {
+      // ignore
+    }
+  }
+  
+  return { configured: false };
+};
+
+/**
+ * 撤销 OpenCode 配置
+ */
+configureOpenCode.revoke = async function() {
+  const paths = getOpenCodeConfigPaths();
+  
+  // 清理配置文件中的 duojie provider
+  if (await fs.pathExists(paths.configFile)) {
+    try {
+      const config = await fs.readJson(paths.configFile);
+      
+      if (config.provider?.duojie) {
+        delete config.provider.duojie;
+      }
+      
+      // 如果默认模型是 duojie 的，清除它
+      if (config.model?.startsWith('duojie/')) {
+        delete config.model;
+      }
+      
+      await fs.writeJson(paths.configFile, config, { spaces: 2 });
+    } catch {
+      // ignore
+    }
+  }
+};
+
+export default configureOpenCode;