duojie-helper 0.2.21 → 0.2.23

package/src/index.js

@@ -1,323 +1,324 @@
-import chalk from 'chalk';
-import ora from 'ora';
-import { configureClaudeCode } from './tools/claude.js';
-import { configureOpenClaw } from './tools/openclaw.js';
-import { configureCursor } from './tools/cursor.js';
-import { configureCline } from './tools/cline.js';
-import { configureCodex } from './tools/codex.js';
-import { configureContinue } from './tools/continue.js';
-import { configureOpenCode } from './tools/opencode.js';
-import { configureDroid } from './tools/droid.js';
-
-// API 配置
-export const API_CONFIG = {
-  baseUrl: 'https://api.duojie.games',
-  // 动态模型列表（运行时填充）
-  models: null,
-  // 原始模型数据（包含 endpoint 信息）
-  rawModels: null,
-};
-
-function normalizeBaseUrl(url) {
-  return `${url || ''}`.replace(/\/+$/, '');
-}
-
-function withV1(url) {
-  const normalized = normalizeBaseUrl(url);
-  return normalized.endsWith('/v1') ? normalized : `${normalized}/v1`;
-}
-
-function withoutV1(url) {
-  const normalized = normalizeBaseUrl(url);
-  return normalized.endsWith('/v1') ? normalized.slice(0, -3) : normalized;
-}
-
-export function getApiBaseForProtocol(protocol = '') {
-  const normalizedProtocol = `${protocol}`.toLowerCase();
-
-  const openaiProtocols = new Set([
-    'openai',
-    'openai-response',
-    'openai-responses',
-    'openai-completion',
-    'openai-completions',
-    'responses',
-    'chat',
-  ]);
-
-  if (openaiProtocols.has(normalizedProtocol)) {
-    return withV1(API_CONFIG.baseUrl);
-  }
-
-  return withoutV1(API_CONFIG.baseUrl);
-}
-
-/**
- * 从 pricing 页面获取公开模型 ID 集合
- * 返回 null 表示获取失败，调用方应跳过过滤
- */
-async function fetchPublicModelIds(apiKey) {
-  try {
-    const response = await fetch(`${API_CONFIG.baseUrl}/pricing`, {
-      headers: { 'Authorization': `Bearer ${apiKey}` },
-    });
-    if (!response.ok) return null;
-    const html = await response.text();
-    const ids = new Set();
-    for (const m of (html.match(/###\s+([^\n\r]+)/g) || [])) {
-      ids.add(m.replace(/^###\s+/, '').trim());
-    }
-    return ids.size > 0 ? ids : null;
-  } catch {
-    return null;
-  }
-}
-
-/**
- * 从 API 获取模型列表
- */
-export async function fetchModels(apiKey) {
-  try {
-    const [response, publicIds] = await Promise.all([
-      fetch(`${API_CONFIG.baseUrl}/v1/models`, {
-        headers: {
-          'Authorization': `Bearer ${apiKey}`,
-          'Content-Type': 'application/json',
-        },
-      }),
-      fetchPublicModelIds(apiKey),
-    ]);
-
-    if (!response.ok) {
-      throw new Error(`HTTP ${response.status}`);
-    }
-
-    const data = await response.json();
-    
-    // 保存原始数据
-    API_CONFIG.rawModels = data.data || [];
-    
-    // 解析模型列表，根据 ID 和 supported_endpoint_types 分类
-    const models = { claude: [], gpt: [], gemini: [], other: [] };
-    
-    if (data.data && Array.isArray(data.data)) {
-      for (const model of data.data) {
-        const id = model.id;
-        // 跳过未在 pricing 页面公开的隐藏模型
-        if (publicIds && !publicIds.has(id)) continue;
-        // 跳过含 image 的模型（图像生成类，不支持对话）
-        if (id.toLowerCase().includes('image')) continue;
-        const endpoints = model.supported_endpoint_types || [];
-        
-        // 判断最佳协议：GPT 用 openai，其余统一用 anthropic-messages
-        let api = 'anthropic-messages'; // 默认
-        if (id.startsWith('gpt')) {
-          // GPT 模型优先用 openai-responses，否则 openai-completions
-          if (endpoints.includes('openai-response')) {
-            api = 'openai-responses';
-          } else {
-            api = 'openai-completions';
-          }
-          models.gpt.push({ id, name: id, api, endpoints });
-        } else if (id.startsWith('claude')) {
-          models.claude.push({ id, name: id, api, endpoints });
-        } else if (id.startsWith('gemini')) {
-          models.gemini.push({ id, name: id, api, endpoints });
-        } else if (id.startsWith('glm')) {
-          models.other.push({ id, name: id, api, endpoints });
-        } else {
-          models.other.push({ id, name: id, api, endpoints });
-        }
-      }
-    }
-
-    // 缓存结果
-    API_CONFIG.models = models;
-    return models;
-  } catch (error) {
-    console.log(chalk.yellow(`⚠ 无法获取模型列表: ${error.message}`));
-    return { claude: [], gpt: [], gemini: [], other: [] };
-  }
-}
-
-/**
- * 获取模型列表（优先使用缓存）
- */
-export function getModels() {
-  return API_CONFIG.models || { claude: [], gpt: [], gemini: [], other: [] };
-}
-
-/**
- * 获取原始模型数据
- */
-export function getRawModels() {
-  return API_CONFIG.rawModels || [];
-}
-
-// 工具配置器映射
-const toolConfigurators = {
-  claude: configureClaudeCode,
-  openclaw: configureOpenClaw,
-  cursor: configureCursor,
-  cline: configureCline,
-  codex: configureCodex,
-  continue: configureContinue,
-  opencode: configureOpenCode,
-  droid: configureDroid,
-};
-
-// 工具信息
-const toolInfo = {
-  claude: {
-    name: 'Claude Code',
-    description: 'Anthropic 官方 CLI 工具（自动配置）',
-    configPath: '~/.claude/settings.json',
-  },
-  droid: {
-    name: 'Droid',
-    description: 'Factory 的 AI 编程工具（自动配置）',
-    configPath: '~/.factory/config.json',
-  },
-  openclaw: {
-    name: 'OpenClaw / Moltbot',
-    description: '多平台 AI 助手网关（自动配置）',
-    configPath: '~/.openclaw/openclaw.json',
-  },
-  opencode: {
-    name: 'OpenCode',
-    description: '终端 AI 编码助手（自动配置）',
-    configPath: '~/.config/opencode/opencode.json',
-  },
-  continue: {
-    name: 'Continue',
-    description: 'VS Code/JetBrains AI 插件（自动配置）',
-    configPath: '~/.continue/config.json',
-  },
-  codex: {
-    name: 'Codex CLI',
-    description: 'OpenAI Codex 命令行工具（自动配置）',
-    configPath: '~/.codex/config.toml',
-  },
-  cursor: {
-    name: 'Cursor',
-    description: 'AI 代码编辑器（需手动配置，需 Pro 订阅）',
-    configPath: 'Cursor Settings → Models',
-  },
-  cline: {
-    name: 'Cline',
-    description: 'VS Code AI 编程插件（需手动配置）',
-    configPath: 'Cline 插件设置',
-  },
-};
-
-/**
- * 配置所有选中的工具
- */
-export async function configureAll(apiKey, tools) {
-  const results = [];
-
-  for (const tool of tools) {
-    const configurator = toolConfigurators[tool];
-    if (configurator) {
-      try {
-        const result = await configurator(apiKey);
-        results.push({ tool: toolInfo[tool]?.name || tool, ...result });
-      } catch (error) {
-        results.push({
-          tool: toolInfo[tool]?.name || tool,
-          success: false,
-          message: error.message,
-        });
-      }
-    } else {
-      results.push({
-        tool,
-        success: false,
-        message: '暂不支持该工具',
-      });
-    }
-  }
-
-  return results;
-}
-
-/**
- * 配置单个工具
- */
-export async function configureTool(tool, apiKey) {
-  const configurator = toolConfigurators[tool];
-  if (!configurator) {
-    return { success: false, message: `不支持的工具: ${tool}` };
-  }
-
-  return await configurator(apiKey);
-}
-
-/**
- * 列出支持的工具
- */
-export function listTools() {
-  console.log(chalk.cyan('\n📋 支持的工具列表:\n'));
-  
-  for (const [key, info] of Object.entries(toolInfo)) {
-    console.log(chalk.green(`  ${key.padEnd(12)}`), chalk.white(info.name));
-    console.log(chalk.gray(`              ${info.description}`));
-    console.log(chalk.gray(`              配置路径: ${info.configPath}\n`));
-  }
-}
-
-/**
- * 显示当前配置状态
- */
-export async function showStatus() {
-  console.log(chalk.cyan('\n📊 配置状态:\n'));
-  
-  for (const [tool, info] of Object.entries(toolInfo)) {
-    const configurator = toolConfigurators[tool];
-    if (configurator && configurator.checkStatus) {
-      try {
-        const status = await configurator.checkStatus();
-        const icon = status.configured ? chalk.green('✓') : chalk.gray('○');
-        console.log(`  ${icon} ${info.name.padEnd(20)} ${status.message || ''}`);
-      } catch {
-        console.log(`  ${chalk.gray('○')} ${info.name.padEnd(20)} ${chalk.gray('未检测到')}`);
-      }
-    } else {
-      console.log(`  ${chalk.gray('?')} ${info.name.padEnd(20)} ${chalk.gray('状态检测暂不支持')}`);
-    }
-  }
-  console.log('');
-}
-
-/**
- * 撤销配置
- */
-export async function revokeKey(tool) {
-  const spinner = ora('正在撤销配置...').start();
-  
-  try {
-    if (tool) {
-      const configurator = toolConfigurators[tool];
-      if (configurator && configurator.revoke) {
-        await configurator.revoke();
-        spinner.succeed(`已撤销 ${toolInfo[tool]?.name || tool} 的配置`);
-      } else {
-        spinner.warn(`${tool} 暂不支持撤销操作`);
-      }
-    } else {
-      // 撤销所有
-      for (const [key, configurator] of Object.entries(toolConfigurators)) {
-        if (configurator.revoke) {
-          try {
-            await configurator.revoke();
-          } catch {
-            // 忽略单个工具的错误
-          }
-        }
-      }
-      spinner.succeed('已撤销所有配置');
-    }
-  } catch (error) {
-    spinner.fail(`撤销失败: ${error.message}`);
-  }
-}
+import chalk from 'chalk';
+import ora from 'ora';
+import { configureClaudeCode } from './tools/claude.js';
+import { configureOpenClaw } from './tools/openclaw.js';
+import { configureCursor } from './tools/cursor.js';
+import { configureCline } from './tools/cline.js';
+import { configureCodex } from './tools/codex.js';
+import { configureContinue } from './tools/continue.js';
+import { configureOpenCode } from './tools/opencode.js';
+import { configureDroid } from './tools/droid.js';
+
+// API 配置
+export const API_CONFIG = {
+  baseUrl: 'https://api.duojie.games',
+  // 动态模型列表（运行时填充）
+  models: null,
+  // 原始模型数据（包含 endpoint 信息）
+  rawModels: null,
+};
+
+function normalizeBaseUrl(url) {
+  return `${url || ''}`.replace(/\/+$/, '');
+}
+
+function withV1(url) {
+  const normalized = normalizeBaseUrl(url);
+  return normalized.endsWith('/v1') ? normalized : `${normalized}/v1`;
+}
+
+function withoutV1(url) {
+  const normalized = normalizeBaseUrl(url);
+  return normalized.endsWith('/v1') ? normalized.slice(0, -3) : normalized;
+}
+
+export function getApiBaseForProtocol(protocol = '') {
+  const normalizedProtocol = `${protocol}`.toLowerCase();
+
+  const openaiProtocols = new Set([
+    'openai',
+    'openai-response',
+    'openai-responses',
+    'openai-completion',
+    'openai-completions',
+    'responses',
+    'chat',
+    'generic-chat-completion-api',
+  ]);
+
+  if (openaiProtocols.has(normalizedProtocol)) {
+    return withV1(API_CONFIG.baseUrl);
+  }
+
+  return withoutV1(API_CONFIG.baseUrl);
+}
+
+/**
+ * 从 pricing 页面获取公开模型 ID 集合
+ * 返回 null 表示获取失败，调用方应跳过过滤
+ */
+async function fetchPublicModelIds(apiKey) {
+  try {
+    const response = await fetch(`${API_CONFIG.baseUrl}/pricing`, {
+      headers: { 'Authorization': `Bearer ${apiKey}` },
+    });
+    if (!response.ok) return null;
+    const html = await response.text();
+    const ids = new Set();
+    for (const m of (html.match(/###\s+([^\n\r]+)/g) || [])) {
+      ids.add(m.replace(/^###\s+/, '').trim());
+    }
+    return ids.size > 0 ? ids : null;
+  } catch {
+    return null;
+  }
+}
+
+/**
+ * 从 API 获取模型列表
+ */
+export async function fetchModels(apiKey) {
+  try {
+    const [response, publicIds] = await Promise.all([
+      fetch(`${API_CONFIG.baseUrl}/v1/models`, {
+        headers: {
+          'Authorization': `Bearer ${apiKey}`,
+          'Content-Type': 'application/json',
+        },
+      }),
+      fetchPublicModelIds(apiKey),
+    ]);
+
+    if (!response.ok) {
+      throw new Error(`HTTP ${response.status}`);
+    }
+
+    const data = await response.json();
+    
+    // 保存原始数据
+    API_CONFIG.rawModels = data.data || [];
+    
+    // 解析模型列表，根据 ID 和 supported_endpoint_types 分类
+    const models = { claude: [], gpt: [], gemini: [], other: [] };
+    
+    if (data.data && Array.isArray(data.data)) {
+      for (const model of data.data) {
+        const id = model.id;
+        // 跳过未在 pricing 页面公开的隐藏模型
+        if (publicIds && !publicIds.has(id)) continue;
+        // 跳过含 image 的模型（图像生成类，不支持对话）
+        if (id.toLowerCase().includes('image')) continue;
+        const endpoints = model.supported_endpoint_types || [];
+        
+        // 判断最佳协议：GPT 用 openai，其余统一用 anthropic-messages
+        let api = 'anthropic-messages'; // 默认
+        if (id.startsWith('gpt')) {
+          // GPT 模型优先用 openai-responses，否则 openai-completions
+          if (endpoints.includes('openai-response')) {
+            api = 'openai-responses';
+          } else {
+            api = 'openai-completions';
+          }
+          models.gpt.push({ id, name: id, api, endpoints });
+        } else if (id.startsWith('claude')) {
+          models.claude.push({ id, name: id, api, endpoints });
+        } else if (id.startsWith('gemini')) {
+          models.gemini.push({ id, name: id, api, endpoints });
+        } else if (id.startsWith('glm')) {
+          models.other.push({ id, name: id, api: 'openai-completions', endpoints });
+        } else {
+          models.other.push({ id, name: id, api, endpoints });
+        }
+      }
+    }
+
+    // 缓存结果
+    API_CONFIG.models = models;
+    return models;
+  } catch (error) {
+    console.log(chalk.yellow(`⚠ 无法获取模型列表: ${error.message}`));
+    return { claude: [], gpt: [], gemini: [], other: [] };
+  }
+}
+
+/**
+ * 获取模型列表（优先使用缓存）
+ */
+export function getModels() {
+  return API_CONFIG.models || { claude: [], gpt: [], gemini: [], other: [] };
+}
+
+/**
+ * 获取原始模型数据
+ */
+export function getRawModels() {
+  return API_CONFIG.rawModels || [];
+}
+
+// 工具配置器映射
+const toolConfigurators = {
+  claude: configureClaudeCode,
+  openclaw: configureOpenClaw,
+  cursor: configureCursor,
+  cline: configureCline,
+  codex: configureCodex,
+  continue: configureContinue,
+  opencode: configureOpenCode,
+  droid: configureDroid,
+};
+
+// 工具信息
+const toolInfo = {
+  claude: {
+    name: 'Claude Code',
+    description: 'Anthropic 官方 CLI 工具（自动配置）',
+    configPath: '~/.claude/settings.json',
+  },
+  droid: {
+    name: 'Droid',
+    description: 'Factory 的 AI 编程工具（自动配置）',
+    configPath: '~/.factory/config.json',
+  },
+  openclaw: {
+    name: 'OpenClaw / Moltbot',
+    description: '多平台 AI 助手网关（自动配置）',
+    configPath: '~/.openclaw/openclaw.json',
+  },
+  opencode: {
+    name: 'OpenCode',
+    description: '终端 AI 编码助手（自动配置）',
+    configPath: '~/.config/opencode/opencode.json',
+  },
+  continue: {
+    name: 'Continue',
+    description: 'VS Code/JetBrains AI 插件（自动配置）',
+    configPath: '~/.continue/config.json',
+  },
+  codex: {
+    name: 'Codex CLI',
+    description: 'OpenAI Codex 命令行工具（自动配置）',
+    configPath: '~/.codex/config.toml',
+  },
+  cursor: {
+    name: 'Cursor',
+    description: 'AI 代码编辑器（需手动配置，需 Pro 订阅）',
+    configPath: 'Cursor Settings → Models',
+  },
+  cline: {
+    name: 'Cline',
+    description: 'VS Code AI 编程插件（需手动配置）',
+    configPath: 'Cline 插件设置',
+  },
+};
+
+/**
+ * 配置所有选中的工具
+ */
+export async function configureAll(apiKey, tools) {
+  const results = [];
+
+  for (const tool of tools) {
+    const configurator = toolConfigurators[tool];
+    if (configurator) {
+      try {
+        const result = await configurator(apiKey);
+        results.push({ tool: toolInfo[tool]?.name || tool, ...result });
+      } catch (error) {
+        results.push({
+          tool: toolInfo[tool]?.name || tool,
+          success: false,
+          message: error.message,
+        });
+      }
+    } else {
+      results.push({
+        tool,
+        success: false,
+        message: '暂不支持该工具',
+      });
+    }
+  }
+
+  return results;
+}
+
+/**
+ * 配置单个工具
+ */
+export async function configureTool(tool, apiKey) {
+  const configurator = toolConfigurators[tool];
+  if (!configurator) {
+    return { success: false, message: `不支持的工具: ${tool}` };
+  }
+
+  return await configurator(apiKey);
+}
+
+/**
+ * 列出支持的工具
+ */
+export function listTools() {
+  console.log(chalk.cyan('\n📋 支持的工具列表:\n'));
+  
+  for (const [key, info] of Object.entries(toolInfo)) {
+    console.log(chalk.green(`  ${key.padEnd(12)}`), chalk.white(info.name));
+    console.log(chalk.gray(`              ${info.description}`));
+    console.log(chalk.gray(`              配置路径: ${info.configPath}\n`));
+  }
+}
+
+/**
+ * 显示当前配置状态
+ */
+export async function showStatus() {
+  console.log(chalk.cyan('\n📊 配置状态:\n'));
+  
+  for (const [tool, info] of Object.entries(toolInfo)) {
+    const configurator = toolConfigurators[tool];
+    if (configurator && configurator.checkStatus) {
+      try {
+        const status = await configurator.checkStatus();
+        const icon = status.configured ? chalk.green('✓') : chalk.gray('○');
+        console.log(`  ${icon} ${info.name.padEnd(20)} ${status.message || ''}`);
+      } catch {
+        console.log(`  ${chalk.gray('○')} ${info.name.padEnd(20)} ${chalk.gray('未检测到')}`);
+      }
+    } else {
+      console.log(`  ${chalk.gray('?')} ${info.name.padEnd(20)} ${chalk.gray('状态检测暂不支持')}`);
+    }
+  }
+  console.log('');
+}
+
+/**
+ * 撤销配置
+ */
+export async function revokeKey(tool) {
+  const spinner = ora('正在撤销配置...').start();
+  
+  try {
+    if (tool) {
+      const configurator = toolConfigurators[tool];
+      if (configurator && configurator.revoke) {
+        await configurator.revoke();
+        spinner.succeed(`已撤销 ${toolInfo[tool]?.name || tool} 的配置`);
+      } else {
+        spinner.warn(`${tool} 暂不支持撤销操作`);
+      }
+    } else {
+      // 撤销所有
+      for (const [key, configurator] of Object.entries(toolConfigurators)) {
+        if (configurator.revoke) {
+          try {
+            await configurator.revoke();
+          } catch {
+            // 忽略单个工具的错误
+          }
+        }
+      }
+      spinner.succeed('已撤销所有配置');
+    }
+  } catch (error) {
+    spinner.fail(`撤销失败: ${error.message}`);
+  }
+}