@mariozechner/pi-coding-agent 0.12.10 → 0.12.12

package/dist/tui/tui-renderer.js

@@ -20,6 +20,7 @@ import { FooterComponent } from "./footer.js";
 import { ModelSelectorComponent } from "./model-selector.js";
 import { OAuthSelectorComponent } from "./oauth-selector.js";
 import { QueueModeSelectorComponent } from "./queue-mode-selector.js";
+import { SessionSelectorComponent } from "./session-selector.js";
 import { ThemeSelectorComponent } from "./theme-selector.js";
 import { ThinkingSelectorComponent } from "./thinking-selector.js";
 import { ToolExecutionComponent } from "./tool-execution.js";
@@ -62,6 +63,8 @@ export class TuiRenderer {
     modelSelector = null;
     // User message selector (for branching)
     userMessageSelector = null;
+    // Session selector (for resume)
+    sessionSelector = null;
     // OAuth selector
     oauthSelector = null;
     // Track if this is the first user message (to skip spacer)
@@ -70,6 +73,8 @@ export class TuiRenderer {
     scopedModels = [];
     // Tool output expansion state
     toolOutputExpanded = false;
+    // Thinking block visibility state
+    hideThinkingBlock = false;
     // Agent subscription unsubscribe function
     unsubscribe;
     // File-based slash commands
@@ -148,6 +153,12 @@ export class TuiRenderer {
             name: "autocompact",
             description: "Toggle automatic context compaction",
         };
+        const resumeCommand = {
+            name: "resume",
+            description: "Resume a different session",
+        };
+        // Load hide thinking block setting
+        this.hideThinkingBlock = settingsManager.getHideThinkingBlock();
         // Load file-based slash commands
         this.fileCommands = loadSlashCommands();
         // Convert file commands to SlashCommand format
@@ -171,6 +182,7 @@ export class TuiRenderer {
             clearCommand,
             compactCommand,
             autocompactCommand,
+            resumeCommand,
             ...fileSlashCommands,
         ], process.cwd(), fdPath);
         this.editor.setAutocompleteProvider(autocompleteProvider);
@@ -201,6 +213,9 @@ export class TuiRenderer {
             theme.fg("dim", "ctrl+o") +
             theme.fg("muted", " to expand tools") +
             "\n" +
+            theme.fg("dim", "ctrl+t") +
+            theme.fg("muted", " to toggle thinking") +
+            "\n" +
             theme.fg("dim", "/") +
             theme.fg("muted", " for commands") +
             "\n" +
@@ -269,6 +284,9 @@ export class TuiRenderer {
         this.editor.onCtrlO = () => {
             this.toggleToolOutputExpansion();
         };
+        this.editor.onCtrlT = () => {
+            this.toggleThinkingBlockVisibility();
+        };
         // Handle editor submission
         this.editor.onSubmit = async (text) => {
             text = text.trim();
@@ -367,6 +385,12 @@ export class TuiRenderer {
                 this.editor.setText("");
                 return;
             }
+            // Check for /resume command
+            if (text === "/resume") {
+                this.showSessionSelector();
+                this.editor.setText("");
+                return;
+            }
             // Check for file-based slash commands
             text = expandSlashCommand(text, this.fileCommands);
             // Normal message submission - validate model and API key first
@@ -398,6 +422,8 @@ export class TuiRenderer {
                 });
                 // Update pending messages display
                 this.updatePendingMessagesDisplay();
+                // Add to history for up/down arrow navigation
+                this.editor.addToHistory(text);
                 // Clear editor
                 this.editor.setText("");
                 this.ui.requestRender();
@@ -407,6 +433,8 @@ export class TuiRenderer {
             if (this.onInputCallback) {
                 this.onInputCallback(text);
             }
+            // Add to history for up/down arrow navigation
+            this.editor.addToHistory(text);
         };
         // Start the UI
         this.ui.start();
@@ -508,7 +536,7 @@ export class TuiRenderer {
                 }
                 else if (event.message.role === "assistant") {
                     // Create assistant component for streaming
-                    this.streamingComponent = new AssistantMessageComponent();
+                    this.streamingComponent = new AssistantMessageComponent(undefined, this.hideThinkingBlock);
                     this.chatContainer.addChild(this.streamingComponent);
                     this.streamingComponent.updateContent(event.message);
                     this.ui.requestRender();
@@ -634,7 +662,7 @@ export class TuiRenderer {
         else if (message.role === "assistant") {
             const assistantMsg = message;
             // Add assistant message component
-            const assistantComponent = new AssistantMessageComponent(assistantMsg);
+            const assistantComponent = new AssistantMessageComponent(assistantMsg, this.hideThinkingBlock);
             this.chatContainer.addChild(assistantComponent);
         }
         // Note: tool calls and results are now handled via tool_execution_start/end events
@@ -675,7 +703,7 @@ export class TuiRenderer {
             }
             else if (message.role === "assistant") {
                 const assistantMsg = message;
-                const assistantComponent = new AssistantMessageComponent(assistantMsg);
+                const assistantComponent = new AssistantMessageComponent(assistantMsg, this.hideThinkingBlock);
                 this.chatContainer.addChild(assistantComponent);
                 // Create tool execution components for any tool calls
                 for (const content of assistantMsg.content) {
@@ -715,6 +743,19 @@ export class TuiRenderer {
         }
         // Clear pending tools after rendering initial messages
         this.pendingTools.clear();
+        // Populate editor history with user messages from the session (oldest first so newest is at index 0)
+        for (const message of state.messages) {
+            if (message.role === "user") {
+                const textBlocks = typeof message.content === "string"
+                    ? [{ type: "text", text: message.content }]
+                    : message.content.filter((c) => c.type === "text");
+                const textContent = textBlocks.map((c) => c.text).join("");
+                // Skip compaction summary messages
+                if (textContent && !textContent.startsWith(SUMMARY_PREFIX)) {
+                    this.editor.addToHistory(textContent);
+                }
+            }
+        }
         this.ui.requestRender();
     }
     async getUserInput() {
@@ -755,7 +796,7 @@ export class TuiRenderer {
             }
             else if (message.role === "assistant") {
                 const assistantMsg = message;
-                const assistantComponent = new AssistantMessageComponent(assistantMsg);
+                const assistantComponent = new AssistantMessageComponent(assistantMsg, this.hideThinkingBlock);
                 this.chatContainer.addChild(assistantComponent);
                 for (const content of assistantMsg.content) {
                     if (content.type === "toolCall") {
@@ -923,6 +964,24 @@ export class TuiRenderer {
         }
         this.ui.requestRender();
     }
+    toggleThinkingBlockVisibility() {
+        this.hideThinkingBlock = !this.hideThinkingBlock;
+        this.settingsManager.setHideThinkingBlock(this.hideThinkingBlock);
+        // Update all assistant message components and rebuild their content
+        for (const child of this.chatContainer.children) {
+            if (child instanceof AssistantMessageComponent) {
+                child.setHideThinkingBlock(this.hideThinkingBlock);
+            }
+        }
+        // Rebuild chat to apply visibility change
+        this.chatContainer.clear();
+        this.rebuildChatFromMessages();
+        // Show brief notification
+        const status = this.hideThinkingBlock ? "hidden" : "visible";
+        this.chatContainer.addChild(new Spacer(1));
+        this.chatContainer.addChild(new Text(theme.fg("dim", `Thinking blocks: ${status}`), 1, 0));
+        this.ui.requestRender();
+    }
     clearEditor() {
         this.editor.setText("");
         this.ui.requestRender();
@@ -939,14 +998,6 @@ export class TuiRenderer {
         this.chatContainer.addChild(new Text(theme.fg("warning", `Warning: ${warningMessage}`), 1, 0));
         this.ui.requestRender();
     }
-    showSuccess(message, detail) {
-        this.chatContainer.addChild(new Spacer(1));
-        const text = detail
-            ? `${theme.fg("success", message)}\n${theme.fg("muted", detail)}`
-            : theme.fg("success", message);
-        this.chatContainer.addChild(new Text(text, 1, 1));
-        this.ui.requestRender();
-    }
     showThinkingSelector() {
         // Create thinking selector with current level
         this.thinkingSelector = new ThinkingSelectorComponent(this.agent.state.thinkingLevel, (level) => {
@@ -1176,6 +1227,76 @@ export class TuiRenderer {
         this.userMessageSelector = null;
         this.ui.setFocus(this.editor);
     }
+    showSessionSelector() {
+        // Create session selector
+        this.sessionSelector = new SessionSelectorComponent(this.sessionManager, async (sessionPath) => {
+            this.hideSessionSelector();
+            await this.handleResumeSession(sessionPath);
+        }, () => {
+            // Just hide the selector
+            this.hideSessionSelector();
+            this.ui.requestRender();
+        });
+        // Replace editor with selector
+        this.editorContainer.clear();
+        this.editorContainer.addChild(this.sessionSelector);
+        this.ui.setFocus(this.sessionSelector.getSessionList());
+        this.ui.requestRender();
+    }
+    async handleResumeSession(sessionPath) {
+        // Unsubscribe first to prevent processing events during transition
+        this.unsubscribe?.();
+        // Abort and wait for completion
+        this.agent.abort();
+        await this.agent.waitForIdle();
+        // Stop loading animation
+        if (this.loadingAnimation) {
+            this.loadingAnimation.stop();
+            this.loadingAnimation = null;
+        }
+        this.statusContainer.clear();
+        // Clear UI state
+        this.queuedMessages = [];
+        this.pendingMessagesContainer.clear();
+        this.streamingComponent = null;
+        this.pendingTools.clear();
+        // Set the selected session as active
+        this.sessionManager.setSessionFile(sessionPath);
+        // Reload the session
+        const loaded = loadSessionFromEntries(this.sessionManager.loadEntries());
+        this.agent.replaceMessages(loaded.messages);
+        // Restore model if saved in session
+        const savedModel = this.sessionManager.loadModel();
+        if (savedModel) {
+            const availableModels = (await getAvailableModels()).models;
+            const match = availableModels.find((m) => m.provider === savedModel.provider && m.id === savedModel.modelId);
+            if (match) {
+                this.agent.setModel(match);
+            }
+        }
+        // Restore thinking level if saved in session
+        const savedThinking = this.sessionManager.loadThinkingLevel();
+        if (savedThinking) {
+            this.agent.setThinkingLevel(savedThinking);
+        }
+        // Resubscribe to agent
+        this.subscribeToAgent();
+        // Clear and re-render the chat
+        this.chatContainer.clear();
+        this.isFirstUserMessage = true;
+        this.renderInitialMessages(this.agent.state);
+        // Show confirmation message
+        this.chatContainer.addChild(new Spacer(1));
+        this.chatContainer.addChild(new Text(theme.fg("dim", "Resumed session"), 1, 0));
+        this.ui.requestRender();
+    }
+    hideSessionSelector() {
+        // Replace selector with editor in the container
+        this.editorContainer.clear();
+        this.editorContainer.addChild(this.editor);
+        this.sessionSelector = null;
+        this.ui.setFocus(this.editor);
+    }
     async showOAuthSelector(mode) {
         // For logout mode, filter to only show logged-in providers
         let providersToShow = [];
@@ -1521,9 +1642,6 @@ export class TuiRenderer {
             this.chatContainer.addChild(compactionComponent);
             // Update footer with new state (fixes context % display)
             this.footer.updateState(this.agent.state);
-            // Show success message
-            const successTitle = isAuto ? "✓ Context auto-compacted" : "✓ Context compacted";
-            this.showSuccess(successTitle, `Reduced from ${compactionEntry.tokensBefore.toLocaleString()} tokens`);
         }
         catch (error) {
             const message = error instanceof Error ? error.message : String(error);