pumpkinai-config 1.3.1 → 1.5.0

package/claude-api.js

@@ -0,0 +1,424 @@
+#!/usr/bin/env node
+
+/**
+ * PPChat Claude Code 一键安装配置工具 (API 版本)
+ * 自动安装 @anthropic-ai/claude-code 并配置环境变量
+ */
+
+const fs = require('fs');
+const path = require('path');
+const os = require('os');
+const { execSync } = require('child_process');
+const readline = require('readline');
+
+// 颜色输出
+const colors = {
+  red: '\x1b[31m',
+  green: '\x1b[32m',
+  yellow: '\x1b[33m',
+  blue: '\x1b[34m',
+  magenta: '\x1b[35m',
+  cyan: '\x1b[36m',
+  reset: '\x1b[0m'
+};
+
+function log(message, color = 'reset') {
+  console.log(`${colors[color]}${message}${colors.reset}`);
+}
+
+// 检测操作系统
+function getOS() {
+  const platform = os.platform();
+  if (platform === 'win32') return 'windows';
+  if (platform === 'darwin') return 'mac';
+  return 'linux';
+}
+
+// 获取用户输入
+function getUserInput(question) {
+  const rl = readline.createInterface({
+    input: process.stdin,
+    output: process.stdout
+  });
+
+  return new Promise((resolve) => {
+    rl.question(question, (answer) => {
+      rl.close();
+      resolve(answer.trim());
+    });
+  });
+}
+
+// 检查 @anthropic-ai/claude-code 是否已安装
+function checkClaudeInstalled() {
+  try {
+    const output = execSync('npm list -g @anthropic-ai/claude-code --depth=0', { encoding: 'utf8' });
+    return output.includes('@anthropic-ai/claude-code');
+  } catch (error) {
+    return false;
+  }
+}
+
+// 获取 @anthropic-ai/claude-code 版本号
+function getClaudeVersion() {
+  try {
+    const output = execSync('npm list -g @anthropic-ai/claude-code --depth=0', { encoding: 'utf8' });
+    const match = output.match(/@anthropic-ai\/claude-code@([\d.]+)/);
+    return match ? match[1] : '未知';
+  } catch (error) {
+    return '未知';
+  }
+}
+
+// 安装 @anthropic-ai/claude-code
+function installClaude() {
+  const osType = getOS();
+  let command;
+
+  log('\n[检查] 检查 @anthropic-ai/claude-code 安装状态...', 'cyan');
+
+  // 先检查是否已安装
+  if (checkClaudeInstalled()) {
+    log('[成功] @anthropic-ai/claude-code 已安装,跳过安装步骤', 'green');
+    return true;
+  }
+
+  log('[安装] 开始安装 @anthropic-ai/claude-code...', 'cyan');
+
+  if (osType === 'windows') {
+    command = 'npm install -g @anthropic-ai/claude-code --registry https://registry.npmmirror.com';
+  } else {
+    // 检查是否已经是 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');
+
+  try {
+    execSync(command, { stdio: 'inherit' });
+    log('[成功] @anthropic-ai/claude-code 安装成功', 'green');
+    return true;
+  } catch (error) {
+    log('[错误] @anthropic-ai/claude-code 安装失败', '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');
+    return false;
+  }
+}
+
+// 设置系统用户环境变量
+function setupEnvironmentVariables(apiKey) {
+  log('\n[配置] 设置系统用户环境变量...', 'cyan');
+
+  const osType = getOS();
+  let homeDir;
+  if (process.platform !== 'win32' && process.env.SUDO_USER) {
+    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 configPath = path.join(claudeDir, 'config.json');
+  const settingsPath = path.join(claudeDir, 'settings.json');
+  try {
+    if (!fs.existsSync(claudeDir)) {
+      fs.mkdirSync(claudeDir, { recursive: true });
+    }
+    // 创建 config.json
+    const config = { primaryApiKey: "1" };
+    fs.writeFileSync(configPath, JSON.stringify(config, null, 2), 'utf8');
+
+    // 创建 settings.json
+    const settings = {
+      env: {
+        ANTHROPIC_AUTH_TOKEN: apiKey,
+        ANTHROPIC_BASE_URL: "https://new.aicode.us.com",
+        ANTHROPIC_SMALL_FAST_MODEL: "claude-3-5-haiku-20241022"
+      }
+    };
+    fs.writeFileSync(settingsPath, JSON.stringify(settings, null, 2), '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 -R ${actualUser}:${group} ${claudeDir}`);
+      } catch (error) {}
+    }
+  } catch (error) {}
+
+  // 需要设置的环境变量
+  const envVars = {
+    ANTHROPIC_AUTH_TOKEN: apiKey,
+    ANTHROPIC_BASE_URL: "https://new.aicode.us.com",
+    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;
+    }
+
+  } catch (error) {
+    log(`[错误] 设置环境变量失败: ${error.message}`, 'red');
+    return false;
+  }
+}
+
+// 配置 Claude
+async function configureEnvironment() {
+  log('\n[配置] 配置 API Key...', 'cyan');
+
+  // 显示粘贴提示
+  if (process.platform === 'win32') {
+    log('[提示] Windows 粘贴方式:', 'yellow');
+    log('       • CMD/PowerShell: 鼠标右键 或 Shift+Insert', 'yellow');
+
+  }
+
+  const apiKey = await getUserInput('请输入你的 Claude API Key (sk-xxx): ');
+
+  if (!apiKey || !apiKey.startsWith('sk-')) {
+    log('[错误] API Key 格式不正确,应该以 sk- 开头', 'red');
+    return false;
+  }
+
+  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;
+    homeDir = process.platform === 'darwin' ? `/Users/${actualUser}` : `/home/${actualUser}`;
+  } else {
+    homeDir = os.homedir();
+  }
+
+  log('\n', 'reset');
+  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');
+  log('  支持系统: Windows 10/11  •  macOS  •  Linux', 'reset');
+  log('\n', 'reset');
+}
+
+// 主函数
+async function main() {
+  console.clear();
+
+  // ASCII Logo - 水平渐变 (蓝→紫→粉)
+  log('\n', 'reset');
+  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 (API)', 'magenta');
+  log('  ' + '='.repeat(50), 'cyan');
+  log('', 'reset');
+
+  try {
+    // 1. 安装 @anthropic-ai/claude-code
+    const installSuccess = installClaude();
+    if (!installSuccess) {
+      log('\n[提示] 安装失败,但你可以继续配置环境变量', 'yellow');
+    }
+
+    // 2. 配置 API Key
+    const configSuccess = await configureEnvironment();
+    if (!configSuccess) {
+      process.exit(1);
+    }
+
+    // 3. 显示完成信息 (不再设置系统环境变量)
+    showCompletionInfo();
+
+  } catch (error) {
+    log(`\n[错误] 安装过程中出现错误: ${error.message}`, 'red');
+    process.exit(1);
+  }
+}
+
+// 如果直接运行此脚本
+if (require.main === module) {
+  main();
+}
+
+// 导出函数供其他模块使用
+module.exports = {
+  main,
+  installClaude,
+  configureEnvironment,
+  setupEnvironmentVariables
+};