@raolin2025/claude-code-node 1.2.0 → 2.1.0

package/src/core/compact.js

@@ -0,0 +1,171 @@
+/**
+ * 上下文压缩 (Compact) — 长对话自动摘要
+ * 
+ * 当对话 token 数接近预算上限时，自动将早期对话压缩为摘要，
+ * 保留最近 N 轮完整对话 + 工具结果的关键信息。
+ * 
+ * 对应原版: src/query/compact.ts
+ */
+
+import { estimateTokens } from './token-budget.js'
+
+/**
+ * 压缩策略：保留最近 N 轮完整对话，早期部分压缩为摘要
+ * 
+ * @param {Array} messages — 完整消息列表
+ * @param {object} options — 配置
+ * @param {number} options.maxTokens — token 预算上限
+ * @param {number} options.keepRecentTurns — 保留最近 N 轮（默认 4）
+ * @param {number} options.maxToolResultChars — 工具结果截断长度（默认 2000）
+ * @returns {Array} 压缩后的消息列表
+ */
+export function compactMessages(messages, options = {}) {
+  const maxTokens = options.maxTokens || 160_000
+  const keepRecentTurns = options.keepRecentTurns || 4
+  const maxToolResultChars = options.maxToolResultChars || 2000
+
+  // 1. 先截断过长的工具结果
+  const trimmed = messages.map(msg => {
+    if (msg.role === 'tool' && msg.content && msg.content.length > maxToolResultChars) {
+      return {
+        ...msg,
+        content: msg.content.slice(0, maxToolResultChars) + '\n[...compact: truncated]'
+      }
+    }
+    return msg
+  })
+
+  // 2. 估算总 token 数
+  const totalTokens = estimateTokens(
+    trimmed.map(m => typeof m.content === 'string' ? m.content : JSON.stringify(m.content)).join('')
+  )
+
+  if (totalTokens <= maxTokens) {
+    return trimmed // 不需要压缩
+  }
+
+  // 3. 找到分界点：保留最近 keepRecentTurns 轮
+  // 一轮 = user + assistant(+tool_calls) + tool 结果们 + assistant 最终回复
+  let turnCount = 0
+  let splitIndex = trimmed.length
+
+  for (let i = trimmed.length - 1; i >= 0; i--) {
+    if (trimmed[i].role === 'user') {
+      turnCount++
+      if (turnCount > keepRecentTurns) {
+        splitIndex = i
+        break
+      }
+    }
+  }
+
+  if (splitIndex === 0 || splitIndex >= trimmed.length) {
+    return trimmed // 无法压缩，全部保留
+  }
+
+  // 4. 将早期消息压缩为摘要
+  const earlyMessages = trimmed.slice(0, splitIndex)
+  const recentMessages = trimmed.slice(splitIndex)
+
+  const summary = generateSummary(earlyMessages)
+
+  // 5. 构建压缩后的消息列表
+  const compacted = []
+
+  // 如果第一条是 system，保留
+  if (recentMessages[0]?.role === 'system') {
+    compacted.push(recentMessages.shift())
+  }
+
+  // 插入摘要作为 system 上下文
+  compacted.push({
+    role: 'system',
+    content: `[Context Summary — ${new Date().toISOString()}]\n${summary}\n[End of Summary — recent conversation follows]`
+  })
+
+  // 追加最近对话
+  compacted.push(...recentMessages)
+
+  return compacted
+}
+
+/**
+ * 从消息列表生成摘要
+ */
+function generateSummary(messages) {
+  const topics = new Set()
+  const toolsUsed = new Set()
+  const keyResults = []
+  let lastUserIntent = ''
+
+  for (const msg of messages) {
+    if (msg.role === 'user' && msg.content) {
+      // 提取用户意图（取第一行或前 80 字符）
+      const intent = typeof msg.content === 'string'
+        ? msg.content.split('\n')[0].slice(0, 80)
+        : ''
+      if (intent) lastUserIntent = intent
+      topics.add(intent)
+    }
+
+    if (msg.role === 'assistant') {
+      // 收集使用的工具
+      if (msg.toolCalls) {
+        for (const tc of msg.toolCalls) {
+          toolsUsed.add(tc.name)
+        }
+      }
+      // 收集关键文本结果（取最后一条重要的 assistant 回复）
+      if (msg.content && typeof msg.content === 'string' && msg.content.length > 20) {
+        keyResults.push(msg.content.slice(0, 300))
+      }
+    }
+
+    if (msg.role === 'tool' && msg.content) {
+      // 记录工具结果摘要
+      const content = typeof msg.content === 'string' ? msg.content : JSON.stringify(msg.content)
+      if (content.length > 100) {
+        keyResults.push(`[tool result]: ${content.slice(0, 150)}...`)
+      }
+    }
+  }
+
+  // 组装摘要
+  const parts = []
+  if (topics.size > 0) {
+    const topicList = [...topics].slice(-5).map(t => `- ${t}`).join('\n')
+    parts.push(`User intents:\n${topicList}`)
+  }
+  if (toolsUsed.size > 0) {
+    parts.push(`Tools used: ${[...toolsUsed].join(', ')}`)
+  }
+  if (keyResults.length > 0) {
+    const lastResult = keyResults[keyResults.length - 1]
+    parts.push(`Last key result: ${lastResult}`)
+  }
+
+  return parts.join('\n\n') || 'Previous conversation context was compacted.'
+}
+
+/**
+ * 自动检查是否需要压缩，需要时执行
+ * 
+ * @param {Array} messages — 当前消息列表
+ * @param {object} tokenBudget — TokenBudget 实例
+ * @param {object} options — 压缩选项
+ * @returns {{ compacted: boolean, messages: Array }} 是否压缩了 + 结果消息列表
+ */
+export function autoCompact(messages, tokenBudget, options = {}) {
+  const threshold = options.threshold || 0.8 // 80% 时触发
+  const usagePercent = tokenBudget.usagePercent / 100
+
+  if (usagePercent >= threshold) {
+    const compacted = compactMessages(messages, {
+      maxTokens: Math.floor(tokenBudget.maxTokens * 0.6), // 压缩到 60%
+      ...options,
+    })
+    return { compacted: true, messages: compacted }
+  }
+
+  return { compacted: false, messages }
+}

package/src/core/config.js

@@ -3,8 +3,7 @@
  * 对应原版: 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 { join } from 'path'
 import { homedir } from 'os'
 
 const PROJECT_CONFIG_FILE = '.claude-code/config.json'

package/src/core/cost-tracker.js

@@ -0,0 +1,171 @@
+/**
+ * 费用追踪 (Cost Tracking) — API 调用费用计算和报告
+ * 
+ * 支持主流模型的价格表，自动根据 model 名称匹配价格。
+ * 对应原版: src/utils/costTracker.ts
+ */
+
+// ============================================================
+// 模型价格表（美元 / 1M tokens）
+// ============================================================
+
+const PRICING = {
+  // DeepSeek
+  'deepseek-chat':           { input: 0.27,  output: 1.10,  cache_read: 0.07 },
+  'deepseek-reasoner':       { input: 0.55,  output: 2.19,  cache_read: 0.14 },
+
+  // OpenAI
+  'gpt-4o':                  { input: 2.50,  output: 10.00 },
+  'gpt-4o-mini':             { input: 0.15,  output: 0.60 },
+  'gpt-4.1':                 { input: 2.00,  output: 8.00 },
+  'gpt-4.1-mini':            { input: 0.40,  output: 1.60 },
+  'gpt-4.1-nano':            { input: 0.10,  output: 0.40 },
+  'o3':                      { input: 2.00,  output: 8.00 },
+  'o3-mini':                 { input: 1.10,  output: 4.40 },
+  'o4-mini':                 { input: 1.10,  output: 4.40 },
+
+  // 通义千问 (Qwen)
+  'qwen-plus':               { input: 0.50,  output: 2.00 },
+  'qwen-turbo':              { input: 0.05,  output: 0.20 },
+  'qwen-max':                { input: 2.00,  output: 6.00 },
+  'qwen-long':               { input: 0.07,  output: 0.28 },
+
+  // 智谱 GLM
+  'glm-4-flash':             { input: 0.10,  output: 0.10 },
+  'glm-4-plus':              { input: 0.50,  output: 0.50 },
+  'glm-4':                   { input: 1.00,  output: 1.00 },
+  'glm-4-long':              { input: 0.10,  output: 0.10 },
+
+  // Moonshot Kimi
+  'moonshot-v1-8k':          { input: 0.50,  output: 2.00 },
+  'moonshot-v1-32k':         { input: 1.00,  output: 4.00 },
+  'moonshot-v1-128k':        { input: 2.00,  output: 8.00 },
+
+  // Ollama (本地免费)
+  'ollama':                  { input: 0,     output: 0 },
+}
+
+// 默认价格（未匹配到的模型）
+const DEFAULT_PRICING = { input: 0.50, output: 2.00 }
+
+/**
+ * 根据模型名查找价格
+ */
+function findPricing(model) {
+  if (!model) return DEFAULT_PRICING
+
+  // 精确匹配
+  if (PRICING[model]) return PRICING[model]
+
+  // 前缀匹配 (e.g. "deepseek-chat-v3" → "deepseek-chat")
+  for (const [key, price] of Object.entries(PRICING)) {
+    if (model.startsWith(key) || model.includes(key)) return price
+  }
+
+  // Ollama 系列全部免费
+  if (model.includes('ollama') || model.includes('local') || model.includes('localhost')) {
+    return PRICING.ollama
+  }
+
+  return DEFAULT_PRICING
+}
+
+/**
+ * 费用追踪器
+ */
+export class CostTracker {
+  constructor(options = {}) {
+    this.model = options.model || ''
+    this.pricing = findPricing(this.model)
+    this.totalInputTokens = 0
+    this.totalOutputTokens = 0
+    this.totalCacheReadTokens = 0
+    this.totalApiCalls = 0
+    this.history = []  // 每次 API 调用的记录
+  }
+
+  /** 切换模型（更新价格表） */
+  setModel(model) {
+    this.model = model
+    this.pricing = findPricing(model)
+  }
+
+  /** 记录一次 API 调用 */
+  recordUsage(usage) {
+    const inputTokens = usage.input_tokens || usage.prompt_tokens || 0
+    const outputTokens = usage.output_tokens || usage.completion_tokens || 0
+    const cacheReadTokens = usage.cache_read_input_tokens || 0
+
+    const cost = this.calculateCost(inputTokens, outputTokens, cacheReadTokens)
+
+    this.totalInputTokens += inputTokens
+    this.totalOutputTokens += outputTokens
+    this.totalCacheReadTokens += cacheReadTokens
+    this.totalApiCalls++
+
+    this.history.push({
+      timestamp: Date.now(),
+      inputTokens,
+      outputTokens,
+      cacheReadTokens,
+      cost,
+    })
+
+    return cost
+  }
+
+  /** 计算单次费用 */
+  calculateCost(inputTokens, outputTokens, cacheReadTokens = 0) {
+    const cost = {
+      input: (inputTokens / 1_000_000) * this.pricing.input,
+      output: (outputTokens / 1_000_000) * this.pricing.output,
+      cacheRead: (cacheReadTokens / 1_000_000) * (this.pricing.cache_read || 0),
+    }
+    cost.total = cost.input + cost.output + cost.cacheRead
+    return cost
+  }
+
+  /** 获取总费用 */
+  getTotalCost() {
+    const cost = this.calculateCost(
+      this.totalInputTokens,
+      this.totalOutputTokens,
+      this.totalCacheReadTokens
+    )
+    return cost
+  }
+
+  /** 格式化费用报告 */
+  formatReport() {
+    const cost = this.getTotalCost()
+    const lines = [
+      `💰 Cost Report — ${this.model}`,
+      `   API Calls:     ${this.totalApiCalls}`,
+      `   Input Tokens:  ${this.totalInputTokens.toLocaleString()} → $${cost.input.toFixed(4)}`,
+      `   Output Tokens: ${this.totalOutputTokens.toLocaleString()} → $${cost.output.toFixed(4)}`,
+    ]
+    if (this.totalCacheReadTokens > 0) {
+      lines.push(`   Cache Read:    ${this.totalCacheReadTokens.toLocaleString()} → $${cost.cacheRead.toFixed(4)}`)
+    }
+    lines.push(`   ───────────────────────────────`)
+    lines.push(`   Total:         $${cost.total.toFixed(4)}`)
+    return lines.join('\n')
+  }
+
+  /** 简短格式（单行） */
+  formatShort() {
+    const cost = this.getTotalCost()
+    return `$${cost.total.toFixed(4)} (${this.totalApiCalls} calls, ${this.totalInputTokens.toLocaleString()}+${this.totalOutputTokens.toLocaleString()} tok)`
+  }
+
+  /** 重置 */
+  reset() {
+    this.totalInputTokens = 0
+    this.totalOutputTokens = 0
+    this.totalCacheReadTokens = 0
+    this.totalApiCalls = 0
+    this.history = []
+  }
+}
+
+export { PRICING, findPricing }

package/src/core/paths.js

@@ -0,0 +1,12 @@
+/**
+ * 共享路径常量 — cc-node 和 cc-notify 共用
+ */
+import { join } from 'path'
+import { homedir } from 'os'
+
+export const SOCK_DIR = join(homedir(), '.cc-node')
+export const SOCK_PATH = join(SOCK_DIR, 'repl.sock')
+export const CC_NODE_PID = join(SOCK_DIR, 'cc-node.pid')
+export const CC_NOTIFY_PID = join(SOCK_DIR, 'cc-notify.pid')
+export const CC_NOTIFY_LOG = join(SOCK_DIR, 'cc-notify.log')
+export const DEFAULT_HTTP_PORT = 3456