codemini-cli 0.2.0 → 0.2.2

package/src/tui/chat-app.js

@@ -63,6 +63,7 @@ const TUI_COPY = {
       waitingForInput: '等待输入',
       ready: '就绪',
       noMessagesYet: '还没有消息',
+      taskCompleted: '已完成任务',
       code: '代码',
       codeActivity: '代码活动',
       textNotes: '说明文本',
@@ -140,6 +141,10 @@ const TUI_COPY = {
       doingCodeGeneration: '正在生成代码',
       doneSkill: '已完成技能',
       doingSkill: '正在执行技能',
+      doneProjectIndex: '已初始化项目索引',
+      doingProjectIndex: '正在初始化项目索引',
+      doneFileIndex: '已刷新文件索引',
+      doingFileIndex: '正在刷新文件索引',
       toolFailed: (name) => `工具执行失败: ${name}`,
       waitingModelContinue: (detail) => `${detail}，等待模型继续`,
       waitingModelAdjust: (detail) => `${detail}，等待模型调整`
@@ -189,6 +194,7 @@ const TUI_COPY = {
       waitingForInput: 'waiting for input',
       ready: 'ready',
       noMessagesYet: 'No messages yet',
+      taskCompleted: 'Task completed',
       code: 'code',
       codeActivity: 'CODE ACTIVITY',
       textNotes: 'NOTES',
@@ -266,6 +272,10 @@ const TUI_COPY = {
       doingCodeGeneration: 'Generating code',
       doneSkill: 'Completed skill',
       doingSkill: 'Running skill',
+      doneProjectIndex: 'Project index initialized',
+      doingProjectIndex: 'Initializing project index',
+      doneFileIndex: 'File index refreshed',
+      doingFileIndex: 'Refreshing file index',
       toolFailed: (name) => `Tool failed: ${name}`,
       waitingModelContinue: (detail) => `${detail}, waiting for model to continue`,
       waitingModelAdjust: (detail) => `${detail}, waiting for model to adjust`
@@ -431,6 +441,12 @@ function getActivityDisplayParts(activity) {
       secondary: `(${activity?.name || 'unknown'})`
     };
   }
+  if ((activity?.type || 'tool') === 'system_tool') {
+    return {
+      primary: 'Index',
+      secondary: parsed.target ? `(${parsed.target})` : parsed.base ? `(${parsed.base})` : ''
+    };
+  }
   const labels = {
     read: 'Read',
     edit: 'Edit',
@@ -455,8 +471,32 @@ function getActivityDisplayParts(activity) {
   };
 }
 
+export function isIndexSystemToolName(name) {
+  const parsed = parseToolDisplayName(name);
+  return parsed.base === 'project_index' || parsed.base === 'file_index';
+}
+
+export function shouldShowCompletionFooter(msg) {
+  return Boolean(msg && msg.label === 'coder' && !msg.loading && !(msg.phase || '').trim());
+}
+
 function describeToolActivity(name, copy, { done = false, blocked = false } = {}) {
   const parsed = parseToolDisplayName(name);
+  if (parsed.base === 'project_index') {
+    return blocked
+      ? `${copy.toolActivity.blocked}: project index`
+      : done
+        ? copy.toolActivity.doneProjectIndex
+        : copy.toolActivity.doingProjectIndex;
+  }
+  if (parsed.base === 'file_index') {
+    const safeTarget = trimText(parsed.target || '.codemini-project/file-index.json', 72);
+    return blocked
+      ? `${copy.toolActivity.blocked}: ${safeTarget}`
+      : done
+        ? `${copy.toolActivity.doneFileIndex}: ${safeTarget}`
+        : `${copy.toolActivity.doingFileIndex}: ${safeTarget}`;
+  }
   if (parsed.base === 'run' || parsed.base === 'start_service') {
     const intent = classifyCommandIntent(parsed.target);
     const target = parsed.target || intent.kind || 'command';
@@ -765,8 +805,11 @@ function PlanStrip({ planState, copy }) {
   );
 }
 
-function Header({ sessionId, model, shellName }) {
+function Header({ sessionId, model, shellName, safeMode = true }) {
   const shortSession = String(sessionId || '').slice(-12) || '-';
+  const modeValue = safeMode ? 'SAFE' : 'OPEN';
+  const modeColor = safeMode ? 'greenBright' : 'redBright';
+  const modeTextColor = safeMode ? 'black' : 'white';
   return h(
     Box,
     { width: '100%', justifyContent: 'center', marginTop: 1, marginBottom: 2 },
@@ -781,12 +824,6 @@ function Header({ sessionId, model, shellName }) {
         alignItems: 'center',
         minWidth: 88
       },
-      h(
-        Box,
-        { width: '100%', justifyContent: 'space-between', marginBottom: 1 },
-        h(Text, { color: 'cyan' }, 'CLI'),
-        h(Text, { color: 'greenBright' }, 'SAFE')
-      ),
       ...BANNER.map((line, idx) =>
         h(
           Box,
@@ -801,8 +838,9 @@ function Header({ sessionId, model, shellName }) {
         Box,
         { flexDirection: 'row', justifyContent: 'center' },
         h(StatusPill, { label: 'MODEL', value: model, color: 'cyanBright', textColor: 'black' }),
-        h(StatusPill, { label: 'SHELL', value: shellName || 'powershell', color: 'greenBright', textColor: 'black' }),
-        h(StatusPill, { label: 'SESSION', value: shortSession, color: 'magentaBright', textColor: 'black' })
+        h(StatusPill, { label: 'SHELL', value: shellName || 'powershell', color: 'yellowBright', textColor: 'black' }),
+        h(StatusPill, { label: 'SESSION', value: shortSession, color: 'magentaBright', textColor: 'black' }),
+        h(StatusPill, { label: 'MODE', value: modeValue, color: modeColor, textColor: modeTextColor })
       )
     )
   );
@@ -878,9 +916,16 @@ export function parsePlanProgressLine(text) {
   };
 }
 
-function getTailPreviewLines(text, maxLines = 3) {
+function getTailPreviewWindow(text, maxLines = 3) {
   const source = String(text || '');
-  if (!source.trim()) return [];
+  if (!source.trim()) return { lines: [], startLine: 1 };
+  const sliceTail = (lines) => {
+    const tail = lines.slice(-Math.max(1, maxLines));
+    return {
+      lines: tail,
+      startLine: Math.max(1, lines.length - tail.length + 1)
+    };
+  };
 
   const lines = source.split('\n').map((line) => line.replace(/\r$/, ''));
   let insideFence = false;
@@ -908,21 +953,21 @@ function getTailPreviewLines(text, maxLines = 3) {
   if (insideFence) {
     const codeLines = fenceLines.filter((line) => line.trim().length > 0);
     if (codeLines.length > 0) {
-      return codeLines.slice(-Math.max(1, maxLines));
+      return sliceTail(codeLines);
     }
   }
 
   const closedFenceLines = latestClosedFenceLines.filter((line) => line.trim().length > 0);
   if (closedFenceLines.length > 0) {
-    return closedFenceLines.slice(-Math.max(1, maxLines));
+    return sliceTail(closedFenceLines);
   }
 
   const tailLines = source
     .split('\n')
     .map((line) => line.replace(/\r$/, ''))
     .filter((line) => line.trim().length > 0);
-  if (tailLines.length === 0) return [];
-  return tailLines.slice(-Math.max(1, maxLines));
+  if (tailLines.length === 0) return { lines: [], startLine: 1 };
+  return sliceTail(tailLines);
 }
 
 function collectPreviewStrings(value, out = []) {
@@ -954,7 +999,6 @@ function collectPreviewStrings(value, out = []) {
 function extractPreviewTextFromRawArguments(raw) {
   const source = String(raw || '');
   if (!source.trim()) return '';
-
   const contentMatch = source.match(/"(content|new_content|new_text|patch|code|body|script|source|value)"\s*:\s*"([\s\S]*)$/);
   if (!contentMatch) return '';
 
@@ -962,7 +1006,7 @@ function extractPreviewTextFromRawArguments(raw) {
     .replace(/\\n/g, '\n')
     .replace(/\\"/g, '"')
     .replace(/\\\\/g, '\\')
-    .replace(/",?\s*$/g, '')
+    .replace(/"\s*[,}\]]*\s*$/g, '')
     .trim();
 }
 
@@ -973,40 +1017,76 @@ function compactPreviewLine(line, maxChars = 56) {
   return `${text.slice(0, Math.max(1, maxChars - 3))}...`;
 }
 
+function extractPreviewLinesFromTool(tool, maxLines = 3) {
+  const previewSource =
+    typeof tool?.arguments === 'string'
+      ? (() => {
+          const parsedArguments = safeJsonParse(tool.arguments);
+          if (parsedArguments) {
+            const parsedPreview = collectPreviewStrings(parsedArguments);
+            if (parsedPreview.length > 0) return parsedPreview;
+          }
+          const rawArgumentPreview = extractPreviewTextFromRawArguments(tool.arguments);
+          return rawArgumentPreview ? [rawArgumentPreview] : [];
+        })()
+      : collectPreviewStrings(tool?.arguments || tool?.content || tool?.summary || []);
+
+  if (previewSource.length === 0) return { lines: [], startLine: 1 };
+  const combined = previewSource.join('\n');
+  const previewWindow = getTailPreviewWindow(combined, maxLines);
+  return {
+    lines: previewWindow.lines.map((line) => compactPreviewLine(line)),
+    startLine: previewWindow.startLine
+  };
+}
+
 function getLatestToolPreviewLines(msg, maxLines = 3) {
-  const toolCalls = [
-    ...(Array.isArray(msg?.pendingToolCalls) ? msg.pendingToolCalls : []),
-    ...(Array.isArray(msg?.toolCalls) ? msg.toolCalls : [])
-  ];
   const codeTools = new Set(['edit', 'write', 'patch', 'generate_diff']);
-  for (let index = toolCalls.length - 1; index >= 0; index -= 1) {
-    const tool = toolCalls[index];
-    const parsed = parseToolDisplayName(tool?.name);
-    if (!codeTools.has(parsed.base)) continue;
-    const rawArgumentPreview =
-      typeof tool?.arguments === 'string' ? extractPreviewTextFromRawArguments(tool.arguments) : '';
-    const previewSource = rawArgumentPreview
-      ? [rawArgumentPreview]
-      : collectPreviewStrings(tool?.arguments || tool?.content || tool?.summary || []);
-    if (previewSource.length === 0) continue;
-    const combined = previewSource.join('\n');
-    const previewLines = getTailPreviewLines(combined, maxLines);
-    if (previewLines.length > 0) return previewLines.map((line) => compactPreviewLine(line));
+  const extractFromCalls = (calls) => {
+    for (let index = calls.length - 1; index >= 0; index -= 1) {
+      const tool = calls[index];
+      const parsed = parseToolDisplayName(tool?.name);
+      if (!codeTools.has(parsed.base)) continue;
+      const previewWindow = extractPreviewLinesFromTool(tool, maxLines);
+      if (previewWindow.lines.length > 0) return previewWindow;
+    }
+    return { lines: [], startLine: 1 };
+  };
+
+  const pendingCodeCalls = (Array.isArray(msg?.pendingToolCalls) ? msg.pendingToolCalls : []).filter((tool) =>
+    codeTools.has(parseToolDisplayName(tool?.name).base)
+  );
+  if (pendingCodeCalls.length > 0) {
+    const latestPendingPreview = extractPreviewLinesFromTool(pendingCodeCalls.at(-1), maxLines);
+    if (latestPendingPreview.lines.length > 0) return latestPendingPreview;
+    return { lines: [], startLine: 1 };
   }
-  return [];
+
+  const toolCalls = (Array.isArray(msg?.toolCalls) ? msg.toolCalls : []).filter(
+    (tool) => tool?.status === 'running' && codeTools.has(parseToolDisplayName(tool?.name).base)
+  );
+  if (toolCalls.length > 0) {
+    return extractFromCalls(toolCalls);
+  }
+
+  return { lines: [], startLine: 1 };
 }
 
 export function getGeneratingCodePlaceholderRows(msg, copy, contentWidth = 72) {
   const liveStatus = String(msg?.liveStatus || '').trim();
   if (!msg?.loading || (msg?.phase !== 'generating' && msg?.phase !== 'tooling')) return [];
-  if (liveStatus !== String(copy?.runtime?.generatingCode || '').trim()) return [];
 
-  const previewLines = getLatestToolPreviewLines(msg, 3);
-  if (previewLines.length === 0) return [];
+  const previewWindow = getLatestToolPreviewLines(msg, 3);
+  if (previewWindow.lines.length === 0) return [];
+  const hasRunningCodeTool = (Array.isArray(msg?.toolCalls) ? msg.toolCalls : []).some(
+    (tool) => tool?.status === 'running' && new Set(['edit', 'write', 'patch', 'generate_diff']).has(parseToolDisplayName(tool?.name).base)
+  );
+  const isCodeGenerationStatus = liveStatus === String(copy?.runtime?.generatingCode || '').trim();
+  if (!isCodeGenerationStatus && !(msg?.phase === 'tooling' && hasRunningCodeTool)) return [];
 
-  return previewLines.map((line, idx) => ({
+  return previewWindow.lines.map((line, idx) => ({
     kind: 'code-placeholder',
-    lineNo: idx + 1,
+    lineNo: previewWindow.startLine + idx,
     text: line,
     color: 'gray'
   }));
@@ -1321,7 +1401,7 @@ export function insertRowsAfterLastCodeRow(rows, extraRows) {
 
   let insertIndex = -1;
   for (let index = source.length - 1; index >= 0; index -= 1) {
-    if (source[index]?.kind === 'code') {
+    if (source[index]?.kind === 'code' || source[index]?.kind === 'code-placeholder') {
       insertIndex = index + 1;
       break;
     }
@@ -1498,10 +1578,18 @@ function buildMessageRows(msg, showToolDetails, contentWidth = 72, copy) {
   };
 
   if (Array.isArray(msg?.segments) && msg.segments.length > 0) {
-    const totalTools = msg.segments.filter((segment) => segment.type === 'tool' || segment.type === 'skill').length;
+    const totalTools = msg.segments.filter(
+      (segment) =>
+        segment.type === 'tool' ||
+        segment.type === 'skill' ||
+        (segment.type === 'system_tool' && (showToolDetails || !isIndexSystemToolName(segment.name)))
+    ).length;
     let toolIndex = 0;
     for (const segment of msg.segments) {
-      if (segment.type === 'tool' || segment.type === 'skill') {
+      if (segment.type === 'tool' || segment.type === 'skill' || segment.type === 'system_tool') {
+        if (segment.type === 'system_tool' && !showToolDetails && isIndexSystemToolName(segment.name)) {
+          continue;
+        }
         pushActivityRows(segment, toolIndex, totalTools);
         toolIndex += 1;
       } else {
@@ -1548,6 +1636,10 @@ function renderMessageRow(msg, row, idx, loaderTick) {
         ? row.status === 'error'
           ? 'redBright'
           : 'cyanBright'
+        : activity.type === 'system_tool'
+          ? row.status === 'error' || row.status === 'blocked'
+            ? 'redBright'
+            : 'blueBright'
         : row.status === 'error' || row.status === 'blocked'
           ? 'redBright'
           : 'cyanBright';
@@ -1783,7 +1875,14 @@ function MessageBubble({ msg, loaderTick, showToolDetails, rowWindow = null, con
           ? h(Text, { color: 'blueBright' }, autoSkillBadge)
           : h(Text, { color: theme.chrome }, ' ')
       ),
-      ...rendered
+      ...rendered,
+      shouldShowCompletionFooter(msg)
+        ? h(
+            Box,
+            { marginTop: 1, marginLeft: 1, key: `row-completion-${msg.id}` },
+            h(Text, { color: 'gray', dimColor: true }, copy.generic.taskCompleted)
+          )
+        : null
     )
   );
 }
@@ -2017,7 +2116,7 @@ function makeIdleStatus(copy, snapshot, variant = 'ready') {
   );
 }
 
-export function ChatApp({ runtime, sessionId, model, language = 'zh', shellName = 'powershell', version = '' }) {
+export function ChatApp({ runtime, sessionId, model, language = 'zh', shellName = 'powershell', version = '', safeMode = true }) {
   const copy = getCopy(language);
   const stdoutCols = Number(process.stdout?.columns || 120);
   const [inputValue, setInputValue] = useState('');
@@ -2060,6 +2159,23 @@ export function ChatApp({ runtime, sessionId, model, language = 'zh', shellName
   const messagesRef = useRef([]);
   const pendingQueueRef = useRef([]);
   const deltaBufferRef = useRef('');
+
+  useEffect(() => {
+    const rawStartupActivities = runtime.consumeStartupEvents?.();
+    const startupActivities = Array.isArray(rawStartupActivities) ? rawStartupActivities : [];
+    if (startupActivities.length === 0) return;
+    setMessages((prev) => [
+      ...prev,
+      {
+        id: nextId(),
+        label: 'system',
+        text: '',
+        color: 'yellowBright',
+        toolCalls: startupActivities,
+        segments: startupActivities
+      }
+    ]);
+  }, [runtime]);
   const deltaFlushTimerRef = useRef(null);
   const escSeqRef = useRef('');
   const planTextBufferRef = useRef('');
@@ -2191,6 +2307,7 @@ export function ChatApp({ runtime, sessionId, model, language = 'zh', shellName
       prev.map((m) => {
         if (m.id !== targetId) return m;
         const toolCalls = Array.isArray(m.toolCalls) ? [...m.toolCalls] : [];
+        const pendingToolCalls = Array.isArray(m.pendingToolCalls) ? [...m.pendingToolCalls] : [];
         const activityType = toolEvent.type || 'tool';
         const idx = findActivityUpdateIndex(toolCalls, toolEvent);
         const startedAt = toolEvent.status === 'running' ? Date.now() : undefined;
@@ -2243,7 +2360,16 @@ export function ChatApp({ runtime, sessionId, model, language = 'zh', shellName
               : {})
           };
         }
-        return { ...m, toolCalls, segments };
+        const shouldClearPending =
+          activityType === 'tool' && ['running', 'done', 'blocked', 'error'].includes(toolEvent.status);
+        const nextPendingToolCalls = shouldClearPending
+          ? pendingToolCalls.filter((entry) => {
+              if (!entry) return false;
+              if (toolEvent.id && entry.id) return entry.id !== toolEvent.id;
+              return parseToolDisplayName(entry.name).base !== parseToolDisplayName(toolEvent.name).base;
+            })
+          : pendingToolCalls;
+        return { ...m, toolCalls, segments, pendingToolCalls: nextPendingToolCalls };
       })
     );
   };
@@ -2582,6 +2708,54 @@ export function ChatApp({ runtime, sessionId, model, language = 'zh', shellName
             arguments: event.arguments
           });
         }
+        if (event?.type === 'system_tool:start') {
+          ensureActiveAssistant();
+          const detail = describeToolActivity(event.name, copy);
+          setRuntimeStatus(makeStatus(copy.runtime.toolRunning, detail, 'blueBright'));
+          setInputStage('tooling');
+          updateActivityStatusOnActiveAssistant({
+            type: 'system_tool',
+            id: event.id,
+            name: event.name,
+            status: 'running',
+            summary: event.summary
+          });
+          setActiveAssistantMeta({ loading: true, phase: 'tooling', liveStatus: detail });
+        }
+        if (event?.type === 'system_tool:end') {
+          const detail = describeToolActivity(event.name, copy, { done: true });
+          setRuntimeStatus(makeStatus(copy.runtime.toolCompleted, copy.toolActivity.waitingModelContinue(detail), 'blueBright'));
+          setInputStage('thinking');
+          updateActivityStatusOnActiveAssistant({
+            type: 'system_tool',
+            id: event.id,
+            name: event.name,
+            status: 'done',
+            summary: event.summary
+          });
+          setActiveAssistantMeta({
+            loading: true,
+            phase: 'thinking',
+            liveStatus: copy.toolActivity.waitingModelContinue(detail)
+          });
+        }
+        if (event?.type === 'system_tool:error') {
+          const detail = copy.toolActivity.toolFailed(event.name);
+          setRuntimeStatus(makeStatus(copy.runtime.toolFailed, event.summary || detail, 'redBright'));
+          setInputStage('thinking');
+          updateActivityStatusOnActiveAssistant({
+            type: 'system_tool',
+            id: event.id,
+            name: event.name,
+            status: 'error',
+            summary: event.summary
+          });
+          setActiveAssistantMeta({
+            loading: true,
+            phase: 'thinking',
+            liveStatus: copy.toolActivity.waitingModelAdjust(detail)
+          });
+        }
         if (event?.type === 'skill:start') {
           ensureActiveAssistant();
           const detail = describeSkillActivity(event.name, copy);
@@ -3087,7 +3261,7 @@ export function ChatApp({ runtime, sessionId, model, language = 'zh', shellName
   return h(
     Box,
     { flexDirection: 'column' },
-    h(Header, { sessionId: displaySessionId, model: displayModel, shellName }),
+    h(Header, { sessionId: displaySessionId, model: displayModel, shellName, safeMode }),
     h(MessageList, {
       messages: visibleMessages,
       loaderTick,