@pikecode/api-key-manager 2.0.0

package/src/commands/edit.js

@@ -0,0 +1,208 @@
+const inquirer = require('inquirer');
+const chalk = require('chalk');
+const { ConfigManager } = require('../config');
+const { validator } = require('../utils/validator');
+const { Logger } = require('../utils/logger');
+const { UIHelper } = require('../utils/ui-helper');
+const { BaseCommand } = require('./BaseCommand');
+
+class ProviderEditor extends BaseCommand {
+  constructor() {
+    super();
+    this.configManager = new ConfigManager();
+  }
+
+  async interactive(providerName) {
+    await this.configManager.load();
+    const providers = this.configManager.listProviders();
+
+    if (providers.length === 0) {
+      Logger.warning('没有可编辑的供应商配置。请先添加一个。');
+      return;
+    }
+
+    let providerToEdit;
+    if (providerName) {
+        providerToEdit = this.configManager.getProvider(providerName);
+        if (!providerToEdit) {
+            Logger.error(`供应商 '${providerName}' 不存在。`);
+            return;
+        }
+    } else {
+        let selection;
+        try {
+          selection = await this.prompt([
+              {
+                  type: 'list',
+                  name: 'selectedProviderName',
+                  message: '请选择要编辑的供应商:',
+                  choices: [
+                      ...providers.map(p => ({ name: p.displayName || p.name, value: p.name })),
+                      new inquirer.Separator(),
+                      { name: '取消', value: null },
+                  ],
+              },
+          ]);
+        } catch (error) {
+          if (this.isEscCancelled(error)) {
+            return;
+          }
+          throw error;
+        }
+
+        const { selectedProviderName } = selection;
+
+        if (!selectedProviderName) {
+            Logger.info('操作已取消。');
+            return;
+        }
+        providerToEdit = this.configManager.getProvider(selectedProviderName);
+    }
+
+    console.log(UIHelper.createTitle(`编辑供应商: ${providerToEdit.displayName}`, UIHelper.icons.edit));
+    console.log();
+    console.log(UIHelper.createTooltip('请更新供应商配置信息。按 Enter 键接受默认值。'));
+    console.log();
+
+    const escListener = this.createESCListener(() => {
+      Logger.info('取消编辑供应商。');
+      const { registry } = require('../CommandRegistry');
+      registry.executeCommand('switch');
+    }, '取消编辑');
+
+    try {
+      let answers;
+      try {
+        answers = await this.prompt([
+          {
+            type: 'input',
+            name: 'displayName',
+            message: '供应商显示名称:',
+            default: providerToEdit.displayName,
+            validate: (input) => validator.validateDisplayName(input) || true,
+          },
+          {
+            type: 'list',
+            name: 'authMode',
+            message: '认证模式:',
+            choices: [
+              { name: '🔑 通用API密钥模式 - 支持 ANTHROPIC_API_KEY 和 ANTHROPIC_AUTH_TOKEN', value: 'api_key' },
+              { name: '🔐 认证令牌模式 (仅 ANTHROPIC_AUTH_TOKEN) - 适用于某些服务商', value: 'auth_token' },
+              { name: '🌐 OAuth令牌模式 (CLAUDE_CODE_OAUTH_TOKEN) - 适用于官方Claude Code', value: 'oauth_token' },
+            ],
+            default: providerToEdit.authMode,
+          },
+          {
+            type: 'list',
+            name: 'tokenType',
+            message: 'Token类型:',
+            choices: [
+              { name: '🔑 ANTHROPIC_API_KEY - 通用API密钥', value: 'api_key' },
+              { name: '🔐 ANTHROPIC_AUTH_TOKEN - 认证令牌', value: 'auth_token' }
+            ],
+            default: providerToEdit.tokenType || 'api_key',
+            when: (answers) => answers.authMode === 'api_key'
+          },
+          {
+            type: 'input',
+            name: 'baseUrl',
+            message: 'API基础URL:',
+            default: providerToEdit.baseUrl,
+            validate: (input) => validator.validateUrl(input) || true,
+            when: (answers) => answers.authMode === 'api_key' || answers.authMode === 'auth_token',
+          },
+          {
+            type: 'input',
+            name: 'authToken',
+            message: (answers) => {
+              switch (answers.authMode) {
+                case 'api_key':
+                  const tokenTypeLabel = answers.tokenType === 'auth_token' ? 'ANTHROPIC_AUTH_TOKEN' : 'ANTHROPIC_API_KEY';
+                  return `Token (${tokenTypeLabel}):`;
+                case 'auth_token': return '认证令牌 (ANTHROPIC_AUTH_TOKEN):';
+                case 'oauth_token': return 'OAuth令牌 (CLAUDE_CODE_OAUTH_TOKEN):';
+                default: return '认证令牌:';
+              }
+            },
+            default: providerToEdit.authToken,
+            validate: (input) => validator.validateToken(input) || true,
+          },
+          {
+              type: 'checkbox',
+              name: 'launchArgs',
+              message: '启动参数:',
+              choices: validator.getAvailableLaunchArgs().map(arg => ({
+                name: `${arg.label || arg.name} (${arg.name})${arg.description ? ' - ' + arg.description : ''}`,
+                value: arg.name,
+                checked: providerToEdit.launchArgs && providerToEdit.launchArgs.includes(arg.name),
+              })),
+          },
+        ]);
+      } catch (error) {
+        this.removeESCListener(escListener);
+        if (this.isEscCancelled(error)) {
+          return;
+        }
+        throw error;
+      }
+
+      this.removeESCListener(escListener);
+      await this.saveProvider(providerToEdit.name, answers);
+
+    } catch (error) {
+      this.removeESCListener(escListener);
+      if (this.isEscCancelled(error)) {
+        return;
+      }
+      throw error;
+    }
+  }
+
+  async saveProvider(name, answers) {
+    try {
+      // Re-use addProvider which can overwrite existing providers
+      await this.configManager.addProvider(name, {
+        displayName: answers.displayName,
+        baseUrl: answers.baseUrl,
+        authToken: answers.authToken,
+        authMode: answers.authMode,
+        tokenType: answers.tokenType, // 仅在 authMode 为 'api_key' 时使用
+        launchArgs: answers.launchArgs,
+        // Retain original model settings unless we add editing for them
+        primaryModel: this.configManager.getProvider(name).models.primary,
+        smallFastModel: this.configManager.getProvider(name).models.smallFast,
+        setAsDefault: false, // Don't change default status on edit
+      });
+
+      Logger.success(`供应商 '${answers.displayName}' 更新成功！`);
+
+      console.log(chalk.green('\n🎉 供应商编辑完成！正在返回主界面...'));
+      await new Promise(resolve => setTimeout(resolve, 1500));
+
+      const { registry } = require('../CommandRegistry');
+      return await registry.executeCommand('switch');
+
+    } catch (error) {
+      if (this.isEscCancelled(error)) {
+        return;
+      }
+      Logger.error(`更新供应商失败: ${error.message}`);
+      throw error;
+    }
+  }
+}
+
+async function editCommand(providerName) {
+  const editor = new ProviderEditor();
+  try {
+    await editor.interactive(providerName);
+  } catch (error) {
+    if (!editor.isEscCancelled(error)) {
+      Logger.error(`编辑供应商失败: ${error.message}`);
+    }
+  } finally {
+    editor.destroy();
+  }
+}
+
+module.exports = { editCommand, ProviderEditor };

package/src/commands/list.js

@@ -0,0 +1,122 @@
+const chalk = require('chalk');
+const { ConfigManager } = require('../config');
+const { Logger } = require('../utils/logger');
+const { ProviderStatusChecker } = require('../utils/provider-status-checker');
+
+class ProviderLister {
+  constructor() {
+    this.configManager = new ConfigManager();
+    this.statusChecker = new ProviderStatusChecker();
+  }
+
+  async list() {
+    try {
+      await this.configManager.ensureLoaded();
+      const providers = this.configManager.listProviders();
+      const currentProvider = this.configManager.getCurrentProvider();
+      const statusMap = await this.statusChecker.checkAll(providers);
+
+      if (providers.length === 0) {
+        Logger.warning('暂无配置的供应商');
+        Logger.info('请使用 "akm add" 添加供应商配置');
+        return;
+      }
+
+      console.log(chalk.blue('\n📋 供应商列表:'));
+      console.log(chalk.gray('═'.repeat(60)));
+
+      providers.forEach((provider, index) => {
+        const isCurrent = provider.name === currentProvider?.name;
+        const status = isCurrent ? '✅' : '🔹';
+        const availability = statusMap[provider.name] || { state: 'unknown', label: '未知', latency: null };
+        const availabilityIcon = this._iconForState(availability.state);
+        const availabilityText = this._formatAvailability(availability);
+        const nameColor = isCurrent ? chalk.green : chalk.white;
+        
+        console.log(`${status} ${availabilityIcon} ${nameColor(provider.name)} (${provider.displayName}) - ${availabilityText}`);
+
+        // 显示认证模式
+        const authModeDisplay = {
+          api_key: '通用API密钥模式',
+          auth_token: '认证令牌模式',
+          oauth_token: 'OAuth令牌模式'
+        };
+        console.log(chalk.gray(`   认证模式: ${authModeDisplay[provider.authMode] || provider.authMode}`));
+
+        // 如果是 api_key 模式，显示 tokenType
+        if (provider.authMode === 'api_key' && provider.tokenType) {
+          const tokenTypeDisplay = provider.tokenType === 'auth_token' ? 'ANTHROPIC_AUTH_TOKEN' : 'ANTHROPIC_API_KEY';
+          console.log(chalk.gray(`   Token类型: ${tokenTypeDisplay}`));
+        }
+
+        if (provider.baseUrl) {
+          console.log(chalk.gray(`   URL: ${provider.baseUrl}`));
+        }
+        console.log(chalk.gray(`   Token: ${provider.authToken}`));
+        if (provider.launchArgs && provider.launchArgs.length > 0) {
+          console.log(chalk.gray(`   启动参数: ${provider.launchArgs.join(' ')}`));
+        }
+        if (provider.models && (provider.models.primary || provider.models.smallFast)) {
+          console.log(chalk.gray(`   主模型: ${provider.models.primary || '未设置'}`));
+          console.log(chalk.gray(`   快速模型: ${provider.models.smallFast || '未设置'}`));
+        }
+        console.log(chalk.gray(`   创建时间: ${new Date(provider.createdAt).toLocaleString()}`));
+        console.log(chalk.gray(`   最后使用: ${new Date(provider.lastUsed).toLocaleString()}`));
+        
+        if (index < providers.length - 1) {
+          console.log(chalk.gray('─'.repeat(60)));
+        }
+      });
+
+      console.log(chalk.gray('═'.repeat(60)));
+      
+      if (currentProvider) {
+        console.log(chalk.green(`\n当前供应商: ${currentProvider.displayName}`));
+      } else {
+        console.log(chalk.yellow('\n未设置当前供应商'));
+      }
+
+      console.log(chalk.blue(`\n总计: ${providers.length} 个供应商`));
+      
+    } catch (error) {
+      Logger.error(`获取供应商列表失败: ${error.message}`);
+      throw error;
+    }
+  }
+
+  _iconForState(state) {
+    if (state === 'online') {
+      return '🟢';
+    }
+    if (state === 'degraded') {
+      return '🟡';
+    }
+    if (state === 'offline') {
+      return '🔴';
+    }
+    return '⚪';
+  }
+
+  _formatAvailability(availability) {
+    if (!availability) {
+      return '未知';
+    }
+    if (availability.state === 'online') {
+      return chalk.green(availability.label);
+    }
+    if (availability.state === 'degraded') {
+      return chalk.yellow(availability.label);
+    }
+    if (availability.state === 'offline') {
+      return chalk.red(availability.label);
+    }
+    return chalk.gray(availability.label || '未知');
+  }
+}
+
+async function listCommand() {
+  const lister = new ProviderLister();
+  await lister.list();
+}
+
+module.exports = { listCommand, ProviderLister };

package/src/commands/remove.js

@@ -0,0 +1,150 @@
+const inquirer = require('inquirer');
+const { ConfigManager } = require('../config');
+const { Logger } = require('../utils/logger');
+const { UIHelper } = require('../utils/ui-helper');
+const { BaseCommand } = require('./BaseCommand');
+
+class ProviderRemover extends BaseCommand {
+  constructor() {
+    super();
+    this.configManager = new ConfigManager();
+  }
+
+  async remove(providerName) {
+    try {
+      await this.configManager.ensureLoaded();
+      
+      // 如果没有指定供应商名称，显示选择列表
+      if (!providerName) {
+        return await this.interactiveRemove();
+      }
+
+      // 直接删除指定供应商
+      if (!this.configManager.getProvider(providerName)) {
+        Logger.error(`供应商 '${providerName}' 不存在`);
+        return;
+      }
+
+      const provider = this.configManager.getProvider(providerName);
+      let confirm;
+      try {
+        confirm = await this.prompt([
+          {
+            type: 'confirm',
+            name: 'confirm',
+            message: `确定要删除供应商 '${provider.displayName}' 吗?`,
+            default: false
+          }
+        ]);
+      } catch (error) {
+        if (this.isEscCancelled(error)) {
+          return;
+        }
+        throw error;
+      }
+
+      if (!confirm.confirm) {
+        Logger.warning('删除操作已取消');
+        return;
+      }
+
+      await this.configManager.removeProvider(providerName);
+      Logger.success(`供应商 '${provider.displayName}' 已删除`);
+      
+    } catch (error) {
+      if (this.isEscCancelled(error)) {
+        return;
+      }
+      Logger.error(`删除供应商失败: ${error.message}`);
+      throw error;
+    }
+  }
+
+  async interactiveRemove() {
+    await this.configManager.ensureLoaded();
+    const providers = this.configManager.listProviders();
+    
+    if (providers.length === 0) {
+      Logger.warning('暂无配置的供应商');
+      return;
+    }
+
+    console.log(UIHelper.createTitle('删除供应商', UIHelper.icons.delete));
+    console.log();
+    console.log(UIHelper.createTooltip('选择要删除的供应商'));
+    console.log();
+
+    const choices = providers.map(provider => ({
+      name: `${provider.current ? '✅' : '🔹'} ${provider.name} (${provider.displayName})${provider.current ? ' - 当前使用中' : ''}`,
+      value: provider.name,
+      short: provider.name
+    }));
+
+    choices.push(
+      new inquirer.Separator(),
+      { name: '❌ 取消删除', value: '__CANCEL__' }
+    );
+
+    // 对于删除操作，不默认选中当前供应商，而是选中第一个非当前的供应商
+    const nonCurrentProvider = providers.find(p => !p.current);
+    const defaultChoice = nonCurrentProvider ? nonCurrentProvider.name : providers[0]?.name;
+
+    // 设置 ESC 键监听
+    const escListener = this.createESCListener(() => {
+      Logger.info('取消删除供应商');
+      // 使用CommandRegistry避免循环引用
+      const { registry } = require('../CommandRegistry');
+      registry.executeCommand('switch');
+    }, '取消删除');
+
+    try {
+      let answer;
+      try {
+        answer = await this.prompt([
+          {
+            type: 'list',
+            name: 'provider',
+            message: '选择要删除的供应商:',
+            choices,
+            default: defaultChoice,
+            pageSize: 10
+          }
+        ]);
+      } catch (error) {
+        this.removeESCListener(escListener);
+        if (this.isEscCancelled(error)) {
+          return;
+        }
+        throw error;
+      }
+
+      this.removeESCListener(escListener);
+
+      if (answer.provider === '__CANCEL__') {
+        Logger.info('删除操作已取消');
+        return;
+      }
+
+      await this.remove(answer.provider);
+    } catch (error) {
+      this.removeESCListener(escListener);
+      throw error;
+    }
+  }
+}
+
+async function removeCommand(providerName) {
+  const remover = new ProviderRemover();
+  try {
+    await remover.remove(providerName);
+  } catch (error) {
+    if (!remover.isEscCancelled(error)) {
+      Logger.error(`删除供应商失败: ${error.message}`);
+    }
+  } finally {
+    // 确保资源清理
+    remover.destroy();
+  }
+}
+
+module.exports = { removeCommand, ProviderRemover };