aihezu 1.8.0 → 2.0.0

package/lib/cache.js

@@ -2,10 +2,7 @@ const path = require('path');
 const os = require('os');
 const fs = require('fs');
 
-const homeDir = os.homedir();
-const claudeDir = path.join(homeDir, '.claude');
-
-// 生成本地时间戳，格式：YYYYMMDDHHMMSS
+// Generate local timestamp: YYYYMMDDHHMMSS
 function getLocalTimestamp() {
   const now = new Date();
   const year = now.getFullYear();
@@ -17,91 +14,74 @@ function getLocalTimestamp() {
   return `${year}${month}${day}${hours}${minutes}${seconds}`;
 }
 
-// 清理缓存函数
 function cleanCache(options = {}) {
-  const { showHeader = true } = options;
+  const { 
+    targetDir, 
+    itemsToClean = [], 
+    showHeader = true 
+  } = options;
 
   if (showHeader) {
     console.log('\n=== 清理缓存 ===\n');
   }
 
-  const timestamp = getLocalTimestamp();
-  const cacheItems = [
-    'history.jsonl',      // 历史记录
-    'debug',              // 调试信息
-    'file-history',       // 文件历史
-    'session-env',        // 会话环境
-    'shell-snapshots',    // Shell 快照
-    'statsig',            // 统计信息
-    'todos'               // 待办事项
-  ];
-
-  // 需要保留的配置和工具（不清理）
-  // - settings.json (配置文件)
-  // - commands/ (自定义命令)
-  // - skills/ (技能)
-  // - mcp/ (MCP 服务器)
-  // - projects/ (项目信息)
-  // - ide/ (IDE 配置)
+  if (!targetDir || !fs.existsSync(targetDir)) {
+    console.log(`ℹ️  未找到目录: ${targetDir}`);
+    return 0;
+  }
 
+  const timestamp = getLocalTimestamp();
   let cleanedCount = 0;
 
-  if (fs.existsSync(claudeDir)) {
-    console.log('📂 开始清理 ~/.claude 目录下的缓存文件...\n');
+  console.log(`📂 正在清理目录: ${targetDir}...
+`);
 
-    // 遍历并清理缓存项
-    for (const item of cacheItems) {
-      const itemPath = path.join(claudeDir, item);
+  // Clean specific items
+  for (const item of itemsToClean) {
+    const itemPath = path.join(targetDir, item);
 
-      try {
-        if (fs.existsSync(itemPath)) {
-          const stat = fs.statSync(itemPath);
-          const backupPath = `${itemPath}-backup-${timestamp}`;
-
-          if (stat.isDirectory()) {
-            console.log(`📦 备份并清理目录: ${item}/`);
-          } else {
-            console.log(`📦 备份并清理文件: ${item}`);
-          }
-
-          // 使用 Node.js 原生 API，跨平台兼容
-          fs.renameSync(itemPath, backupPath);
-          cleanedCount++;
+    try {
+      if (fs.existsSync(itemPath)) {
+        const stat = fs.statSync(itemPath);
+        const backupPath = `${itemPath}-backup-${timestamp}`;
+
+        if (stat.isDirectory()) {
+          console.log(`📦 备份并清理目录: ${item}/`);
+        } else {
+          console.log(`📦 备份并清理文件: ${item}`);
         }
-      } catch (e) {
-        console.log(`⚠️  处理 ${item} 时出错: ${e.message}`);
+
+        fs.renameSync(itemPath, backupPath);
+        cleanedCount++;
       }
+    } catch (e) {
+      console.log(`⚠️  处理 ${item} 时出错: ${e.message}`);
     }
+  }
 
-    // 清理旧的备份文件夹（.claude-* 格式）
-    try {
-      const items = fs.readdirSync(claudeDir);
-      for (const item of items) {
-        if (item.startsWith('.claude-') || item.startsWith('backup-')) {
-          const itemPath = path.join(claudeDir, item);
-          const stat = fs.statSync(itemPath);
-
-          if (stat.isDirectory()) {
-            console.log(`🗑️  删除旧备份: ${item}/`);
-            // 使用 Node.js 原生 API，跨平台兼容
-            fs.rmSync(itemPath, { recursive: true, force: true });
-            cleanedCount++;
-          }
+  // Clean old backups (generic logic: starting with 'backup-' or hidden backups)
+  try {
+    const items = fs.readdirSync(targetDir);
+    for (const item of items) {
+      // Logic for backups: usually we renamed them to 'name-backup-timestamp'
+      // Or if the original logic was specific: '.claude-*' or 'backup-*'
+      // Let's stick to generic 'backup-*' or if the file ends with '-backup-\d+'
+      
+      const isBackup = item.startsWith('backup-') || /backup-\d{14}$/.test(item);
+
+      if (isBackup) {
+        const itemPath = path.join(targetDir, item);
+        const stat = fs.statSync(itemPath);
+
+        if (stat.isDirectory()) {
+          console.log(`🗑️  删除旧备份: ${item}/`);
+          fs.rmSync(itemPath, { recursive: true, force: true });
+          cleanedCount++;
         }
       }
-    } catch (e) {
-      // 忽略错误
     }
-
-    console.log('\n✅ 已保留以下配置和工具:');
-    console.log('   - settings.json (配置文件)');
-    console.log('   - commands/ (自定义命令)');
-    console.log('   - skills/ (技能)');
-    console.log('   - mcp/ (MCP 服务器)');
-    console.log('   - projects/ (项目信息)');
-    console.log('   - ide/ (IDE 配置)');
-  } else {
-    console.log('ℹ️  未找到 ~/.claude 目录');
+  } catch (e) {
+    // ignore
   }
 
   return cleanedCount;

package/lib/hosts.js

@@ -2,7 +2,7 @@ const { execSync } = require('child_process');
 const os = require('os');
 const fs = require('fs');
 
-// 生成本地时间戳，格式：YYYYMMDDHHMMSS
+// Generate local timestamp: YYYYMMDDHHMMSS
 function getLocalTimestamp() {
   const now = new Date();
   const year = now.getFullYear();
@@ -14,31 +14,37 @@ function getLocalTimestamp() {
   return `${year}${month}${day}${hours}${minutes}${seconds}`;
 }
 
-// 修改 hosts 文件的函数
-function modifyHostsFile() {
+/**
+ * Modifies the system hosts file.
+ * @param {Array<{ip: string, domain: string}>} hostEntries - Array of entries to add.
+ * @param {string} markerText - Comment marker to identify managed lines.
+ */
+function modifyHostsFile(hostEntries = [], markerText = '# Added by aihezu CLI') {
+  if (!hostEntries || hostEntries.length === 0) {
+    console.log('ℹ️  此服务无需修改 hosts 文件。');
+    return true;
+  }
+
   const isWindows = os.platform() === 'win32';
   const hostsPath = isWindows
     ? 'C:\\Windows\\System32\\drivers\\etc\\hosts'
     : '/etc/hosts';
 
-  const domains = [
-    'statsig.anthropic.com',
-    'api.anthropic.com'
-  ];
+  // Extract just the domain names for filtering
+  const targetDomains = hostEntries.map(e => e.domain);
 
   try {
-    console.log('🔧 开始修改 hosts 文件...');
+    console.log(`🔧 正在修改 hosts 文件 (${hostsPath})...`);
 
-    // 读取现有 hosts 文件内容
     let hostsContent = '';
     try {
       hostsContent = fs.readFileSync(hostsPath, 'utf8');
     } catch (error) {
-      console.error('❌ 无法读取 hosts 文件，请确保以管理员/root权限运行');
+      console.error('❌ 无法读取 hosts 文件。请确保以管理员/root权限运行。');
       return false;
     }
 
-    // 备份 hosts 文件
+    // Backup
     const timestamp = getLocalTimestamp();
     const hostsBackup = `${hostsPath}.backup-${timestamp}`;
     try {
@@ -48,90 +54,86 @@ function modifyHostsFile() {
       console.error('⚠️  备份 hosts 文件失败:', error.message);
     }
 
-    // 移除所有包含目标域名的现有条目（无论指向什么IP）
+    // Filter existing
     const lines = hostsContent.split('\n');
     const filteredLines = lines.filter(line => {
       const trimmed = line.trim();
-      // 跳过注释行（但保留其他注释）
-      if (trimmed.startsWith('#')) {
-        // 如果是之前添加的标记注释，也移除
-        if (trimmed.includes('Added by aihezu ccclear tool')) {
-          return false;
-        }
-        return true;
+      
+      // Remove our managed marker comments
+      // Also remove legacy markers for backward compatibility
+      if (trimmed === markerText || 
+          trimmed === '# Added by aihezu ccclear tool' || 
+          (trimmed.startsWith('#') && trimmed.includes('aihezu'))) {
+        return false;
       }
-      // 移除任何包含目标域名的行
-      return !domains.some(domain => {
-        const regex = new RegExp('\\s+' + domain.replace('.', '\\.') + '(\\s|$)', 'i');
+
+      // Keep other comments
+      if (trimmed.startsWith('#')) return true;
+
+      // Remove lines matching our target domains
+      return !targetDomains.some(domain => {
+        const regex = new RegExp('\\s+' + domain.replace('.', '\\.') + '(\s|$)', 'i');
         return regex.test(trimmed);
       });
     });
 
-    // 添加新的条目
-    const newEntries = [
+    // Build new entries
+    const newLines = [
       '',
-      '# Added by aihezu ccclear tool',
-      '127.0.0.1 statsig.anthropic.com',
-      '127.0.0.1 api.anthropic.com'
+      markerText,
+      ...hostEntries.map(entry => `${entry.ip} ${entry.domain}`)
     ];
 
-    const newHostsContent = filteredLines.join('\n') + '\n' + newEntries.join('\n') + '\n';
+    const newHostsContent = filteredLines.join('\n') + '\n' + newLines.join('\n') + '\n';
 
-    // 写入 hosts 文件
     try {
       fs.writeFileSync(hostsPath, newHostsContent);
-      console.log('✅ hosts 文件修改成功！');
-      console.log('   已添加/更新以下域名解析：');
-      console.log('   - statsig.anthropic.com -> 127.0.0.1');
-      console.log('   - api.anthropic.com -> 127.0.0.1');
+      console.log('✅ hosts 文件更新成功！');
+      hostEntries.forEach(entry => {
+        console.log(`   - ${entry.domain} -> ${entry.ip}`);
+      });
 
-      // 刷新 DNS 缓存
+      // Flush DNS
       console.log('🔄 刷新 DNS 缓存...');
-      try {
-        if (isWindows) {
-          // Windows: 需要管理员权限
-          execSync('ipconfig /flushdns', { stdio: 'pipe' });
-          console.log('   ✅ DNS 缓存已刷新');
-        } else if (os.platform() === 'darwin') {
-          // macOS: 如果已经是 root/sudo 运行，直接执行；否则会失败但不影响主要功能
-          try {
-            execSync('dscacheutil -flushcache; killall -HUP mDNSResponder', { stdio: 'pipe' });
-            console.log('   ✅ DNS 缓存已刷新');
-          } catch (e) {
-            console.log('   ⚠️  DNS 缓存刷新失败，请稍后手动执行:');
-            console.log('      sudo dscacheutil -flushcache; sudo killall -HUP mDNSResponder');
-          }
-        } else {
-          // Linux
-          try {
-            execSync('systemd-resolve --flush-caches 2>/dev/null || service nscd restart 2>/dev/null', { stdio: 'pipe' });
-            console.log('   ✅ DNS 缓存已刷新');
-          } catch (e) {
-            console.log('   ⚠️  DNS 缓存刷新失败，请稍后手动执行:');
-            console.log('      sudo systemd-resolve --flush-caches');
-          }
-        }
-      } catch (e) {
-        if (isWindows) {
-          console.log('   ⚠️  DNS 缓存刷新失败，请以管理员身份运行命令提示符');
-        } else {
-          console.log('   ℹ️  DNS 缓存刷新失败（可能需要 root 权限）');
-        }
-      }
-
+      flushDNS(isWindows);
       return true;
     } catch (error) {
-      console.error('❌ 无法写入 hosts 文件，请确保以管理员/root权限运行');
-      console.error('   macOS/Linux: 使用 sudo 运行');
-      console.error('   Windows: 以管理员身份运行命令提示符');
+      console.error('❌ 无法写入 hosts 文件。权限不足？');
       return false;
     }
   } catch (error) {
-    console.error('❌ 修改 hosts 文件失败:', error.message);
+    console.error('❌ 修改 hosts 文件出错:', error.message);
     return false;
   }
 }
 
+function flushDNS(isWindows) {
+  try {
+    if (isWindows) {
+      execSync('ipconfig /flushdns', { stdio: 'pipe' });
+      console.log('   ✅ DNS 缓存已刷新。');
+    } else if (os.platform() === 'darwin') {
+      try {
+        execSync('dscacheutil -flushcache; killall -HUP mDNSResponder', { stdio: 'pipe' });
+        console.log('   ✅ DNS 缓存已刷新。');
+      } catch (e) {
+        console.log('   ⚠️  DNS 刷新失败 (请尝试 sudo): sudo dscacheutil -flushcache; sudo killall -HUP mDNSResponder');
+      }
+    } else {
+      // Linux
+      try {
+        execSync('systemd-resolve --flush-caches 2>/dev/null || service nscd restart 2>/dev/null', { stdio: 'pipe' });
+        console.log('   ✅ DNS 缓存已刷新。');
+      } catch (e) {
+        console.log('   ⚠️  DNS 刷新失败。请尝试: sudo systemd-resolve --flush-caches');
+      }
+    }
+  } catch (e) {
+    console.log('   ⚠️  DNS 刷新失败。');
+  }
+}
+
 module.exports = {
-  modifyHostsFile
+  modifyHostsFile,
+  getLocalTimestamp
 };

package/package.json

@@ -1,24 +1,26 @@
 {
   "name": "aihezu",
-  "version": "1.8.0",
-  "description": "Claude Code CLI 清理工具 - 快速备份和清理 Claude Code 的本地配置和缓存，同时修改 hosts 文件实现本地代理",
-  "main": "bin/ccclear.js",
+  "version": "2.0.0",
+  "description": "AI 开发环境配置工具 - 支持 Claude Code, Codex, Google Gemini 的本地化配置、代理设置与缓存清理",
+  "main": "bin/aihezu.js",
   "bin": {
     "aihezu": "bin/aihezu.js",
     "ccclear": "bin/ccclear.js",
     "ccinstall": "bin/ccinstall.js"
   },
   "scripts": {
-    "test": "node bin/ccclear.js"
+    "test": "node bin/aihezu.js help"
   },
   "keywords": [
+    "ai",
+    "cli",
+    "proxy",
     "claude",
     "claude-code",
-    "cli",
-    "clear",
-    "cache",
-    "cleanup",
-    "anthropic"
+    "codex",
+    "gemini",
+    "configuration",
+    "hosts"
   ],
   "author": "aihezu",
   "license": "MIT",
@@ -33,4 +35,4 @@
     "url": "https://github.com/aihezu/npm-ccclear/issues"
   },
   "homepage": "https://github.com/aihezu/npm-ccclear#readme"
-}
+}

package/services/claude.js

@@ -0,0 +1,64 @@
+const path = require('path');
+const os = require('os');
+const fs = require('fs');
+
+const homeDir = os.homedir();
+const configDir = path.join(homeDir, '.claude');
+const settingsPath = path.join(configDir, 'settings.json');
+
+module.exports = {
+  name: 'claude',
+  displayName: 'Claude Code',
+  defaultApiUrl: 'https://cc.aihezu.dev/api',
+  
+  // Cache cleaning configuration
+  cacheConfig: {
+    dir: configDir,
+    items: [
+      'history.jsonl',
+      'debug',
+      'file-history',
+      'session-env',
+      'shell-snapshots',
+      'statsig',
+      'todos'
+    ]
+  },
+
+  // Hosts file configuration
+  hostsConfig: [
+    { domain: 'statsig.anthropic.com', ip: '127.0.0.1' },
+    { domain: 'api.anthropic.com', ip: '127.0.0.1' }
+  ],
+
+  // Configuration setup logic
+  setupConfig: async (apiKey, apiUrl, options = {}) => {
+    // Ensure directory exists
+    if (!fs.existsSync(configDir)) {
+      console.log('📁 创建 ~/.claude 目录...');
+      fs.mkdirSync(configDir, { recursive: true });
+    }
+
+    // Read existing settings
+    let settings = {};
+    if (fs.existsSync(settingsPath)) {
+      try {
+        const content = fs.readFileSync(settingsPath, 'utf8');
+        settings = JSON.parse(content);
+      } catch (e) {
+        console.log('⚠️  现有配置文件格式错误，将创建新的配置。');
+      }
+    }
+
+    // Update settings
+    settings.env = {
+      ANTHROPIC_AUTH_TOKEN: apiKey,
+      ANTHROPIC_BASE_URL: apiUrl
+    };
+
+    // Write file
+    fs.writeFileSync(settingsPath, JSON.stringify(settings, null, 2), 'utf8');
+    
+    return [settingsPath];
+  }
+};

package/services/codex.js

@@ -0,0 +1,119 @@
+const path = require('path');
+const os = require('os');
+const fs = require('fs');
+
+const homeDir = os.homedir();
+const configDir = path.join(homeDir, '.codex');
+const configPath = path.join(configDir, 'config.toml');
+const authPath = path.join(configDir, 'auth.json');
+
+module.exports = {
+  name: 'codex',
+  displayName: 'Codex',
+  defaultApiUrl: 'https://cc.aihezu.dev/openai',
+  
+  // Cache cleaning configuration (minimal for Codex usually)
+  cacheConfig: {
+    dir: configDir,
+    items: [] // Add items if needed in future
+  },
+
+  // Hosts file configuration (Codex usually doesn't need hosts mod if config allows URL change)
+  hostsConfig: [],
+
+  // Configuration setup logic
+  setupConfig: async (apiKey, apiUrl, options = {}) => {
+    const { modelName = 'gpt-5-codex' } = options;
+
+    if (!fs.existsSync(configDir)) {
+      console.log('📁 创建 ~/.codex 目录...');
+      fs.mkdirSync(configDir, { recursive: true });
+    }
+
+    // Handle config.toml
+    let configContent = '';
+    let existingConfig = '';
+    let providerName = 'aihezu';
+
+    if (fs.existsSync(configPath)) {
+      existingConfig = fs.readFileSync(configPath, 'utf8');
+      
+      const providerMatch = existingConfig.match(/model_provider\s*=\s*"([^"]+)"/);
+      if (providerMatch) {
+        providerName = providerMatch[1];
+      }
+
+      // Update logic (simplified from original)
+      let newConfig = existingConfig;
+      
+      // Update model
+      const modelPattern = /^model\s*=\s*"[^"]*"/m;
+      if (modelPattern.test(newConfig)) {
+        newConfig = newConfig.replace(modelPattern, `model = "${modelName}"`);
+      }
+
+      // Update base_url
+      const baseUrlPattern = new RegExp(
+        `(\[model_providers\.${providerName}\][\s\S]*?base_url\s*=\s*)"[^"]*"`,
+        'm'
+      );
+
+      if (baseUrlPattern.test(newConfig)) {
+        newConfig = newConfig.replace(baseUrlPattern, `$1"${apiUrl}"`);
+      } else {
+        // If provider block exists but no base_url (rare), or provider block missing
+        const providerSectionPattern = new RegExp(`\[model_providers\.${providerName}\]`, 'm');
+        if (providerSectionPattern.test(newConfig)) {
+           newConfig = newConfig.replace(
+            providerSectionPattern,
+            `[model_providers.${providerName}]\nbase_url = "${apiUrl}"`
+          );
+        } else {
+          // Append new provider
+           newConfig = newConfig.trim() + '\n\n' +
+            `[model_providers.${providerName}]\n` +
+            `name = "${providerName}"\n` +
+            `base_url = "${apiUrl}"\n` +
+            `wire_api = "responses"\n` +
+            `requires_openai_auth = true\n` +
+            `env_key = "AIHEZU_OAI_KEY"\n`;
+        }
+      }
+      configContent = newConfig;
+
+    } else {
+      // New Config
+      configContent = [
+        'model_provider = "aihezu"',
+        `model = "${modelName}"`, 
+        'model_reasoning_effort = "high"',
+        'disable_response_storage = true',
+        'preferred_auth_method = "apikey"',
+        '',
+        '[model_providers.aihezu]',
+        'name = "aihezu"',
+        `base_url = "${apiUrl}"`, 
+        'wire_api = "responses"',
+        'requires_openai_auth = true',
+        'env_key = "AIHEZU_OAI_KEY"',
+        ''
+      ].join('\n');
+    }
+
+    fs.writeFileSync(configPath, configContent, 'utf8');
+
+    // Handle auth.json
+    let authData = {};
+    if (fs.existsSync(authPath)) {
+      try {
+        authData = JSON.parse(fs.readFileSync(authPath, 'utf8'));
+      } catch (e) {
+        authData = {};
+      }
+    }
+    authData.AIHEZU_OAI_KEY = apiKey;
+    fs.writeFileSync(authPath, JSON.stringify(authData, null, 2), 'utf8');
+
+    return [configPath, authPath];
+  }
+};

package/services/gemini.js

@@ -0,0 +1,50 @@
+const path = require('path');
+const os = require('os');
+const fs = require('fs');
+
+const homeDir = os.homedir();
+const configDir = path.join(homeDir, '.gemini');
+const envFilePath = path.join(configDir, '.env');
+const settingsPath = path.join(configDir, 'settings.json');
+
+module.exports = {
+  name: 'gemini',
+  displayName: 'Google Gemini',
+  defaultApiUrl: 'https://cc.aihezu.dev/gemini', 
+  
+  // Cache cleaning configuration
+  cacheConfig: {
+    dir: configDir,
+    items: ['cache', 'logs']
+  },
+
+  // Hosts file configuration
+  // Since we use .env for GOOGLE_GEMINI_BASE_URL, we don't need hosts mod anymore!
+  hostsConfig: [],
+
+  // Configuration setup logic
+  setupConfig: async (apiKey, apiUrl, options = {}) => {
+    if (!fs.existsSync(configDir)) {
+      console.log('📁 创建 ~/.gemini 目录...');
+      fs.mkdirSync(configDir, { recursive: true });
+    }
+
+    // 1. Write to .env (Preferred method verified by user)
+    const envContent = [
+      `GEMINI_API_KEY="${apiKey}"`,
+      `GOOGLE_GEMINI_BASE_URL="${apiUrl}"`,
+      '' // End with newline
+    ].join('\n');
+
+    fs.writeFileSync(envFilePath, envContent, 'utf8');
+
+    // 2. Also keep settings.json but keep it clean (no duplicate sensitive data)
+    // We only store extra options here if any.
+    const settings = {
+      ...options 
+    };
+    fs.writeFileSync(settingsPath, JSON.stringify(settings, null, 2), 'utf8');
+    
+    return [envFilePath, settingsPath];
+  }
+};