@pikecode/api-key-manager 1.0.38 → 1.0.42

package/src/commands/validate.js

@@ -0,0 +1,310 @@
+/**
+ * Provider Validation Command
+ * 验证供应商配置的有效性
+ * @module commands/validate
+ */
+
+const chalk = require('chalk');
+const { configManager } = require('../config');
+const { Logger } = require('../utils/logger');
+const { ProviderStatusChecker } = require('../utils/provider-status-checker');
+const { UIHelper } = require('../utils/ui-helper');
+
+/**
+ * 供应商验证器
+ * 用于验证 API 供应商的 Token 和配置是否有效
+ */
+class ProviderValidator {
+  /**
+   * 创建供应商验证器实例
+   */
+  constructor() {
+    this.configManager = configManager;
+    this.statusChecker = new ProviderStatusChecker();
+  }
+
+  /**
+   * 验证单个供应商
+   * @param {string} providerName - 供应商名称或别名
+   * @returns {Promise<void>}
+   */
+  async validateOne(providerName) {
+    try {
+      await this.configManager.ensureLoaded();
+
+      // 支持别名查找
+      let provider = this.configManager.getProvider(providerName);
+      if (!provider) {
+        provider = this.configManager.getProviderByNameOrAlias(providerName);
+      }
+
+      if (!provider) {
+        Logger.error(`供应商 '${providerName}' 不存在`);
+        Logger.info('使用 "akm list" 查看所有已配置的供应商');
+        return;
+      }
+
+      console.log(chalk.blue(`\n🔍 正在验证供应商: ${provider.displayName} (${provider.name})`));
+      console.log(chalk.gray('═'.repeat(60)));
+
+      const loadingInterval = UIHelper.createLoadingAnimation('验证中...');
+
+      try {
+        const status = await this.statusChecker.check(provider, { skipCache: true });
+        UIHelper.clearLoadingAnimation(loadingInterval);
+
+        this._printValidationResult(provider, status);
+      } catch (error) {
+        UIHelper.clearLoadingAnimation(loadingInterval);
+        Logger.error(`验证失败: ${error.message}`);
+      }
+
+    } catch (error) {
+      Logger.error(`验证供应商失败: ${error.message}`);
+    }
+  }
+
+  /**
+   * 验证所有供应商
+   * @param {string|null} filter - 过滤器 ('codex', 'claude', 或 null 表示全部)
+   * @returns {Promise<void>}
+   */
+  async validateAll(filter = null) {
+    try {
+      await this.configManager.ensureLoaded();
+      let providers = this.configManager.listProviders();
+
+      // 应用过滤器
+      if (filter === 'codex') {
+        providers = providers.filter(p => p.ideName === 'codex');
+      } else if (filter === 'claude') {
+        providers = providers.filter(p => p.ideName !== 'codex');
+      }
+
+      if (providers.length === 0) {
+        if (filter) {
+          const filterName = filter === 'codex' ? 'Codex CLI' : 'Claude Code';
+          Logger.warning(`暂无 ${filterName} 供应商配置`);
+        } else {
+          Logger.warning('暂无配置的供应商');
+        }
+        Logger.info('请使用 "akm add" 添加供应商配置');
+        return;
+      }
+
+      const titleSuffix = filter === 'codex' ? ' (Codex CLI)' : (filter === 'claude' ? ' (Claude Code)' : '');
+      console.log(chalk.blue(`\n🔍 正在验证供应商配置${titleSuffix}...`));
+      console.log(chalk.gray('═'.repeat(60)));
+      console.log();
+
+      let completedCount = 0;
+      const total = providers.length;
+
+      // 创建进度显示
+      const progressInterval = setInterval(() => {
+        process.stdout.write(`\r验证中... ${completedCount}/${total}`);
+      }, 100);
+
+      try {
+        // 使用流式检查，实时显示结果
+        const results = await this.statusChecker.checkAllStreaming(providers, (providerName, status) => {
+          completedCount++;
+        });
+
+        clearInterval(progressInterval);
+        process.stdout.write('\r' + ' '.repeat(50) + '\r'); // 清除进度显示
+
+        // 显示结果
+        this._printBatchResults(providers, results);
+      } catch (error) {
+        clearInterval(progressInterval);
+        Logger.error(`批量验证失败: ${error.message}`);
+      }
+
+    } catch (error) {
+      Logger.error(`验证供应商失败: ${error.message}`);
+    }
+  }
+
+  /**
+   * 打印单个验证结果
+   * @private
+   * @param {Object} provider - 供应商配置
+   * @param {Object} status - 状态结果
+   */
+  _printValidationResult(provider, status) {
+    const stateIcon = this._getStateIcon(status.state);
+    const stateLabel = this._getStateLabel(status.state);
+    const stateColor = this._getStateColor(status.state);
+
+    console.log();
+    console.log(`${stateIcon} 状态: ${stateColor(stateLabel)}`);
+
+    if (status.message) {
+      console.log(`   消息: ${status.message}`);
+    }
+
+    if (status.latency !== null) {
+      console.log(`   响应时间: ${status.latency.toFixed(0)}ms`);
+    }
+
+    console.log();
+
+    // 配置详情
+    console.log(chalk.gray('配置详情:'));
+    console.log(chalk.gray(`   供应商名称: ${provider.name}`));
+    console.log(chalk.gray(`   显示名称: ${provider.displayName}`));
+    if (provider.alias) {
+      console.log(chalk.gray(`   别名: ${provider.alias}`));
+    }
+
+    if (provider.ideName === 'codex') {
+      console.log(chalk.gray('   IDE: Codex CLI'));
+      if (provider.baseUrl) {
+        console.log(chalk.gray(`   OPENAI_BASE_URL: ${provider.baseUrl}`));
+      }
+    } else {
+      console.log(chalk.gray(`   IDE: Claude Code`));
+      console.log(chalk.gray(`   认证模式: ${provider.authMode}`));
+      if (provider.baseUrl) {
+        console.log(chalk.gray(`   基础URL: ${provider.baseUrl}`));
+      }
+    }
+
+    // 根据状态提供建议
+    if (status.state === 'offline') {
+      console.log();
+      console.log(chalk.yellow('💡 建议:'));
+      console.log(chalk.yellow('   1. 检查 Token 是否正确'));
+      console.log(chalk.yellow('   2. 检查基础URL是否可访问'));
+      console.log(chalk.yellow('   3. 检查网络连接是否正常'));
+    } else if (status.state === 'unknown') {
+      console.log();
+      console.log(chalk.yellow('💡 建议:'));
+      console.log(chalk.yellow('   1. 检查配置是否完整'));
+      console.log(chalk.yellow('   2. 使用 "akm edit" 更新配置'));
+    }
+  }
+
+  /**
+   * 打印批量验证结果
+   * @private
+   * @param {Array} providers - 供应商列表
+   * @param {Object} results - 验证结果
+   */
+  _printBatchResults(providers, results) {
+    const currentProvider = this.configManager.getCurrentProvider();
+
+    let onlineCount = 0;
+    let offlineCount = 0;
+    let unknownCount = 0;
+
+    providers.forEach(provider => {
+      const status = results[provider.name];
+      const isCurrent = provider.name === currentProvider?.name;
+      const statusIcon = isCurrent ? '✅' : '🔹';
+      const stateIcon = this._getStateIcon(status.state);
+      const stateColor = this._getStateColor(status.state);
+      const nameColor = isCurrent ? chalk.green : chalk.white;
+
+      // IDE 类型标签
+      const ideTag = provider.ideName === 'codex'
+        ? chalk.cyan('[Codex]')
+        : chalk.magenta('[Claude]');
+
+      // 别名
+      const aliasText = provider.alias ? chalk.yellow(` [别名: ${provider.alias}]`) : '';
+
+      const statusText = status.message || this._getStateLabel(status.state);
+
+      console.log(`${statusIcon} ${stateIcon} ${ideTag} ${nameColor(provider.name)} (${provider.displayName})${aliasText}`);
+      console.log(`   ${stateColor(statusText)}`);
+
+      // 统计
+      if (status.state === 'online') onlineCount++;
+      else if (status.state === 'offline') offlineCount++;
+      else unknownCount++;
+    });
+
+    // 显示统计
+    console.log();
+    console.log(chalk.gray('═'.repeat(60)));
+    console.log(chalk.blue('📊 验证统计:'));
+    console.log(`   总计: ${providers.length} 个供应商`);
+    console.log(`   ${chalk.green('✓')} 可用: ${onlineCount}`);
+    console.log(`   ${chalk.red('✗')} 不可用: ${offlineCount}`);
+    console.log(`   ${chalk.yellow('?')} 未知: ${unknownCount}`);
+  }
+
+  /**
+   * 获取状态图标
+   * @private
+   * @param {string} state - 状态
+   * @returns {string} 图标
+   */
+  _getStateIcon(state) {
+    switch (state) {
+      case 'online': return '✓';
+      case 'offline': return '✗';
+      case 'degraded': return '⚠';
+      case 'pending': return '⋯';
+      default: return '?';
+    }
+  }
+
+  /**
+   * 获取状态标签
+   * @private
+   * @param {string} state - 状态
+   * @returns {string} 标签
+   */
+  _getStateLabel(state) {
+    switch (state) {
+      case 'online': return '可用';
+      case 'offline': return '不可用';
+      case 'degraded': return '降级';
+      case 'pending': return '检测中';
+      default: return '未知';
+    }
+  }
+
+  /**
+   * 获取状态颜色
+   * @private
+   * @param {string} state - 状态
+   * @returns {Function} 颜色函数
+   */
+  _getStateColor(state) {
+    switch (state) {
+      case 'online': return chalk.green;
+      case 'offline': return chalk.red;
+      case 'degraded': return chalk.yellow;
+      case 'pending': return chalk.blue;
+      default: return chalk.gray;
+    }
+  }
+}
+
+/**
+ * 验证命令
+ * @param {string|null} providerName - 供应商名称或别名
+ * @param {Object} options - 选项
+ * @param {string|null} options.filter - 过滤器
+ * @returns {Promise<void>}
+ */
+async function validateCommand(providerName, options = {}) {
+  const validator = new ProviderValidator();
+
+  try {
+    if (providerName) {
+      await validator.validateOne(providerName);
+    } else {
+      await validator.validateAll(options.filter || null);
+    }
+  } catch (error) {
+    Logger.error(`验证失败: ${error.message}`);
+    throw error;
+  }
+}
+
+module.exports = { validateCommand, ProviderValidator };

package/src/config.js

@@ -3,15 +3,54 @@ const path = require('path');
 const os = require('os');
 const chalk = require('chalk');
 
+/**
+ * @typedef {Object} ProviderConfig
+ * @property {string} name - 供应商名称
+ * @property {string} displayName - 显示名称
+ * @property {string} ideName - IDE 名称 ('claude' 或 'codex')
+ * @property {string} authMode - 认证模式
+ * @property {string} authToken - 认证令牌
+ * @property {string|null} baseUrl - API 基础 URL
+ * @property {string|null} tokenType - Token 类型
+ * @property {Object|null} models - 模型配置
+ * @property {string[]} launchArgs - 启动参数
+ * @property {boolean} current - 是否为当前供应商
+ * @property {number} usageCount - 使用次数
+ * @property {string} lastUsed - 最后使用时间
+ * @property {string} createdAt - 创建时间
+ */
+
+/**
+ * @typedef {Object} Config
+ * @property {string} version - 配置版本
+ * @property {string|null} currentProvider - 当前供应商名称
+ * @property {Object.<string, ProviderConfig>} providers - 供应商配置对象
+ */
+
+/**
+ * 配置管理器
+ * 管理 API 供应商配置的加载、保存和操作
+ */
 class ConfigManager {
   constructor() {
+    /** @type {string} 配置文件路径 */
     this.configPath = path.join(os.homedir(), '.akm-config.json');
-    this.config = null; // 延迟加载
+    /** @type {Config|null} 配置数据 */
+    this.config = null;
+    /** @type {boolean} 是否已加载 */
     this.isLoaded = false;
+    /** @type {Date|null} 最后修改时间 */
     this.lastModified = null;
-    this.loadPromise = null; // 防止并发加载
+    /** @type {Promise<Config>|null} 加载 Promise，防止并发加载 */
+    this.loadPromise = null;
   }
 
+  /**
+   * 标准化可选字符串值
+   * @private
+   * @param {*} value - 输入值
+   * @returns {string|null} 标准化后的值
+   */
   _normalizeOptionalString(value) {
     if (value === null || value === undefined) {
       return null;
@@ -23,6 +62,10 @@ class ConfigManager {
     return trimmed.length === 0 ? null : trimmed;
   }
 
+  /**
+   * 获取默认配置
+   * @returns {Config} 默认配置对象
+   */
   getDefaultConfig() {
     return {
       version: '1.0.0',
@@ -31,6 +74,11 @@ class ConfigManager {
     };
   }
 
+  /**
+   * 加载配置文件
+   * @param {boolean} [forceReload=false] - 是否强制重新加载
+   * @returns {Promise<Config>} 配置对象
+   */
   async load(forceReload = false) {
     // 如果正在加载，等待当前加载完成
     if (this.loadPromise) {
@@ -248,14 +296,21 @@ class ConfigManager {
       ? providerConfig.launchArgs
       : (existing?.launchArgs || []);
 
+    // 处理别名
+    const alias = providerConfig.alias !== undefined
+      ? this._normalizeOptionalString(providerConfig.alias)
+      : (existing?.alias || null);
+
     // 基础字段
     this.config.providers[name] = {
       name,
       displayName: providerConfig.displayName || existing?.displayName || name,
+      alias,
       ideName,
       baseUrl,
       authToken,
       launchArgs,
+      lastUsedArgs: existing?.lastUsedArgs || null,
       createdAt: existing?.createdAt || now,
       lastUsed: existing?.lastUsed || now,
       usageCount: existing?.usageCount || 0,
@@ -338,6 +393,168 @@ class ConfigManager {
     return await this.save();
   }
 
+  /**
+   * 更新供应商的上次使用启动参数
+   * @param {string} name - 供应商名称
+   * @param {string[]} args - 启动参数数组
+   * @returns {Promise<void>}
+   */
+  async updateLastUsedArgs(name, args) {
+    await this.ensureLoaded();
+
+    if (!this.config.providers[name]) {
+      throw new Error(`供应商 '${name}' 不存在`);
+    }
+
+    // 更新上次使用的启动参数
+    this.config.providers[name].lastUsedArgs = args;
+    this.config.providers[name].lastUsed = new Date().toISOString();
+
+    // 增加使用次数
+    this.config.providers[name].usageCount = (this.config.providers[name].usageCount || 0) + 1;
+
+    return await this.save();
+  }
+
+  /**
+   * 记录供应商使用会话
+   * @param {string} name - 供应商名称
+   * @param {number} durationMs - 使用时长（毫秒）
+   */
+  async recordUsageSession(name, durationMs = 0) {
+    await this.ensureLoaded();
+
+    if (!this.config.providers[name]) {
+      throw new Error(`供应商 '${name}' 不存在`);
+    }
+
+    const provider = this.config.providers[name];
+
+    // 初始化统计数据
+    if (!provider.stats) {
+      provider.stats = {
+        totalSessions: 0,
+        totalDurationMs: 0,
+        averageDurationMs: 0,
+        lastSessionDuration: 0,
+        firstUsed: new Date().toISOString()
+      };
+    }
+
+    // 更新统计
+    provider.stats.totalSessions = (provider.stats.totalSessions || 0) + 1;
+    provider.stats.totalDurationMs = (provider.stats.totalDurationMs || 0) + durationMs;
+    provider.stats.lastSessionDuration = durationMs;
+    provider.stats.averageDurationMs = Math.round(
+      provider.stats.totalDurationMs / provider.stats.totalSessions
+    );
+
+    // 更新最后使用时间
+    provider.lastUsed = new Date().toISOString();
+
+    return await this.save();
+  }
+
+  /**
+   * 获取使用统计信息
+   * @param {string|null} name - 供应商名称，null 表示获取所有
+   * @returns {Object} 统计信息
+   */
+  getUsageStats(name = null) {
+    if (!this.isLoaded) {
+      throw new Error('配置未加载，请先调用 load() 方法');
+    }
+
+    if (name) {
+      const provider = this.getProvider(name);
+      if (!provider) {
+        return null;
+      }
+      return {
+        name: provider.name,
+        displayName: provider.displayName,
+        usageCount: provider.usageCount || 0,
+        lastUsed: provider.lastUsed || null,
+        stats: provider.stats || null
+      };
+    }
+
+    // 返回所有供应商的统计
+    const allStats = Object.entries(this.config.providers).map(([name, provider]) => ({
+      name,
+      displayName: provider.displayName,
+      usageCount: provider.usageCount || 0,
+      lastUsed: provider.lastUsed || null,
+      stats: provider.stats || null
+    }));
+
+    // 按使用次数降序排序
+    return allStats.sort((a, b) => (b.usageCount || 0) - (a.usageCount || 0));
+  }
+
+  /**
+   * 获取智能推荐的供应商列表
+   * @param {Object} options - 选项
+   * @param {number} options.limit - 返回数量限制
+   * @param {string|null} options.filter - 过滤器 ('codex', 'claude', 或 null)
+   * @returns {Array} 推荐的供应商列表
+   */
+  getRecommendedProviders(options = {}) {
+    if (!this.isLoaded) {
+      throw new Error('配置未加载，请先调用 load() 方法');
+    }
+
+    const { limit = 5, filter = null } = options;
+    let providers = this.listProviders();
+
+    // 应用过滤器
+    if (filter === 'codex') {
+      providers = providers.filter(p => p.ideName === 'codex');
+    } else if (filter === 'claude') {
+      providers = providers.filter(p => p.ideName !== 'codex');
+    }
+
+    // 计算推荐分数
+    const scoredProviders = providers.map(provider => {
+      let score = 0;
+
+      // 使用次数权重 (40%)
+      const usageCount = provider.usageCount || 0;
+      score += usageCount * 0.4;
+
+      // 最近使用时间权重 (30%)
+      if (provider.lastUsed) {
+        const daysSinceLastUse = (Date.now() - new Date(provider.lastUsed).getTime()) / (1000 * 60 * 60 * 24);
+        // 越近使用分数越高，超过30天分数衰减
+        const recencyScore = Math.max(0, 30 - daysSinceLastUse) / 30;
+        score += recencyScore * 30;
+      }
+
+      // 会话平均时长权重 (20%)
+      if (provider.stats?.averageDurationMs) {
+        // 平均使用时长越长，说明使用越频繁，最高20分
+        const avgMinutes = provider.stats.averageDurationMs / (1000 * 60);
+        const durationScore = Math.min(avgMinutes / 60, 1) * 20; // 最多1小时算满分
+        score += durationScore;
+      }
+
+      // 总会话数权重 (10%)
+      if (provider.stats?.totalSessions) {
+        score += Math.min(provider.stats.totalSessions / 10, 1) * 10;
+      }
+
+      return {
+        ...provider,
+        recommendScore: Math.round(score * 100) / 100
+      };
+    });
+
+    // 按推荐分数降序排序并限制数量
+    return scoredProviders
+      .sort((a, b) => b.recommendScore - a.recommendScore)
+      .slice(0, limit);
+  }
+
   getProvider(name) {
     // 同步方法，但需要先确保配置已加载
     if (!this.isLoaded) {
@@ -346,6 +563,30 @@ class ConfigManager {
     return this.config.providers[name];
   }
 
+  /**
+   * 通过名称或别名获取供应商
+   * @param {string} nameOrAlias - 供应商名称或别名
+   * @returns {ProviderConfig|null} 供应商配置对象，未找到返回 null
+   */
+  getProviderByNameOrAlias(nameOrAlias) {
+    // 同步方法，但需要先确保配置已加载
+    if (!this.isLoaded) {
+      throw new Error('配置未加载，请先调用 load() 方法');
+    }
+
+    // 先尝试按名称查找
+    if (this.config.providers[nameOrAlias]) {
+      return this.config.providers[nameOrAlias];
+    }
+
+    // 再尝试按别名查找
+    const providerEntry = Object.entries(this.config.providers).find(
+      ([_, provider]) => provider.alias && provider.alias.toLowerCase() === nameOrAlias.toLowerCase()
+    );
+
+    return providerEntry ? providerEntry[1] : null;
+  }
+
   listProviders() {
     // 同步方法，但需要先确保配置已加载
     if (!this.isLoaded) {