@raolin2025/claude-code-node 1.0.0

package/src/core/config.js

@@ -0,0 +1,131 @@
+/**
+ * 配置管理
+ * 对应原版: src/query/config.ts + src/utils/config.ts
+ */
+import { readFile, writeFile, mkdir } from 'fs/promises'
+import { resolve, join } from 'path'
+import { existsSync } from 'fs'
+import { homedir } from 'os'
+
+const PROJECT_CONFIG_FILE = '.claude-code/config.json'
+const USER_CONFIG_FILE = join(homedir(), '.claude-code/config.json')
+
+/**
+ * 默认配置
+ */
+const DEFAULTS = {
+  model: 'deepseek-chat',
+  apiBase: 'https://api.deepseek.com/v1',
+  maxTurns: 100,
+  maxBudgetTokens: 1_000_000,
+  permissionMode: 'ask',
+  verbose: false,
+  apiKey: '',
+  sessionsDir: '.claude-code/sessions',
+  tools: {
+    bash: { timeout: 120, allowed: true },
+    fileRead: { maxLines: 2000, maxSizeKB: 256 },
+    webFetch: { timeout: 30, maxChars: 100000 },
+  },
+  mcp: {
+    servers: {},
+  },
+}
+
+export class Config {
+  constructor() {
+    this.data = { ...DEFAULTS }
+    this._projectPath = null
+    this._userPath = USER_CONFIG_FILE
+  }
+
+  /** 从项目目录加载配置 */
+  async loadFromProject(projectDir) {
+    this._projectPath = join(projectDir, PROJECT_CONFIG_FILE)
+    await this._load(this._projectPath)
+  }
+
+  /** 从用户目录加载配置 */
+  async loadFromUser() {
+    await this._load(this._userPath)
+  }
+
+  /** 完整加载流程：用户级 → 项目级（项目级覆盖用户级） */
+  async load(projectDir) {
+    await this.loadFromUser()
+    if (projectDir) await this.loadFromProject(projectDir)
+  }
+
+  async _load(filePath) {
+    try {
+      const raw = await readFile(filePath, 'utf-8')
+      const data = JSON.parse(raw)
+      this.data = this._deepMerge(this.data, data)
+    } catch {
+      // 文件不存在或不合法 — 使用默认值
+    }
+  }
+
+  /** 保存到项目配置 */
+  async saveToProject(projectDir) {
+    const dir = join(projectDir, '.claude-code')
+    await mkdir(dir, { recursive: true })
+    const filePath = join(dir, 'config.json')
+    await writeFile(filePath, JSON.stringify(this.data, null, 2), 'utf-8')
+  }
+
+  /** 保存到用户配置 */
+  async saveToUser() {
+    const dir = join(homedir(), '.claude-code')
+    await mkdir(dir, { recursive: true })
+    await writeFile(this._userPath, JSON.stringify(this.data, null, 2), 'utf-8')
+  }
+
+  /** 获取配置值（支持点号路径，如 "tools.bash.timeout"） */
+  get(key) {
+    if (!key) return this.data
+    const parts = key.split('.')
+    let current = this.data
+    for (const part of parts) {
+      if (current == null) return undefined
+      current = current[part]
+    }
+    return current
+  }
+
+  /** 设置配置值 */
+  set(key, value) {
+    const parts = key.split('.')
+    let current = this.data
+    for (let i = 0; i < parts.length - 1; i++) {
+      if (current[parts[i]] == null) current[parts[i]] = {}
+      current = current[parts[i]]
+    }
+    current[parts[parts.length - 1]] = value
+  }
+
+  /** 深度合并 */
+  _deepMerge(target, source) {
+    const result = { ...target }
+    for (const key of Object.keys(source)) {
+      if (
+        source[key] &&
+        typeof source[key] === 'object' &&
+        !Array.isArray(source[key]) &&
+        target[key] &&
+        typeof target[key] === 'object' &&
+        !Array.isArray(target[key])
+      ) {
+        result[key] = this._deepMerge(target[key], source[key])
+      } else {
+        result[key] = source[key]
+      }
+    }
+    return result
+  }
+
+  /** 导出为 JSON */
+  toJSON() {
+    return { ...this.data }
+  }
+}

package/src/core/index.js

@@ -0,0 +1,9 @@
+/**
+ * 核心模块统一导出
+ */
+export { QueryEngine, QueryEngineConfig } from './query-engine.js'
+export { TokenBudget, estimateTokens } from './token-budget.js'
+export { SessionManager } from './session.js'
+export { Config } from './config.js'
+export { parseStream, parseNonStreamResponse } from './streaming.js'
+export { main } from './cli.js'

package/src/core/query-engine.js

@@ -0,0 +1,344 @@
+/**
+ * QueryEngine — Claude Code 核心引擎的 Node.js 重构
+ *
+ * 原版: src/QueryEngine.ts (46K 行)
+ * 职责: LLM API 调用 → 工具调用循环 → 流式响应 → 重试逻辑
+ *
+ * 核心循环:
+ * 用户输入 → 构建消息列表 → 调用 LLM → 解析工具调用 →
+ * 执行工具 → 把结果喂回 LLM → 循环直到无工具调用 → 输出
+ *
+ * API 协议: OpenAI 兼容（全行业通用）
+ * 适用于: OpenAI / DeepSeek / Qwen / GLM / Kimi / Ollama / vLLM / LM Studio / 任何兼容接口
+ */
+import crypto from 'crypto'
+import { Message, UserMessage, AssistantMessage, SystemMessage, ToolCall, ToolResult, SessionState } from '../types/index.js'
+import { EnhancedPermissionChecker } from '../security/enhanced-permission.js'
+
+/**
+ * 配置选项
+ */
+export class QueryEngineConfig {
+  constructor(options = {}) {
+    this.cwd = options.cwd || process.cwd()
+    this.tools = options.tools || []
+    this.commands = options.commands || []
+    this.systemPrompt = options.systemPrompt || ''
+    // 默认模型 — 不绑定任何厂商，用户必须通过 --model 或配置指定
+    this.model = options.model || ''
+    this.maxTurns = options.maxTurns || 100
+    this.maxBudgetTokens = options.maxBudgetTokens || 1_000_000
+    this.permissionMode = options.permissionMode || 'ask'
+    this.verbose = options.verbose || false
+    // API 配置 — 通用 OpenAI 兼容协议
+    // 优先级: 构造参数 > LLM_API_KEY > 厂商专用 Key
+    this.apiKey = options.apiKey || process.env.LLM_API_KEY || process.env.DEEPSEEK_API_KEY || process.env.OPENAI_API_KEY || process.env.QWEN_API_KEY || process.env.GLM_API_KEY || process.env.KIMI_API_KEY || ''
+    // API Base — DeepSeek 为默认
+    this.apiBase = options.apiBase || process.env.LLM_API_BASE || 'https://api.deepseek.com/v1'
+    this.initialMessages = options.initialMessages || []
+  }
+}
+
+/**
+ * 查询引擎 — 核心循环
+ */
+export class QueryEngine {
+  constructor(config) {
+    this.config = config instanceof QueryEngineConfig ? config : new QueryEngineConfig(config)
+    this.state = new SessionState()
+    this.permissionChecker = new EnhancedPermissionChecker(this.config.permissionMode, {
+      cwd: this.config.cwd,
+      projectDir: this.config.cwd,
+    })
+    this.abortController = null
+  }
+
+  /**
+   * 主入口 — 处理用户消息
+   */
+  async processMessage(userInput) {
+    if (this.state.isRunning) {
+      throw new Error('引擎正在运行中，请等待当前回合完成')
+    }
+    this.state.isRunning = true
+    this.state.turnCount++
+    this.abortController = new AbortController()
+    const userMsg = new UserMessage(userInput)
+    this.state.messages.push(userMsg)
+
+    try {
+      const result = await this._runToolLoop(userMsg)
+      return result
+    } finally {
+      this.state.isRunning = false
+    }
+  }
+
+  /**
+   * 工具调用循环 — 核心逻辑
+   *
+   * LLM 回复可能包含工具调用 → 执行工具 → 把工具结果喂回 LLM
+   * 最多跑 maxTurns 次
+   */
+  async _runToolLoop(userMessage) {
+    let currentMessages = [...this.state.messages]
+    let finalResponse = ''
+
+    for (let turn = 0; turn < this.config.maxTurns; turn++) {
+      const requestMessages = this._buildRequest(currentMessages)
+      const response = await this._callLLM(requestMessages, currentMessages)
+
+      if (this.abortController.signal.aborted) {
+        throw new Error('操作已取消')
+      }
+
+      // 没有工具调用 → 最终回复
+      if (!response.toolCalls || response.toolCalls.length === 0) {
+        finalResponse = response.content
+        const assistantMsg = new AssistantMessage(response.content)
+        this.state.messages.push(assistantMsg)
+        break
+      }
+
+      // 有工具调用 → 记录助手消息
+      const assistantMsg = new AssistantMessage(response.content, response.toolCalls)
+      this.state.messages.push(assistantMsg)
+
+      // 执行工具
+      const toolResults = await this._executeToolCalls(response.toolCalls)
+
+      // 工具结果加入消息流（OpenAI 兼容格式）
+      for (const result of toolResults) {
+        // 先加 assistant 的 tool_calls 消息
+        currentMessages.push({
+          role: 'assistant',
+          content: null,
+          tool_calls: [{
+            id: result.toolCallId,
+            type: 'function',
+            function: {
+              name: result.toolName || '',
+              arguments: '{}',
+            },
+          }],
+        })
+        // 再加 tool 结果消息
+        currentMessages.push({
+          role: 'tool',
+          tool_call_id: result.toolCallId,
+          content: result.isError
+            ? `[ERROR] ${result.content}`
+            : result.content,
+        })
+        this.state.toolResults.set(result.toolCallId, result)
+      }
+
+      if (this.config.verbose) {
+        console.error(`[QueryEngine] 工具循环第 ${turn + 1} 轮完成，执行了 ${toolResults.length} 个工具`)
+      }
+    }
+
+    if (!finalResponse && this.state.turnCount >= this.config.maxTurns) {
+      finalResponse = `[达到最大回合数限制 (${this.config.maxTurns})，停止响应]`
+    }
+
+    return {
+      response: finalResponse,
+      turns: this.state.turnCount,
+      toolResults: Array.from(this.state.toolResults.values()),
+    }
+  }
+
+  /**
+   * 构建 LLM 请求消息列表
+   */
+  _buildRequest(messages) {
+    const request = []
+
+    // 系统提示
+    if (this.config.systemPrompt) {
+      request.push({ role: 'system', content: this.config.systemPrompt })
+    }
+
+    // 历史消息
+    for (const msg of messages) {
+      if (msg.role === 'system') {
+        request.push({ role: 'system', content: msg.content })
+      } else if (msg.role === 'user') {
+        request.push({ role: 'user', content: this._formatContent(msg.content) })
+      } else if (msg.role === 'assistant') {
+        const content = []
+        if (msg.content) content.push({ type: 'text', text: msg.content })
+        if (msg.toolCalls) {
+          for (const tc of msg.toolCalls) {
+            content.push({ type: 'tool_use', id: tc.id, name: tc.name, input: tc.input })
+          }
+        }
+        request.push({ role: 'assistant', content })
+      }
+    }
+
+    return request
+  }
+
+  /**
+   * 执行工具调用
+   */
+  async _executeToolCalls(toolCalls) {
+    const results = []
+    for (const tc of toolCalls) {
+      // 安全检查（一票否决）
+      const permResult = await this.permissionChecker.check(tc.name, tc.input)
+      if (!permResult.allowed) {
+        results.push(new ToolResult(tc.id, `工具调用被安全策略拒绝: ${tc.name} — ${permResult.reason || ''}`, true))
+        continue
+      }
+
+      // 查找工具
+      const tool = this.config.tools.find(t => t.name === tc.name)
+      if (!tool) {
+        results.push(new ToolResult(tc.id, `未找到工具: ${tc.name}`, true))
+        continue
+      }
+
+      try {
+        tc.status = 'running'
+        const content = await tool.handler(tc.input, { cwd: this.config.cwd, engine: this })
+        tc.status = 'done'
+        results.push(new ToolResult(tc.id, typeof content === 'string' ? content : JSON.stringify(content), false))
+        // 记录工具名用于消息构建
+        results[results.length - 1].toolName = tc.name
+      } catch (err) {
+        tc.status = 'error'
+        results.push(new ToolResult(tc.id, `工具执行错误: ${err.message}`, true))
+        results[results.length - 1].toolName = tc.name
+      }
+    }
+    return results
+  }
+
+  /**
+   * 调用 LLM API — OpenAI 兼容协议（全行业通用）
+   *
+   * 支持的提供商（举例）:
+   * - DeepSeek:      https://api.deepseek.com/v1
+   * - 通义千问:       https://dashscope.aliyuncs.com/compatible-mode/v1
+   * - 智谱 GLM:       https://open.bigmodel.cn/api/paas/v4
+   * - Moonshot Kimi:  https://api.moonshot.cn/v1
+   * - Ollama 本地:    http://localhost:11434/v1
+   * - vLLM:          http://localhost:8000/v1
+   * - LM Studio:     http://localhost:1234/v1
+   * - OpenAI:        https://api.openai.com/v1
+   */
+  async _callLLM(messages, contextMessages) {
+    const apiKey = this.config.apiKey
+    const apiBase = this.config.apiBase
+
+    if (!apiKey) {
+      throw new Error(
+        `未设置 API Key。请设置以下环境变量之一:\n` +
+        `  LLM_API_KEY=xxx          (通用，推荐)\n` +
+        `  DEEPSEEK_API_KEY=xxx     (DeepSeek)\n` +
+        `  OPENAI_API_KEY=xxx       (OpenAI)\n` +
+        `  QWEN_API_KEY=xxx         (通义千问)\n` +
+        `  GLM_API_KEY=xxx          (智谱 GLM)\n` +
+        `  KIMI_API_KEY=xxx         (Moonshot Kimi)\n` +
+        `或通过 --api-key 参数传入`
+      )
+    }
+
+    if (!apiBase) {
+      throw new Error(
+        `未设置 API Base URL。默认使用 https://api.deepseek.com/v1
+` +
+        `可通过 LLM_API_BASE 或 --api-base 参数切换其他提供商，例如:\n` +
+        `  --api-base https://dashscope.aliyuncs.com/compatible-mode/v1   (通义千问)\n` +
+        `  --api-base https://open.bigmodel.cn/api/paas/v4              (智谱 GLM)\n` +
+        `  --api-base https://api.moonshot.cn/v1                         (Moonshot Kimi)\n` +
+        `  --api-base http://localhost:11434/v1                          (Ollama 本地)\n` +
+        `  --api-base http://localhost:8000/v1                           (vLLM 本地)\n` +
+        `  --api-base https://api.openai.com/v1                          (OpenAI)`
+      )
+    }
+
+    // 构建工具定义 — OpenAI function-calling 格式
+    const tools = this.config.tools.map(t => ({
+      type: 'function',
+      function: {
+        name: t.name,
+        description: t.description,
+        parameters: t.parameters,
+      },
+    }))
+
+    const body = {
+      model: this.config.model,
+      messages,
+      max_tokens: 4096,
+      ...(tools.length && { tools }),
+    }
+
+    const url = apiBase.replace(/\/+$/, '') + '/chat/completions'
+
+    const response = await fetch(url, {
+      method: 'POST',
+      headers: {
+        'Content-Type': 'application/json',
+        'Authorization': `Bearer ${apiKey}`,
+      },
+      body: JSON.stringify(body),
+      signal: this.abortController?.signal,
+    })
+
+    if (!response.ok) {
+      const errText = await response.text()
+      throw new Error(`API 错误 ${response.status}: ${errText}`)
+    }
+
+    const data = await response.json()
+    return this._parseResponse(data)
+  }
+
+  /**
+   * 解析 OpenAI 兼容响应
+   */
+  _parseResponse(data) {
+    const result = { content: '', toolCalls: [] }
+    const choice = data.choices?.[0]
+    if (!choice) return result
+
+    const message = choice.message
+    if (message.content) {
+      result.content = message.content
+    }
+
+    for (const tc of (message.tool_calls || [])) {
+      let input = {}
+      try {
+        input = JSON.parse(tc.function.arguments || '{}')
+      } catch {
+        input = { _raw: tc.function.arguments }
+      }
+      result.toolCalls.push(new ToolCall(tc.id, tc.function.name, input))
+    }
+
+    return result
+  }
+
+  /** 格式化内容 */
+  _formatContent(content) {
+    if (typeof content === 'string') return content
+    if (typeof content === 'object') return JSON.stringify(content)
+    return String(content)
+  }
+
+  /** 取消当前运行 */
+  abort() {
+    this.abortController?.abort()
+  }
+
+  /** 重置引擎状态 */
+  reset() {
+    this.state = new SessionState()
+  }
+}

package/src/core/session.js

@@ -0,0 +1,120 @@
+/**
+ * 会话管理
+ * 对应原版: src/utils/sessionState.ts + src/utils/sessionStorage.ts
+ */
+import { readFile, writeFile, mkdir, readdir, rm } from 'fs/promises'
+import { resolve, join } from 'path'
+import { existsSync } from 'fs'
+import { SessionState } from '../types/index.js'
+
+const DEFAULT_SESSIONS_DIR = '.claude-code/sessions'
+
+export class SessionManager {
+  constructor(options = {}) {
+    this.sessionsDir = options.sessionsDir || resolve(process.cwd(), DEFAULT_SESSIONS_DIR)
+    this.currentSession = null
+  }
+
+  /** 确保会话目录存在 */
+  async ensureDir() {
+    await mkdir(this.sessionsDir, { recursive: true })
+  }
+
+  /** 创建新会话 */
+  async create(title = '') {
+    await this.ensureDir()
+    const id = `session-${Date.now()}-${Math.random().toString(36).slice(2, 8)}`
+    const session = {
+      id,
+      title: title || `Session ${new Date().toISOString().slice(0, 19)}`,
+      created: new Date().toISOString(),
+      updated: new Date().toISOString(),
+      messages: [],
+      state: { turnCount: 0, budgetUsed: 0 },
+    }
+    await this.save(session)
+    this.currentSession = session
+    return session
+  }
+
+  /** 保存会话 */
+  async save(session) {
+    await this.ensureDir()
+    session.updated = new Date().toISOString()
+    const filePath = join(this.sessionsDir, `${session.id}.json`)
+    await writeFile(filePath, JSON.stringify(session, null, 2), 'utf-8')
+    return session
+  }
+
+  /** 加载会话 */
+  async load(sessionId) {
+    const filePath = join(this.sessionsDir, `${sessionId}.json`)
+    try {
+      const data = await readFile(filePath, 'utf-8')
+      const session = JSON.parse(data)
+      this.currentSession = session
+      return session
+    } catch (err) {
+      if (err.code === 'ENOENT') {
+        return null
+      }
+      throw err
+    }
+  }
+
+  /** 列出所有会话 */
+  async list() {
+    await this.ensureDir()
+    const files = await readdir(this.sessionsDir)
+    const sessions = []
+    for (const file of files) {
+      if (!file.endsWith('.json')) continue
+      try {
+        const data = await readFile(join(this.sessionsDir, file), 'utf-8')
+        const session = JSON.parse(data)
+        sessions.push({
+          id: session.id,
+          title: session.title,
+          created: session.created,
+          updated: session.updated,
+          messageCount: session.messages?.length || 0,
+        })
+      } catch { /* skip corrupted */ }
+    }
+    // 按更新时间倒序
+    sessions.sort((a, b) => new Date(b.updated) - new Date(a.updated))
+    return sessions
+  }
+
+  /** 删除会话 */
+  async delete(sessionId) {
+    const filePath = join(this.sessionsDir, `${sessionId}.json`)
+    try {
+      await rm(filePath)
+      if (this.currentSession?.id === sessionId) {
+        this.currentSession = null
+      }
+      return true
+    } catch {
+      return false
+    }
+  }
+
+  /** 获取或创建当前会话 */
+  async getOrCreate(title) {
+    if (this.currentSession) return this.currentSession
+    return this.create(title)
+  }
+
+  /** 追加消息到当前会话 */
+  async appendMessage(message) {
+    if (!this.currentSession) await this.getOrCreate()
+    this.currentSession.messages.push({
+      role: message.role,
+      content: typeof message.content === 'string' ? message.content : JSON.stringify(message.content),
+      timestamp: new Date().toISOString(),
+    })
+    await this.save(this.currentSession)
+    return this.currentSession
+  }
+}