ai-account-switch 1.9.0 → 1.11.0

package/src/commands/helpers.js

@@ -9,6 +9,37 @@ function maskApiKey(apiKey) {
   return `${apiKey.substring(0, 4)}****${apiKey.substring(apiKey.length - 4)}`;
 }
 
+/**
+ * Mask sensitive value for display
+ * Shows first 2 chars + fixed 6 stars + last 2 chars for sensitive variables
+ * 用于显示敏感环境变量值，前2字符+固定6星号+后2字符
+ * @param {string} key - Variable name to check if sensitive
+ * @param {string} value - Value to mask
+ * @returns {string} Masked value or original value if not sensitive
+ */
+function maskEnvValue(key, value) {
+  if (!key || !value) return value;
+
+  // Check if variable name contains sensitive keywords
+  const isSensitive = key.includes('KEY') || key.includes('TOKEN') || key.includes('SECRET') || key.includes('PASSWORD');
+
+  if (!isSensitive) {
+    return value;
+  }
+
+  // For sensitive values, show first 2 + fixed 6 stars + last 2
+  const strValue = String(value);
+  if (strValue.length <= 4) {
+    // If value is too short, show all stars
+    return '*'.repeat(strValue.length);
+  }
+
+  const firstTwo = strValue.substring(0, 2);
+  const lastTwo = strValue.substring(strValue.length - 2);
+
+  return firstTwo + '******' + lastTwo;
+}
+
 /**
  * Validate account by testing API key
  * 验证账号的 API key 是否有效
@@ -31,5 +62,6 @@ async function validateAccount(apiKey, apiUrl) {
 
 module.exports = {
   maskApiKey,
+  maskEnvValue,
   validateAccount
 };

package/src/commands/index.js

@@ -42,6 +42,17 @@ const {
   testMcpServer
 } = require('./mcp');
 
+const {
+  listEnv,
+  addEnv,
+  setEnv,
+  removeEnv,
+  unsetEnv,
+  showEnv,
+  clearEnv,
+  editEnv
+} = require('./env');
+
 module.exports = {
   // Account management commands
   addAccount,
@@ -74,5 +85,15 @@ module.exports = {
   disableMcpServer,
   showEnabledMcpServers,
   syncMcpConfig,
-  testMcpServer
+  testMcpServer,
+
+  // Environment variable management commands
+  listEnv,
+  addEnv,
+  setEnv,
+  removeEnv,
+  unsetEnv,
+  showEnv,
+  clearEnv,
+  editEnv
 };

package/src/commands/mcp.js

@@ -10,6 +10,8 @@ const config = new ConfigManager();
  */
 async function addMcpServer(name) {
   try {
+    const { MCP_SCOPES, DEFAULT_MCP_SCOPE } = require('../config');
+
     if (!name) {
       const { serverName } = await inquirer.prompt([{
         type: 'input',
@@ -79,8 +81,22 @@ async function addMcpServer(name) {
     const { description } = await inquirer.prompt([{ type: 'input', name: 'description', message: 'Enter description:', default: '' }]);
     serverData.description = description;
 
+    // Ask for default scope
+    const { scope } = await inquirer.prompt([{
+      type: 'list',
+      name: 'scope',
+      message: 'Select default scope (默认作用范围):',
+      choices: [
+        { name: 'local - Only current project (仅当前项目)', value: MCP_SCOPES.LOCAL },
+        { name: 'project - Share with project members (与项目成员共享)', value: MCP_SCOPES.PROJECT },
+        { name: 'user - All projects for current user (当前用户所有项目)', value: MCP_SCOPES.USER }
+      ],
+      default: DEFAULT_MCP_SCOPE
+    }]);
+    serverData.scope = scope;
+
     config.addMcpServer(name, serverData);
-    console.log(chalk.green(`✓ MCP server '${name}' added successfully!`));
+    console.log(chalk.green(`✓ MCP server '${name}' added successfully with scope: ${scope}!`));
 
     // Auto-test the server
     console.log(chalk.cyan('\nTesting server availability...'));
@@ -101,7 +117,7 @@ async function addMcpServer(name) {
  */
 async function listMcpServers() {
   try {
-    const servers = config.getAllMcpServers();
+    const servers = config.getAllAvailableMcpServers();
     const projectServers = config.getEnabledMcpServers();
 
     if (Object.keys(servers).length === 0) {
@@ -110,12 +126,16 @@ async function listMcpServers() {
     }
 
     console.log(chalk.bold.cyan('\n📋 Available MCP servers:\n'));
-    console.log('  Name         Type    Active   Description');
-    console.log('  ─────────────────────────────────────────────────────────────');
-
-    Object.values(servers).forEach(server => {
-      const isActive = projectServers.includes(server.name) ? chalk.green('✓') : ' ';
-      console.log(`  ${server.name.padEnd(12)} ${server.type.padEnd(7)} ${isActive}        ${server.description || ''}`);
+    console.log('  Name         Type    Active   Scope      Description');
+    console.log('  ───────────────────────────────────────────────────────────────────────');
+
+    Object.entries(servers).forEach(([key, server]) => {
+      const name = server.name || key;
+      const type = server.type || 'unknown';
+      const isActive = projectServers.includes(name) ? chalk.green('✓') : ' ';
+      const scope = server.scope || 'local';
+      const scopeDisplay = scope.padEnd(10);
+      console.log(`  ${name.padEnd(12)} ${type.padEnd(7)} ${isActive}        ${scopeDisplay} ${server.description || ''}`);
     });
 
     const activeCount = projectServers.length;
@@ -133,7 +153,7 @@ async function listMcpServers() {
 async function showMcpServer(name) {
   try {
     if (!name) {
-      const servers = config.getAllMcpServers();
+      const servers = config.getAllAvailableMcpServers();
       if (Object.keys(servers).length === 0) {
         console.log(chalk.yellow('No MCP servers configured'));
         return;
@@ -155,6 +175,7 @@ async function showMcpServer(name) {
 
     console.log(chalk.bold.cyan(`\n📋 MCP Server: ${name}\n`));
     console.log(chalk.bold('Type:'), server.type);
+    console.log(chalk.bold('Scope:'), server.scope || 'local');
     console.log(chalk.bold('Description:'), server.description || 'N/A');
 
     if (server.command) console.log(chalk.bold('Command:'), server.command);
@@ -322,8 +343,17 @@ async function removeMcpServer(name) {
 /**
  * Enable MCP server for current project
  */
-async function enableMcpServer(name) {
+async function enableMcpServer(name, options = {}) {
   try {
+    const { MCP_SCOPES, DEFAULT_MCP_SCOPE } = require('../config');
+    let scope = options.scope || DEFAULT_MCP_SCOPE;
+
+    // Validate scope
+    if (!Object.values(MCP_SCOPES).includes(scope)) {
+      console.log(chalk.red(`✗ Invalid scope '${scope}'. Valid scopes: local, project, user`));
+      return;
+    }
+
     if (!name) {
       const servers = config.getAllMcpServers();
       const enabled = config.getEnabledMcpServers();
@@ -341,12 +371,38 @@ async function enableMcpServer(name) {
         choices: available
       }]);
       name = serverName;
+
+      // Ask for scope if not provided
+      if (!options.scope) {
+        const { selectedScope } = await inquirer.prompt([{
+          type: 'list',
+          name: 'selectedScope',
+          message: 'Select scope (作用范围):',
+          choices: [
+            { name: 'local - Only current project (仅当前项目)', value: MCP_SCOPES.LOCAL },
+            { name: 'project - Share with project members via .mcp.json (通过 .mcp.json 与项目成员共享)', value: MCP_SCOPES.PROJECT },
+            { name: 'user - All projects for current user (当前用户所有项目)', value: MCP_SCOPES.USER }
+          ],
+          default: DEFAULT_MCP_SCOPE
+        }]);
+        scope = selectedScope;
+      }
     }
 
-    if (config.enableProjectMcpServer(name)) {
+    if (config.enableProjectMcpServer(name, scope)) {
       config.syncMcpConfig();
-      console.log(chalk.green(`✓ MCP server '${name}' activated for current project`));
+      console.log(chalk.green(`✓ MCP server '${name}' activated for current project with scope: ${scope}`));
       console.log(chalk.green('✓ Claude configuration updated'));
+
+      // Show scope-specific information
+      if (scope === MCP_SCOPES.LOCAL) {
+        console.log(chalk.cyan('  Scope: local - Only available in this project'));
+      } else if (scope === MCP_SCOPES.PROJECT) {
+        console.log(chalk.cyan('  Scope: project - Configuration stored in project, shared with team members'));
+        console.log(chalk.gray('  Note: Make sure to commit .ais-project-config to share with your team'));
+      } else if (scope === MCP_SCOPES.USER) {
+        console.log(chalk.cyan('  Scope: user - Available to all your projects'));
+      }
     } else {
       console.log(chalk.red(`✗ MCP server '${name}' not found`));
     }
@@ -408,7 +464,9 @@ async function showEnabledMcpServers() {
     enabled.forEach(name => {
       const server = servers[name];
       if (server) {
-        console.log(`  ${server.name.padEnd(12)} ${server.type.padEnd(7)} ${server.description || ''}`);
+        const serverName = server.name || name;
+        const type = server.type || 'unknown';
+        console.log(`  ${String(serverName).padEnd(12)} ${String(type).padEnd(7)} ${server.description || ''}`);
       }
     });
 

package/src/config.js

@@ -5,7 +5,8 @@ const os = require('os');
 // Constants for wire API modes
 const WIRE_API_MODES = {
   CHAT: 'chat',
-  RESPONSES: 'responses'
+  RESPONSES: 'responses',
+  ENV: 'env'
 };
 
 const DEFAULT_WIRE_API = WIRE_API_MODES.CHAT;
@@ -18,6 +19,15 @@ const ACCOUNT_TYPES = {
   DROIDS: 'Droids'
 };
 
+// Constants for MCP server scopes
+const MCP_SCOPES = {
+  LOCAL: 'local',      // Only available in current project
+  PROJECT: 'project',  // Shared with project members via .mcp.json
+  USER: 'user'         // Available to all projects for current user (global)
+};
+
+const DEFAULT_MCP_SCOPE = MCP_SCOPES.LOCAL;
+
 /**
  * Cross-platform configuration manager
  * Stores global accounts in user home directory
@@ -698,23 +708,20 @@ class ConfigManager {
 
         // Update auth.json with API key
         this.updateCodexAuthJson(account.apiKey);
+      } else if (wireApi === WIRE_API_MODES.ENV) {
+        // Env mode: use environment variable for authentication
+        profileConfig += `wire_api = "${WIRE_API_MODES.CHAT}"\n`;
+        const envKey = account.envKey || 'AIS_USER_API_KEY';
+        profileConfig += `env_key = "${envKey}"\n`;
+
+        // Clear auth.json to ensure env mode is used
+        this.clearCodexAuthJson();
       }
     }
 
     // Remove all old profiles with the same name (including duplicates)
-    // Pattern 1: Remove profiles with AIS comment header
-    const commentedProfilePattern = new RegExp(
-      `# AIS Profile for project: [^\\n]*${escapedProjectName}[^\\n]*\\n\\[profiles\\.${escapedProfileName}\\]\\n(?:[^\\[].*\\n)*`,
-      'g'
-    );
-    existingConfig = existingConfig.replace(commentedProfilePattern, '');
-
-    // Pattern 2: Remove standalone profiles without comment
-    const standaloneProfilePattern = new RegExp(
-      `\\n\\[profiles\\.${escapedProfileName}\\]\\n(?:[^\\[].*\\n)*(?=\\n\\[|$)`,
-      'g'
-    );
-    existingConfig = existingConfig.replace(standaloneProfilePattern, '');
+    // Use line-by-line parsing for more reliable cleanup
+    existingConfig = this._removeProfileFromConfig(existingConfig, profileName);
 
     // Append new profile
     const newConfig = existingConfig.trimEnd() + '\n' + profileConfig;
@@ -787,6 +794,68 @@ class ConfigManager {
     }
   }
 
+  /**
+   * Remove a profile from TOML config string
+   * Uses line-by-line parsing for reliable removal of all instances
+   * @private
+   * @param {string} configContent - The TOML config content
+   * @param {string} profileName - The profile name to remove (e.g., "ais_myproject")
+   * @returns {string} Cleaned config content
+   */
+  _removeProfileFromConfig(configContent, profileName) {
+    const lines = configContent.split('\n');
+    const cleanedLines = [];
+    let skipUntilNextSection = false;
+    const profileSectionHeader = `[profiles.${profileName}]`;
+
+    for (let i = 0; i < lines.length; i++) {
+      const line = lines[i];
+      const trimmedLine = line.trim();
+
+      // Check if this is the profile section we want to remove
+      if (trimmedLine === profileSectionHeader) {
+        skipUntilNextSection = true;
+
+        // Remove the AIS comment line before it if present
+        if (cleanedLines.length > 0) {
+          const lastLine = cleanedLines[cleanedLines.length - 1].trim();
+          if (lastLine.startsWith('# AIS Profile for project:')) {
+            cleanedLines.pop();
+          }
+        }
+
+        // Remove trailing empty lines before the profile
+        while (cleanedLines.length > 0 && cleanedLines[cleanedLines.length - 1].trim() === '') {
+          cleanedLines.pop();
+        }
+
+        continue; // Skip the profile header line
+      }
+
+      // If we're in skip mode, check if we've reached the next section
+      if (skipUntilNextSection) {
+        // A new section starts with '[' at the beginning (after trimming)
+        if (trimmedLine.startsWith('[')) {
+          skipUntilNextSection = false;
+          // Don't skip this line - it's the start of a new section
+        } else {
+          // Skip this line as it belongs to the profile we're removing
+          continue;
+        }
+      }
+
+      cleanedLines.push(line);
+    }
+
+    // Join lines and clean up excessive empty lines
+    let result = cleanedLines.join('\n');
+
+    // Replace 3+ consecutive newlines with just 2 (one blank line)
+    result = result.replace(/\n{3,}/g, '\n\n');
+
+    return result;
+  }
+
   /**
    * Clear OPENAI_API_KEY in ~/.codex/auth.json for chat mode
    * @deprecated This method is no longer called automatically.
@@ -923,6 +992,12 @@ class ConfigManager {
   addMcpServer(name, serverData) {
     const config = this.readGlobalConfig();
     if (!config.mcpServers) config.mcpServers = {};
+
+    // Set default scope if not specified
+    if (!serverData.scope) {
+      serverData.scope = DEFAULT_MCP_SCOPE;
+    }
+
     config.mcpServers[name] = {
       ...serverData,
       createdAt: config.mcpServers[name]?.createdAt || new Date().toISOString(),
@@ -1043,9 +1118,11 @@ class ConfigManager {
   }
 
   /**
-   * Enable MCP server for current project
+   * Enable MCP server for current project with scope
+   * @param {string} serverName - Name of the MCP server
+   * @param {string} scope - Scope: 'local', 'project', or 'user'
    */
-  enableProjectMcpServer(serverName) {
+  enableProjectMcpServer(serverName, scope = DEFAULT_MCP_SCOPE) {
     const server = this.getMcpServer(serverName);
     if (!server) return false;
 
@@ -1063,11 +1140,42 @@ class ConfigManager {
       const data = fs.readFileSync(projectConfigFile, 'utf8');
       const projectConfig = JSON.parse(data);
 
-      if (!projectConfig.enabledMcpServers) projectConfig.enabledMcpServers = [];
-      if (!projectConfig.enabledMcpServers.includes(serverName)) {
-        projectConfig.enabledMcpServers.push(serverName);
-        fs.writeFileSync(projectConfigFile, JSON.stringify(projectConfig, null, 2), 'utf8');
+      // Update server scope in global config
+      const globalConfig = this.readGlobalConfig();
+      if (globalConfig.mcpServers[serverName]) {
+        globalConfig.mcpServers[serverName].scope = scope;
+        this.saveGlobalConfig(globalConfig);
       }
+
+      // Handle different scopes
+      if (scope === MCP_SCOPES.LOCAL) {
+        // Local scope: only enable for current project
+        if (!projectConfig.enabledMcpServers) projectConfig.enabledMcpServers = [];
+        if (!projectConfig.enabledMcpServers.includes(serverName)) {
+          projectConfig.enabledMcpServers.push(serverName);
+        }
+      } else if (scope === MCP_SCOPES.PROJECT) {
+        // Project scope: store in project config for sharing
+        if (!projectConfig.projectMcpServers) projectConfig.projectMcpServers = {};
+        projectConfig.projectMcpServers[serverName] = {
+          ...server,
+          scope: MCP_SCOPES.PROJECT
+        };
+
+        // Also add to enabled list
+        if (!projectConfig.enabledMcpServers) projectConfig.enabledMcpServers = [];
+        if (!projectConfig.enabledMcpServers.includes(serverName)) {
+          projectConfig.enabledMcpServers.push(serverName);
+        }
+      } else if (scope === MCP_SCOPES.USER) {
+        // User scope: mark as globally enabled
+        if (!projectConfig.enabledMcpServers) projectConfig.enabledMcpServers = [];
+        if (!projectConfig.enabledMcpServers.includes(serverName)) {
+          projectConfig.enabledMcpServers.push(serverName);
+        }
+      }
+
+      fs.writeFileSync(projectConfigFile, JSON.stringify(projectConfig, null, 2), 'utf8');
       return true;
     } catch (error) {
       throw new Error(`Failed to enable MCP server: ${error.message}`);
@@ -1108,13 +1216,58 @@ class ConfigManager {
 
   /**
    * Get enabled MCP servers for current project
+   * Includes local, project, and user-scoped servers
    */
   getEnabledMcpServers() {
-    return this.getProjectMcpServers();
+    const projectServers = this.getProjectMcpServers();
+    const globalServers = this.getAllMcpServers();
+
+    // Add user-scoped servers that are globally enabled
+    const userScopedServers = Object.keys(globalServers).filter(name =>
+      globalServers[name].scope === MCP_SCOPES.USER && !projectServers.includes(name)
+    );
+
+    return [...projectServers, ...userScopedServers];
+  }
+
+  /**
+   * Get all available MCP servers including project-scoped ones
+   */
+  getAllAvailableMcpServers() {
+    const globalServers = this.getAllMcpServers();
+    const projectRoot = this.findProjectRoot();
+
+    if (!projectRoot) {
+      return globalServers;
+    }
+
+    try {
+      const projectConfigFile = path.join(projectRoot, this.projectConfigFilename);
+      if (!fs.existsSync(projectConfigFile)) {
+        return globalServers;
+      }
+
+      const data = fs.readFileSync(projectConfigFile, 'utf8');
+      const projectConfig = JSON.parse(data);
+
+      // Merge global and project servers
+      const allServers = { ...globalServers };
+
+      if (projectConfig.projectMcpServers) {
+        Object.entries(projectConfig.projectMcpServers).forEach(([name, server]) => {
+          allServers[name] = server;
+        });
+      }
+
+      return allServers;
+    } catch (error) {
+      return globalServers;
+    }
   }
 
   /**
    * Get Claude Code user config path (cross-platform)
+   * Priority: ~/.claude/settings.json > platform-specific paths > legacy paths
    */
   getClaudeUserConfigPath() {
     const platform = process.platform;
@@ -1122,12 +1275,30 @@ class ConfigManager {
 
     if (!home) return null;
 
-    // Try common locations
-    const locations = [
-      path.join(home, '.claude.json'),
-      path.join(home, '.config', 'claude', 'config.json')
-    ];
+    // Priority order for Claude user config
+    const locations = [];
+
+    // Primary location: ~/.claude/settings.json (modern Claude Code)
+    locations.push(path.join(home, '.claude', 'settings.json'));
 
+    // Platform-specific locations
+    if (platform === 'win32') {
+      // Windows: %APPDATA%\claude\settings.json
+      const appData = process.env.APPDATA;
+      if (appData) {
+        locations.push(path.join(appData, 'claude', 'settings.json'));
+        locations.push(path.join(appData, 'claude', 'config.json'));
+      }
+    } else {
+      // macOS/Linux: ~/.config/claude/settings.json
+      locations.push(path.join(home, '.config', 'claude', 'settings.json'));
+      locations.push(path.join(home, '.config', 'claude', 'config.json'));
+    }
+
+    // Legacy fallback: ~/.claude.json
+    locations.push(path.join(home, '.claude.json'));
+
+    // Return first existing location
     for (const loc of locations) {
       if (fs.existsSync(loc)) {
         return loc;
@@ -1353,14 +1524,22 @@ class ConfigManager {
 
       // Get enabled MCP servers
       const enabledServers = this.getEnabledMcpServers();
-      const allServers = this.getAllMcpServers();
+      const allServers = this.getAllAvailableMcpServers();
+
+      // Filter servers by scope:
+      // - Only 'project' scoped servers should be in .mcp.json (shared with team)
+      // - 'local' and 'user' scoped servers should NOT be in .mcp.json
+      const projectScopedServers = enabledServers.filter(serverName => {
+        const server = allServers[serverName];
+        return server && server.scope === MCP_SCOPES.PROJECT;
+      });
 
-      if (enabledServers.length > 0) {
+      if (projectScopedServers.length > 0) {
         const mcpConfig = {
           mcpServers: {}
         };
 
-        enabledServers.forEach(serverName => {
+        projectScopedServers.forEach(serverName => {
           const server = allServers[serverName];
           if (server) {
             const serverConfig = {};
@@ -1395,7 +1574,7 @@ class ConfigManager {
 
         fs.writeFileSync(mcpConfigFile, JSON.stringify(mcpConfig, null, 2), 'utf8');
       } else {
-        // Remove .mcp.json if no servers are enabled
+        // Remove .mcp.json if no project-scoped servers are enabled
         if (fs.existsSync(mcpConfigFile)) {
           fs.unlinkSync(mcpConfigFile);
         }
@@ -1410,3 +1589,5 @@ module.exports = ConfigManager;
 module.exports.WIRE_API_MODES = WIRE_API_MODES;
 module.exports.DEFAULT_WIRE_API = DEFAULT_WIRE_API;
 module.exports.ACCOUNT_TYPES = ACCOUNT_TYPES;
+module.exports.MCP_SCOPES = MCP_SCOPES;
+module.exports.DEFAULT_MCP_SCOPE = DEFAULT_MCP_SCOPE;

package/src/index.js

@@ -27,7 +27,15 @@ const {
   disableMcpServer,
   showEnabledMcpServers,
   syncMcpConfig,
-  testMcpServer
+  testMcpServer,
+  listEnv,
+  addEnv,
+  setEnv,
+  removeEnv,
+  unsetEnv,
+  showEnv,
+  clearEnv,
+  editEnv
 } = require('./commands');
 
 // Package info
@@ -167,6 +175,7 @@ mcpCommand
 mcpCommand
   .command('enable [name]')
   .description('Activate MCP server for current project (为当前项目激活 MCP 服务器)')
+  .option('-s, --scope <scope>', 'Scope: local (default), project, or user (作用范围: local(默认), project, user)')
   .action(enableMcpServer);
 
 mcpCommand
@@ -189,6 +198,58 @@ mcpCommand
   .description('Test MCP server availability (测试 MCP 服务器可用性)')
   .action(testMcpServer);
 
+// Environment variable management commands
+const envCommand = program
+  .command('env')
+  .description('Manage Claude environment variables (管理 Claude 环境变量)');
+
+envCommand
+  .command('list')
+  .alias('ls')
+  .description('List all environment variables from project and user configs (列出项目和用户配置中的所有环境变量)')
+  .action(listEnv);
+
+envCommand
+  .command('add')
+  .description('Add or update an environment variable interactively (交互式添加或更新环境变量)')
+  .action(addEnv);
+
+envCommand
+  .command('set <key> <value>')
+  .description('Set an environment variable (non-interactive) (设置环境变量,非交互式)')
+  .option('-l, --level <level>', 'Configuration level: project or user (default: user) (配置级别: project 或 user,默认: user)')
+  .action(setEnv);
+
+envCommand
+  .command('remove')
+  .alias('rm')
+  .description('Remove an environment variable interactively (交互式删除环境变量)')
+  .action(removeEnv);
+
+envCommand
+  .command('unset <key>')
+  .description('Remove an environment variable by key (non-interactive) (通过键名删除环境变量,非交互式)')
+  .option('-l, --level <level>', 'Configuration level: project or user (default: user) (配置级别: project 或 user,默认: user)')
+  .action(unsetEnv);
+
+envCommand
+  .command('show <key>')
+  .description('Show an environment variable value (显示环境变量值)')
+  .option('-l, --level <level>', 'Configuration level: project or user (if not specified, searches both) (配置级别: project 或 user,未指定则搜索两者)')
+  .action(showEnv);
+
+envCommand
+  .command('clear')
+  .description('Clear all environment variables at a level (清空某个级别的所有环境变量)')
+  .option('-l, --level <level>', 'Configuration level: project or user (default: user) (配置级别: project 或 user,默认: user)')
+  .action(clearEnv);
+
+envCommand
+  .command('edit')
+  .description('Edit environment variables interactively (交互式编辑环境变量)')
+  .option('-l, --level <level>', 'Configuration level: project or user (default: user) (配置级别: project 或 user,默认: user)')
+  .action(editEnv);
+
 // Help command
 program
   .command('help')
@@ -211,6 +272,7 @@ program
     console.log('  ui                     Start web-based account manager UI (启动基于 Web 的账号管理界面)');
     console.log('  model                  Manage model groups (管理模型组)');
     console.log('  mcp                    Manage MCP servers (管理 MCP 服务器)');
+    console.log('  env                    Manage Claude environment variables (管理 Claude 环境变量)');
     console.log('  help                   Display this help message (显示此帮助信息)');
     console.log('  version                Show version number (显示版本号)\n');