pumpkinai-config 1.3.1 → 1.5.0

package/claude-setup.js

@@ -88,7 +88,13 @@ function installClaude() {
   if (osType === 'windows') {
     command = 'npm install -g @anthropic-ai/claude-code --registry https://registry.npmmirror.com';
   } else {
-    command = 'sudo npm install -g @anthropic-ai/claude-code --registry https://registry.npmmirror.com';
+    // 检查是否已经是 root 用户（通过 sudo 运行或直接 root）
+    const isRoot = process.getuid && process.getuid() === 0;
+    if (isRoot) {
+      command = 'npm install -g @anthropic-ai/claude-code --registry https://registry.npmmirror.com';
+    } else {
+      command = 'sudo npm install -g @anthropic-ai/claude-code --registry https://registry.npmmirror.com';
+    }
   }
 
   log(`[执行] ${command}`, 'yellow');
@@ -99,7 +105,28 @@ function installClaude() {
     return true;
   } catch (error) {
     log('[错误] @anthropic-ai/claude-code 安装失败', 'red');
-    log(`[错误] ${error.message}`, 'red');
+    log('\n[错误详情]', 'red');
+
+    // 显示错误信息
+    if (error.message) {
+      log(`错误信息: ${error.message}`, 'red');
+    }
+
+    // 显示退出码
+    if (error.status !== undefined) {
+      log(`退出码: ${error.status}`, 'red');
+    }
+
+    // 显示错误输出
+    if (error.stderr) {
+      log(`错误输出:\n${error.stderr.toString()}`, 'red');
+    }
+
+    // 显示标准输出（可能包含有用信息）
+    if (error.stdout) {
+      log(`标准输出:\n${error.stdout.toString()}`, 'yellow');
+    }
+
     log('\n[提示] 你可以手动安装:', 'yellow');
     log(`       ${command}`, 'yellow');
     log('       然后重新运行此工具', 'yellow');
@@ -107,61 +134,40 @@ function installClaude() {
   }
 }
 
-// 创建 .claude 目录和配置文件
-function createClaudeConfig(apiKey) {
-  log('\n[配置] 创建 Claude 配置文件...', 'cyan');
+// 设置系统用户环境变量
+function setupEnvironmentVariables(apiKey) {
+  log('\n[配置] 设置系统用户环境变量...', 'cyan');
 
-  // 获取实际用户的 home 目录
+  const osType = getOS();
   let homeDir;
   if (process.platform !== 'win32' && process.env.SUDO_USER) {
-    // Linux/Mac 使用 sudo 时，获取实际用户的 home 目录
     const actualUser = process.env.SUDO_USER;
     homeDir = process.platform === 'darwin' ? `/Users/${actualUser}` : `/home/${actualUser}`;
   } else {
     homeDir = os.homedir();
   }
 
+  // 创建 .claude 目录和配置文件
   const claudeDir = path.join(homeDir, '.claude');
-  const settingsPath = path.join(claudeDir, 'settings.json');
   const configPath = path.join(claudeDir, 'config.json');
-  const oldConfigPath = path.join(homeDir, '.claude.json');
-  const oldConfigBackupPath = path.join(homeDir, '.claude.json.backup');
-
+  const settingsPath = path.join(claudeDir, 'settings.json');
   try {
-    // 删除旧的配置文件（不提示用户）
-    if (fs.existsSync(oldConfigPath)) {
-      fs.unlinkSync(oldConfigPath);
-    }
-    if (fs.existsSync(oldConfigBackupPath)) {
-      fs.unlinkSync(oldConfigBackupPath);
-    }
-
-    // 创建 .claude 目录
     if (!fs.existsSync(claudeDir)) {
       fs.mkdirSync(claudeDir, { recursive: true });
-      log(`[成功] 创建目录: ${claudeDir}`, 'green');
-    } else {
-      log(`[提示] 目录已存在: ${claudeDir}`, 'yellow');
     }
+    // 创建 config.json
+    const config = { primaryApiKey: "1" };
+    fs.writeFileSync(configPath, JSON.stringify(config, null, 2), 'utf8');
 
-    // 创建 settings.json 配置
+    // 创建 settings.json
     const settings = {
       env: {
         ANTHROPIC_AUTH_TOKEN: apiKey,
-        ANTHROPIC_BASE_URL: "https://code.ppchat.vip"
+        ANTHROPIC_BASE_URL: "https://code.ppchat.vip",
+        ANTHROPIC_SMALL_FAST_MODEL: "claude-3-5-haiku-20241022"
       }
     };
-
     fs.writeFileSync(settingsPath, JSON.stringify(settings, null, 2), 'utf8');
-    log(`[成功] 创建配置文件: ${settingsPath}`, 'green');
-
-    // 创建 config.json 配置
-    const config = {
-      primaryApiKey: "1"
-    };
-
-    fs.writeFileSync(configPath, JSON.stringify(config, null, 2), 'utf8');
-    log(`[成功] 创建配置文件: ${configPath}`, 'green');
 
     // 如果是通过 sudo 运行的，修改文件所有者
     if (process.platform !== 'win32' && process.env.SUDO_USER) {
@@ -169,15 +175,122 @@ function createClaudeConfig(apiKey) {
       const group = process.platform === 'darwin' ? 'staff' : actualUser;
       try {
         execSync(`chown -R ${actualUser}:${group} ${claudeDir}`);
-        log(`[成功] 设置文件所有者为: ${actualUser}`, 'green');
-      } catch (error) {
-        log(`[警告] 无法设置文件所有者: ${error.message}`, 'yellow');
+      } catch (error) {}
+    }
+  } catch (error) {}
+
+  // 需要设置的环境变量
+  const envVars = {
+    ANTHROPIC_AUTH_TOKEN: apiKey,
+    ANTHROPIC_BASE_URL: "https://code.ppchat.vip",
+    CLAUDE_CODE_DISABLE_EXPERIMENTAL_BETAS:"1",
+    ANTHROPIC_SMALL_FAST_MODEL:"claude-3-5-haiku-20241022"
+  };
+
+  try {
+    if (osType === 'windows') {
+      // Windows: 使用 setx 设置用户环境变量（永久生效）
+      for (const [key, value] of Object.entries(envVars)) {
+        try {
+          execSync(`setx ${key} "${value}"`, { stdio: 'pipe' });
+          log(`[成功] 设置环境变量: ${key}`, 'green');
+        } catch (error) {
+          log(`[错误] 设置环境变量 ${key} 失败: ${error.message}`, 'red');
+          return false;
+        }
+      }
+      log('[提示] 请重新打开终端使环境变量生效', 'yellow');
+      return true;
+
+    } else if (osType === 'mac') {
+      // macOS: 写入 ~/.zshrc
+      const zshrcPath = path.join(homeDir, '.zshrc');
+      let zshrcContent = '';
+      if (fs.existsSync(zshrcPath)) {
+        zshrcContent = fs.readFileSync(zshrcPath, 'utf8');
+      }
+
+      // 构建环境变量行
+      let envLines = '\n# Claude Code 环境变量 (pumpkinai)\n';
+      for (const [key, value] of Object.entries(envVars)) {
+        const envLine = `export ${key}="${value}"`;
+        if (!zshrcContent.includes(`export ${key}=`)) {
+          envLines += envLine + '\n';
+          log(`[成功] 添加环境变量: ${key}`, 'green');
+        } else {
+          // 替换已存在的环境变量
+          const regex = new RegExp(`export ${key}=.*`, 'g');
+          zshrcContent = zshrcContent.replace(regex, envLine);
+          log(`[更新] 更新环境变量: ${key}`, 'yellow');
+        }
+      }
+
+      // 写入文件
+      if (envLines !== '\n# Claude Code 环境变量 (pumpkinai)\n') {
+        const newContent = zshrcContent + envLines;
+        fs.writeFileSync(zshrcPath, newContent, 'utf8');
+      } else {
+        fs.writeFileSync(zshrcPath, zshrcContent, 'utf8');
+      }
+
+      // 修改文件所有者
+      if (process.env.SUDO_USER) {
+        const actualUser = process.env.SUDO_USER;
+        try {
+          execSync(`chown ${actualUser}:staff ${zshrcPath}`);
+        } catch (error) {}
+      }
+
+      log(`[成功] 环境变量已写入: ${zshrcPath}`, 'green');
+      log('[提示] 请重新打开终端或运行: source ~/.zshrc', 'yellow');
+      return true;
+
+    } else {
+      // Linux: 写入 ~/.bashrc
+      const bashrcPath = path.join(homeDir, '.bashrc');
+      let bashrcContent = '';
+      if (fs.existsSync(bashrcPath)) {
+        bashrcContent = fs.readFileSync(bashrcPath, 'utf8');
+      }
+
+      // 构建环境变量行
+      let envLines = '\n# Claude Code 环境变量 (pumpkinai)\n';
+      for (const [key, value] of Object.entries(envVars)) {
+        const envLine = `export ${key}="${value}"`;
+        if (!bashrcContent.includes(`export ${key}=`)) {
+          envLines += envLine + '\n';
+          log(`[成功] 添加环境变量: ${key}`, 'green');
+        } else {
+          // 替换已存在的环境变量
+          const regex = new RegExp(`export ${key}=.*`, 'g');
+          bashrcContent = bashrcContent.replace(regex, envLine);
+          log(`[更新] 更新环境变量: ${key}`, 'yellow');
+        }
+      }
+
+      // 写入文件
+      if (envLines !== '\n# Claude Code 环境变量 (pumpkinai)\n') {
+        const newContent = bashrcContent + envLines;
+        fs.writeFileSync(bashrcPath, newContent, 'utf8');
+      } else {
+        fs.writeFileSync(bashrcPath, bashrcContent, 'utf8');
+      }
+
+      // 修改文件所有者
+      if (process.env.SUDO_USER) {
+        const actualUser = process.env.SUDO_USER;
+        try {
+          execSync(`chown ${actualUser}:${actualUser} ${bashrcPath}`);
+        } catch (error) {}
       }
+
+      log(`[成功] 环境变量已写入: ${bashrcPath}`, 'green');
+      log('[提示] 请重新打开终端或运行: source ~/.bashrc', 'yellow');
+      return true;
     }
 
-    return true;
   } catch (error) {
-    log(`[错误] 创建配置文件失败: ${error.message}`, 'red');
+    log(`[错误] 设置环境变量失败: ${error.message}`, 'red');
     return false;
   }
 }
@@ -200,14 +313,14 @@ async function configureEnvironment() {
     return false;
   }
 
-  return createClaudeConfig(apiKey);
+  return setupEnvironmentVariables(apiKey);
 }
 
 // 显示完成信息
 function showCompletionInfo() {
   const osType = getOS();
 
-  // 获取配置文件路径
+  // 获取 home 目录
   let homeDir;
   if (process.platform !== 'win32' && process.env.SUDO_USER) {
     const actualUser = process.env.SUDO_USER;
@@ -215,40 +328,43 @@ function showCompletionInfo() {
   } else {
     homeDir = os.homedir();
   }
-  const claudeDir = path.join(homeDir, '.claude');
-  const settingsPath = path.join(claudeDir, 'settings.json');
-  const configPath = path.join(claudeDir, 'config.json');
 
   log('\n', 'reset');
-  log('╔' + '═'.repeat(68) + '╗', 'yellow');
-  log('║' + ' '.repeat(9) + '欢迎使用 Claude 开始您的 AI 编程之旅' + ' '.repeat(9) + '║', 'yellow');
-  log('╚' + '═'.repeat(68) + '╝', 'yellow');
-
-  log('\n  ┌─ 已完成操作', 'cyan');
-  log('  │', 'cyan');
-  log('  │  ✓  安装 @anthropic-ai/claude-code', 'green');
-  log('  │  ✓  创建配置文件 settings.json', 'green');
-  log('  │  ✓  创建配置文件 config.json', 'green');
-  log('  │', 'cyan');
-  log('  └─────────────────────────────────────────────', 'cyan');
-
-  log('\n  ┌─ 配置文件路径', 'cyan');
-  log('  │', 'cyan');
-  log(`  │  settings.json(可在此处修改key和域名url)`, 'reset');
-  log(`  │  ${settingsPath}`, 'yellow');
-  log('  │', 'cyan');
-  log(`  │  config.json`, 'reset');
-  log(`  │  ${configPath}`, 'yellow');
-  log('  │', 'cyan');
-  log('  └─────────────────────────────────────────────', 'cyan');
-
-  log('\n  ┌─ 终端的使用方式', 'cyan');
-  log('  │', 'cyan');
-  log('  │  在终端输入以下命令启动 Claude Code:', 'reset');
-  log('  │', 'cyan');
-  log('  │    claude', 'yellow');
-  log('  │', 'cyan');
-  log('  └─────────────────────────────────────────────', 'cyan');
+  log('=' + '='.repeat(68) + '=', 'yellow');
+  log('  欢迎使用 Claude 开始您的 AI 编程之旅', 'yellow');
+  log('=' + '='.repeat(68) + '=', 'yellow');
+
+  log('\n[已完成操作]', 'cyan');
+  log('  * 安装 @anthropic-ai/claude-code', 'green');
+  log('  * 设置用户环境变量 ANTHROPIC_AUTH_TOKEN', 'green');
+  log('  * 设置用户环境变量 ANTHROPIC_BASE_URL', 'green');
+  log('  * 设置用户环境变量 ANTHROPIC_SMALL_FAST_MODEL', 'green');
+
+  log('\n[环境变量配置]', 'cyan');
+  if (osType === 'windows') {
+    log('  环境变量已通过 setx 设置到用户环境变量', 'reset');
+    log('  可在 系统属性 -> 高级 -> 环境变量 中查看', 'yellow');
+  } else if (osType === 'mac') {
+    log('  环境变量已写入: ~/.zshrc', 'reset');
+    log(`  ${path.join(homeDir, '.zshrc')}`, 'yellow');
+  } else {
+    log('  环境变量已写入: ~/.bashrc', 'reset');
+    log(`  ${path.join(homeDir, '.bashrc')}`, 'yellow');
+  }
+
+  log('\n[启动方式]', 'cyan');
+
+  if (osType === 'windows') {
+    log('  请重新打开终端后运行:', 'reset');
+    log('    claude', 'yellow');
+  } else if (osType === 'mac') {
+    log('  请重新打开终端后运行:', 'reset');
+    log('  claude', 'yellow');
+
+  } else {
+    log('  请重新打开终端后运行:', 'reset');
+    log('  claude', 'yellow');
+  }
 
   const claudeVersion = getClaudeVersion();
   log('\n  Claude Code 版本: v' + claudeVersion, 'reset');
@@ -260,15 +376,15 @@ function showCompletionInfo() {
 async function main() {
   console.clear();
 
-  // ASCII 艺术 Logo
+  // ASCII Logo - 水平渐变 (蓝→紫→粉)
   log('\n', 'reset');
-  log('   ██████╗██╗      █████╗ ██╗   ██╗██████╗ ███████╗', 'cyan');
-  log('  ██╔════╝██║     ██╔══██╗██║   ██║██╔══██╗██╔════╝', 'cyan');
-  log('  ██║     ██║     ███████║██║   ██║██║  ██║█████╗  ', 'cyan');
-  log('  ██║     ██║     ██╔══██║██║   ██║██║  ██║██╔══╝  ', 'cyan');
-  log('  ╚██████╗███████╗██║  ██║╚██████╔╝██████╔╝███████╗', 'cyan');
-  log('   ╚═════╝╚══════╝╚═╝  ╚═╝ ╚═════╝ ╚═════╝ ╚══════╝', 'cyan');
-  log('\n  PPChat Claude Code 一键安装配置工具', 'magenta');
+  console.log('   \x1b[38;5;39m██████╗\x1b[38;5;75m██╗      \x1b[38;5;111m█████╗ \x1b[38;5;147m██╗   ██╗\x1b[38;5;183m██████╗ \x1b[38;5;219m███████╗\x1b[0m');
+  console.log('  \x1b[38;5;39m██╔════╝\x1b[38;5;75m██║     \x1b[38;5;111m██╔══██╗\x1b[38;5;147m██║   ██║\x1b[38;5;183m██╔══██╗\x1b[38;5;219m██╔════╝\x1b[0m');
+  console.log('  \x1b[38;5;45m██║     \x1b[38;5;81m██║     \x1b[38;5;117m███████║\x1b[38;5;153m██║   ██║\x1b[38;5;189m██║  ██║\x1b[38;5;225m█████╗  \x1b[0m');
+  console.log('  \x1b[38;5;75m██║     \x1b[38;5;111m██║     \x1b[38;5;147m██╔══██║\x1b[38;5;183m██║   ██║\x1b[38;5;219m██║  ██║\x1b[38;5;225m██╔══╝  \x1b[0m');
+  console.log('  \x1b[38;5;111m╚██████╗\x1b[38;5;147m███████╗\x1b[38;5;183m██║  ██║\x1b[38;5;219m╚██████╔╝\x1b[38;5;225m██████╔╝\x1b[38;5;231m███████╗\x1b[0m');
+  console.log('   \x1b[38;5;147m╚═════╝\x1b[38;5;183m╚══════╝\x1b[38;5;219m╚═╝  ╚═╝\x1b[38;5;225m ╚═════╝ \x1b[38;5;231m╚═════╝ ╚══════╝\x1b[0m');
+  log('\n         pumpkinai Claude Code config Tool', 'magenta');
   log('  ' + '='.repeat(50), 'cyan');
   log('', 'reset');
 
@@ -279,7 +395,7 @@ async function main() {
       log('\n[提示] 安装失败，但你可以继续配置环境变量', 'yellow');
     }
 
-    // 2. 配置环境变量
+    // 2. 配置 API Key (会自动设置环境变量)
     const configSuccess = await configureEnvironment();
     if (!configSuccess) {
       process.exit(1);
@@ -303,5 +419,6 @@ if (require.main === module) {
 module.exports = {
   main,
   installClaude,
-  configureEnvironment
+  configureEnvironment,
+  setupEnvironmentVariables
 };