aihezu 1.6.0 → 1.7.0

package/README.md

@@ -1,10 +1,10 @@
 # aihezu - Claude Code CLI 工具集
 
-快速配置或清理 Claude Code CLI，本地修改 hosts 并提供企业独立域名支持。
+快速配置或清理 Claude Code / Codex，本地修改 hosts 并提供企业独立域名支持。
 
 ## 功能特性
 
-- 🔑 一键配置 Claude Code API Key，默认使用 `cc.aihezu.dev`，支持企业独立域名（`--api` / `--api-url`）
+- 🔑 一键配置 Claude（默认）或 Codex，支持企业独立域名（`--api` / `--api-url`）
 - 🧹 一键清理 Claude Code CLI 的所有本地数据
 - 🌐 自动修改 hosts 文件，将 Anthropic 域名指向本地
 - 📦 自动备份配置文件和 hosts 文件（带时间戳）
@@ -14,9 +14,9 @@
 
 ## 命令列表
 
-- `ccinstall` / `install`：配置 Claude Code API Key
-  - 默认 API 域名：`https://cc.aihezu.dev/api`
-  - 企业用户可通过 `--api` 或 `--api-url` 指定独立域名（如 `--api jfz.aihezu.dev`）
+- `ccinstall` / `install`：交互式配置 Claude（默认）或 Codex
+  - Claude：默认 `https://cc.aihezu.dev/api`，企业可用 `--api` / `--api-url` 指定独立域名
+  - Codex：写入 `~/.codex/config.toml` 和 `auth.json`，使用 `AIHEZU_OAI_KEY`
 - `ccclear`：清理 Claude Code 缓存和配置
 
 ## 清理内容
@@ -49,9 +49,9 @@ sudo npx aihezu install
 - 企业独立域名（支持传入域名或完整 URL，自动补全为 `/api`）：
 
 ```bash
-sudo npx aihezu install --api https://jfz.aihezu.dev
+sudo npx aihezu install --api https://your-org.aihezu.dev
 # 或
-sudo npx aihezu install --api jfz.aihezu.dev
+sudo npx aihezu install --api your-org.aihezu.dev
 ```
 
 也可以全局安装后运行：
@@ -61,6 +61,12 @@ npm install -g aihezu
 sudo ccinstall --api your-company.aihezu.dev
 ```
 
+- 直接选择 Codex（可在交互中选择，也可用参数指定）：
+
+```bash
+sudo npx aihezu install --provider codex
+```
+
 ### 清理 Claude Code（ccclear）
 
 - 使用 npx（推荐）：

package/bin/aihezu.js

@@ -9,17 +9,18 @@ const commandArgs = args.slice(1);
 
 // 显示帮助信息
 function showHelp() {
-  console.log('🌐 aihezu - Claude Code 工具集');
+  console.log('🌐 aihezu - Claude / Codex 工具集');
   console.log('Powered by https://aihezu.dev\n');
   console.log('使用方法:');
   console.log('  npx aihezu <command>\n');
   console.log('可用命令:');
-  console.log('  ccinstall / install  配置 Claude Code API Key（企业可自定义域名）');
+  console.log('  ccinstall / install  配置 Claude (默认) 或 Codex，支持企业自定义域名');
   console.log('  ccclear              清理 Claude Code 缓存和配置');
   console.log('  help       显示帮助信息\n');
   console.log('示例:');
-  console.log('  npx aihezu ccinstall   # 配置 API Key');
-  console.log('  npx aihezu install --api jfz.aihezu.dev   # 企业用户自定义域名');
+  console.log('  npx aihezu ccinstall   # 配置 API Key (默认 Claude)');
+  console.log('  npx aihezu install --api your-org.aihezu.dev   # 企业用户自定义域名');
+  console.log('  npx aihezu install --provider codex   # 直接配置 Codex');
   console.log('  npx aihezu ccclear     # 清理缓存');
 }
 

package/bin/ccinstall.js

@@ -8,11 +8,39 @@ const { modifyHostsFile } = require('../lib/hosts');
 const { cleanCache } = require('../lib/cache');
 
 const homeDir = os.homedir();
-const settingsPath = path.join(homeDir, '.claude', 'settings.json');
+const claudeSettingsPath = path.join(homeDir, '.claude', 'settings.json');
 const claudeDir = path.join(homeDir, '.claude');
-const DEFAULT_API_BASE = 'https://cc.aihezu.dev/api';
+const codexDir = path.join(homeDir, '.codex');
+const codexConfigPath = path.join(codexDir, 'config.toml');
+const codexAuthPath = path.join(codexDir, 'auth.json');
+const DEFAULT_CLAUDE_API_BASE = 'https://cc.aihezu.dev/api';
+const DEFAULT_CODEX_BASE_URL = 'https://cc.aihezu.dev/openai';
+const PROVIDERS = {
+  CLAUDE: 'claude',
+  CODEX: 'codex'
+};
 const cliArgs = process.argv.slice(2);
 
+function askQuestion(rl, question) {
+  return new Promise(resolve => rl.question(question, answer => resolve(answer.trim())));
+}
+
+function getArgValue(argv, keys) {
+  for (let i = 0; i < argv.length; i++) {
+    const arg = argv[i];
+    const [key, inlineValue] = arg.split('=');
+    if (keys.includes(key)) {
+      return (inlineValue || argv[i + 1] || '').trim();
+    }
+  }
+  return '';
+}
+
+function parseProviderArg(argv) {
+  const value = getArgValue(argv, ['--provider', '--product', '--service', '--mode']);
+  return value.toLowerCase();
+}
+
 function parseApiBaseInput(argv) {
   let apiBaseInput = '';
 
@@ -34,7 +62,7 @@ function parseApiBaseInput(argv) {
 }
 
 function normalizeApiBaseUrl(input) {
-  if (!input) return DEFAULT_API_BASE;
+  if (!input) return DEFAULT_CLAUDE_API_BASE;
 
   let normalized = input;
 
@@ -46,7 +74,7 @@ function normalizeApiBaseUrl(input) {
   try {
     url = new URL(normalized);
   } catch (error) {
-    throw new Error('无效的域名或 URL，请输入类似 jfz.aihezu.dev 或 https://jfz.aihezu.dev');
+    throw new Error('无效的域名或 URL，请输入类似 your-org.aihezu.dev 或 https://your-org.aihezu.dev');
   }
 
   let pathname = url.pathname.replace(/\/+$/, '');
@@ -61,115 +89,210 @@ function normalizeApiBaseUrl(input) {
   return `${url.origin}${url.pathname}`;
 }
 
-const apiBaseInput = parseApiBaseInput(cliArgs);
-let apiBaseUrl;
+function resolveProvider(input) {
+  const value = (input || '').toLowerCase();
+  if (['2', 'codex', 'c'].includes(value)) return PROVIDERS.CODEX;
+  return PROVIDERS.CLAUDE;
+}
 
-try {
-  apiBaseUrl = normalizeApiBaseUrl(apiBaseInput);
-} catch (error) {
-  console.error(`❌ ${error.message}`);
-  process.exit(1);
+function backupFile(filePath) {
+  if (!fs.existsSync(filePath)) return null;
+  const timestamp = new Date().toISOString().replace(/[:.]/g, '-');
+  const backupPath = `${filePath}.backup-${timestamp}`;
+  fs.copyFileSync(filePath, backupPath);
+  return backupPath;
 }
 
-const usingCustomDomain = !!apiBaseInput;
+function writeClaudeSettings(apiKey, apiBaseUrl) {
+  // 确保 .claude 目录存在
+  if (!fs.existsSync(claudeDir)) {
+    console.log('📁 创建 ~/.claude 目录...');
+    fs.mkdirSync(claudeDir, { recursive: true });
+  }
 
-console.log('🔧 Claude Code API 配置工具');
-console.log('🌐 Powered by https://aihezu.dev\n');
-if (usingCustomDomain) {
-  console.log(`🏢 已启用企业独立域名: ${apiBaseUrl}\n`);
-} else {
-  console.log(`ℹ️  未指定域名，将使用默认地址: ${DEFAULT_API_BASE}`);
-  console.log('    企业用户可使用 --api 或 --api-url 指定独立域名，例如:');
-  console.log('    sudo npx aihezu install --api https://jfz.aihezu.dev\n');
-}
+  // 读取或创建 settings.json
+  let settings = {};
+  if (fs.existsSync(claudeSettingsPath)) {
+    console.log('📖 读取现有配置文件...');
+    const content = fs.readFileSync(claudeSettingsPath, 'utf8');
+    try {
+      settings = JSON.parse(content);
+    } catch (e) {
+      console.log('⚠️  现有配置文件格式错误，将创建新的配置');
+      // 备份错误的配置文件
+      const timestamp = new Date().toISOString().replace(/[:.]/g, '-');
+      const backupPath = `${claudeSettingsPath}.backup-${timestamp}`;
+      fs.writeFileSync(backupPath, content);
+      console.log(`📦 已备份原配置文件到: ${path.basename(backupPath)}`);
+    }
+  }
 
-// 创建 readline 接口
-const rl = readline.createInterface({
-  input: process.stdin,
-  output: process.stdout
-});
+  // 完全替换 env 对象，清除所有旧的环境变量配置
+  settings.env = {
+    ANTHROPIC_AUTH_TOKEN: apiKey,
+    ANTHROPIC_BASE_URL: apiBaseUrl
+  };
+
+  // 写入配置文件
+  fs.writeFileSync(claudeSettingsPath, JSON.stringify(settings, null, 2), 'utf8');
+
+  console.log('\n=== API 配置 (Claude) ===\n');
+  console.log('✅ 配置成功！');
+  console.log('📝 已更新配置文件:', claudeSettingsPath);
+  console.log('\n配置内容:');
+  console.log('  ANTHROPIC_AUTH_TOKEN:', apiKey);
+  console.log('  ANTHROPIC_BASE_URL:', apiBaseUrl);
+}
 
+function writeCodexConfig(apiKey) {
+  if (!fs.existsSync(codexDir)) {
+    console.log('📁 创建 ~/.codex 目录...');
+    fs.mkdirSync(codexDir, { recursive: true });
+  }
 
-// 询问是否需要清理缓存
-rl.question('您是从其他服务切换过来的吗？(y/n，如果是首次使用请输入 n): ', (answer) => {
-  const needClean = answer.toLowerCase() === 'y' || answer.toLowerCase() === 'yes';
+  const configBackup = backupFile(codexConfigPath);
+  if (configBackup) {
+    console.log(`📦 已备份原配置文件到: ${path.basename(configBackup)}`);
+  }
 
-  if (needClean) {
+  const configContent = [
+    'model_provider = "aihezu"',
+    'model = "gpt-5-codex"',
+    'model_reasoning_effort = "high"',
+    'disable_response_storage = true',
+    'preferred_auth_method = "apikey"',
+    '',
+    '[model_providers.aihezu]',
+    'name = "aihezu"',
+    `base_url = "${DEFAULT_CODEX_BASE_URL}"`,
+    'wire_api = "responses"',
+    'requires_openai_auth = true',
+    'env_key = "AIHEZU_OAI_KEY"',
+    ''
+  ].join('\n');
+
+  fs.writeFileSync(codexConfigPath, configContent, 'utf8');
+
+  let authData = {};
+  if (fs.existsSync(codexAuthPath)) {
     try {
-      console.log('\n=== 清理旧缓存 ===');
-      const count = cleanCache({ showHeader: false });
-      console.log(`\n✅ 缓存清理完成！（共处理 ${count} 项）`);
-      console.log('💡 配置文件已保留，即将配置新的 API Key\n');
-    } catch (error) {
-      console.error('⚠️  清理缓存时出错:', error.message);
-      console.log('继续配置流程...\n');
+      const content = fs.readFileSync(codexAuthPath, 'utf8');
+      authData = JSON.parse(content);
+    } catch (e) {
+      const backupPath = backupFile(codexAuthPath);
+      console.log('⚠️  现有 auth.json 解析失败，已备份旧文件:', backupPath ? path.basename(backupPath) : '未备份');
+      authData = {};
     }
   }
 
-  // 提示用户输入 API Key
-  rl.question('请输入您的 API Key: ', (apiKey) => {
-  if (!apiKey || apiKey.trim() === '') {
-    console.error('❌ API Key 不能为空');
-    rl.close();
-    process.exit(1);
-  }
+  authData.OPENAI_API_KEY = null;
+  authData.AIHEZU_OAI_KEY = apiKey;
+
+  fs.writeFileSync(codexAuthPath, JSON.stringify(authData, null, 2), 'utf8');
+
+  console.log('\n=== API 配置 (Codex) ===\n');
+  console.log('✅ 配置成功！');
+  console.log('📝 已更新配置文件:');
+  console.log('  -', codexConfigPath);
+  console.log('  -', codexAuthPath);
+  console.log('\n配置内容:');
+  console.log('  AIHEZU_OAI_KEY:', apiKey);
+  console.log('  base_url:', DEFAULT_CODEX_BASE_URL);
+  console.log('\n💡 已在 auth.json 中写入密钥，并在 config.toml 中指定 env_key = "AIHEZU_OAI_KEY"');
+  console.log('   如需在 shell 中直接使用，可运行: export AIHEZU_OAI_KEY="<你的密钥>"');
+}
+
+async function main() {
+  console.log('🔧 Claude / Codex API 配置工具');
+  console.log('🌐 Powered by https://aihezu.dev\n');
 
-  apiKey = apiKey.trim();
+  const rl = readline.createInterface({
+    input: process.stdin,
+    output: process.stdout
+  });
 
   try {
-    // 确保 .claude 目录存在
-    if (!fs.existsSync(claudeDir)) {
-      console.log('📁 创建 ~/.claude 目录...');
-      fs.mkdirSync(claudeDir, { recursive: true });
+    const cliProviderInput = parseProviderArg(cliArgs);
+    let provider = resolveProvider(cliProviderInput);
+
+    if (!cliProviderInput) {
+      const providerAnswer = await askQuestion(rl, '请选择服务类型 [1] Claude (默认) / [2] Codex: ');
+      provider = resolveProvider(providerAnswer);
     }
 
-    // 读取或创建 settings.json
-    let settings = {};
-    if (fs.existsSync(settingsPath)) {
-      console.log('📖 读取现有配置文件...');
-      const content = fs.readFileSync(settingsPath, 'utf8');
+    const apiBaseInput = provider === PROVIDERS.CLAUDE ? parseApiBaseInput(cliArgs) : '';
+    let apiBaseUrl = DEFAULT_CLAUDE_API_BASE;
+    if (provider === PROVIDERS.CLAUDE) {
       try {
-        settings = JSON.parse(content);
-      } catch (e) {
-        console.log('⚠️  现有配置文件格式错误，将创建新的配置');
-        // 备份错误的配置文件
-        const timestamp = new Date().toISOString().replace(/[:.]/g, '-');
-        const backupPath = `${settingsPath}.backup-${timestamp}`;
-        fs.writeFileSync(backupPath, content);
-        console.log(`📦 已备份原配置文件到: ${path.basename(backupPath)}`);
+        apiBaseUrl = normalizeApiBaseUrl(apiBaseInput);
+      } catch (error) {
+        console.error(`❌ ${error.message}`);
+        process.exit(1);
       }
     }
 
-    // 完全替换 env 对象，清除所有旧的环境变量配置
-    // 这样可以避免旧配置（如 ANTHROPIC_MODEL 等）干扰新配置
-    settings.env = {
-      ANTHROPIC_AUTH_TOKEN: apiKey,
-      ANTHROPIC_BASE_URL: apiBaseUrl
-    };
+    const usingCustomDomain = provider === PROVIDERS.CLAUDE && !!apiBaseInput;
 
-    // 写入配置文件
-    fs.writeFileSync(settingsPath, JSON.stringify(settings, null, 2), 'utf8');
+    if (provider === PROVIDERS.CLAUDE) {
+      if (usingCustomDomain) {
+        console.log(`🏢 已启用企业独立域名: ${apiBaseUrl}\n`);
+      } else {
+        console.log(`ℹ️  未指定域名，将使用默认地址: ${DEFAULT_CLAUDE_API_BASE}`);
+        console.log('    企业用户可使用 --api 或 --api-url 指定独立域名，例如:');
+        console.log('    sudo npx aihezu install --api your-org.aihezu.dev\n');
+      }
+    } else {
+      console.log(`🤖 已选择 Codex，默认网关: ${DEFAULT_CODEX_BASE_URL}\n`);
+    }
 
-    console.log('\n=== API 配置 ===\n');
-    console.log('✅ 配置成功！');
-    console.log('📝 已更新配置文件:', settingsPath);
-    console.log('\n配置内容:');
-    console.log('  ANTHROPIC_AUTH_TOKEN:', apiKey);
-    console.log('  ANTHROPIC_BASE_URL:', apiBaseUrl);
+    const needCleanAnswer = await askQuestion(
+      rl,
+      '您是从其他服务切换过来的吗？(y/n，如果是首次使用请输入 n): '
+    );
+    const needClean = ['y', 'yes'].includes(needCleanAnswer.toLowerCase());
 
-    // 修改 hosts 文件
-    console.log('\n=== 修改 hosts 文件 ===\n');
-    modifyHostsFile();
+    if (needClean) {
+      try {
+        console.log('\n=== 清理旧缓存 ===');
+        const count = cleanCache({ showHeader: false });
+        console.log(`\n✅ 缓存清理完成！（共处理 ${count} 项）`);
+        console.log('💡 配置文件已保留，即将配置新的 API Key\n');
+      } catch (error) {
+        console.error('⚠️  清理缓存时出错:', error.message);
+        console.log('继续配置流程...\n');
+      }
+    }
+
+    const apiKey = await askQuestion(rl, '请输入您的 API Key: ');
+    if (!apiKey) {
+      console.error('❌ API Key 不能为空');
+      process.exit(1);
+    }
+
+    if (provider === PROVIDERS.CLAUDE) {
+      writeClaudeSettings(apiKey.trim(), apiBaseUrl);
+    } else {
+      writeCodexConfig(apiKey.trim());
+    }
+
+    if (provider === PROVIDERS.CLAUDE) {
+      console.log('\n=== 修改 hosts 文件 ===\n');
+      modifyHostsFile();
+    }
 
     console.log('\n=== 全部完成 ===');
-    console.log('💡 您现在可以开始使用 Claude Code 了！');
+    if (provider === PROVIDERS.CLAUDE) {
+      console.log('💡 您现在可以开始使用 Claude Code 了！');
+    } else {
+      console.log('💡 您现在可以开始使用 Codex 了！');
+    }
     console.log('🌐 更多服务请访问: https://aihezu.dev');
-
   } catch (error) {
     console.error('❌ 配置失败:', error.message);
     process.exit(1);
   } finally {
     rl.close();
   }
-  });
-});
+}
+
+main();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "aihezu",
-  "version": "1.6.0",
+  "version": "1.7.0",
   "description": "Claude Code CLI 清理工具 - 快速备份和清理 Claude Code 的本地配置和缓存，同时修改 hosts 文件实现本地代理",
   "main": "bin/ccclear.js",
   "bin": {