@pikecode/api-key-manager 1.0.38 → 1.0.42

package/src/constants/index.js

@@ -0,0 +1,246 @@
+/**
+ * 全局常量定义
+ * 统一管理项目中重复出现的常量，提高可维护性
+ */
+
+// ============ 认证模式常量 ============
+
+/**
+ * 认证模式的显示名称映射
+ */
+const AUTH_MODE_DISPLAY = {
+  api_key: '通用API密钥模式',
+  auth_token: '认证令牌模式',
+  oauth_token: 'OAuth令牌模式'
+};
+
+/**
+ * 认证模式详细说明（用于添加供应商时）
+ */
+const AUTH_MODE_DISPLAY_DETAILED = {
+  api_key: '通用API密钥模式',
+  auth_token: '认证令牌模式 (仅 ANTHROPIC_AUTH_TOKEN)',
+  oauth_token: 'OAuth令牌模式 (CLAUDE_CODE_OAUTH_TOKEN)'
+};
+
+/**
+ * 所有支持的认证模式列表
+ */
+const AUTH_MODES = ['api_key', 'auth_token', 'oauth_token'];
+
+/**
+ * Token 类型显示映射
+ */
+const TOKEN_TYPE_DISPLAY = {
+  auth_token: 'ANTHROPIC_AUTH_TOKEN',
+  api_key: 'ANTHROPIC_API_KEY'
+};
+
+// ============ ESC 返回提示常量 ============
+
+/**
+ * ESC 导航提示消息
+ */
+const ESC_HINTS = {
+  BACK_TO_PROVIDER_SELECTION: '返回供应商选择',
+  BACK_TO_MANAGE_LIST: '返回管理列表',
+  BACK_TO_QUICK_SETTINGS: '返回快速设置',
+  BACK_TO_MAIN_MENU: '返回主菜单',
+  EXIT_PROGRAM: '退出程序',
+  CANCEL_ADD: '取消添加'
+};
+
+/**
+ * ESC 日志消息
+ */
+const ESC_LOGS = {
+  BACK_TO_PROVIDER_SELECTION: '返回供应商选择',
+  BACK_TO_MANAGE_LIST: '返回管理列表',
+  BACK_TO_QUICK_SETTINGS: '返回快速设置',
+  BACK_TO_MAIN_MENU: '返回主菜单',
+  EXIT_PROGRAM: '退出程序',
+  CANCEL_ADD: '取消添加供应商',
+  CANCEL_LAUNCH: '已取消启动'
+};
+
+// ============ IDE 类型常量 ============
+
+/**
+ * IDE 名称常量
+ */
+const IDE_NAMES = {
+  CLAUDE: 'claude',
+  CODEX: 'codex'
+};
+
+/**
+ * IDE 显示名称
+ */
+const IDE_DISPLAY_NAMES = {
+  claude: 'Claude Code',
+  codex: 'Codex CLI'
+};
+
+// ============ 状态检查常量 ============
+
+/**
+ * 供应商状态
+ */
+const PROVIDER_STATUS = {
+  ONLINE: 'online',
+  DEGRADED: 'degraded',
+  OFFLINE: 'offline',
+  PENDING: 'pending',
+  UNKNOWN: 'unknown'
+};
+
+/**
+ * 状态图标映射
+ */
+const STATUS_ICONS = {
+  online: '🟢',
+  degraded: '🟡',
+  offline: '🔴',
+  pending: '⏳',
+  unknown: '⚪'
+};
+
+/**
+ * 状态颜色配置 (chalk 颜色名称)
+ */
+const STATUS_COLORS = {
+  online: 'green',
+  degraded: 'yellow',
+  offline: 'red',
+  pending: 'gray',
+  unknown: 'gray'
+};
+
+// ============ 操作类型常量 ============
+
+/**
+ * 特殊选择值
+ */
+const SPECIAL_SELECTION = {
+  ADD_PROVIDER: '__ADD__',
+  MANAGE_PROVIDER: '__MANAGE__',
+  OPEN_CONFIG: '__OPEN_CONFIG__',
+  EXIT: '__EXIT__',
+  BACK: 'back'
+};
+
+// ============ 配置相关常量 ============
+
+/**
+ * 基础 URL 相关的常量
+ */
+const BASE_URL = {
+  OFFICIAL_DEFAULT: '✨ 官方默认服务器',
+  NOT_SET: '⚠️ 未设置'
+};
+
+/**
+ * 当前状态显示
+ */
+const CURRENT_STATUS = {
+  IN_USE: '✅ 使用中',
+  NOT_IN_USE: '⚫ 未使用'
+};
+
+// ============ 模型相关常量 ============
+
+/**
+ * 默认模型值
+ */
+const DEFAULT_MODELS = {
+  NOT_SET: '未设置'
+};
+
+/**
+ * 模型类型
+ */
+const MODEL_TYPES = {
+  PRIMARY: 'primary',
+  SMALL_FAST: 'smallFast'
+};
+
+// ============ 消息常量 ============
+
+/**
+ * 系统消息
+ */
+const MESSAGES = {
+  CONFIGURATION_COMPLETE: '🎉 供应商添加完成！正在返回主界面...',
+  NO_PROVIDER_CONFIGURED: '暂无配置的供应商',
+  NO_CODEX_PROVIDER: '暂无 Codex CLI 供应商配置',
+  NO_CLAUDE_PROVIDER: '暂无 Claude Code 供应商配置',
+  ADD_FIRST_PROVIDER: '请先运行 "akm add" 添加供应商配置'
+};
+
+// ============ API 相关常量 ============
+
+/**
+ * API 请求配置
+ */
+const API_CONFIG = {
+  // 请求超时时间（毫秒）
+  DEFAULT_TIMEOUT: 5000,
+  // 状态缓存时间（毫秒）
+  CACHE_TTL: 30000,
+  // 默认测试消息
+  TEST_MESSAGE: '你好',
+  // 默认最大 Token 数
+  MAX_TOKENS: 32,
+  // 默认模型
+  DEFAULT_MODEL: 'claude-haiku-4-5-20251001'
+};
+
+/**
+ * 环境变量名称
+ */
+const ENV_VARS = {
+  ANTHROPIC_AUTH_TOKEN: 'ANTHROPIC_AUTH_TOKEN',
+  ANTHROPIC_API_KEY: 'ANTHROPIC_API_KEY',
+  OPENAI_API_KEY: 'OPENAI_API_KEY',
+  OPENAI_BASE_URL: 'OPENAI_BASE_URL',
+  CLAUDE_CODE_OAUTH_TOKEN: 'CLAUDE_CODE_OAUTH_TOKEN'
+};
+
+module.exports = {
+  // 认证模式
+  AUTH_MODE_DISPLAY,
+  AUTH_MODE_DISPLAY_DETAILED,
+  AUTH_MODES,
+  TOKEN_TYPE_DISPLAY,
+
+  // ESC 提示
+  ESC_HINTS,
+  ESC_LOGS,
+
+  // IDE
+  IDE_NAMES,
+  IDE_DISPLAY_NAMES,
+
+  // 状态
+  PROVIDER_STATUS,
+  STATUS_ICONS,
+  STATUS_COLORS,
+
+  // 操作
+  SPECIAL_SELECTION,
+
+  // 配置
+  BASE_URL,
+  CURRENT_STATUS,
+
+  // 模型
+  DEFAULT_MODELS,
+  MODEL_TYPES,
+
+  // 消息
+  MESSAGES,
+
+  // API
+  API_CONFIG,
+  ENV_VARS
+};

package/src/index.js

@@ -2,14 +2,21 @@ require('./utils/inquirer-setup');
 const { switchCommand } = require('./commands/switch');
 const { Logger } = require('./utils/logger');
 
-async function main(providerName) {
+/**
+ * 主入口函数
+ * @param {string} providerName - 供应商名称
+ * @param {Object} options - 选项
+ * @param {boolean} options.quick - 快速启动
+ * @param {boolean} options.noArgs - 不使用任何启动参数
+ */
+async function main(providerName, options = {}) {
   try {
     if (providerName) {
       // 直接切换到指定供应商
-      await switchCommand(providerName);
+      await switchCommand(providerName, options);
     } else {
       // 显示供应商选择界面
-      await switchCommand();
+      await switchCommand(null, options);
     }
   } catch (error) {
     Logger.fatal(`程序执行失败: ${error.message}`);

package/src/utils/codex-files.js

@@ -121,21 +121,68 @@ function ensureApiKeyAuthMethod(configToml) {
     return 'preferred_auth_method = "apikey"\n';
   }
 
-  // 检查是否已经设置了 preferred_auth_method
-  const authMethodRegex = /^preferred_auth_method\s*=\s*["']?([^"'\n]+)["']?\s*$/m;
-  const match = configToml.match(authMethodRegex);
-
-  if (match) {
-    if (match[1].trim() === 'apikey') {
-      // 已经是 apikey，无需修改
-      return configToml;
-    }
-    // 替换为 apikey
-    return configToml.replace(authMethodRegex, 'preferred_auth_method = "apikey"');
+  // 移除所有已存在的 preferred_auth_method 行（防止重复）
+  const authMethodRegex = /^preferred_auth_method\s*=\s*["']?[^"'\n]*["']?\s*\n?/gm;
+  let cleanedConfig = configToml.replace(authMethodRegex, '');
+
+  // 清理可能存在的损坏字符（如孤立的 { 或空行堆积）
+  cleanedConfig = cleanedConfig.replace(/^[{}]+\s*\n?/gm, '');
+  cleanedConfig = cleanedConfig.replace(/^\n{3,}/gm, '\n\n');
+
+  // 在文件开头添加 preferred_auth_method
+  const newLine = 'preferred_auth_method = "apikey"\n';
+
+  // 如果清理后为空，直接返回新配置
+  if (!cleanedConfig.trim()) {
+    return newLine;
   }
 
-  // 没有找到 preferred_auth_method，在文件开头添加
-  return 'preferred_auth_method = "apikey"\n' + configToml;
+  return newLine + cleanedConfig;
+}
+
+/**
+ * 更新 config.toml 中的 api_base_url
+ * @param {string} configToml - 现有的 config.toml 内容
+ * @param {string|null} baseUrl - API base URL，null 时移除该配置
+ * @returns {string} 更新后的 config.toml 内容
+ */
+function updateApiBaseUrl(configToml, baseUrl) {
+  if (!configToml) {
+    configToml = '';
+  }
+
+  // 匹配 api_base_url 配置行
+  const baseUrlRegex = /^api_base_url\s*=\s*["']?[^"'\n]*["']?\s*\n?/m;
+
+  if (baseUrl) {
+    // 需要设置 api_base_url
+    const newLine = `api_base_url = "${baseUrl}"`;
+
+    if (configToml.match(baseUrlRegex)) {
+      // 替换现有的
+      return configToml.replace(baseUrlRegex, newLine + '\n');
+    }
+    // 空配置，直接返回新行
+    if (configToml.length === 0) {
+      return newLine + '\n';
+    }
+    // 找到第一个 section [xxx]，在它之前插入
+    const sectionMatch = configToml.match(/^\[/m);
+    if (sectionMatch) {
+      const insertPos = configToml.indexOf(sectionMatch[0]);
+      const before = configToml.slice(0, insertPos);
+      const after = configToml.slice(insertPos);
+      // 确保前面有换行分隔
+      const separator = before.endsWith('\n') ? '' : '\n';
+      return before + separator + newLine + '\n\n' + after;
+    }
+    // 没有 section，在文件末尾添加
+    const separator = configToml.endsWith('\n') ? '' : '\n';
+    return configToml + separator + newLine + '\n';
+  } else {
+    // 移除 api_base_url（使用官方 API）
+    return configToml.replace(baseUrlRegex, '');
+  }
 }
 
 /**
@@ -172,7 +219,11 @@ async function applyCodexConfig(config, options = {}) {
   }
 
   // 确保设置了 preferred_auth_method = "apikey"
-  const updatedConfigToml = ensureApiKeyAuthMethod(existingConfigToml);
+  let updatedConfigToml = ensureApiKeyAuthMethod(existingConfigToml);
+
+  // 更新 api_base_url（如果有则设置，没有则移除）
+  updatedConfigToml = updateApiBaseUrl(updatedConfigToml, config.baseUrl || null);
+
   await fs.writeFile(configTomlPath, updatedConfigToml, 'utf8');
   await setSecurePermissions(configTomlPath);
 
@@ -192,6 +243,7 @@ module.exports = {
   applyCodexConfig,
   backupCodexFiles,
   ensureApiKeyAuthMethod,
+  updateApiBaseUrl,
   buildAuthJson
 };
 

package/src/utils/codex-launcher.js

@@ -48,8 +48,11 @@ async function executeCodexWithEnv(config, launchArgs = []) {
 
   // 写入 ~/.codex/config.toml 和 ~/.codex/auth.json
   // 确保 Codex CLI 使用 API Key 认证方式
+  // 这样用户也可以直接运行 `codex` 命令而无需通过 akm
   await applyCodexConfig(config);
 
+  // 同时设置环境变量，确保兼容性
+  // 环境变量优先级更高，作为双重保障
   const env = buildCodexEnvVariables(config);
 
   // 处理参数：子命令放前面，选项放后面

package/src/utils/config-opener.js

@@ -1,3 +1,9 @@
+/**
+ * Config Opener Utility
+ * 使用默认应用打开配置文件
+ * @module utils/config-opener
+ */
+
 const fs = require('fs-extra');
 const path = require('path');
 const os = require('os');
@@ -5,10 +11,19 @@ const { spawn } = require('child_process');
 
 const AKM_CONFIG_FILE = path.join(os.homedir(), '.akm-config.json');
 
+/**
+ * 检查配置文件是否存在
+ * @returns {Promise<boolean>} 文件是否存在
+ */
 function ensureConfigExists() {
   return fs.pathExists(AKM_CONFIG_FILE);
 }
 
+/**
+ * 使用系统默认应用打开文件
+ * @param {string} filePath - 文件路径
+ * @returns {Promise<void>}
+ */
 function openFileWithDefaultApp(filePath) {
   return new Promise((resolve, reject) => {
     let command;

package/src/utils/env-utils.js

@@ -14,7 +14,7 @@ function sanitizeEnvValue(value) {
   }
 
   // 移除控制字符
-  let cleaned = value.replace(/[\x00-\x08\x0B\x0C\x0E-\x1F]/g, '');
+  const cleaned = value.replace(/[\x00-\x08\x0B\x0C\x0E-\x1F]/g, '');
 
   // 检测可能的 shell 命令注入（允许 $ 因为 token 可能包含）
   // 只禁止明确的命令分隔符和反引号执行

package/src/utils/error-handler.js

@@ -4,32 +4,32 @@ const { Logger } = require('./logger');
 class ErrorHandler {
   static handle(error, context = '') {
     const contextStr = context ? `[${context}] ` : '';
-    
+
     switch (error.type) {
-      case 'CONFIG_NOT_FOUND':
-        Logger.error(`${contextStr}配置文件不存在，请先添加供应商`);
-        break;
-      case 'PROVIDER_NOT_FOUND':
-        Logger.error(`${contextStr}指定的供应商不存在`);
-        break;
-      case 'INVALID_TOKEN':
-        Logger.error(`${contextStr}Token格式无效`);
-        break;
-      case 'INVALID_URL':
-        Logger.error(`${contextStr}URL格式无效`);
-        break;
-      case 'INVALID_NAME':
-        Logger.error(`${contextStr}供应商名称格式无效`);
-        break;
-      case 'FILE_PERMISSION':
-        Logger.error(`${contextStr}文件权限不足`);
-        break;
-      case 'NETWORK_ERROR':
-        Logger.error(`${contextStr}网络连接错误`);
-        break;
-      default:
-        Logger.error(`${contextStr}${error.message || '未知错误'}`);
-        break;
+    case 'CONFIG_NOT_FOUND':
+      Logger.error(`${contextStr}配置文件不存在，请先添加供应商`);
+      break;
+    case 'PROVIDER_NOT_FOUND':
+      Logger.error(`${contextStr}指定的供应商不存在`);
+      break;
+    case 'INVALID_TOKEN':
+      Logger.error(`${contextStr}Token格式无效`);
+      break;
+    case 'INVALID_URL':
+      Logger.error(`${contextStr}URL格式无效`);
+      break;
+    case 'INVALID_NAME':
+      Logger.error(`${contextStr}供应商名称格式无效`);
+      break;
+    case 'FILE_PERMISSION':
+      Logger.error(`${contextStr}文件权限不足`);
+      break;
+    case 'NETWORK_ERROR':
+      Logger.error(`${contextStr}网络连接错误`);
+      break;
+    default:
+      Logger.error(`${contextStr}${error.message || '未知错误'}`);
+      break;
     }
   }
 
@@ -50,4 +50,4 @@ class ErrorHandler {
   }
 }
 
-module.exports = { ErrorHandler };
+module.exports = { ErrorHandler };