@icode-js/icode 3.0.4 → 3.2.0

package/src/core/tools/interactive.js

@@ -0,0 +1,368 @@
+import { createInterface } from 'node:readline/promises'
+import { stdin, stdout } from 'node:process'
+
+/**
+ * 判断当前 stdin/stdout 是否连接到交互式终端。
+ * 输入为空，输出为布尔值。
+ */
+function isInteractive() {
+  return Boolean(stdin.isTTY && stdout.isTTY)
+}
+
+/**
+ * 对外暴露交互终端判断，供工作流决定是否进入交互模式。
+ * 输入为空，输出为布尔值。
+ */
+export function isInteractiveTerminal() {
+  return isInteractive()
+}
+
+/**
+ * 归一化 confirm 的默认值配置，兼容布尔值与 options.defaultValue 两种调用方式。
+ * 输入为 confirm 第二参数，输出为布尔默认值。
+ */
+function resolveConfirmDefaultValue(optionsOrDefaultValue) {
+  if (typeof optionsOrDefaultValue === 'boolean') {
+    return optionsOrDefaultValue
+  }
+
+  if (optionsOrDefaultValue && typeof optionsOrDefaultValue === 'object') {
+    return optionsOrDefaultValue.defaultValue === true
+  }
+
+  return false
+}
+
+/**
+ * 读取确认输入并返回布尔结果；非交互环境直接返回默认值。
+ * 输入为提示文案与默认值配置，输出为用户是否确认。
+ */
+export async function confirm(message, optionsOrDefaultValue = false) {
+  const defaultValue = resolveConfirmDefaultValue(optionsOrDefaultValue)
+  if (!isInteractive()) {
+    return defaultValue
+  }
+
+  const rl = createInterface({ input: stdin, output: stdout })
+  const suffix = defaultValue ? '[Y/n]' : '[y/N]'
+  const answer = (await rl.question(`${message} ${suffix} `)).trim().toLowerCase()
+  rl.close()
+
+  if (!answer) {
+    return defaultValue
+  }
+
+  if (['y', 'yes', '1', 'true', 'ok', 'sure', '是'].includes(answer)) {
+    return true
+  }
+
+  if (['n', 'no', '0', 'false', '否'].includes(answer)) {
+    return false
+  }
+
+  return defaultValue
+}
+
+/**
+ * 读取一行文本输入；非交互环境直接返回默认值。
+ * 输入为提示文案与默认值，输出为归一化后的文本。
+ */
+export async function input(message, defaultValue = '') {
+  if (!isInteractive()) {
+    return defaultValue
+  }
+
+  const rl = createInterface({ input: stdin, output: stdout })
+  const answer = await rl.question(`${message}${defaultValue ? ` (${defaultValue})` : ''}: `)
+  rl.close()
+  const normalized = answer.trim()
+  return normalized || defaultValue
+}
+
+/**
+ * 在给定选项中返回单个值；非交互环境返回默认索引对应值。
+ * 输入为提示文案、选项列表与默认索引，输出为选中的 value。
+ */
+export async function chooseOne(message, choices, defaultIndex = 0) {
+  if (!Array.isArray(choices) || choices.length === 0) {
+    throw new Error('chooseOne 需要至少一个可选项')
+  }
+
+  const safeDefaultIndex = Math.min(Math.max(defaultIndex, 0), choices.length - 1)
+  if (!isInteractive()) {
+    return choices[safeDefaultIndex].value
+  }
+
+  stdout.write(`${message}\n`)
+  choices.forEach((choice, index) => {
+    stdout.write(`  ${index + 1}. ${choice.label}\n`)
+  })
+
+  const rl = createInterface({ input: stdin, output: stdout })
+  const answer = (await rl.question(`请选择 [${safeDefaultIndex + 1}]: `)).trim()
+  rl.close()
+
+  if (!answer) {
+    return choices[safeDefaultIndex].value
+  }
+
+  const numeric = Number(answer)
+  if (!Number.isInteger(numeric) || numeric < 1 || numeric > choices.length) {
+    return choices[safeDefaultIndex].value
+  }
+
+  return choices[numeric - 1].value
+}
+
+/**
+ * 按选项顺序输出已选 value。
+ * 输入为选项与已选集合，输出为稳定顺序的 value 数组。
+ */
+function collectSelectedValues(choices, selected, maxSelections) {
+  return choices
+    .map((choice) => choice.value)
+    .filter((value) => selected.has(value))
+    .slice(0, maxSelections)
+}
+
+/**
+ * 使用旧版数字菜单进行多选，供不支持 raw mode 的终端回退。
+ * 输入为多选上下文，输出为已选 value 数组或 null。
+ */
+async function chooseManyByNumberInput(message, choices, context) {
+  const selected = new Set(context.selectedValues)
+  const doneValue = '__prompt_done__'
+  const cancelValue = '__prompt_cancel__'
+
+  while (true) {
+    const menuChoices = choices.map((choice) => ({
+      value: choice.value,
+      label: `${selected.has(choice.value) ? '[x]' : '[ ]'} ${choice.label}`
+    }))
+    menuChoices.push({
+      value: doneValue,
+      label: `${context.doneLabel}（已选 ${selected.size} 项）`
+    })
+    menuChoices.push({
+      value: cancelValue,
+      label: context.cancelLabel
+    })
+
+    const defaultChoiceValue = selected.size >= context.minSelections ? doneValue : choices[0].value
+    const resolvedDefaultIndex = Math.max(0, menuChoices.findIndex((item) => item.value === defaultChoiceValue))
+    const selectedAction = await chooseOne(message, menuChoices, resolvedDefaultIndex)
+
+    if (selectedAction === cancelValue) {
+      return null
+    }
+
+    if (selectedAction === doneValue) {
+      if (selected.size < context.minSelections) {
+        stdout.write(`至少需要选择 ${context.minSelections} 项。\n`)
+        continue
+      }
+      return collectSelectedValues(choices, selected, context.maxSelections)
+    }
+
+    if (selected.has(selectedAction)) {
+      selected.delete(selectedAction)
+      continue
+    }
+
+    if (selected.size >= context.maxSelections) {
+      stdout.write(`最多只能选择 ${context.maxSelections} 项。\n`)
+      continue
+    }
+
+    selected.add(selectedAction)
+  }
+}
+
+/**
+ * 渲染 checkbox 多选列表。
+ * 输入为当前光标和选择状态，输出为本次渲染占用的行数。
+ */
+function renderCheckboxChoices(message, choices, context, cursorIndex, selected, statusMessage, renderedLines) {
+  if (renderedLines > 0) {
+    stdout.write(`\x1b[${renderedLines}F`)
+    stdout.write('\x1b[J')
+  }
+
+  const lines = [
+    message,
+    `↑/↓ 移动，Space 选中/取消，Enter ${context.doneLabel}，q ${context.cancelLabel}（已选 ${selected.size} 项）`
+  ]
+
+  if (statusMessage) {
+    lines.push(statusMessage)
+  }
+
+  choices.forEach((choice, index) => {
+    const cursor = index === cursorIndex ? '>' : ' '
+    const checkbox = selected.has(choice.value) ? '[x]' : '[ ]'
+    lines.push(`${cursor} ${checkbox} ${choice.label}`)
+  })
+
+  stdout.write(`${lines.join('\n')}\n`)
+  return lines.length
+}
+
+/**
+ * 使用键盘 checkbox 进行多选。
+ * 输入为多选上下文，输出为已选 value 数组或 null。
+ */
+function chooseManyByCheckbox(message, choices, context) {
+  const selected = new Set(context.selectedValues)
+  let cursorIndex = 0
+  let renderedLines = 0
+  let statusMessage = ''
+  let closed = false
+  const wasRaw = stdin.isRaw === true
+
+  return new Promise((resolve) => {
+    const finish = (value) => {
+      if (closed) {
+        return
+      }
+
+      closed = true
+      stdin.off('data', onData)
+      stdin.setRawMode(wasRaw)
+      stdin.pause()
+      resolve(value)
+    }
+
+    const render = () => {
+      renderedLines = renderCheckboxChoices(message, choices, context, cursorIndex, selected, statusMessage, renderedLines)
+    }
+
+    const moveCursor = (offset) => {
+      statusMessage = ''
+      cursorIndex = (cursorIndex + offset + choices.length) % choices.length
+      render()
+    }
+
+    const toggleCurrent = () => {
+      statusMessage = ''
+      const currentValue = choices[cursorIndex].value
+      if (selected.has(currentValue)) {
+        selected.delete(currentValue)
+        render()
+        return
+      }
+
+      if (selected.size >= context.maxSelections) {
+        statusMessage = `最多只能选择 ${context.maxSelections} 项。`
+        render()
+        return
+      }
+
+      selected.add(currentValue)
+      render()
+    }
+
+    const complete = () => {
+      if (selected.size < context.minSelections) {
+        statusMessage = `至少需要选择 ${context.minSelections} 项。`
+        render()
+        return
+      }
+      finish(collectSelectedValues(choices, selected, context.maxSelections))
+    }
+
+    function onData(chunk) {
+      const input = chunk.toString('utf8')
+
+      for (let index = 0; index < input.length; index += 1) {
+        if (closed) {
+          return
+        }
+
+        const sequence = input.slice(index, index + 3)
+        if (sequence === '\u001b[A') {
+          moveCursor(-1)
+          index += 2
+          continue
+        }
+
+        if (sequence === '\u001b[B') {
+          moveCursor(1)
+          index += 2
+          continue
+        }
+
+        const key = input[index]
+        if (key === '\u0003' || key === '\u001b' || key.toLowerCase() === 'q') {
+          finish(null)
+          return
+        }
+
+        if (key === '\r' || key === '\n') {
+          complete()
+          continue
+        }
+
+        if (key === ' ') {
+          toggleCurrent()
+          continue
+        }
+
+        if (key.toLowerCase() === 'k') {
+          moveCursor(-1)
+          continue
+        }
+
+        if (key.toLowerCase() === 'j') {
+          moveCursor(1)
+        }
+      }
+    }
+
+    stdin.setRawMode(true)
+    stdin.resume()
+    stdin.on('data', onData)
+    render()
+  })
+}
+
+/**
+ * 在给定选项中进行多选；非交互环境返回默认值集合。
+ * 输入为提示文案、选项列表与多选配置，输出为选中的 value 数组或 null。
+ */
+export async function chooseMany(message, choices, options = {}) {
+  if (!Array.isArray(choices) || choices.length === 0) {
+    throw new Error('chooseMany 需要至少一个可选项')
+  }
+
+  const parsedMin = Number(options.minSelections ?? 0)
+  const parsedMax = Number(options.maxSelections ?? choices.length)
+  const minSelections = Number.isFinite(parsedMin) ? Math.max(0, Math.floor(parsedMin)) : 0
+  const maxCap = Number.isFinite(parsedMax) ? Math.max(0, Math.floor(parsedMax)) : choices.length
+  const maxSelections = Math.max(minSelections, Math.min(choices.length, maxCap))
+  const doneLabel = options.doneLabel || '完成选择'
+  const cancelLabel = options.cancelLabel || '取消'
+  const defaultValues = Array.isArray(options.defaultValues) ? options.defaultValues : []
+  const defaultSet = new Set(defaultValues)
+  const selectedValues = choices
+    .map((choice) => choice.value)
+    .filter((value) => defaultSet.has(value))
+    .slice(0, maxSelections)
+
+  if (!isInteractive()) {
+    return selectedValues
+  }
+
+  const context = {
+    minSelections,
+    maxSelections,
+    doneLabel,
+    cancelLabel,
+    selectedValues
+  }
+
+  if (typeof stdin.setRawMode !== 'function') {
+    return chooseManyByNumberInput(message, choices, context)
+  }
+
+  return chooseManyByCheckbox(message, choices, context)
+}

package/src/core/{loading.js → tools/loading.js}

@@ -1,9 +1,11 @@
 const SPINNER_FRAMES = ['-', '\\', '|', '/']
 
+// 判断当前进程是否适合渲染动态等待效果，输入为空，输出布尔值。
 function canRenderSpinner() {
   return Boolean(process.stderr.isTTY) && process.env.ICODE_NO_SPINNER !== '1'
 }
 
+// 清理当前终端行，输入为空，输出为空，仅在 TTY 中生效。
 function clearLine() {
   if (!canRenderSpinner()) {
     return
@@ -12,6 +14,7 @@ function clearLine() {
   process.stderr.write('\r\x1b[2K')
 }
 
+// 执行异步任务时展示等待动画，输入展示文案和任务函数，输出任务结果。
 export async function withSpinner(text, task) {
   if (!canRenderSpinner()) {
     return task()
@@ -21,7 +24,8 @@ export async function withSpinner(text, task) {
   const render = () => {
     const frame = SPINNER_FRAMES[frameIndex % SPINNER_FRAMES.length]
     frameIndex += 1
-    process.stderr.write(`\r[icode] ${frame} ${text}`)
+    clearLine()
+    process.stderr.write(`[icode] ${frame} ${text}`)
   }
 
   render()

package/src/core/{logger.js → tools/logger.js}

@@ -25,6 +25,11 @@ function print(stream, level, message) {
 }
 
 export const logger = {
+  // 按指定颜色渲染短文本，输入颜色名和文本，输出可直接拼进日志的 ANSI 字符串。
+  color(color, message) {
+    return colorize(color, String(message ?? ''))
+  },
+
   setVerbose(enabled) {
     verboseEnabled = Boolean(enabled)
   },

package/src/workflows/ai-codereview-workflow.js

@@ -1,8 +1,9 @@
-import { askAi } from '../core/ai-client.js'
-import { resolveAiDiffRange } from '../core/ai-diff-range.js'
+import { askAi } from '../core/ai/client.js'
+import { buildLimitedDiffPrompt, formatDiffCoverage, summarizeDiffCoverage } from '../core/ai/diff-prompt.js'
+import { resolveAiDiffRange } from '../core/ai/diff-range.js'
 import { IcodeError } from '../core/errors.js'
-import { resolveGitContext } from '../core/git-context.js'
-import { GitService } from '../core/git-service.js'
+import { resolveGitContext } from '../core/git/context.js'
+import { GitService } from '../core/git/service.js'
 
 function truncate(value, limit) {
   const text = value || ''
@@ -19,6 +20,14 @@ function joinSections(sections) {
     .join('\n\n')
 }
 
+// 构建 AI code review prompt 中的 diff 片段，输入完整 diff，输出受限 diff 与覆盖范围。
+function buildReviewDiffPrompt(diff) {
+  return buildLimitedDiffPrompt(diff, {
+    limit: 18000
+  })
+}
+
+// 执行 AI 代码审查工作流，输入 diff 范围和 AI 选项，输出审查文本与 diff 覆盖范围。
 export async function runAiCodeReviewWorkflow(options) {
   const context = await resolveGitContext({
     cwd: options.cwd,
@@ -101,10 +110,12 @@ export async function runAiCodeReviewWorkflow(options) {
     ])
   }
 
+  const limitedDiff = buildReviewDiffPrompt(diff)
+
   const review = await askAi(
     {
       systemPrompt: '你是严格的软件代码审查工程师，请优先关注 bug、安全风险、行为回归、缺失测试。输出中文 Markdown。',
-      userPrompt: `请按如下结构输出：\n1. Findings（按严重度从高到低）\n2. Open Questions\n3. Summary\n\nFocus: ${options.focus || 'general'}\nRange: ${rangeSpec}\nDiff Source: ${diffSource}\n\nDiff Stat:\n${truncate(stat, 3000)}\n\nName Status:\n${truncate(nameStatus, 3000)}\n\nUnified Diff:\n${truncate(diff, 18000)}`
+      userPrompt: `请按如下结构输出：\n1. Findings（按严重度从高到低）\n2. Open Questions\n3. Summary\n\nFocus: ${options.focus || 'general'}\nRange: ${rangeSpec}\nDiff Source: ${diffSource}\n\n${formatDiffCoverage(limitedDiff.coverage)}\n\nDiff Stat:\n${truncate(stat, 3000)}\n\nName Status:\n${truncate(nameStatus, 3000)}\n\nUnified Diff:\n${limitedDiff.diff}`
     },
     {
       profile: options.profile,
@@ -121,6 +132,7 @@ export async function runAiCodeReviewWorkflow(options) {
 
   return {
     rangeSpec,
-    review
+    review,
+    diffCoverage: summarizeDiffCoverage(limitedDiff.coverage)
   }
 }

package/src/workflows/ai-commit-workflow.js

@@ -1,10 +1,11 @@
 import { IcodeError } from '../core/errors.js'
-import { resolveGitContext } from '../core/git-context.js'
-import { GitService } from '../core/git-service.js'
-import { logger } from '../core/logger.js'
-import { askAiJson } from '../core/ai-client.js'
-import { scanCommitConventions } from '../core/commit-conventions.js'
-import { confirm } from '../core/prompts.js'
+import { askAiJson } from '../core/ai/client.js'
+import { buildLimitedDiffPrompt, formatDiffCoverage, summarizeDiffCoverage } from '../core/ai/diff-prompt.js'
+import { resolveGitContext } from '../core/git/context.js'
+import { scanCommitConventions } from '../core/git/commit-conventions.js'
+import { GitService } from '../core/git/service.js'
+import { confirm } from '../core/tools/interactive.js'
+import { logger } from '../core/tools/logger.js'
 
 function normalizeCommitType(value) {
   const allowed = new Set(['feat', 'fix', 'docs', 'style', 'refactor', 'perf', 'test', 'chore', 'build', 'ci', 'revert'])
@@ -74,7 +75,9 @@ export async function runAiCommitWorkflow(options) {
     })
   }
 
-  const limitedDiff = diff.length > 12000 ? `${diff.slice(0, 12000)}\n\n...<truncated>` : diff
+  const limitedDiff = buildLimitedDiffPrompt(diff, {
+    limit: 12000
+  })
   const conventionContext = scanCommitConventions(context)
   const conventionPrompt = conventionContext.hasConventions
     ? `Local commit conventions were detected from repository hooks/config files. Follow these local rules first when generating the commit message.\n\n${conventionContext.summary}\n\n`
@@ -89,7 +92,7 @@ export async function runAiCommitWorkflow(options) {
   const { parsed, text } = await askAiJson(
     {
       systemPrompt: `You are a senior software engineer. Generate a concise Conventional Commit message. Output JSON only. Language: ${language}.`,
-      userPrompt: `${conventionPrompt}Based on the following git diff, return JSON with fields:\n{\"type\":\"feat|fix|docs|style|refactor|perf|test|chore|build|ci|revert\",\"scope\":\"optional\",\"subject\":\"required one-line summary\",\"body\":\"optional details\"}\n\nDiff Source: ${diffSource}\n\nDiff:\n${limitedDiff}`
+      userPrompt: `${conventionPrompt}Based on the following git diff, return JSON with fields:\n{\"type\":\"feat|fix|docs|style|refactor|perf|test|chore|build|ci|revert\",\"scope\":\"optional\",\"subject\":\"required one-line summary\",\"body\":\"optional details\"}\n\nDiff Source: ${diffSource}\n\n${formatDiffCoverage(limitedDiff.coverage)}\n\nDiff:\n${limitedDiff.diff}`
     },
     {
       profile: options.profile
@@ -98,13 +101,16 @@ export async function runAiCommitWorkflow(options) {
 
   const commitMessage = buildCommitMessage(parsed)
 
-  logger.success(`AI 建议提交信息:\n${commitMessage}`)
+  if (!options.silentResultLog) {
+    logger.success(`AI 建议提交信息:\n${commitMessage}`)
+  }
 
   if (!options.apply) {
     return {
       applied: false,
       commitMessage,
-      raw: text
+      raw: text,
+      diffCoverage: summarizeDiffCoverage(limitedDiff.coverage)
     }
   }
 
@@ -115,6 +121,7 @@ export async function runAiCommitWorkflow(options) {
         applied: false,
         commitMessage,
         raw: text,
+        diffCoverage: summarizeDiffCoverage(limitedDiff.coverage),
         canceled: true
       }
     }
@@ -130,7 +137,7 @@ export async function runAiCommitWorkflow(options) {
   })
 
   const commitId = await git.revParseShort('HEAD')
-  if (commitId) {
+  if (commitId && !options.silentResultLog) {
     logger.success(`AI commit 已创建: ${commitId}`)
   }
 
@@ -138,6 +145,7 @@ export async function runAiCommitWorkflow(options) {
     applied: true,
     commitId,
     commitMessage,
-    raw: text
+    raw: text,
+    diffCoverage: summarizeDiffCoverage(limitedDiff.coverage)
   }
 }

package/src/workflows/ai-conflict-workflow.js

@@ -1,10 +1,10 @@
 import fs from 'node:fs'
 import path from 'node:path'
-import { askAi } from '../core/ai-client.js'
+import { askAi } from '../core/ai/client.js'
 import { IcodeError } from '../core/errors.js'
-import { resolveGitContext } from '../core/git-context.js'
-import { GitService } from '../core/git-service.js'
-import { logger } from '../core/logger.js'
+import { resolveGitContext } from '../core/git/context.js'
+import { GitService } from '../core/git/service.js'
+import { logger } from '../core/tools/logger.js'
 
 function extractConflictBlocks(content, maxBlocks = 3, maxLines = 80) {
   const lines = content.split('\n')

package/src/workflows/ai-explain-workflow.js

@@ -1,8 +1,8 @@
-import { askAi } from '../core/ai-client.js'
-import { resolveAiDiffRange } from '../core/ai-diff-range.js'
+import { askAi } from '../core/ai/client.js'
+import { resolveAiDiffRange } from '../core/ai/diff-range.js'
 import { IcodeError } from '../core/errors.js'
-import { resolveGitContext } from '../core/git-context.js'
-import { GitService } from '../core/git-service.js'
+import { resolveGitContext } from '../core/git/context.js'
+import { GitService } from '../core/git/service.js'
 
 function truncate(value, limit) {
   const text = value || ''

package/src/workflows/ai-risk-review-workflow.js

@@ -1,4 +1,4 @@
-import { askAiJson } from '../core/ai-client.js'
+import { askAiJson } from '../core/ai/client.js'
 
 function normalizeDecision(value) {
   const normalized = (value || '').trim().toLowerCase()

package/src/workflows/checkout-workflow.js

@@ -1,8 +1,18 @@
 import { IcodeError } from '../core/errors.js'
-import { GitService } from '../core/git-service.js'
-import { resolveGitContext } from '../core/git-context.js'
-import { logger } from '../core/logger.js'
+import { resolveGitContext } from '../core/git/context.js'
+import {
+  assertCleanWorktree,
+  assertNoConflictedFiles,
+  assertNoPendingOperation,
+  readGitSafetyState
+} from '../core/git/preconditions.js'
+import { GitService } from '../core/git/service.js'
+import { logger } from '../core/tools/logger.js'
 
+/**
+ * 执行 checkout 工作流。
+ * 输入为目标分支、基线分支和执行选项，输出分支切换结果；实际切换前要求工作区干净且没有未完成 Git 操作。
+ */
 export async function runCheckoutWorkflow(input) {
   if (!input.branchName) {
     throw new IcodeError('缺少分支名: icode checkout <branch> [base]', {
@@ -19,6 +29,10 @@ export async function runCheckoutWorkflow(input) {
   const git = new GitService(context)
   const branchName = input.branchName.trim()
   const baseBranchName = input.baseBranchName?.trim() || context.defaultBranch
+  const safetyState = await readGitSafetyState(git)
+  assertNoPendingOperation(safetyState.operation, 'checkout', 'CHECKOUT_GIT_OPERATION_IN_PROGRESS')
+  assertNoConflictedFiles(safetyState.changes, 'checkout', 'CHECKOUT_CONFLICTED_FILES')
+  assertCleanWorktree(safetyState.changes, 'checkout', 'CHECKOUT_DIRTY_WORKTREE')
 
   logger.info(`仓库根目录: ${context.topLevelPath}`)
   if (context.inheritedFromParent) {
@@ -55,7 +69,6 @@ export async function runCheckoutWorkflow(input) {
   if (input.pullMain && context.defaultBranch !== branchName) {
     logger.info(`同步主分支到当前分支: ${context.defaultBranch}`)
     await git.pull(context.defaultBranch, {
-      allowUnrelatedHistories: true,
       noRebase: true
     })
   }
@@ -63,7 +76,6 @@ export async function runCheckoutWorkflow(input) {
   if (remoteExists) {
     logger.info(`拉取远程分支: ${branchName}`)
     await git.pull(branchName, {
-      allowUnrelatedHistories: true,
       noRebase: true
     })
   }