@pikecode/api-key-manager 2.0.0

package/src/config.js

@@ -0,0 +1,250 @@
+const fs = require('fs-extra');
+const path = require('path');
+const os = require('os');
+const chalk = require('chalk');
+
+class ConfigManager {
+  constructor() {
+    this.configPath = path.join(os.homedir(), '.akm-config.json');
+    this.config = null; // 延迟加载
+    this.isLoaded = false;
+    this.lastModified = null;
+    this.loadPromise = null; // 防止并发加载
+  }
+
+  getDefaultConfig() {
+    return {
+      version: '1.0.0',
+      currentProvider: null,
+      providers: {}
+    };
+  }
+
+  async load(forceReload = false) {
+    // 如果正在加载，等待当前加载完成
+    if (this.loadPromise) {
+      return await this.loadPromise;
+    }
+
+    // 如果已经加载且不是强制重载，直接返回缓存
+    if (this.isLoaded && !forceReload) {
+      // 检查文件是否被外部修改
+      const needsReload = await this.checkIfModified();
+      if (!needsReload) {
+        return this.config;
+      }
+    }
+
+    // 创建加载Promise
+    this.loadPromise = this._performLoad();
+    try {
+      const result = await this.loadPromise;
+      this.loadPromise = null;
+      return result;
+    } catch (error) {
+      this.loadPromise = null;
+      throw error;
+    }
+  }
+
+  async _performLoad() {
+    try {
+      if (await fs.pathExists(this.configPath)) {
+        const data = await fs.readJSON(this.configPath);
+
+        if (!data || typeof data !== 'object' || Array.isArray(data)) {
+          // 配置文件被写成非对象内容时，重置为默认配置
+          this.config = this.getDefaultConfig();
+          await this._performSave();
+        } else {
+          this.config = { ...this.getDefaultConfig(), ...data };
+        }
+
+        // 迁移旧的认证模式
+        this._migrateAuthModes();
+
+        const stat = await fs.stat(this.configPath);
+        this.lastModified = stat.mtime;
+      } else {
+        this.config = this.getDefaultConfig();
+        await this._performSave();
+      }
+      this.isLoaded = true;
+      return this.config;
+    } catch (error) {
+      if (error.message.includes('Unexpected end of JSON input')) {
+        // 处理空文件或损坏的JSON文件
+        this.config = this.getDefaultConfig();
+        await this._performSave();
+        this.isLoaded = true;
+        return this.config;
+      }
+      console.error(chalk.red('❌ 加载配置失败:'), error.message);
+      throw error;
+    }
+  }
+
+  _migrateAuthModes() {
+    // 迁移旧的 api_token 模式到新的 auth_token 模式
+    if (this.config.providers) {
+      Object.keys(this.config.providers).forEach(key => {
+        const provider = this.config.providers[key];
+        if (provider.authMode === 'api_token') {
+          provider.authMode = 'auth_token';
+        }
+      });
+    }
+  }
+
+  async checkIfModified() {
+    try {
+      if (!this.lastModified || !await fs.pathExists(this.configPath)) {
+        return true;
+      }
+      const stat = await fs.stat(this.configPath);
+      return stat.mtime > this.lastModified;
+    } catch (error) {
+      return true; // 出错时重新加载
+    }
+  }
+
+  async save(config = this.config) {
+    // 确保配置已加载
+    await this.ensureLoaded();
+    if (config) {
+      this.config = config;
+    }
+    return await this._performSave();
+  }
+
+  async _performSave() {
+    try {
+      await fs.writeJSON(this.configPath, this.config, { spaces: 2 });
+      // 更新最后修改时间
+      const stat = await fs.stat(this.configPath);
+      this.lastModified = stat.mtime;
+      return true;
+    } catch (error) {
+      console.error(chalk.red('❌ 保存配置失败:'), error.message);
+      throw error;
+    }
+  }
+
+  // 确保配置已加载的辅助方法
+  async ensureLoaded() {
+    if (!this.isLoaded) {
+      await this.load();
+    }
+  }
+
+  async addProvider(name, providerConfig) {
+    await this.ensureLoaded();
+
+    this.config.providers[name] = {
+      name,
+      displayName: providerConfig.displayName || name,
+      baseUrl: providerConfig.baseUrl,
+      authToken: providerConfig.authToken,
+      authMode: providerConfig.authMode || 'api_key',
+      tokenType: providerConfig.tokenType || 'api_key', // 'api_key' 或 'auth_token' - 仅在 authMode 为 'api_key' 时使用
+      launchArgs: providerConfig.launchArgs || [],
+      models: {
+        primary: providerConfig.primaryModel || null,
+        smallFast: providerConfig.smallFastModel || null
+      },
+      createdAt: new Date().toISOString(),
+      lastUsed: new Date().toISOString(),
+      current: false
+    };
+
+    // 如果是第一个供应商或设置为默认，则设为当前供应商
+    if (Object.keys(this.config.providers).length === 1 || providerConfig.setAsDefault) {
+      // 重置所有供应商的current状态
+      Object.keys(this.config.providers).forEach(key => {
+        this.config.providers[key].current = false;
+      });
+      
+      // 设置新的当前供应商
+      this.config.providers[name].current = true;
+      this.config.providers[name].lastUsed = new Date().toISOString();
+      this.config.currentProvider = name;
+    }
+
+    return await this.save();
+  }
+
+  async removeProvider(name) {
+    await this.ensureLoaded();
+    
+    if (!this.config.providers[name]) {
+      throw new Error(`供应商 '${name}' 不存在`);
+    }
+
+    delete this.config.providers[name];
+
+    // 如果删除的是当前供应商，清空当前供应商
+    if (this.config.currentProvider === name) {
+      this.config.currentProvider = null;
+    }
+
+    return await this.save();
+  }
+
+  async setCurrentProvider(name) {
+    await this.ensureLoaded();
+    
+    if (!this.config.providers[name]) {
+      throw new Error(`供应商 '${name}' 不存在`);
+    }
+
+    // 重置所有供应商的current状态
+    Object.keys(this.config.providers).forEach(key => {
+      this.config.providers[key].current = false;
+    });
+
+    // 设置新的当前供应商
+    this.config.providers[name].current = true;
+    this.config.providers[name].lastUsed = new Date().toISOString();
+    this.config.currentProvider = name;
+
+    return await this.save();
+  }
+
+  getProvider(name) {
+    // 同步方法，但需要先确保配置已加载
+    if (!this.isLoaded) {
+      throw new Error('配置未加载，请先调用 load() 方法');
+    }
+    return this.config.providers[name];
+  }
+
+  listProviders() {
+    // 同步方法，但需要先确保配置已加载
+    if (!this.isLoaded) {
+      throw new Error('配置未加载，请先调用 load() 方法');
+    }
+    return Object.keys(this.config.providers).map(name => ({
+      name,
+      ...this.config.providers[name]
+    }));
+  }
+
+  getCurrentProvider() {
+    // 同步方法，但需要先确保配置已加载
+    if (!this.isLoaded) {
+      throw new Error('配置未加载，请先调用 load() 方法');
+    }
+    if (!this.config.currentProvider) {
+      return null;
+    }
+    return this.getProvider(this.config.currentProvider);
+  }
+
+  async reset() {
+    this.config = this.getDefaultConfig();
+    this.isLoaded = true;
+    return await this._performSave();
+  }
+}
+
+module.exports = { ConfigManager };

package/src/index.js

@@ -0,0 +1,19 @@
+require('./utils/inquirer-setup');
+const { switchCommand } = require('./commands/switch');
+const { Logger } = require('./utils/logger');
+
+async function main(providerName) {
+  try {
+    if (providerName) {
+      // 直接切换到指定供应商
+      await switchCommand(providerName);
+    } else {
+      // 显示供应商选择界面
+      await switchCommand();
+    }
+  } catch (error) {
+    Logger.fatal(`程序执行失败: ${error.message}`);
+  }
+}
+
+module.exports = { main };

package/src/navigation/EscNavigationManager.js

@@ -0,0 +1,213 @@
+const readline = require('readline');
+
+class EscNavigationManager {
+  constructor(input = process.stdin, options = {}) {
+    this.input = input;
+    this.handlers = [];
+    this.pendingTimeout = null;
+    this.escapePending = false;
+    this.triggerDelay = typeof options.triggerDelay === 'number' ? options.triggerDelay : 100;
+    this.listenerBound = false;
+    this.rawModeEnabled = false;
+    this.previousRawMode = null;
+    this.postCallbackDelay = typeof options.postCallbackDelay === 'number' ? options.postCallbackDelay : 50;
+
+    this.supported = Boolean(
+      this.input &&
+      typeof this.input.on === 'function' &&
+      typeof this.input.removeListener === 'function' &&
+      typeof this.input.setRawMode === 'function'
+    );
+
+    this.handleData = this.handleData.bind(this);
+  }
+
+  isSupported() {
+    return this.supported;
+  }
+
+  register(options = {}) {
+    if (!this.isSupported()) {
+      return null;
+    }
+
+    const handler = {
+      id: Symbol('esc-handler'),
+      onTrigger: typeof options.onTrigger === 'function' ? options.onTrigger : null,
+      once: options.once !== false,
+      postDelay: typeof options.postDelay === 'number' ? options.postDelay : this.postCallbackDelay
+    };
+
+    this.handlers.push(handler);
+    this.ensureListening();
+    return handler;
+  }
+
+  unregister(handler) {
+    if (!this.isSupported() || !handler) {
+      return;
+    }
+
+    const index = this.handlers.indexOf(handler);
+    if (index === -1) {
+      return;
+    }
+
+    this.handlers.splice(index, 1);
+    if (this.handlers.length === 0) {
+      this.teardown();
+    }
+  }
+
+  reset() {
+    if (!this.isSupported()) {
+      return;
+    }
+
+    this.handlers = [];
+    this.teardown();
+  }
+
+  handleData(chunk) {
+    if (!this.handlers.length) {
+      return;
+    }
+
+    const data = typeof chunk === 'string' ? chunk : chunk.toString('utf8');
+    if (!data) {
+      return;
+    }
+
+    for (const char of data) {
+      if (char === '\u001b') {
+        this.escapePending = true;
+        this.scheduleTrigger();
+      } else if (this.escapePending) {
+        // 遇到其它字符说明这是组合键 (例如方向键)
+        this.cancelPending();
+      }
+    }
+  }
+
+  scheduleTrigger() {
+    if (this.pendingTimeout) {
+      clearTimeout(this.pendingTimeout);
+    }
+
+    this.pendingTimeout = setTimeout(() => {
+      this.pendingTimeout = null;
+      const shouldTrigger = this.escapePending;
+      this.escapePending = false;
+      if (shouldTrigger) {
+        this.triggerTopHandler();
+      }
+    }, this.triggerDelay);
+  }
+
+  cancelPending() {
+    if (this.pendingTimeout) {
+      clearTimeout(this.pendingTimeout);
+      this.pendingTimeout = null;
+    }
+    this.escapePending = false;
+  }
+
+  triggerTopHandler() {
+    if (!this.handlers.length) {
+      return;
+    }
+
+    const handler = this.handlers[this.handlers.length - 1];
+    if (!handler) {
+      return;
+    }
+
+    if (handler.once) {
+      this.handlers.pop();
+    }
+
+    if (!this.handlers.length) {
+      this.teardown();
+    }
+
+    if (handler.onTrigger) {
+      setTimeout(() => {
+        try {
+          handler.onTrigger();
+        } catch (error) {
+          // 在ESC回调中抛出的错误不应终止进程
+          // 交由调用方自行处理记录
+        }
+      }, handler.postDelay);
+    }
+  }
+
+  ensureListening() {
+    if (this.listenerBound || !this.isSupported()) {
+      return;
+    }
+
+    readline.emitKeypressEvents(this.input);
+
+    if (typeof this.input.setMaxListeners === 'function') {
+      try {
+        const currentMax = this.input.getMaxListeners();
+        if (currentMax !== 0 && currentMax < 50) {
+          this.input.setMaxListeners(50);
+        }
+      } catch (error) {
+        // 某些输入流可能不支持获取/设置监听器上限
+      }
+    }
+
+    if (this.input.isTTY && !this.rawModeEnabled) {
+      try {
+        this.previousRawMode = typeof this.input.isRaw === 'boolean' ? this.input.isRaw : null;
+        this.input.setRawMode(true);
+        this.rawModeEnabled = true;
+      } catch (error) {
+        this.supported = false;
+        this.handlers = [];
+        return;
+      }
+    }
+
+    if (typeof this.input.resume === 'function') {
+      try {
+        this.input.resume();
+      } catch (error) {
+        // 某些输入流不支持 resume，忽略即可
+      }
+    }
+
+    this.input.on('data', this.handleData);
+    this.listenerBound = true;
+  }
+
+  teardown() {
+    if (!this.isSupported()) {
+      this.cancelPending();
+      return;
+    }
+
+    if (this.listenerBound) {
+      this.input.removeListener('data', this.handleData);
+      this.listenerBound = false;
+    }
+
+    this.cancelPending();
+
+    if (this.rawModeEnabled && this.input.isTTY) {
+      try {
+        const restoreRaw = typeof this.previousRawMode === 'boolean' ? this.previousRawMode : false;
+        this.input.setRawMode(restoreRaw);
+      } catch (error) {
+        // 忽略还原失败
+      }
+
+      this.rawModeEnabled = false;
+    }
+  }
+}
+
+module.exports = { EscNavigationManager };

package/src/utils/claude-settings.js

@@ -0,0 +1,150 @@
+const fs = require('fs-extra');
+const path = require('path');
+const os = require('os');
+
+const CONFLICT_ENV_KEYS = [
+  'ANTHROPIC_API_KEY',
+  'ANTHROPIC_AUTH_TOKEN',
+  'ANTHROPIC_BASE_URL',
+  'CLAUDE_CODE_OAUTH_TOKEN',
+  'ANTHROPIC_MODEL',
+  'ANTHROPIC_SMALL_FAST_MODEL'
+];
+
+const SETTINGS_FILE_NAMES = ['settings.local.json', 'settings.json'];
+
+function unique(array) {
+  return Array.from(new Set(array.filter(Boolean)));
+}
+
+function timestampSuffix() {
+  const now = new Date();
+  const pad = (num) => String(num).padStart(2, '0');
+  return [
+    now.getFullYear(),
+    pad(now.getMonth() + 1),
+    pad(now.getDate()),
+    '_',
+    pad(now.getHours()),
+    pad(now.getMinutes()),
+    pad(now.getSeconds())
+  ].join('');
+}
+
+function resolveCandidatePaths() {
+  const candidates = [];
+
+  if (process.env.CLAUDE_SETTINGS_PATH) {
+    candidates.push(process.env.CLAUDE_SETTINGS_PATH);
+  }
+
+  const cwd = process.cwd();
+  if (cwd) {
+    for (const fileName of SETTINGS_FILE_NAMES) {
+      candidates.push(path.join(cwd, '.claude', fileName));
+    }
+  }
+
+  const homeDir = os.homedir();
+  if (homeDir) {
+    for (const fileName of SETTINGS_FILE_NAMES) {
+      candidates.push(path.join(homeDir, '.claude', fileName));
+    }
+  }
+
+  if (process.platform === 'win32') {
+    const baseDirs = unique([process.env.APPDATA, process.env.LOCALAPPDATA]);
+    for (const baseDir of baseDirs) {
+      if (!baseDir) {
+        continue;
+      }
+      for (const fileName of SETTINGS_FILE_NAMES) {
+        candidates.push(path.join(baseDir, 'claude', fileName));
+      }
+    }
+  }
+
+  return unique(candidates);
+}
+
+async function loadSettingsFile() {
+  const candidates = resolveCandidatePaths();
+  for (const candidate of candidates) {
+    try {
+      if (await fs.pathExists(candidate)) {
+        const data = await fs.readJson(candidate);
+        return { path: candidate, data };
+      }
+    } catch (error) {
+      throw new Error(`读取 Claude 设置文件失败 (${candidate}): ${error.message}`);
+    }
+  }
+  return null;
+}
+
+function detectConflictKeys(settings) {
+  if (!settings || typeof settings !== 'object') {
+    return [];
+  }
+  const env = settings.env;
+  if (!env || typeof env !== 'object') {
+    return [];
+  }
+
+  return CONFLICT_ENV_KEYS.filter((key) => Object.prototype.hasOwnProperty.call(env, key));
+}
+
+async function backupSettingsFile(filePath) {
+  const dir = path.dirname(filePath);
+  const backupName = `settings.backup-${timestampSuffix()}.json`;
+  const backupPath = path.join(dir, backupName);
+  await fs.copy(filePath, backupPath, { overwrite: false, errorOnExist: false });
+  return backupPath;
+}
+
+function clearConflictKeys(settings, keys) {
+  if (!settings || !settings.env) {
+    return settings;
+  }
+
+  for (const key of keys) {
+    if (Object.prototype.hasOwnProperty.call(settings.env, key)) {
+      delete settings.env[key];
+    }
+  }
+
+  if (Object.keys(settings.env).length === 0) {
+    delete settings.env;
+  }
+
+  return settings;
+}
+
+async function saveSettingsFile(filePath, data) {
+  await fs.writeJson(filePath, data, { spaces: 2 });
+}
+
+async function findSettingsConflict() {
+  const fileInfo = await loadSettingsFile();
+  if (!fileInfo) {
+    return null;
+  }
+
+  const conflictKeys = detectConflictKeys(fileInfo.data);
+  if (!conflictKeys.length) {
+    return null;
+  }
+
+  return {
+    filePath: fileInfo.path,
+    settings: fileInfo.data,
+    keys: conflictKeys
+  };
+}
+
+module.exports = {
+  findSettingsConflict,
+  backupSettingsFile,
+  clearConflictKeys,
+  saveSettingsFile
+};

package/src/utils/config-opener.js

@@ -0,0 +1,44 @@
+const fs = require('fs-extra');
+const path = require('path');
+const os = require('os');
+const { spawn } = require('child_process');
+
+const AKM_CONFIG_FILE = path.join(os.homedir(), '.akm-config.json');
+
+function ensureConfigExists() {
+  return fs.pathExists(AKM_CONFIG_FILE);
+}
+
+function openFileWithDefaultApp(filePath) {
+  return new Promise((resolve, reject) => {
+    let command;
+    let args = [];
+
+    if (process.platform === 'win32') {
+      command = 'cmd';
+      args = ['/c', 'start', '', filePath];
+    } else if (process.platform === 'darwin') {
+      command = 'open';
+      args = [filePath];
+    } else {
+      command = 'xdg-open';
+      args = [filePath];
+    }
+
+    const child = spawn(command, args, { stdio: 'ignore', detached: true });
+    child.on('error', reject);
+    child.unref();
+    resolve();
+  });
+}
+
+async function openAKMConfigFile() {
+  if (!(await ensureConfigExists())) {
+    throw new Error('未找到 ~/.akm-config.json，请先运行 akm add 创建配置');
+  }
+  await openFileWithDefaultApp(AKM_CONFIG_FILE);
+}
+
+module.exports = {
+  openAKMConfigFile
+};