whale-code 6.4.0

package/dist/cli/chat/ChatApp.js

@@ -0,0 +1,267 @@
+import { jsx as _jsx, jsxs as _jsxs, Fragment as _Fragment } from "react/jsx-runtime";
+/**
+ * ChatApp — whale code CLI
+ *
+ * Uses Ink's <Static> for completed messages — written to stdout once,
+ * never re-rendered. Only the active area (streaming, tools, input)
+ * is managed by Ink's render loop. This prevents scroll bounce when
+ * content exceeds the terminal height.
+ */
+import { useState, useEffect, useRef, useMemo, useCallback } from "react";
+import { Box, Text, Static, useApp, useInput } from "ink";
+import Spinner from "ink-spinner";
+import { canUseAgent, getServerToolCount, getModelShortName, setModel, getPermissionMode, mcpClientManager, } from "../services/agent-loop.js";
+import { CompletedMessage } from "./MessageList.js";
+import { ToolIndicator } from "./ToolIndicator.js";
+import { SubagentPanel } from "./SubagentPanel.js";
+import { TeamPanel } from "./TeamPanel.js";
+import { StreamingText } from "./StreamingText.js";
+import { ChatInput } from "./ChatInput.js";
+import { StoreSelector } from "./StoreSelector.js";
+import { ModelSelector } from "./ModelSelector.js";
+import { RewindViewer, RewindOutcome } from "./RewindViewer.js";
+import { RewindManager } from "../services/rewind.js";
+import { colors, symbols } from "../shared/Theme.js";
+import { loadKeybindings, matchesBinding } from "../services/keybinding-manager.js";
+import { loadConfig, updateConfig } from "../services/config-store.js";
+import { createRequire } from "module";
+import { fileURLToPath } from "url";
+import { dirname, join } from "path";
+// Extracted hooks
+import { useSlashCommands } from "./hooks/useSlashCommands.js";
+import { useAgentLoop } from "./hooks/useAgentLoop.js";
+// Thinking verbs — rotate randomly each render (Claude Code parity)
+const THINKING_VERBS = [
+    "thinking…",
+    "reasoning…",
+    "considering…",
+    "analyzing…",
+    "evaluating…",
+    "pondering…",
+    "processing…",
+    "reflecting…",
+    "examining…",
+    "working…",
+];
+function randomVerb() {
+    return THINKING_VERBS[Math.floor(Math.random() * THINKING_VERBS.length)];
+}
+const PKG_NAME = "whale-code";
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = dirname(__filename);
+const PKG_VERSION = createRequire(import.meta.url)(join(__dirname, "..", "..", "..", "package.json")).version;
+// ── Component ──
+export function ChatApp() {
+    const { exit } = useApp();
+    // Core state
+    const [ready, setReady] = useState(false);
+    const [error, setError] = useState("");
+    const [messages, setMessages] = useState([]);
+    const [streamingText, setStreamingText] = useState("");
+    const [isStreaming, setIsStreaming] = useState(false);
+    const [activeTools, setActiveTools] = useState([]);
+    const [subagentActivity, setSubagentActivity] = useState(new Map());
+    const [completedSubagents, setCompletedSubagents] = useState([]);
+    const [teamState, setTeamState] = useState(null);
+    // UI state
+    const [userLabel, setUserLabel] = useState("");
+    const [toolsExpanded, setToolsExpanded] = useState(false);
+    const [serverToolsAvailable, setServerToolsAvailable] = useState(0);
+    const [storeSelectMode, setStoreSelectMode] = useState(false);
+    const [storeList, setStoreList] = useState([]);
+    const [modelSelectMode, setModelSelectMode] = useState(false);
+    const [currentModel, setCurrentModel] = useState(getModelShortName());
+    const [sessionId, setSessionId] = useState(null);
+    const [thinkingEnabled, setThinkingEnabled] = useState(() => {
+        try {
+            return loadConfig().thinking_enabled ?? true;
+        }
+        catch {
+            return true;
+        }
+    });
+    // Stream activity indicator — show spinner when no new content for 800ms+
+    const [streamStale, setStreamStale] = useState(false);
+    const lastContentUpdateRef = useRef(Date.now());
+    // Poll-based staleness check — avoids re-renders on every content change
+    useEffect(() => {
+        if (!isStreaming) {
+            setStreamStale(false);
+            return;
+        }
+        lastContentUpdateRef.current = Date.now();
+        const interval = setInterval(() => {
+            setStreamStale(Date.now() - lastContentUpdateRef.current > 800);
+        }, 400);
+        return () => clearInterval(interval);
+    }, [isStreaming]);
+    // Rewind state
+    const [showRewind, setShowRewind] = useState(false);
+    const rewindManagerRef = useRef(new RewindManager());
+    const turnIndexRef = useRef(0);
+    // Refs
+    const conversationRef = useRef([]);
+    const abortRef = useRef(null);
+    const accTextRef = useRef("");
+    const thinkingVerbRef = useRef(randomVerb());
+    // No timer cleanup needed — unified flush timer is managed inside useAgentLoop,
+    // stale polling interval is cleaned up by its own effect.
+    // ── Init ──
+    useEffect(() => {
+        const check = canUseAgent();
+        if (!check.ready) {
+            setError(check.reason || "Run 'whale login' to authenticate.");
+        }
+        else {
+            const config = loadConfig();
+            if (config.email)
+                setUserLabel(config.store_name || config.email);
+            setReady(true);
+            getServerToolCount().then((count) => setServerToolsAvailable(count));
+            mcpClientManager.connectAll().catch(() => { });
+        }
+        return () => {
+            mcpClientManager.disconnectAll().catch(() => { });
+        };
+    }, []);
+    // ── Keys (configurable via ~/.swagmanager/keybindings.json) ──
+    const kb = useMemo(() => loadKeybindings(), []);
+    useInput((input, key) => {
+        if (matchesBinding(kb.exit, input, key)) {
+            if (abortRef.current)
+                abortRef.current.abort();
+            exit();
+        }
+        if (matchesBinding(kb.cancel_stream, input, key) && isStreaming && abortRef.current) {
+            abortRef.current.abort();
+        }
+        if (matchesBinding(kb.toggle_expand, input, key)) {
+            setToolsExpanded((prev) => !prev);
+        }
+        if (matchesBinding(kb.toggle_thinking, input, key)) {
+            setThinkingEnabled((prev) => {
+                const next = !prev;
+                try {
+                    updateConfig({ thinking_enabled: next });
+                }
+                catch { /* best effort */ }
+                setMessages((msgs) => [...msgs, {
+                        role: "assistant",
+                        text: `  Thinking: ${next ? "on" : "off"}`,
+                    }]);
+                return next;
+            });
+        }
+    });
+    // ── Extracted hooks ──
+    const { handleCommand, handleStoreSelect, handleStoreCancel } = useSlashCommands({
+        exit, toolsExpanded, serverToolsAvailable, sessionId, thinkingEnabled,
+        conversationRef,
+        setMessages, setStreamingText, setActiveTools, setTeamState,
+        setStoreList, setStoreSelectMode, setModelSelectMode, setCurrentModel,
+        setSessionId, setThinkingEnabled, setUserLabel, setServerToolsAvailable,
+        setShowRewind,
+        rewindCheckpointCount: rewindManagerRef.current.getCheckpointCount(),
+        PKG_NAME, PKG_VERSION,
+    });
+    const { handleSend } = useAgentLoop({
+        isStreaming, thinkingEnabled, conversationRef, abortRef,
+        accTextRef, lastContentUpdateRef, thinkingVerbRef,
+        rewindManagerRef, turnIndexRef,
+        setMessages, setStreamingText, setIsStreaming, setActiveTools,
+        setSubagentActivity, setCompletedSubagents, setTeamState,
+    });
+    // ── Rewind handler ──
+    const handleRewind = useCallback((checkpointIndex, outcome) => {
+        const rm = rewindManagerRef.current;
+        if (outcome === RewindOutcome.Cancel) {
+            setShowRewind(false);
+            return;
+        }
+        if (outcome === RewindOutcome.RevertOnly) {
+            // Revert files only, keep conversation
+            const result = rm.revertFilesFrom(checkpointIndex);
+            setShowRewind(false);
+            const total = result.filesReverted.length + result.filesDeleted.length;
+            const parts = [];
+            if (result.filesReverted.length > 0)
+                parts.push(`${result.filesReverted.length} reverted`);
+            if (result.filesDeleted.length > 0)
+                parts.push(`${result.filesDeleted.length} deleted`);
+            if (result.errors.length > 0)
+                parts.push(`${result.errors.length} errors`);
+            setMessages(prev => [...prev, {
+                    role: "assistant",
+                    text: total > 0
+                        ? `  ${symbols.check} File changes reverted (${parts.join(", ")})`
+                        : `  ${symbols.check} No file changes to revert.`,
+                }]);
+            return;
+        }
+        // RewindOnly or RewindAndRevert
+        const shouldRevertFiles = outcome === RewindOutcome.RewindAndRevert;
+        const result = rm.rewindTo(checkpointIndex);
+        // Truncate messages
+        setMessages(prev => prev.slice(0, result.messageCount));
+        // Truncate conversation history to match
+        // We need to figure out how many Anthropic messages correspond to the checkpoint.
+        // The checkpoint stores messageCount which is the UI messages count at that point.
+        // Rebuild conversation from remaining messages by keeping only the conversation
+        // entries up to the corresponding turn.
+        const targetTurnIndex = rm.getCheckpoints()[checkpointIndex]?.turnIndex ?? 0;
+        // Each user/assistant exchange is roughly 2 conversation entries (user + assistant).
+        // But the actual conversation is maintained independently.
+        // We slice it to 2*(turnIndex+1) entries (user+assistant pairs)
+        const conversationSlicePoint = Math.min((targetTurnIndex + 1) * 2, conversationRef.current.length);
+        conversationRef.current = conversationRef.current.slice(0, conversationSlicePoint);
+        // Reset turn index to match
+        turnIndexRef.current = targetTurnIndex + 1;
+        setShowRewind(false);
+        const parts = [];
+        if (shouldRevertFiles) {
+            if (result.filesReverted.length > 0)
+                parts.push(`${result.filesReverted.length} files reverted`);
+            if (result.filesDeleted.length > 0)
+                parts.push(`${result.filesDeleted.length} files deleted`);
+        }
+        if (result.errors.length > 0)
+            parts.push(`${result.errors.length} revert errors`);
+        const detail = parts.length > 0 ? ` (${parts.join(", ")})` : "";
+        setMessages(prev => [...prev, {
+                role: "assistant",
+                text: `  ${symbols.check} Rewound to turn ${targetTurnIndex + 1}${detail}`,
+            }]);
+    }, []);
+    // ── Render ──
+    const termWidth = process.stdout.columns || 80;
+    const contentWidth = Math.max(20, termWidth - 2);
+    const { staticItems, dynamicMessages } = useMemo(() => {
+        const items = [{ id: "header", type: "header" }];
+        const cutoff = Math.max(0, messages.length - 1);
+        for (let i = 0; i < cutoff; i++) {
+            items.push({ id: `msg-${i}`, type: "message", msg: messages[i], index: i });
+        }
+        const tail = messages.length > 0 ? [messages[messages.length - 1]] : [];
+        return { staticItems: items, dynamicMessages: tail };
+    }, [messages]);
+    if (error) {
+        return (_jsxs(Box, { flexDirection: "column", paddingLeft: 1, paddingRight: 1, children: [_jsx(Text, { children: " " }), _jsx(Text, { color: colors.brand, bold: true, children: "\u25C6 whale code" }), _jsx(Text, { children: " " }), _jsx(Text, { color: colors.error, children: error })] }));
+    }
+    if (!ready) {
+        return (_jsx(Box, { flexDirection: "column", paddingLeft: 1, paddingRight: 1, children: _jsx(Text, { color: colors.tertiary, children: "loading..." }) }));
+    }
+    return (_jsxs(Box, { flexDirection: "column", paddingLeft: 1, paddingRight: 1, children: [_jsx(Static, { items: staticItems, children: (item) => {
+                    if (item.type === "header") {
+                        return (_jsxs(Box, { flexDirection: "column", children: [_jsx(Text, { children: " " }), _jsxs(Text, { children: [_jsx(Text, { color: colors.brand, bold: true, children: "\u25C6 whale code" }), userLabel ? _jsxs(Text, { color: colors.dim, children: ["  ", userLabel] }) : null, _jsxs(Text, { color: colors.dim, children: ["  ", currentModel] }), thinkingEnabled ? _jsx(Text, { color: colors.warning, children: "  thinking" }) : null, getPermissionMode() !== "default" && (_jsxs(Text, { color: getPermissionMode() === "yolo" ? colors.error : colors.info, children: ["  ", getPermissionMode()] })), serverToolsAvailable > 0 ? (_jsxs(Text, { color: colors.tertiary, children: ["  ", symbols.dot, " ", serverToolsAvailable, " server tools"] })) : null] }), _jsx(Text, { color: colors.separator, children: "─".repeat(contentWidth) })] }, item.id));
+                    }
+                    return _jsx(CompletedMessage, { msg: item.msg, index: item.index, toolsExpanded: toolsExpanded }, item.id);
+                } }), dynamicMessages.map((msg) => (_jsx(CompletedMessage, { msg: msg, index: messages.length - 1, toolsExpanded: toolsExpanded }, `dynamic-${messages.length - 1}`))), teamState ? (_jsx(TeamPanel, { team: teamState })) : (_jsxs(_Fragment, { children: [isStreaming && !streamingText && activeTools.length === 0 && (_jsxs(Box, { marginLeft: 2, marginY: 1, children: [_jsx(Text, { color: colors.brand, children: _jsx(Spinner, { type: "dots" }) }), _jsxs(Text, { color: colors.dim, children: ["  ", thinkingVerbRef.current] })] })), activeTools.length > 0 && (_jsx(Box, { flexDirection: "column", marginLeft: 2, children: activeTools.map((tc, i) => (_jsxs(Box, { flexDirection: "column", children: [_jsx(ToolIndicator, { id: `live-${tc.name}-${i}`, name: tc.name, status: tc.status, input: tc.input, expanded: toolsExpanded }), tc.name === "task" && tc.status === "running" && (subagentActivity.size > 0 || completedSubagents.length > 0) && (_jsx(SubagentPanel, { running: subagentActivity, completed: completedSubagents }))] }, `live-${tc.name}-${i}`))) })), streamingText && (_jsxs(Box, { marginLeft: 2, flexDirection: "column", children: [_jsx(StreamingText, { text: streamingText }), streamStale && (_jsx(Box, { marginTop: 0, children: _jsx(Text, { color: colors.brand, children: _jsx(Spinner, { type: "dots" }) }) }))] }))] })), showRewind ? (_jsx(RewindViewer, { checkpoints: rewindManagerRef.current.getCheckpoints(), onRewind: handleRewind, onCancel: () => setShowRewind(false) })) : storeSelectMode ? (_jsx(StoreSelector, { stores: storeList, currentStoreId: loadConfig().store_id || "", onSelect: handleStoreSelect, onCancel: handleStoreCancel })) : modelSelectMode ? (_jsx(ModelSelector, { currentModel: currentModel, onSelect: (model) => {
+                    setModelSelectMode(false);
+                    setModel(model.value);
+                    setCurrentModel(model.value);
+                    setMessages((prev) => [...prev, {
+                            role: "assistant",
+                            text: `  ${symbols.check} Model: ${model.label}  (${model.modelId})`,
+                        }]);
+                }, onCancel: () => setModelSelectMode(false) })) : (_jsx(ChatInput, { onSubmit: handleSend, onCommand: handleCommand, disabled: isStreaming, agentName: "whale code" }))] }));
+}

package/dist/cli/chat/ChatInput.d.ts

@@ -0,0 +1,39 @@
+/**
+ * ChatInput — Custom input with bracketed paste + image attachments
+ *
+ * Replaces ink-text-input with raw stdin handling to fix:
+ * - Paste losing content on enter (newlines in paste triggered submit)
+ * - Multi-line paste mangled/unformatted
+ * - No drag-drop image support
+ *
+ * Features:
+ * - Bracketed paste mode — clean multi-line paste
+ * - Drag-drop images — detects image file paths, attaches as chips
+ * - Image chips above input like Claude Code: [image1.png] [image2.jpg]
+ * - Backspace on empty input removes last image
+ * - Slash command menu preserved
+ * - Multi-line input with ⎸ continuation markers
+ */
+export interface ImageAttachment {
+    path: string;
+    name: string;
+    base64: string;
+    mediaType: string;
+}
+export interface FileAttachment {
+    path: string;
+    name: string;
+}
+export interface SlashCommand {
+    name: string;
+    description: string;
+}
+export declare const SLASH_COMMANDS: SlashCommand[];
+interface ChatInputProps {
+    onSubmit: (message: string, images?: ImageAttachment[]) => void;
+    onCommand: (command: string) => void;
+    disabled: boolean;
+    agentName?: string;
+}
+export declare function ChatInput({ onSubmit, onCommand, disabled, agentName }: ChatInputProps): import("react/jsx-runtime").JSX.Element;
+export {};