aicodeswitch 1.2.6 → 1.3.1

package/bin/cli.js

@@ -34,6 +34,7 @@ const commands = {
   stop: () => require(path.join(binDir, 'stop')),
   restart: () => require(path.join(binDir, 'restart')),
   update: () => require(path.join(binDir, 'update')),
+  restore: () => require(path.join(binDir, 'restore')),
   version: () => require(path.join(binDir, 'version')),
 };
 
@@ -46,6 +47,7 @@ Commands:
   stop       Stop the AI Code Switch server
   restart    Restart the AI Code Switch server
   update     Update to the latest version and restart
+  restore    Restore original configuration files
   version    Show current version information
 
 Example:
@@ -53,6 +55,9 @@ Example:
   aicos stop
   aicos restart
   aicos update
+  aicos restore
+  aicos restore claude-code
+  aicos restore codex
   aicos version
   `);
   process.exit(1);

package/bin/restore.js

@@ -0,0 +1,220 @@
+const fs = require('fs');
+const path = require('path');
+const os = require('os');
+const chalk = require('chalk');
+const boxen = require('boxen');
+const ora = require('ora');
+
+// 恢复 Claude Code 配置
+const restoreClaudeConfig = () => {
+  const results = {
+    restored: [],
+    notFound: [],
+    errors: []
+  };
+
+  try {
+    const homeDir = os.homedir();
+    const claudeDir = path.join(homeDir, '.claude');
+    const claudeSettingsPath = path.join(claudeDir, 'settings.json');
+    const claudeSettingsBakPath = path.join(claudeDir, 'settings.json.bak');
+
+    // Restore settings.json
+    if (fs.existsSync(claudeSettingsBakPath)) {
+      if (fs.existsSync(claudeSettingsPath)) {
+        fs.unlinkSync(claudeSettingsPath);
+      }
+      fs.renameSync(claudeSettingsBakPath, claudeSettingsPath);
+      results.restored.push('settings.json');
+    } else {
+      results.notFound.push('settings.json.bak');
+    }
+  } catch (err) {
+    results.errors.push({ file: 'settings.json', error: err.message });
+  }
+
+  try {
+    const homeDir = os.homedir();
+    const claudeJsonPath = path.join(homeDir, '.claude.json');
+    const claudeJsonBakPath = path.join(homeDir, '.claude.json.bak');
+
+    // Restore .claude.json
+    if (fs.existsSync(claudeJsonBakPath)) {
+      if (fs.existsSync(claudeJsonPath)) {
+        fs.unlinkSync(claudeJsonPath);
+      }
+      fs.renameSync(claudeJsonBakPath, claudeJsonPath);
+      results.restored.push('.claude.json');
+    } else {
+      results.notFound.push('.claude.json.bak');
+    }
+  } catch (err) {
+    results.errors.push({ file: '.claude.json', error: err.message });
+  }
+
+  return results;
+};
+
+// 恢复 Codex 配置
+const restoreCodexConfig = () => {
+  const results = {
+    restored: [],
+    notFound: [],
+    errors: []
+  };
+
+  try {
+    const homeDir = os.homedir();
+    const codexDir = path.join(homeDir, '.codex');
+    const codexConfigPath = path.join(codexDir, 'config.toml');
+    const codexConfigBakPath = path.join(codexDir, 'config.toml.bak');
+
+    // Restore config.toml
+    if (fs.existsSync(codexConfigBakPath)) {
+      if (fs.existsSync(codexConfigPath)) {
+        fs.unlinkSync(codexConfigPath);
+      }
+      fs.renameSync(codexConfigBakPath, codexConfigPath);
+      results.restored.push('config.toml');
+    } else {
+      results.notFound.push('config.toml.bak');
+    }
+  } catch (err) {
+    results.errors.push({ file: 'config.toml', error: err.message });
+  }
+
+  try {
+    const homeDir = os.homedir();
+    const codexAuthPath = path.join(homeDir, '.codex', 'auth.json');
+    const codexAuthBakPath = path.join(homeDir, '.codex', 'auth.json.bak');
+
+    // Restore auth.json
+    if (fs.existsSync(codexAuthBakPath)) {
+      if (fs.existsSync(codexAuthPath)) {
+        fs.unlinkSync(codexAuthPath);
+      }
+      fs.renameSync(codexAuthBakPath, codexAuthPath);
+      results.restored.push('auth.json');
+    } else {
+      results.notFound.push('auth.json.bak');
+    }
+  } catch (err) {
+    results.errors.push({ file: 'auth.json', error: err.message });
+  }
+
+  return results;
+};
+
+// 显示恢复结果
+const showRestoreResult = (target, results) => {
+  const targetName = target === 'claude-code' ? 'Claude Code' : 'Codex';
+  const targetColor = target === 'claude-code' ? chalk.cyan : chalk.magenta;
+
+  let message = targetColor.bold(`${targetName} Configuration Restore\n\n`);
+
+  if (results.restored.length > 0) {
+    message += chalk.green('Restored files:\n');
+    results.restored.forEach(file => {
+      message += chalk.white(`  ✓ ${file}\n`);
+    });
+    message += '\n';
+  }
+
+  if (results.notFound.length > 0) {
+    message += chalk.gray('No backup found:\n');
+    results.notFound.forEach(file => {
+      message += chalk.gray(`  - ${file}\n`);
+    });
+    message += '\n';
+  }
+
+  if (results.errors.length > 0) {
+    message += chalk.red('Errors:\n');
+    results.errors.forEach(({ file, error }) => {
+      message += chalk.red(`  ✗ ${file}: ${error}\n`);
+    });
+    message += '\n';
+  }
+
+  // 添加重启提示
+  if (results.restored.length > 0) {
+    message += chalk.yellow.bold('⚠️  Important:\n\n');
+    if (target === 'claude-code') {
+      message += chalk.white('Please restart ') + chalk.cyan.bold('Claude Code') + chalk.white(' to apply the restored configuration.\n');
+    } else {
+      message += chalk.white('Please restart ') + chalk.magenta.bold('Codex') + chalk.white(' to apply the restored configuration.\n');
+    }
+  }
+
+  const borderColor = results.errors.length > 0 ? 'red' :
+                     results.restored.length > 0 ? 'green' : 'gray';
+
+  console.log(boxen(message, {
+    padding: 1,
+    margin: 1,
+    borderStyle: 'round',
+    borderColor
+  }));
+};
+
+// 主恢复逻辑
+const restore = async () => {
+  const args = process.argv.slice(2);
+  const target = args[0]; // claude-code, codex, or undefined (all)
+
+  console.log('\n');
+
+  const validTargets = ['claude-code', 'codex', undefined];
+
+  if (target && !validTargets.includes(target)) {
+    console.log(boxen(
+      chalk.red.bold('✗ Invalid target\n\n') +
+      chalk.white('Usage: ') + chalk.cyan('aicos restore [target]\n\n') +
+      chalk.white('Targets:\n') +
+      chalk.white('  claude-code   Restore Claude Code configuration\n') +
+      chalk.white('  codex         Restore Codex configuration\n') +
+      chalk.white('  (no arg)      Restore all configurations\n'),
+      {
+        padding: 1,
+        margin: 1,
+        borderStyle: 'round',
+        borderColor: 'red'
+      }
+    ));
+    console.log('');
+    process.exit(1);
+  }
+
+  // 恢复配置
+  if (target === 'claude-code' || !target) {
+    const spinner = ora({
+      text: chalk.cyan('Restoring Claude Code configuration...'),
+      color: 'cyan'
+    }).start();
+
+    const claudeResults = restoreClaudeConfig();
+    spinner.succeed(chalk.green('Claude Code restore complete'));
+    showRestoreResult('claude-code', claudeResults);
+  }
+
+  if (target === 'codex' || !target) {
+    if (!target) console.log(''); // Add spacing when restoring both
+
+    const spinner = ora({
+      text: chalk.magenta('Restoring Codex configuration...'),
+      color: 'magenta'
+    }).start();
+
+    const codexResults = restoreCodexConfig();
+    spinner.succeed(chalk.green('Codex restore complete'));
+    showRestoreResult('codex', codexResults);
+  }
+
+  console.log('');
+  console.log(chalk.cyan('💡 Tips:\n'));
+  console.log(chalk.white('  • Write config:  ') + chalk.cyan('aicos write-config [target]'));
+  console.log(chalk.white('  • Start server:   ') + chalk.cyan('aicos start'));
+  console.log('\n');
+};
+
+module.exports = restore();

package/bin/update.js

@@ -268,19 +268,19 @@ const update = async () => {
     color: 'cyan'
   }).start();
 
-  try {
-    const npmArgs = ['install', '-g', `${PACKAGE_NAME}@latest`];
-    if (needSudo) {
-      npmArgs.unshift('sudo');
-    }
+  const npmArgs = ['npm', 'install', '-g', `${PACKAGE_NAME}@latest`];
+  if (needSudo) {
+    npmArgs.unshift('sudo');
+  }
 
-    await execCommand(needSudo ? 'sudo' : 'npm', npmArgs);
+  try {
+    await execCommand(npmArgs);
     updateSpinner.succeed(chalk.green('Update successful!'));
   } catch (err) {
     updateSpinner.fail(chalk.red('Update failed'));
     console.log(chalk.yellow(`\nUpdate failed with error code ${err.code || 'unknown'}\n`));
     console.log(chalk.white('You can try manually updating:\n'));
-    console.log(chalk.cyan(`  ${needSudo ? 'sudo ' : ''}npm install -g ${PACKAGE_NAME}@latest\n`));
+    console.log(chalk.cyan(`  ${npmArgs.join(' ')}\n`));
 
     // 尝试重新启动服务器
     console.log(chalk.yellow('Attempting to restart server...\n'));

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "aicodeswitch",
-  "version": "1.2.6",
+  "version": "1.3.1",
   "description": "A tool to help you manage AI programming tools to access large language models locally. It allows your Claude Code, Codex and other tools to no longer be limited to official models.",
   "author": "tangshuang",
   "license": "MIT",