whale-code 6.4.0

package/dist/cli/chat/ToolIndicator.js

@@ -0,0 +1,436 @@
+import { jsx as _jsx, jsxs as _jsxs, Fragment as _Fragment } from "react/jsx-runtime";
+/**
+ * ToolIndicator — polished tool call rendering
+ *
+ * Every result gets syntax highlighting via MarkdownText.
+ * Financial-aware: green for gains, red for deductions/negatives.
+ * Params: purple keys, typed values (blue dates, green money, red negatives).
+ * Duration badge, tool type glyph.
+ */
+import React, { useMemo, useState, useEffect } from "react";
+import { Box, Text } from "ink";
+import Spinner from "ink-spinner";
+import { MarkdownText } from "./MarkdownText.js";
+import { colors } from "../shared/Theme.js";
+import { contentWidth } from "../shared/markdown.js";
+import { isLocalTool } from "../services/local-tools.js";
+import os from "os";
+// ============================================================================
+// CONSTANTS
+// ============================================================================
+const AUTO_EXPAND_THRESHOLD = 12;
+const PREVIEW_LINES = 6;
+// ============================================================================
+// HELPERS
+// ============================================================================
+/** Shorten an absolute path for display: strip cwd, collapse home, truncate */
+export function shortenPath(fullPath, maxLen = 40) {
+    let p = fullPath;
+    const cwd = process.cwd();
+    if (p.startsWith(cwd + "/"))
+        p = p.slice(cwd.length + 1);
+    else if (p.startsWith(cwd))
+        p = p.slice(cwd.length);
+    else {
+        const home = os.homedir();
+        if (p.startsWith(home))
+            p = "~" + p.slice(home.length);
+    }
+    if (p.length <= maxLen)
+        return p;
+    const parts = p.split("/");
+    const file = parts.pop();
+    if (file.length >= maxLen - 4)
+        return "…/" + file.slice(-(maxLen - 4));
+    const parent = parts.pop();
+    return parent ? "…/" + parent + "/" + file : "…/" + file;
+}
+function formatDuration(ms) {
+    if (ms < 1000)
+        return `${ms}ms`;
+    if (ms < 10000)
+        return `${(ms / 1000).toFixed(1)}s`;
+    return `${Math.round(ms / 1000)}s`;
+}
+/** Compact one-line context */
+function formatContext(name, input) {
+    if (!input)
+        return "";
+    if (name === "read_file" || name === "write_file" || name === "edit_file" || name === "list_directory") {
+        return input.path ? shortenPath(String(input.path)) : "";
+    }
+    if (name === "run_command") {
+        let cmd = String(input.command || "");
+        // Mask passwords/secrets in display
+        cmd = cmd.replace(/(?:PASSWORD|SECRET|TOKEN|KEY)=['"][^'"]*['"]/gi, (m) => {
+            const eq = m.indexOf("=");
+            return m.slice(0, eq + 1) + "'••••'";
+        });
+        // Adaptive truncation — fit within one terminal line
+        const maxCtx = Math.min(60, Math.max(20, contentWidth() - 40));
+        return cmd.length > maxCtx ? cmd.slice(0, maxCtx - 1) + "…" : cmd;
+    }
+    if (name === "search_files") {
+        const parts = [input.pattern, input.path ? shortenPath(String(input.path)) : null].filter(Boolean);
+        return parts.join(" in ");
+    }
+    if (name === "search_content") {
+        const q = String(input.query || "");
+        const p = input.path ? ` in ${shortenPath(String(input.path))}` : "";
+        return q.length > 30 ? q.slice(0, 27) + "…" + p : q + p;
+    }
+    if (name === "grep") {
+        const parts = [input.pattern, input.path ? shortenPath(String(input.path)) : null].filter(Boolean);
+        return parts.join(" in ");
+    }
+    if (name === "glob") {
+        const parts = [input.pattern, input.path ? shortenPath(String(input.path)) : null].filter(Boolean);
+        return parts.join(" in ");
+    }
+    if (name === "multi_edit") {
+        const edits = (input.edits ?? []);
+        return input.file_path ? `${shortenPath(String(input.file_path))} (${edits.length} edits)` : "";
+    }
+    if (name === "task") {
+        const type = input.subagent_type || "";
+        const model = input.model || "sonnet";
+        return `${type} (${model})`;
+    }
+    if (name === "lsp") {
+        const op = input.operation || "";
+        const fp = input.filePath ? shortenPath(String(input.filePath)) : "";
+        const ln = input.line ? `:${input.line}` : "";
+        return `${op} ${fp}${ln}`;
+    }
+    // Kali — show the command being executed (most useful context)
+    if (name === "kali" && input.command) {
+        let cmd = String(input.command);
+        const maxCtx = Math.min(60, Math.max(20, contentWidth() - 40));
+        cmd = cmd.length > maxCtx ? cmd.slice(0, maxCtx - 1) + "…" : cmd;
+        const action = input.action ? String(input.action) + " " : "";
+        return action + cmd;
+    }
+    // Server tools → action + key param
+    if (input.action) {
+        const parts = [String(input.action)];
+        if (input.query)
+            parts.push(String(input.query).slice(0, 25));
+        else if (input.command)
+            parts.push(String(input.command).slice(0, 40));
+        else if (input.name)
+            parts.push(String(input.name));
+        else if (input.period)
+            parts.push(String(input.period));
+        else if (input.product_id)
+            parts.push(String(input.product_id).slice(0, 12));
+        else if (input.location_id)
+            parts.push(String(input.location_id).slice(0, 12));
+        else if (input.customer_id)
+            parts.push(String(input.customer_id).slice(0, 12));
+        else if (input.order_id)
+            parts.push(String(input.order_id).slice(0, 12));
+        return parts.join(" ");
+    }
+    return "";
+}
+const LANG_MAP = {
+    ts: "typescript", tsx: "tsx", js: "javascript", jsx: "jsx",
+    py: "python", rs: "rust", go: "go", rb: "ruby",
+    swift: "swift", kt: "kotlin", java: "java", c: "c", cpp: "cpp",
+    css: "css", html: "html", json: "json", yaml: "yaml", yml: "yaml",
+    toml: "toml", md: "markdown", sh: "bash", zsh: "bash",
+    sql: "sql", xml: "xml",
+};
+const CATEGORY_STYLES = {
+    read: { icon: "◇", color: colors.info },
+    write: { icon: "◆", color: colors.success },
+    edit: { icon: "±", color: colors.warning },
+    search: { icon: "⊕", color: colors.purple },
+    command: { icon: "$", color: colors.pink },
+    directory: { icon: "▤", color: colors.indigo },
+    web: { icon: "◎", color: colors.brand },
+    agent: { icon: "⊛", color: colors.purple },
+    todo: { icon: "☐", color: colors.success },
+    notebook: { icon: "◫", color: colors.warning },
+    server: { icon: "▹", color: colors.pink },
+    lsp: { icon: "⊞", color: colors.info },
+    interactive: { icon: "▹", color: colors.indigo },
+};
+const TOOL_CATEGORY_MAP = {
+    read_file: "read",
+    write_file: "write",
+    edit_file: "edit",
+    multi_edit: "edit",
+    search_files: "search",
+    search_content: "search",
+    glob: "search",
+    grep: "search",
+    run_command: "command",
+    list_directory: "directory",
+    web_fetch: "web",
+    web_search: "web",
+    task: "agent",
+    team_create: "agent",
+    tasks: "todo",
+    config: "command",
+    ask_user: "command",
+    bash_output: "command",
+    kill_shell: "command",
+    list_shells: "command",
+    notebook_edit: "notebook",
+    task_output: "agent",
+    task_stop: "agent",
+    lsp: "lsp",
+    enter_plan_mode: "interactive",
+    exit_plan_mode: "interactive",
+    ask_user_question: "interactive",
+    skill: "command",
+};
+function getToolCategory(name) {
+    return TOOL_CATEGORY_MAP[name] || (isLocalTool(name) ? "command" : "server");
+}
+// Human-readable tool display names (Claude Code parity)
+const TOOL_DISPLAY_NAMES = {
+    read_file: "Read",
+    write_file: "Write",
+    edit_file: "Edit",
+    multi_edit: "MultiEdit",
+    search_files: "Search",
+    search_content: "Search",
+    glob: "Glob",
+    grep: "Grep",
+    run_command: "Bash",
+    list_directory: "List",
+    web_fetch: "WebFetch",
+    web_search: "WebSearch",
+    task: "Task",
+    team_create: "Team",
+    tasks: "Tasks",
+    config: "Config",
+    ask_user: "AskUser",
+    bash_output: "TaskOutput",
+    kill_shell: "TaskStop",
+    list_shells: "Tasks",
+    notebook_edit: "NotebookEdit",
+    task_output: "TaskOutput",
+    task_stop: "TaskStop",
+    lsp: "LSP",
+    enter_plan_mode: "PlanMode",
+    exit_plan_mode: "PlanMode",
+    ask_user_question: "AskUser",
+    skill: "Skill",
+};
+export function getDisplayName(name) {
+    return TOOL_DISPLAY_NAMES[name] || name;
+}
+/** Shorten absolute paths in search/grep results for readable display */
+function formatSearchResult(result) {
+    return result.split("\n").map(line => {
+        // Match "path:line:col:content" or "path:line:content" (grep/ripgrep output)
+        const grepMatch = line.match(/^(\/[^:]+):(\d+(?::\d+)?):(.*)$/);
+        if (grepMatch) {
+            const [, filePath, lineCol, rest] = grepMatch;
+            return `${shortenPath(filePath, 35)}:${lineCol}:${rest}`;
+        }
+        // Match plain absolute paths (glob results)
+        if (line.startsWith("/")) {
+            return shortenPath(line.trim(), 60);
+        }
+        return line;
+    }).join("\n");
+}
+const PLAIN_TEXT_TOOLS = new Set(["enter_plan_mode", "exit_plan_mode", "ask_user_question", "skill"]);
+function detectLang(toolName, input) {
+    if (toolName === "edit_file" || toolName === "multi_edit")
+        return "diff";
+    if (toolName === "read_file" && input?.path) {
+        const p = String(input.path);
+        const base = p.split("/").pop() || "";
+        // .env, .env.local, .env.production, etc. → bash (KEY=VALUE syntax)
+        if (base.startsWith(".env"))
+            return "bash";
+        const ext = base.split(".").pop()?.toLowerCase() || "";
+        return LANG_MAP[ext] || "";
+    }
+    if (toolName === "run_command" || toolName === "list_directory")
+        return "bash";
+    // Interactive/plain tools — render as plain text or markdown (not JSON fences)
+    if (PLAIN_TEXT_TOOLS.has(toolName))
+        return "";
+    if (!isLocalTool(toolName))
+        return "json";
+    return "";
+}
+function wrapInFence(content, lang, subtitle) {
+    if (content.includes("```"))
+        return content;
+    // Detect formatted markdown (tables, bold, headings) — render as markdown, not fenced
+    const t = content.trim();
+    if (t.startsWith("**") || t.startsWith("| ") || t.startsWith("# "))
+        return content;
+    // For server tools, detect JSON
+    if (!lang) {
+        if ((t.startsWith("{") || t.startsWith("[")) && (t.endsWith("}") || t.endsWith("]"))) {
+            lang = "json";
+        }
+    }
+    // Only add :subtitle when lang is non-empty (prevents "```:path" which has no valid lang)
+    const fence = (lang && subtitle) ? lang + ":" + subtitle : lang;
+    return "```" + fence + "\n" + content + "\n```";
+}
+// ============================================================================
+// COMPONENT
+// ============================================================================
+export const ToolIndicator = React.memo(function ToolIndicator({ id: _id, name, status, result, input, durationMs, expanded = false, count }) {
+    const context = useMemo(() => formatContext(name, input), [name, input]);
+    const lineCount = useMemo(() => result ? result.split("\n").length : 0, [result]);
+    // Elapsed time counter for running tools
+    const [elapsed, setElapsed] = useState(0);
+    useEffect(() => {
+        if (status !== "running")
+            return;
+        setElapsed(0);
+        const t = setInterval(() => setElapsed(e => e + 1), 1000);
+        return () => clearInterval(t);
+    }, [status]);
+    // Detect lang — writes with diffs get "diff" treatment
+    const lang = useMemo(() => {
+        const base = detectLang(name, input);
+        if (base)
+            return base;
+        if (name === "write_file" && result?.includes("\n@@"))
+            return "diff";
+        return base;
+    }, [name, input, result]);
+    // Category-based styling
+    const category = getToolCategory(name);
+    const catStyle = CATEGORY_STYLES[category];
+    // Extract file path for code block subtitle
+    const filePath = useMemo(() => {
+        if (!input)
+            return undefined;
+        const p = input.path || input.file_path;
+        return p ? String(p) : undefined;
+    }, [input]);
+    // Category-specific summary metrics
+    const summary = useMemo(() => {
+        if (!result || status !== "success")
+            return null;
+        if (category === "read") {
+            const lc = result.split("\n").length;
+            return { type: "read", label: `Read ${lc} line${lc !== 1 ? "s" : ""}` };
+        }
+        if (category === "write") {
+            // Overwrite with diff — show +N -N badge
+            const diffMatch = result.match(/Added (\d+) lines?, removed (\d+) lines?/i);
+            if (diffMatch) {
+                return { type: "edit", added: parseInt(diffMatch[1]), removed: parseInt(diffMatch[2]) };
+            }
+            const lineMatch = result.match(/\((\d+) lines?, (\d+) chars\)/);
+            if (lineMatch) {
+                const lines = parseInt(lineMatch[1]);
+                const label = `Wrote ${lines} line${lines !== 1 ? "s" : ""}`;
+                return { type: "write", label };
+            }
+            const charMatch = result.match(/\((\d+) chars\)/);
+            if (charMatch) {
+                const chars = parseInt(charMatch[1]);
+                const label = chars >= 1000 ? `Wrote ${(chars / 1000).toFixed(1)}K chars` : `Wrote ${chars} chars`;
+                return { type: "write", label };
+            }
+            return { type: "write", label: "Written" };
+        }
+        if (category === "edit") {
+            let added = 0, removed = 0;
+            for (const line of result.split("\n")) {
+                if (line.startsWith("+"))
+                    added++;
+                else if (line.startsWith("-"))
+                    removed++;
+            }
+            if (added > 0 || removed > 0)
+                return { type: "edit", added, removed };
+        }
+        if (category === "search") {
+            const lines = result.split("\n").filter(l => l.trim());
+            const files = new Set(lines.map(l => l.split(":")[0]).filter(f => f.includes("/") || f.includes(".")));
+            if (lines.length > 0)
+                return { type: "search", matches: lines.length, files: files.size };
+        }
+        if (category === "command") {
+            const lines = result.split("\n").filter(l => l.trim());
+            if (lines.length > 0)
+                return { type: "command", label: `${lines.length} line${lines.length !== 1 ? "s" : ""} output` };
+        }
+        if (category === "directory") {
+            const items = result.split("\n").filter(l => l.trim());
+            if (items.length > 0)
+                return { type: "directory", label: `${items.length} item${items.length !== 1 ? "s" : ""}` };
+        }
+        if (category === "web") {
+            const chars = result.length;
+            const label = chars >= 1000 ? `${(chars / 1000).toFixed(1)}K chars` : `${chars} chars`;
+            return { type: "web", label: `fetched ${label}` };
+        }
+        // Server tools with formatted markdown — extract bold title as summary
+        if (category === "server" && result.trim().startsWith("**")) {
+            const firstLine = result.trim().split("\n")[0];
+            const match = firstLine.match(/^\*\*(.+?)\*\*/);
+            if (match)
+                return { type: "server", label: match[1] };
+        }
+        return null;
+    }, [result, status, category]);
+    // Live output lines (last 6) for running commands
+    const liveLines = useMemo(() => {
+        if (status !== "running" || !result)
+            return [];
+        return result.split("\n").filter(l => l.trim()).slice(-6);
+    }, [status, result]);
+    // ── RUNNING ──
+    if (status === "running") {
+        return (_jsxs(Box, { flexDirection: "column", children: [_jsxs(Box, { children: [_jsx(Text, { color: catStyle.color, children: _jsx(Spinner, { type: "dots" }) }), _jsxs(Text, { color: catStyle.color, children: [" ", catStyle.icon] }), _jsxs(Text, { color: catStyle.color, bold: true, children: [" ", getDisplayName(name)] }), context ? _jsxs(Text, { color: colors.dim, children: ["  ", context] }) : null, elapsed >= 2 && _jsxs(Text, { color: elapsed >= 30 ? colors.warning : colors.dim, children: ["  ", formatDuration(elapsed * 1000)] })] }), liveLines.length > 0 && (_jsx(Box, { flexDirection: "column", marginLeft: 4, children: liveLines.map((line, i) => (_jsx(Text, { color: colors.tertiary, wrap: "truncate", children: line }, i))) }))] }));
+    }
+    // ── ERROR ──
+    if (status === "error") {
+        return (_jsxs(Box, { flexDirection: "column", children: [_jsxs(Box, { children: [_jsx(Text, { color: colors.error, bold: true, children: "\u2715" }), _jsxs(Text, { color: catStyle.color, children: [" ", catStyle.icon] }), _jsxs(Text, { color: catStyle.color, bold: true, children: [" ", getDisplayName(name)] }), context ? _jsxs(Text, { color: colors.dim, children: ["  ", context] }) : null, durationMs !== undefined && _jsxs(Text, { color: colors.dim, children: ["  ", formatDuration(durationMs)] })] }), result && (_jsx(Box, { marginLeft: 2, children: _jsx(MarkdownText, { text: "```\n" + result.split("\n").slice(0, PREVIEW_LINES).join("\n") + "\n```" }) }))] }));
+    }
+    // ── SUCCESS ──
+    const hasResult = !!(result && result.trim());
+    const isShort = lineCount <= AUTO_EXPAND_THRESHOLD;
+    // Read/write results stay collapsed (just header line) — Claude Code parity.
+    // Edit/search/command results auto-expand if short enough.
+    // Writes with diffs auto-expand like edits.
+    const hasDiff = lang === "diff";
+    const collapseByDefault = category === "read" || (category === "write" && !hasDiff) || category === "directory";
+    // Interactive tools (plan mode) always expand to show their content
+    const alwaysExpand = category === "interactive";
+    const isGrouped = (count ?? 0) > 1;
+    const showFull = hasResult && !isGrouped && (expanded || alwaysExpand || (isShort && !collapseByDefault));
+    // For grouped tools, extract a one-line summary so they aren't completely invisible
+    const groupSummaryLine = useMemo(() => {
+        if (!isGrouped || !result)
+            return null;
+        const firstLine = result.split("\n").find(l => l.trim() && !l.startsWith("**Success**") && !l.startsWith("**Killed**"));
+        if (!firstLine)
+            return null;
+        const trimmed = firstLine.replace(/^\*\*\w+\*\*:\s*/, "").trim();
+        return trimmed.length > 80 ? trimmed.slice(0, 77) + "…" : trimmed;
+    }, [isGrouped, result]);
+    return (_jsxs(Box, { flexDirection: "column", children: [_jsxs(Box, { children: [_jsx(Text, { color: colors.success, children: "\u2713" }), _jsxs(Text, { color: catStyle.color, children: [" ", catStyle.icon] }), _jsxs(Text, { color: catStyle.color, bold: true, children: [" ", getDisplayName(name)] }), context ? _jsxs(Text, { color: colors.dim, children: ["  ", context] }) : null, durationMs !== undefined && (durationMs > 3000
+                        ? _jsxs(Text, { color: colors.warning, children: ["  ", formatDuration(durationMs)] })
+                        : _jsxs(Text, { color: colors.dim, children: ["  ", formatDuration(durationMs)] })), summary?.type === "edit" && category === "edit" && (_jsxs(_Fragment, { children: [_jsxs(Text, { color: colors.success, children: ["  +", summary.added] }), _jsxs(Text, { color: colors.error, children: [" -", summary.removed] })] })), summary?.type === "search" && (_jsxs(Text, { color: colors.dim, children: ["  ", summary.matches, " match", summary.matches !== 1 ? "es" : "", summary.files > 0 ? ` in ${summary.files} file${summary.files !== 1 ? "s" : ""}` : ""] })), summary && "label" in summary && summary.type !== "server" && (_jsxs(Text, { color: colors.dim, children: ["  ", summary.label] })), summary?.type === "server" && (_jsxs(Text, { color: colors.info, children: ["  ", summary.label] })), !summary && hasResult && !showFull && _jsxs(Text, { color: colors.dim, children: ["  ", lineCount, " lines"] }), (count ?? 0) > 1 && _jsxs(Text, { color: colors.dim, dimColor: true, children: ["  \u00D7 ", count] })] }), isGrouped && groupSummaryLine && (_jsx(Box, { marginLeft: 4, children: _jsx(Text, { color: colors.tertiary, wrap: "truncate", children: groupSummaryLine }) })), category === "write" && hasDiff && summary?.type === "edit" && (_jsxs(Box, { marginLeft: 2, children: [_jsx(Text, { color: colors.tertiary, children: "\u2514 Added " }), _jsx(Text, { color: colors.success, children: summary.added }), _jsx(Text, { color: colors.tertiary, children: " lines, removed " }), _jsx(Text, { color: colors.error, children: summary.removed }), _jsx(Text, { color: colors.tertiary, children: " lines" })] })), showFull && (_jsx(Box, { marginLeft: 2, flexDirection: "column", children: _jsx(MarkdownText, { text: category === "agent"
+                        ? result
+                        : wrapInFence(category === "search" ? formatSearchResult(result) : result, lang, filePath) }) })), hasResult && !showFull && !collapseByDefault && !isGrouped && (_jsxs(Box, { flexDirection: "column", marginLeft: 2, children: [_jsx(MarkdownText, { text: category === "agent"
+                            ? result.split("\n").slice(0, PREVIEW_LINES).join("\n")
+                            : wrapInFence((category === "search" ? formatSearchResult(result) : result).split("\n").slice(0, PREVIEW_LINES).join("\n"), lang, filePath) }), _jsxs(Text, { color: colors.quaternary, children: ["    \u2514 +", lineCount - PREVIEW_LINES, " lines  "] }), _jsx(Text, { color: colors.tertiary, dimColor: true, children: "^E" })] }))] }));
+}, (prev, next) => {
+    // Custom comparator: skip deep-comparing input object reference
+    return prev.id === next.id
+        && prev.status === next.status
+        && prev.expanded === next.expanded
+        && prev.result === next.result
+        && prev.durationMs === next.durationMs
+        && prev.name === next.name
+        && prev.count === next.count;
+});

package/dist/cli/chat/hooks/useAgentLoop.d.ts

@@ -0,0 +1,39 @@
+/**
+ * useAgentLoop — agent loop runner extracted from ChatApp
+ *
+ * All consumers should import from ChatApp (re-export facade).
+ */
+import type Anthropic from "@anthropic-ai/sdk";
+import { RewindManager } from "../../services/rewind.js";
+import type { ChatMessage, ToolCall } from "../MessageList.js";
+import type { SubagentActivityState, CompletedSubagentInfo } from "../SubagentPanel.js";
+import type { ImageAttachment } from "../ChatInput.js";
+export interface AgentLoopDeps {
+    isStreaming: boolean;
+    thinkingEnabled: boolean;
+    conversationRef: React.MutableRefObject<Anthropic.MessageParam[]>;
+    abortRef: React.MutableRefObject<AbortController | null>;
+    accTextRef: React.MutableRefObject<string>;
+    lastContentUpdateRef: React.MutableRefObject<number>;
+    thinkingVerbRef: React.MutableRefObject<string>;
+    rewindManagerRef: React.MutableRefObject<RewindManager>;
+    turnIndexRef: React.MutableRefObject<number>;
+    setMessages: React.Dispatch<React.SetStateAction<ChatMessage[]>>;
+    setStreamingText: React.Dispatch<React.SetStateAction<string>>;
+    setIsStreaming: React.Dispatch<React.SetStateAction<boolean>>;
+    setActiveTools: React.Dispatch<React.SetStateAction<ToolCall[]>>;
+    setSubagentActivity: React.Dispatch<React.SetStateAction<Map<string, SubagentActivityState>>>;
+    setCompletedSubagents: React.Dispatch<React.SetStateAction<CompletedSubagentInfo[]>>;
+    setTeamState: React.Dispatch<React.SetStateAction<{
+        name: string;
+        tasksCompleted: number;
+        tasksTotal: number;
+        teammates: Map<string, {
+            name: string;
+            status: string;
+        }>;
+    } | null>>;
+}
+export declare function useAgentLoop(deps: AgentLoopDeps): {
+    handleSend: (userMessage: string, images?: ImageAttachment[]) => Promise<void>;
+};