@bastani/atomic 0.8.25 → 0.8.26-alpha.10

package/dist/modes/interactive/interactive-mode.js

@@ -31,6 +31,7 @@ import { parseGitUrl } from "../../utils/git.js";
 import { getCwdRelativePath } from "../../utils/paths.js";
 import { getPiUserAgent } from "../../utils/pi-user-agent.js";
 import { killTrackedDetachedChildren } from "../../utils/shell.js";
+import { recordTimeSinceReset } from "../../core/timings.js";
 import { ensureTool } from "../../utils/tools-manager.js";
 import { checkForNewPiVersion } from "../../utils/version-check.js";
 import { ArminComponent } from "./components/armin.js";
@@ -41,6 +42,7 @@ import { BranchSummaryMessageComponent } from "./components/branch-summary-messa
 import { chatEntriesFromAgentMessages, renderChatMessageEntry, } from "./components/chat-message-renderer.js";
 import { addChatTranscriptEntry } from "./components/chat-transcript.js";
 import { CompactionSummaryMessageComponent } from "./components/compaction-summary-message.js";
+import { ContextCompactionSummaryMessageComponent } from "./components/context-compaction-summary-message.js";
 import { CountdownTimer } from "./components/countdown-timer.js";
 import { CustomEditor } from "./components/custom-editor.js";
 import { CustomMessageComponent } from "./components/custom-message.js";
@@ -87,6 +89,13 @@ class ExpandableText extends Text {
         this.setText(this.expanded ? this.getExpandedText() : this.getCollapsedText());
     }
 }
+function isContextCompactionResult(result) {
+    return (typeof result === "object" &&
+        result !== null &&
+        "stats" in result &&
+        "deletedTargets" in result &&
+        "protectedEntryIds" in result);
+}
 const DEAD_TERMINAL_ERROR_CODES = new Set(["EIO", "EPIPE", "ENOTCONN"]);
 function isDeadTerminalError(error) {
     if (!error || typeof error !== "object" || !("code" in error)) {
@@ -404,10 +413,6 @@ export class InteractiveMode {
         this.registerSignalHandlers();
         // Load changelog (only show new entries, skip for resumed sessions)
         this.changelogMarkdown = this.getChangelogForDisplay();
-        // Ensure fd and rg are available (downloads if missing, adds to PATH via getBinDir)
-        // Both are needed: fd for autocomplete, rg for grep tool and bash commands
-        const [fdPath] = await Promise.all([ensureTool("fd"), ensureTool("rg")]);
-        this.fdPath = fdPath;
         // Add header container as first child
         this.ui.addChild(this.headerContainer);
         // Add the quiet startup identity (unless silenced). Resource details are
@@ -444,9 +449,21 @@ export class InteractiveMode {
         this.ui.setFocus(this.editor);
         this.setupKeyHandlers();
         this.setupEditorSubmitHandler();
-        // Start the UI before initializing extensions so session_start handlers can use interactive dialogs
+        // Start the UI before initializing extensions so session_start handlers can use interactive dialogs.
+        // fd/rg readiness is intentionally checked after first paint because ensureTool may spawn
+        // or download tools on cold machines.
         this.ui.start();
+        recordTimeSinceReset("time-to-first-frame");
         this.isInitialized = true;
+        void Promise.all([ensureTool("fd"), ensureTool("rg")])
+            .then(([fdPath]) => {
+            this.fdPath = fdPath;
+            this.setupAutocompleteProvider();
+        })
+            .catch((error) => {
+            const message = error instanceof Error ? error.message : String(error);
+            console.error(`Tool readiness check failed: ${message}`);
+        });
         // Initialize extensions first so resources are shown before messages
         await this.rebindCurrentSession();
         // Render initial messages AFTER showing loaded resources
@@ -461,8 +478,11 @@ export class InteractiveMode {
         this.footerDataProvider.onBranchChange(() => {
             this.ui.requestRender();
         });
-        // Initialize available provider count for footer display
-        await this.updateAvailableProviderCount();
+        // Initialize available provider count for footer display without delaying first-frame startup.
+        void this.updateAvailableProviderCount().catch((error) => {
+            const message = error instanceof Error ? error.message : String(error);
+            console.error(`Failed to update provider count: ${message}`);
+        });
     }
     /**
      * Update terminal title with session name and cwd.
@@ -2237,12 +2257,13 @@ export class InteractiveMode {
                 await this.handleClearCommand();
                 return;
             }
-            if (text === "/compact" || text.startsWith("/compact ")) {
-                const customInstructions = text.startsWith("/compact ")
-                    ? text.slice(9).trim()
-                    : undefined;
+            if (/^\/compact(?:\s|$)/.test(text)) {
                 this.editor.setText("");
-                await this.handleCompactCommand(customInstructions);
+                if (text !== "/compact") {
+                    this.showWarning("Usage: /compact");
+                    return;
+                }
+                await this.handleCompactCommand();
                 return;
             }
             if (text === "/reload") {
@@ -2541,10 +2562,11 @@ export class InteractiveMode {
                 };
                 this.statusContainer.clear();
                 const cancelHint = `(${keyText("app.interrupt")} Cancel)`;
+                const isOverflowAutoCompaction = event.reason === "overflow";
                 const label = event.reason === "manual"
                     ? `Compacting context... ${cancelHint}`
-                    : `${event.reason === "overflow" ? "Context overflow detected, " : ""}Auto-compacting... ${cancelHint}`;
-                this.autoCompactionLoader = new Loader(this.ui, (spinner) => theme.fg("accent", spinner), (text) => theme.fg("muted", text), label);
+                    : `${isOverflowAutoCompaction ? "Context overflow detected. " : ""}Auto-compacting... ${cancelHint}`;
+                this.autoCompactionLoader = new Loader(this.ui, (spinner) => theme.fg("accent", spinner), (text) => theme.fg(isOverflowAutoCompaction ? "warning" : "muted", text), label);
                 this.statusContainer.addChild(this.autoCompactionLoader);
                 this.ui.requestRender();
                 break;
@@ -2573,7 +2595,12 @@ export class InteractiveMode {
                 else if (event.result) {
                     this.chatContainer.clear();
                     this.rebuildChatFromMessages();
-                    this.addMessageToChat(createCompactionSummaryMessage(event.result.summary, event.result.tokensBefore, new Date().toISOString()));
+                    if (isContextCompactionResult(event.result)) {
+                        this.addContextCompactionSummaryToChat(event.result);
+                    }
+                    else {
+                        this.addMessageToChat(createCompactionSummaryMessage(event.result.summary, event.result.tokensBefore, new Date().toISOString()));
+                    }
                     this.footer.invalidate();
                 }
                 else if (event.errorMessage) {
@@ -2589,6 +2616,50 @@ export class InteractiveMode {
                 this.ui.requestRender();
                 break;
             }
+            case "context_compaction_start": {
+                if (this.settingsManager.getShowTerminalProgress()) {
+                    this.ui.terminal.setProgress(true);
+                }
+                this.autoCompactionEscapeHandler = this.defaultEditor.onEscape;
+                this.defaultEditor.onEscape = () => {
+                    this.session.abortCompaction();
+                };
+                this.statusContainer.clear();
+                const cancelHint = `(${keyText("app.interrupt")} Cancel)`;
+                this.autoCompactionLoader = new Loader(this.ui, (spinner) => theme.fg("accent", spinner), (text) => theme.fg("muted", text), `Compacting context... ${cancelHint}`);
+                this.statusContainer.addChild(this.autoCompactionLoader);
+                this.ui.requestRender();
+                break;
+            }
+            case "context_compaction_end": {
+                if (this.settingsManager.getShowTerminalProgress()) {
+                    this.ui.terminal.setProgress(false);
+                }
+                if (this.autoCompactionEscapeHandler) {
+                    this.defaultEditor.onEscape = this.autoCompactionEscapeHandler;
+                    this.autoCompactionEscapeHandler = undefined;
+                }
+                if (this.autoCompactionLoader) {
+                    this.autoCompactionLoader.stop();
+                    this.autoCompactionLoader = undefined;
+                    this.statusContainer.clear();
+                }
+                if (event.aborted) {
+                    this.showError("Context compaction cancelled");
+                }
+                else if (event.result) {
+                    this.chatContainer.clear();
+                    this.rebuildChatFromMessages();
+                    this.addContextCompactionSummaryToChat(event.result);
+                    this.footer.invalidate();
+                }
+                else if (event.errorMessage) {
+                    this.showError(event.errorMessage);
+                }
+                void this.flushCompactionQueue({ willRetry: event.willRetry });
+                this.ui.requestRender();
+                break;
+            }
             case "auto_retry_start": {
                 // Set up escape to abort retry
                 this.retryEscapeHandler = this.defaultEditor.onEscape;
@@ -2688,6 +2759,12 @@ export class InteractiveMode {
         addChatTranscriptEntry(this.chatContainer, component, entry.role);
         return component;
     }
+    addContextCompactionSummaryToChat(result) {
+        this.chatContainer.addChild(new Spacer(1));
+        const component = new ContextCompactionSummaryMessageComponent(result, this.getMarkdownThemeWithSettings());
+        component.setExpanded(this.toolOutputExpanded);
+        this.chatContainer.addChild(component);
+    }
     addMessageToChat(message, options) {
         switch (message.role) {
             case "bashExecution": {
@@ -4821,7 +4898,7 @@ export class InteractiveMode {
         this.bashComponent = undefined;
         this.ui.requestRender();
     }
-    async handleCompactCommand(customInstructions) {
+    async handleCompactCommand() {
         const entries = this.sessionManager.getEntries();
         const messageCount = entries.filter((e) => e.type === "message").length;
         if (messageCount < 2) {
@@ -4834,7 +4911,7 @@ export class InteractiveMode {
         }
         this.statusContainer.clear();
         try {
-            await this.session.compact(customInstructions);
+            await this.session.compact();
         }
         catch {
             // Ignore, will be emitted as an event