pumpkinai-config 1.0.20 → 1.3.0

package/codex-api.js

@@ -61,16 +61,6 @@ function getUserInput(question) {
   });
 }
 
-// 检查 @openai/codex 是否已安装
-function checkCodexInstalled() {
-  try {
-    const output = execSync('npm list -g @openai/codex --depth=0', { encoding: 'utf8' });
-    return output.includes('@openai/codex');
-  } catch (error) {
-    return false;
-  }
-}
-
 // 获取 @openai/codex 版本号
 function getCodexVersion() {
   try {
@@ -87,15 +77,7 @@ function installCodex() {
   const osType = getOS();
   let command;
 
-  log('\n[检查] 检查 @openai/codex 安装状态...', 'cyan');
-
-  // 先检查是否已安装
-  if (checkCodexInstalled()) {
-    log('[成功] @openai/codex 已安装，跳过安装步骤', 'green');
-    return true;
-  }
-
-  log('[安装] 开始安装 @openai/codex...', 'cyan');
+  log('\n[安装] 开始安装 @openai/codex...', 'cyan');
 
   if (osType === 'windows') {
     command = 'npm install -g @openai/codex --registry https://registry.npmmirror.com';

package/codex-network.js

@@ -0,0 +1,307 @@
+#!/usr/bin/env node
+
+/**
+ * Codex 网络修复工具
+ * 检测当前域名并允许切换到备用域名
+ * 只修改 config.toml 配置文件，不修改环境变量
+ */
+
+const fs = require('fs');
+const path = require('path');
+const os = require('os');
+const { execSync } = require('child_process');
+
+// 颜色输出
+const colors = {
+  red: '\x1b[31m',
+  green: '\x1b[32m',
+  yellow: '\x1b[33m',
+  blue: '\x1b[34m',
+  magenta: '\x1b[35m',
+  cyan: '\x1b[36m',
+  reset: '\x1b[0m',
+  dim: '\x1b[2m',
+  bright: '\x1b[1m'
+};
+
+function log(message, color = 'reset') {
+  console.log(`${colors[color]}${message}${colors.reset}`);
+}
+
+// 可用域名列表 (Codex 使用 /v1 后缀)
+const domainOptions = [
+  { name: '主域名', url: 'https://code.ppchat.vip/v1' },
+  { name: '备用域名1', url: 'https://code2.ppchat.vip/v1' },
+  { name: '备用域名2', url: 'https://code.pumpkinai.vip/v1' },
+  { name: '清理DNS缓存-推荐', url: 'dns_flush' }
+];
+
+// 检测操作系统
+function getOS() {
+  const platform = os.platform();
+  if (platform === 'win32') return 'windows';
+  if (platform === 'darwin') return 'mac';
+  return 'linux';
+}
+
+// 获取用户目录
+function getHomeDir() {
+  if (process.platform !== 'win32' && process.env.SUDO_USER) {
+    const actualUser = process.env.SUDO_USER;
+    return process.platform === 'darwin' ? `/Users/${actualUser}` : `/home/${actualUser}`;
+  }
+  return os.homedir();
+}
+
+// 获取当前使用的域名 (从 config.toml 读取)
+function getCurrentDomain() {
+  const homeDir = getHomeDir();
+  const configPath = path.join(homeDir, '.codex', 'config.toml');
+
+  if (fs.existsSync(configPath)) {
+    try {
+      const content = fs.readFileSync(configPath, 'utf8');
+      // 匹配 base_url = "xxx" 格式
+      const match = content.match(/base_url\s*=\s*"([^"]+)"/);
+      if (match) {
+        return match[1];
+      }
+    } catch (error) {}
+  }
+
+  return null;
+}
+
+// 更新 config.toml 中的域名
+function updateConfigToml(newUrl) {
+  const homeDir = getHomeDir();
+  const configPath = path.join(homeDir, '.codex', 'config.toml');
+
+  try {
+    if (fs.existsSync(configPath)) {
+      let content = fs.readFileSync(configPath, 'utf8');
+
+      // 替换 base_url 的值
+      const regex = /base_url\s*=\s*"[^"]+"/;
+      if (content.match(regex)) {
+        content = content.replace(regex, `base_url = "${newUrl}"`);
+        fs.writeFileSync(configPath, content, 'utf8');
+
+        // 如果是通过 sudo 运行的，修改文件所有者
+        if (process.platform !== 'win32' && process.env.SUDO_USER) {
+          const actualUser = process.env.SUDO_USER;
+          const group = process.platform === 'darwin' ? 'staff' : actualUser;
+          try {
+            execSync(`chown ${actualUser}:${group} ${configPath}`);
+          } catch (error) {}
+        }
+
+        return true;
+      } else {
+        log('[警告] config.toml 中未找到 base_url 配置', 'yellow');
+        return false;
+      }
+    } else {
+      log('[警告] config.toml 文件不存在，请先运行安装配置', 'yellow');
+      return false;
+    }
+  } catch (error) {
+    log(`[警告] 更新 config.toml 失败: ${error.message}`, 'yellow');
+  }
+  return false;
+}
+
+// 清理 DNS 缓存
+function flushDnsCache() {
+  const osType = getOS();
+  log('\n[DNS] 正在清理 DNS 缓存...', 'cyan');
+
+  try {
+    if (osType === 'windows') {
+      execSync('ipconfig /flushdns', { stdio: 'pipe' });
+      log('[成功] DNS 缓存已清理 (ipconfig /flushdns)', 'green');
+    } else if (osType === 'mac') {
+      // macOS 需要 sudo 权限
+      const isRoot = process.getuid && process.getuid() === 0;
+      if (isRoot) {
+        execSync('dscacheutil -flushcache && killall -HUP mDNSResponder', { stdio: 'pipe' });
+      } else {
+        execSync('sudo dscacheutil -flushcache && sudo killall -HUP mDNSResponder', { stdio: 'pipe' });
+      }
+      log('[成功] DNS 缓存已清理 (dscacheutil -flushcache)', 'green');
+    } else {
+      // Linux - 不同发行版命令不同，尝试常见的
+      const isRoot = process.getuid && process.getuid() === 0;
+      try {
+        if (isRoot) {
+          execSync('systemd-resolve --flush-caches 2>/dev/null || resolvectl flush-caches 2>/dev/null || true', { stdio: 'pipe' });
+        } else {
+          execSync('sudo systemd-resolve --flush-caches 2>/dev/null || sudo resolvectl flush-caches 2>/dev/null || true', { stdio: 'pipe' });
+        }
+        log('[成功] DNS 缓存已清理', 'green');
+      } catch (e) {
+        log('[提示] DNS 缓存清理命令不适用于此系统，已跳过', 'yellow');
+      }
+    }
+    return true;
+  } catch (error) {
+    log(`[警告] DNS 缓存清理失败: ${error.message}`, 'yellow');
+    return false;
+  }
+}
+
+// 切换域名
+function switchDomain(newUrl) {
+  log(`\n[切换] 正在切换到: ${newUrl}`, 'cyan');
+
+  // 更新 config.toml
+  const configUpdated = updateConfigToml(newUrl);
+  if (configUpdated) {
+    log('[成功] config.toml 已更新', 'green');
+    log('\n[完成] 域名切换成功！', 'green');
+    log('[提示] 新域名将在下次启动 Codex 时生效', 'yellow');
+  } else {
+    log('\n[警告] 配置文件更新失败，请先运行安装配置', 'yellow');
+  }
+}
+
+let selectedIndex = 0;
+
+// 渲染域名选择菜单
+function renderMenu() {
+  console.clear();
+  log('\n', 'reset');
+  log('  ' + '='.repeat(54), 'cyan');
+  log('              Codex 网络修复 - 域名切换', 'magenta');
+  log('  ' + '='.repeat(54), 'cyan');
+  log('', 'reset');
+
+  const currentDomain = getCurrentDomain();
+
+  log('  请选择要使用的域名 (↑↓选择, Enter确认, ESC返回):\n', 'cyan');
+
+  domainOptions.forEach((option, index) => {
+    const isSelected = index === selectedIndex;
+    const prefix = isSelected ? '  ▶ ' : '    ';
+    const nameColor = isSelected ? '\x1b[1m\x1b[33m' : '\x1b[0m';
+    const numLabel = `${index + 1}. `;
+
+    // 检查是否是当前使用的域名
+    const isCurrent = option.url && currentDomain === option.url;
+    const currentIndicator = isCurrent ? ' \x1b[32m← 当前使用\x1b[0m' : '';
+
+    console.log(`${prefix}${nameColor}${numLabel}${option.name}${currentIndicator}\x1b[0m`);
+    if (option.url && option.url !== 'dns_flush') {
+      const descColor = isSelected ? '\x1b[36m' : '\x1b[2m';
+      console.log(`      ${descColor}${option.url}\x1b[0m`);
+    }
+    console.log('');
+  });
+
+  log('  ' + '='.repeat(54), 'cyan');
+
+  if (currentDomain) {
+    log(`\n  当前域名: ${currentDomain}`, 'dim');
+  } else {
+    log('\n  当前域名: 未配置', 'dim');
+  }
+}
+
+// 主函数
+async function main() {
+  // 启用原始模式以捕获按键
+  if (process.stdin.isTTY) {
+    process.stdin.setRawMode(true);
+  }
+  process.stdin.resume();
+  process.stdin.setEncoding('utf8');
+
+  renderMenu();
+
+  return new Promise((resolve) => {
+    const handleKeyPress = (key) => {
+      // Ctrl+C 退出
+      if (key === '\u0003') {
+        console.clear();
+        process.exit(0);
+      }
+
+      // ESC 返回
+      if (key === '\u001b') {
+        if (process.stdin.isTTY) {
+          process.stdin.setRawMode(false);
+        }
+        process.stdin.removeAllListeners('data');
+        resolve(false);
+        return;
+      }
+
+      // 上箭头
+      if (key === '\u001b[A' || key === 'k') {
+        selectedIndex = selectedIndex > 0 ? selectedIndex - 1 : domainOptions.length - 1;
+        renderMenu();
+      }
+
+      // 下箭头
+      if (key === '\u001b[B' || key === 'j') {
+        selectedIndex = selectedIndex < domainOptions.length - 1 ? selectedIndex + 1 : 0;
+        renderMenu();
+      }
+
+      // Enter 确认
+      if (key === '\r' || key === '\n') {
+        const selected = domainOptions[selectedIndex];
+
+        // 切换域名
+        if (process.stdin.isTTY) {
+          process.stdin.setRawMode(false);
+        }
+        process.stdin.removeAllListeners('data');
+
+        // 根据选项执行不同操作
+        if (selected.url === 'dns_flush') {
+          flushDnsCache();
+        } else {
+          switchDomain(selected.url);
+        }
+
+        // 等待用户按键返回
+        log('\n按任意键返回菜单...', 'dim');
+
+        if (process.stdin.isTTY) {
+          process.stdin.setRawMode(true);
+        }
+        process.stdin.resume();
+        process.stdin.once('data', () => {
+          if (process.stdin.isTTY) {
+            process.stdin.setRawMode(false);
+          }
+          selectedIndex = 0;
+          resolve(false);
+        });
+        return;
+      }
+
+      // 数字键快捷选择
+      const num = parseInt(key);
+      if (num >= 1 && num <= domainOptions.length) {
+        selectedIndex = num - 1;
+        renderMenu();
+      }
+    };
+
+    process.stdin.on('data', handleKeyPress);
+  });
+}
+
+// 如果直接运行此脚本
+if (require.main === module) {
+  main().then(() => {
+    process.exit(0);
+  }).catch(error => {
+    log(`\n[错误] ${error.message}`, 'red');
+    process.exit(1);
+  });
+}
+
+module.exports = { main, getCurrentDomain, switchDomain };

package/codex-setup.js

@@ -61,16 +61,6 @@ function getUserInput(question) {
   });
 }
 
-// 检查 @openai/codex 是否已安装
-function checkCodexInstalled() {
-  try {
-    const output = execSync('npm list -g @openai/codex --depth=0', { encoding: 'utf8' });
-    return output.includes('@openai/codex');
-  } catch (error) {
-    return false;
-  }
-}
-
 // 获取 @openai/codex 版本号
 function getCodexVersion() {
   try {
@@ -87,15 +77,7 @@ function installCodex() {
   const osType = getOS();
   let command;
 
-  log('\n[检查] 检查 @openai/codex 安装状态...', 'cyan');
-
-  // 先检查是否已安装
-  if (checkCodexInstalled()) {
-    log('[成功] @openai/codex 已安装，跳过安装步骤', 'green');
-    return true;
-  }
-
-  log('[安装] 开始安装 @openai/codex...', 'cyan');
+  log('\n[安装] 开始安装 @openai/codex...', 'cyan');
 
   if (osType === 'windows') {
     command = 'npm install -g @openai/codex --registry https://registry.npmmirror.com';