bhg-helper 1.0.0

package/cli/cli.js

@@ -0,0 +1,454 @@
+#!/usr/bin/env node
+/**
+ * BHG-helper CLI — 命令行交互菜单
+ * 一级菜单：模型选择 / 工具配置 / 语言配置 / BHG-helper UI / 退出
+ */
+
+import { select, confirm, input } from '@inquirer/prompts'
+import { execSync } from 'node:child_process'
+import os from 'node:os'
+import path from 'node:path'
+import fs from 'node:fs'
+
+// ── 色彩 ──────────────────────────────────────────────────────
+const C = { r: '\x1b[0m', bold: '\x1b[1m', cyan: '\x1b[36m', green: '\x1b[32m', yellow: '\x1b[33m', dim: '\x1b[2m', red: '\x1b[31m' }
+const theme = { prefix: `${C.cyan}?${C.r}` }
+
+// ── 常量 ──────────────────────────────────────────────────────
+const RELAY_PORT = 8787
+const UI_URL = 'http://localhost:5173'
+const BHG_HELPER_DIR = path.join(os.homedir(), '.bhg-helper')
+const RELAY_CONFIG_FILE = path.join(BHG_HELPER_DIR, 'relay.config.json')
+const CLAUDE_SETTINGS_FILE = path.join(os.homedir(), '.claude', 'settings.json')
+
+// ── 工具定义 ──────────────────────────────────────────────────
+const TOOLS = {
+  claude: {
+    name: 'Claude Code',
+    cmd: 'claude',
+    desc: 'Anthropic 官方 CLI 编程助手',
+    install: 'npm install -g @anthropic-ai/claude-code',
+  },
+  gemini: {
+    name: 'Gemini CLI',
+    cmd: 'gemini',
+    desc: 'Google 官方 CLI 编程助手',
+    install: 'npm install -g @anthropic-ai/gemini-cli',
+  },
+  opencode: {
+    name: 'OpenCode',
+    cmd: 'opencode',
+    desc: '开源 AI 编程助手',
+    install: 'npm install -g opencode-ai',
+    configFile: (() => {
+      if (process.platform === 'win32') return path.join(os.homedir(), 'AppData', 'Roaming', 'opencode', 'opencode.json')
+      return path.join(os.homedir(), '.config', 'opencode', 'opencode.json')
+    })(),
+  },
+  cursor: {
+    name: 'Cursor',
+    cmd: null,
+    desc: 'AI 代码编辑器',
+    install: '从 https://cursor.com 下载',
+  },
+  continue: {
+    name: 'Continue (VS Code)',
+    cmd: null,
+    desc: 'VS Code / JetBrains AI 扩展',
+    install: 'VS Code 扩展搜索 "Continue"',
+    configFile: path.join(os.homedir(), '.continue', 'config.json'),
+  },
+  trae: {
+    name: 'Trae',
+    cmd: null,
+    desc: '字节跳动 AI 代码编辑器',
+    install: '从 https://trae.ai 下载',
+  },
+  codex: {
+    name: 'OpenAI Codex CLI',
+    cmd: 'codex',
+    desc: 'OpenAI 官方 CLI 编程助手',
+    install: 'npm install -g @openai/codex',
+  },
+}
+
+// ── 读取 / 写入 relay config ─────────────────────────────────
+function loadConfig() {
+  try {
+    if (fs.existsSync(RELAY_CONFIG_FILE)) {
+      return JSON.parse(fs.readFileSync(RELAY_CONFIG_FILE, 'utf-8'))
+    }
+  } catch { /* */ }
+  return null
+}
+
+function saveConfig(cfg) {
+  fs.mkdirSync(BHG_HELPER_DIR, { recursive: true })
+  fs.writeFileSync(RELAY_CONFIG_FILE, JSON.stringify(cfg, null, 2) + '\n', 'utf-8')
+}
+
+function hasApiConfig() {
+  const cfg = loadConfig()
+  return !!(cfg && cfg.deepseekApiKey && cfg.deepseekApiKey.length > 0)
+}
+
+function getCurrentModel() {
+  const cfg = loadConfig()
+  if (!cfg) return 'deepseek-v4-pro'
+  // 优先读取当前激活的模型
+  return cfg.currentModel || 'deepseek-v4-pro'
+}
+
+// ── 辅助 ──────────────────────────────────────────────────────
+function checkInstalled(cmd) {
+  if (!cmd) return null
+  try {
+    execSync(`where ${cmd}`, { stdio: 'pipe' })
+    return true
+  } catch { return false }
+}
+
+function openBrowser(url) {
+  try {
+    execSync(`start "" "${url}"`, { shell: true })
+  } catch { /* */ }
+}
+
+// ── 打印头部 ──────────────────────────────────────────────────
+function printHeader() {
+  console.clear()
+  console.log(`${C.bold}${C.cyan}  ◆ BHG-helper${C.r}`)
+  const apiOk = hasApiConfig()
+  const model = getCurrentModel()
+  console.log(`  ${C.dim}API${C.r}   ${apiOk ? `${C.green}已配置${C.r}` : `${C.red}未配置${C.r}`}`)
+  console.log(`  ${C.dim}模型${C.r}   ${C.cyan}${model}${C.r}`)
+  console.log(`  ${C.dim}UI${C.r}    ${UI_URL}\n`)
+}
+
+// ── 一级菜单 ──────────────────────────────────────────────────
+async function mainMenu() {
+  const choice = await select({
+    message: '请选择操作',
+    choices: [
+      { name: '模型选择', value: 'model', description: '切换模型 / 输入 API Key' },
+      { name: '工具配置', value: 'tools', description: '配置各 AI 编程工具' },
+      { name: '语言配置', value: 'lang', description: '界面语言设置' },
+      { name: `${C.yellow}BHG-helper 网页界面${C.r}`, value: 'ui', description: '打开网页控制台' },
+      { name: `${C.dim}──────────${C.r}`, value: '---', disabled: true },
+      { name: '退出', value: 'exit' },
+    ],
+    theme,
+  })
+  return choice
+}
+
+// ── 模型选择子菜单 ───────────────────────────────────────────
+async function modelMenu() {
+  const cfg = loadConfig()
+  const currentModel = getCurrentModel()
+
+  const choice = await select({
+    message: '模型选择',
+    choices: [
+      { name: `切换模型 ${C.dim}（当前：${currentModel}）${C.r}`, value: 'switch', description: '选择要使用的模型' },
+      { name: `输入 API ${C.dim}（${hasApiConfig() ? C.green + '已配置' + C.r : C.red + '未配置' + C.r}）${C.r}`, value: 'apikey', description: '设置 DeepSeek API Key' },
+      { name: `${C.dim}──────────${C.r}`, value: '---', disabled: true },
+      { name: '返回', value: 'back' },
+      { name: '退出', value: 'exit' },
+    ],
+    theme,
+  })
+
+  if (choice === 'exit') return 'exit'
+  if (choice === 'back') return 'back'
+
+  if (choice === 'switch') {
+    // 从 relay config 读取模型列表
+    const models = []
+    if (cfg && cfg.modelMap) {
+      const seen = new Set()
+      for (const [from, to] of Object.entries(cfg.modelMap)) {
+        if (!seen.has(to)) {
+          seen.add(to)
+          models.push({ from, to })
+        }
+      }
+    } else {
+      // 默认列表
+      models.push(
+        { from: 'deepseek-v4-pro', to: 'deepseek-v4-pro' },
+        { from: 'deepseek-v4-flash', to: 'deepseek-v4-flash' },
+        { from: 'deepseek-chat', to: 'deepseek-v4-pro' },
+        { from: 'deepseek-reasoner', to: 'deepseek-v4-pro' },
+      )
+    }
+
+    const modelChoice = await select({
+      message: '选择模型',
+      choices: [
+        ...models.map(m => ({
+          name: m.from === currentModel
+            ? `${C.green}● ${m.from}${C.r} ${C.dim}→ ${m.to}${C.r}`
+            : `  ${m.from} ${C.dim}→ ${m.to}${C.r}`,
+          value: m.from,
+        })),
+        { name: `${C.dim}──────────${C.r}`, value: '---2', disabled: true },
+        { name: '返回', value: 'back' },
+      ],
+      theme,
+    })
+
+    if (modelChoice && modelChoice !== 'back') {
+      const next = { ...cfg, currentModel: modelChoice }
+      saveConfig(next)
+      console.log(`\n  ${C.green}✓ 已切换模型：${modelChoice}${C.r}\n`)
+      await confirm({ message: '按回车继续...', theme })
+    }
+    return 'back'
+  }
+
+  if (choice === 'apikey') {
+    const currentKey = cfg?.deepseekApiKey || ''
+    const masked = currentKey.length > 8
+      ? currentKey.slice(0, 4) + '****' + currentKey.slice(-4)
+      : currentKey || '(空)'
+
+    console.log(`\n  当前 API Key：${C.dim}${masked}${C.r}`)
+    console.log(`  ${C.dim}前往 https://platform.deepseek.com 申请${C.r}\n`)
+
+    const newKey = await input({
+      message: '输入 DeepSeek API Key（sk-...）',
+      theme,
+      validate: (v) => v.length > 0 ? true : 'API Key 不能为空',
+    })
+
+    const next = { ...(cfg || {}), deepseekApiKey: newKey, port: cfg?.port || RELAY_PORT, deepseekBaseUrl: cfg?.deepseekBaseUrl || 'https://api.deepseek.com', modelMap: cfg?.modelMap || { 'deepseek-v4-pro': 'deepseek-v4-pro', 'deepseek-chat': 'deepseek-v4-pro' }, spoofProvider: cfg?.spoofProvider || 'deepseek', verbose: cfg?.verbose ?? true }
+    saveConfig(next)
+    console.log(`\n  ${C.green}✓ API Key 已保存${C.r}\n`)
+    await confirm({ message: '按回车继续...', theme })
+    return 'back'
+  }
+
+  return 'back'
+}
+
+// ── 工具配置子菜单 ───────────────────────────────────────────
+async function toolsMenu() {
+  const choice = await select({
+    message: '工具配置 — 选择要配置的 AI 编程工具',
+    choices: [
+      { name: `${C.bold}Claude Code${C.r}`, value: 'claude', description: 'Anthropic 官方 CLI 编程助手' },
+      { name: `${C.bold}Gemini CLI${C.r}`, value: 'gemini', description: 'Google 官方 CLI 编程助手' },
+      { name: 'OpenCode', value: 'opencode', description: '开源 AI 编程助手' },
+      { name: 'OpenAI Codex CLI', value: 'codex', description: 'OpenAI 官方 CLI 编程助手' },
+      { name: 'Cursor', value: 'cursor', description: 'AI 代码编辑器' },
+      { name: 'Continue (VS Code)', value: 'continue', description: 'VS Code / JetBrains 扩展' },
+      { name: 'Trae', value: 'trae', description: '字节跳动 AI 编辑器' },
+      { name: `${C.dim}──────────${C.r}`, value: '---', disabled: true },
+      { name: '返回', value: 'back' },
+      { name: '退出', value: 'exit' },
+    ],
+    theme,
+  })
+
+  if (choice === 'exit') return 'exit'
+  if (choice === 'back') return 'back'
+
+  if (choice === 'ui') {
+    openBrowser(UI_URL)
+    console.log(`\n  ${C.green}✓ 浏览器已打开 → ${UI_URL}${C.r}\n`)
+    await confirm({ message: '按回车继续...', theme })
+    return 'back'
+  }
+
+  return await toolDetailMenu(choice)
+}
+
+// ── 单个工具详情 ──────────────────────────────────────────────
+async function toolDetailMenu(toolKey) {
+  const tool = TOOLS[toolKey]
+
+  // Cursor / Continue / Trae — 无 CLI，只引导去 UI
+  if (!tool.cmd) {
+    console.log(`\n  ${C.bold}${tool.name}${C.r} — ${tool.desc}`)
+    console.log(`  ${C.dim}安装方式：${tool.install}${C.r}`)
+    console.log(`  ${C.yellow}${tool.name} 需要在 BHG-helper 网页界面配置${C.r}\n`)
+
+    const openUi = await confirm({ message: '打开 BHG-helper 网页界面？', default: true, theme })
+    if (openUi) {
+      openBrowser(UI_URL)
+      console.log(`  ${C.green}✓ 浏览器已打开${C.r}`)
+    }
+    console.log()
+    await confirm({ message: '按回车继续...', theme })
+    return 'back'
+  }
+
+  // 有 CLI 的工具
+  const installed = checkInstalled(tool.cmd)
+
+  const choices = []
+  if (!installed) {
+    choices.push({ name: `安装 ${tool.name}`, value: 'install', description: tool.install })
+  } else {
+    choices.push({ name: `${C.green}启动 ${tool.name}${C.r}`, value: 'launch', description: '启动工具' })
+    choices.push({ name: `${C.yellow}重装 ${tool.name}${C.r}`, value: 'reinstall', description: tool.install })
+  }
+
+  choices.push(
+    { name: `${C.dim}──────────${C.r}`, value: '---', disabled: true },
+    { name: '返回', value: 'back' },
+    { name: '退出', value: 'exit' },
+  )
+
+  const action = await select({
+    message: `${tool.name} — ${tool.desc}${installed ? ` ${C.green}(已安装)${C.r}` : ` ${C.red}(未安装)${C.r}`}`,
+    choices,
+    theme,
+  })
+
+  if (action === 'exit') return 'exit'
+  if (action === 'back') return 'back'
+
+  if (action === 'install' || action === 'reinstall') {
+    console.log(`\n  正在执行：${tool.install}`)
+    try {
+      execSync(tool.install, { stdio: 'inherit', shell: true })
+      console.log(`\n  ${C.green}✓ ${tool.name} 安装完成${C.r}\n`)
+    } catch {
+      console.log(`\n  ${C.red}安装失败，请手动执行：${tool.install}${C.r}\n`)
+    }
+    await confirm({ message: '按回车继续...', theme })
+    return 'back'
+  }
+
+  if (action === 'launch') {
+    // 启动前检查 API 配置
+    if (!hasApiConfig()) {
+      console.log(`\n  ${C.red}API 未配置！请先在「模型选择」→「输入 API」中配置。${C.r}\n`)
+      await confirm({ message: '按回车继续...', theme })
+      return 'back'
+    }
+
+    // 写入 ~/.claude/settings.json（Claude Code 专用）
+    if (toolKey === 'claude') {
+      const currentModel = getCurrentModel()
+      const dir = path.dirname(CLAUDE_SETTINGS_FILE)
+      fs.mkdirSync(dir, { recursive: true })
+      const existing = fs.existsSync(CLAUDE_SETTINGS_FILE)
+        ? JSON.parse(fs.readFileSync(CLAUDE_SETTINGS_FILE, 'utf-8'))
+        : {}
+      fs.writeFileSync(CLAUDE_SETTINGS_FILE, JSON.stringify({
+        ...existing,
+        hasCompletedOnboarding: true,
+        env: {
+          ...(existing.env || {}),
+          CLAUDE_CODE_DISABLE_NONESSENTIAL_TRAFFIC: '1',
+          ANTHROPIC_BASE_URL: `http://127.0.0.1:${RELAY_PORT}`,
+          ANTHROPIC_AUTH_TOKEN: 'bhg-helper',
+          ANTHROPIC_DEFAULT_OPUS_MODEL: currentModel,
+          ANTHROPIC_DEFAULT_SONNET_MODEL: currentModel,
+          ANTHROPIC_DEFAULT_HAIKU_MODEL: currentModel,
+          ANTHROPIC_MODEL: currentModel,
+        },
+      }, null, 2) + '\n')
+      console.log(`  ${C.green}✓ Claude Code 配置已写入${C.r}`)
+    }
+
+    // 写入 opencode 配置
+    if (toolKey === 'opencode') {
+      const file = TOOLS.opencode.configFile
+      const dir = path.dirname(file)
+      fs.mkdirSync(dir, { recursive: true })
+      const existing = fs.existsSync(file) ? JSON.parse(fs.readFileSync(file, 'utf-8')) : {}
+      const currentModel = getCurrentModel()
+      fs.writeFileSync(file, JSON.stringify({
+        ...existing,
+        model: { provider: 'openai', name: currentModel },
+        providers: [
+          ...(existing.providers || []).filter(p => p.name !== 'bhg-deepseek'),
+          { name: 'bhg-deepseek', type: 'openai', baseURL: `http://127.0.0.1:${RELAY_PORT}/v1`, apiKey: 'bhg-helper' },
+        ],
+      }, null, 2) + '\n')
+      console.log(`  ${C.green}✓ OpenCode 配置已写入${C.r}`)
+    }
+
+    console.log(`\n  正在启动 ${tool.cmd}...`)
+    try {
+      execSync(tool.cmd, { stdio: 'inherit', shell: true })
+    } catch { /* 退出 */ }
+    return 'back'
+  }
+
+  return 'back'
+}
+
+// ── 语言配置 ──────────────────────────────────────────────────
+async function langMenu() {
+  const choice = await select({
+    message: '语言配置',
+    choices: [
+      { name: '中文', value: 'zh' },
+      { name: 'English', value: 'en' },
+      { name: `${C.dim}──────────${C.r}`, value: '---', disabled: true },
+      { name: '返回', value: 'back' },
+    ],
+    theme,
+  })
+
+  if (choice === 'exit') return 'exit'
+  if (choice === 'back') return 'back'
+
+  console.log(`\n  ${C.green}✓ 语言已设为：${choice === 'zh' ? '中文' : 'English'}${C.r}\n`)
+  await confirm({ message: '按回车继续...', theme })
+  return 'back'
+}
+
+// ── 主循环 ────────────────────────────────────────────────────
+async function main() {
+  let running = true
+  while (running) {
+    printHeader()
+    const choice = await mainMenu()
+
+    switch (choice) {
+      case 'exit':
+        console.log(`\n  ${C.dim}再见。${C.r}\n`)
+        running = false
+        break
+
+      case 'model': {
+        const result = await modelMenu()
+        if (result === 'exit') { console.log(`\n  ${C.dim}再见。${C.r}\n`); running = false }
+        break
+      }
+
+      case 'tools': {
+        const result = await toolsMenu()
+        if (result === 'exit') { console.log(`\n  ${C.dim}再见。${C.r}\n`); running = false }
+        break
+      }
+
+      case 'lang': {
+        const result = await langMenu()
+        if (result === 'exit') { console.log(`\n  ${C.dim}再见。${C.r}\n`); running = false }
+        break
+      }
+
+      case 'ui':
+        openBrowser(UI_URL)
+        console.log(`\n  ${C.green}✓ 浏览器已打开 → ${UI_URL}${C.r}\n`)
+        await confirm({ message: '按回车继续...', theme })
+        break
+    }
+  }
+}
+
+main().catch((err) => {
+  if (err.name === 'ExitPromptError') {
+    console.log(`\n  ${C.dim}再见。${C.r}\n`)
+    process.exit(0)
+  }
+  console.error('Error:', err.message)
+  process.exit(1)
+})