@mariozechner/pi-coding-agent 0.13.2 → 0.14.1

package/dist/tui/tui-renderer.js

@@ -1,18 +1,27 @@
+import { randomBytes } from "node:crypto";
 import * as fs from "node:fs";
+import { createWriteStream } from "node:fs";
+import { tmpdir } from "node:os";
 import * as path from "node:path";
+import { join } from "node:path";
 import { CombinedAutocompleteProvider, Container, Input, Loader, Markdown, ProcessTerminal, Spacer, Text, TruncatedText, TUI, visibleWidth, } from "@mariozechner/pi-tui";
-import { exec } from "child_process";
+import { exec, spawn } from "child_process";
+import stripAnsi from "strip-ansi";
 import { getChangelogPath, parseChangelog } from "../changelog.js";
 import { copyToClipboard } from "../clipboard.js";
 import { calculateContextTokens, compact, shouldCompact } from "../compaction.js";
 import { APP_NAME, getDebugLogPath, getModelsPath, getOAuthPath } from "../config.js";
 import { exportSessionToHtml } from "../export-html.js";
+import { isBashExecutionMessage } from "../messages.js";
 import { getApiKeyForModel, getAvailableModels, invalidateOAuthCache } from "../model-config.js";
 import { listOAuthProviders, login, logout } from "../oauth/index.js";
 import { getLatestCompactionEntry, loadSessionFromEntries, SUMMARY_PREFIX, SUMMARY_SUFFIX, } from "../session-manager.js";
+import { getShellConfig, killProcessTree } from "../shell.js";
 import { expandSlashCommand, loadSlashCommands } from "../slash-commands.js";
 import { getEditorTheme, getMarkdownTheme, onThemeChange, setTheme, theme } from "../theme/theme.js";
+import { DEFAULT_MAX_BYTES, truncateTail } from "../tools/truncate.js";
 import { AssistantMessageComponent } from "./assistant-message.js";
+import { BashExecutionComponent } from "./bash-execution.js";
 import { CompactionComponent } from "./compaction.js";
 import { CustomEditor } from "./custom-editor.js";
 import { DynamicBorder } from "./dynamic-border.js";
@@ -47,6 +56,7 @@ export class TuiRenderer {
     lastSigintTime = 0;
     lastEscapeTime = 0;
     changelogMarkdown = null;
+    collapseChangelog = false;
     // Message queueing
     queuedMessages = [];
     // Streaming message tracking
@@ -79,12 +89,19 @@ export class TuiRenderer {
     unsubscribe;
     // File-based slash commands
     fileCommands = [];
-    constructor(agent, sessionManager, settingsManager, version, changelogMarkdown = null, scopedModels = [], fdPath = null) {
+    // Track if editor is in bash mode (text starts with !)
+    isBashMode = false;
+    // Track running bash command process for cancellation
+    bashProcess = null;
+    // Track current bash execution component
+    bashComponent = null;
+    constructor(agent, sessionManager, settingsManager, version, changelogMarkdown = null, collapseChangelog = false, scopedModels = [], fdPath = null) {
         this.agent = agent;
         this.sessionManager = sessionManager;
         this.settingsManager = settingsManager;
         this.version = version;
         this.changelogMarkdown = changelogMarkdown;
+        this.collapseChangelog = collapseChangelog;
         this.scopedModels = scopedModels;
         this.ui = new TUI(new ProcessTerminal());
         this.chatContainer = new Container();
@@ -218,6 +235,9 @@ export class TuiRenderer {
             theme.fg("dim", "/") +
             theme.fg("muted", " for commands") +
             "\n" +
+            theme.fg("dim", "!") +
+            theme.fg("muted", " to run bash") +
+            "\n" +
             theme.fg("dim", "drop files") +
             theme.fg("muted", " to attach");
         const header = new Text(logo + "\n" + instructions, 1, 0);
@@ -228,10 +248,19 @@ export class TuiRenderer {
         // Add changelog if provided
         if (this.changelogMarkdown) {
             this.ui.addChild(new DynamicBorder());
-            this.ui.addChild(new Text(theme.bold(theme.fg("accent", "What's New")), 1, 0));
-            this.ui.addChild(new Spacer(1));
-            this.ui.addChild(new Markdown(this.changelogMarkdown.trim(), 1, 0, getMarkdownTheme()));
-            this.ui.addChild(new Spacer(1));
+            if (this.collapseChangelog) {
+                // Show condensed version with hint to use /changelog
+                const versionMatch = this.changelogMarkdown.match(/##\s+\[?(\d+\.\d+\.\d+)\]?/);
+                const latestVersion = versionMatch ? versionMatch[1] : this.version;
+                const condensedText = `Updated to v${latestVersion}. Use ${theme.bold("/changelog")} to view full changelog.`;
+                this.ui.addChild(new Text(condensedText, 1, 0));
+            }
+            else {
+                this.ui.addChild(new Text(theme.bold(theme.fg("accent", "What's New")), 1, 0));
+                this.ui.addChild(new Spacer(1));
+                this.ui.addChild(new Markdown(this.changelogMarkdown.trim(), 1, 0, getMarkdownTheme()));
+                this.ui.addChild(new Spacer(1));
+            }
             this.ui.addChild(new DynamicBorder());
         }
         this.ui.addChild(this.chatContainer);
@@ -261,6 +290,19 @@ export class TuiRenderer {
                 // Abort
                 this.agent.abort();
             }
+            else if (this.bashProcess) {
+                // Kill running bash command
+                if (this.bashProcess.pid) {
+                    killProcessTree(this.bashProcess.pid);
+                }
+                this.bashProcess = null;
+            }
+            else if (this.isBashMode) {
+                // Cancel bash mode and clear editor
+                this.editor.setText("");
+                this.isBashMode = false;
+                this.updateEditorBorderColor();
+            }
             else if (!this.editor.getText().trim()) {
                 // Double-escape with empty editor triggers /branch
                 const now = Date.now();
@@ -288,6 +330,14 @@ export class TuiRenderer {
         this.editor.onCtrlT = () => {
             this.toggleThinkingBlockVisibility();
         };
+        // Handle editor text changes for bash mode detection
+        this.editor.onChange = (text) => {
+            const wasBashMode = this.isBashMode;
+            this.isBashMode = text.trimStart().startsWith("!");
+            if (wasBashMode !== this.isBashMode) {
+                this.updateEditorBorderColor();
+            }
+        };
         // Handle editor submission
         this.editor.onSubmit = async (text) => {
             text = text.trim();
@@ -392,6 +442,26 @@ export class TuiRenderer {
                 this.editor.setText("");
                 return;
             }
+            // Check for bash command (!<command>)
+            if (text.startsWith("!")) {
+                const command = text.slice(1).trim();
+                if (command) {
+                    // Block if bash already running
+                    if (this.bashProcess) {
+                        this.showWarning("A bash command is already running. Press Esc to cancel it first.");
+                        // Restore text since editor clears on submit
+                        this.editor.setText(text);
+                        return;
+                    }
+                    // Add to history for up/down arrow navigation
+                    this.editor.addToHistory(text);
+                    this.handleBashCommand(command);
+                    // Reset bash mode since editor is now empty
+                    this.isBashMode = false;
+                    this.updateEditorBorderColor();
+                    return;
+                }
+            }
             // Check for file-based slash commands
             text = expandSlashCommand(text, this.fileCommands);
             // Normal message submission - validate model and API key first
@@ -647,6 +717,17 @@ export class TuiRenderer {
         }
     }
     addMessageToChat(message) {
+        // Handle bash execution messages
+        if (isBashExecutionMessage(message)) {
+            const bashMsg = message;
+            const component = new BashExecutionComponent(bashMsg.command, this.ui);
+            if (bashMsg.output) {
+                component.appendOutput(bashMsg.output);
+            }
+            component.setComplete(bashMsg.exitCode, bashMsg.cancelled, bashMsg.truncated ? { truncated: true } : undefined, bashMsg.fullOutputPath);
+            this.chatContainer.addChild(component);
+            return;
+        }
         if (message.role === "user") {
             const userMsg = message;
             // Extract text content from content blocks
@@ -681,6 +762,11 @@ export class TuiRenderer {
         // Render messages
         for (let i = 0; i < state.messages.length; i++) {
             const message = state.messages[i];
+            // Handle bash execution messages
+            if (isBashExecutionMessage(message)) {
+                this.addMessageToChat(message);
+                continue;
+            }
             if (message.role === "user") {
                 const userMsg = message;
                 const textBlocks = typeof userMsg.content === "string"
@@ -774,6 +860,11 @@ export class TuiRenderer {
         // Get compaction info if any
         const compactionEntry = getLatestCompactionEntry(this.sessionManager.loadEntries());
         for (const message of this.agent.state.messages) {
+            // Handle bash execution messages
+            if (isBashExecutionMessage(message)) {
+                this.addMessageToChat(message);
+                continue;
+            }
             if (message.role === "user") {
                 const userMsg = message;
                 const textBlocks = typeof userMsg.content === "string"
@@ -838,8 +929,13 @@ export class TuiRenderer {
         }
     }
     updateEditorBorderColor() {
-        const level = this.agent.state.thinkingLevel || "off";
-        this.editor.borderColor = theme.getThinkingBorderColor(level);
+        if (this.isBashMode) {
+            this.editor.borderColor = theme.getBashModeBorderColor();
+        }
+        else {
+            const level = this.agent.state.thinkingLevel || "off";
+            this.editor.borderColor = theme.getThinkingBorderColor(level);
+        }
         this.ui.requestRender();
     }
     cycleThinkingLevel() {
@@ -850,7 +946,12 @@ export class TuiRenderer {
             this.ui.requestRender();
             return;
         }
-        const levels = ["off", "minimal", "low", "medium", "high"];
+        // xhigh is only available for codex-max models
+        const modelId = this.agent.state.model?.id || "";
+        const supportsXhigh = modelId.includes("codex-max");
+        const levels = supportsXhigh
+            ? ["off", "minimal", "low", "medium", "high", "xhigh"]
+            : ["off", "minimal", "low", "medium", "high"];
         const currentLevel = this.agent.state.thinkingLevel || "off";
         const currentIndex = levels.indexOf(currentLevel);
         const nextIndex = (currentIndex + 1) % levels.length;
@@ -954,7 +1055,7 @@ export class TuiRenderer {
     }
     toggleToolOutputExpansion() {
         this.toolOutputExpanded = !this.toolOutputExpanded;
-        // Update all tool execution and compaction components
+        // Update all tool execution, compaction, and bash execution components
         for (const child of this.chatContainer.children) {
             if (child instanceof ToolExecutionComponent) {
                 child.setExpanded(this.toolOutputExpanded);
@@ -962,6 +1063,9 @@ export class TuiRenderer {
             else if (child instanceof CompactionComponent) {
                 child.setExpanded(this.toolOutputExpanded);
             }
+            else if (child instanceof BashExecutionComponent) {
+                child.setExpanded(this.toolOutputExpanded);
+            }
         }
         this.ui.requestRender();
     }
@@ -1594,6 +1698,117 @@ export class TuiRenderer {
         this.chatContainer.addChild(new Text(theme.fg("accent", "✓ Debug log written") + "\n" + theme.fg("muted", debugLogPath), 1, 1));
         this.ui.requestRender();
     }
+    async handleBashCommand(command) {
+        // Create component and add to chat
+        this.bashComponent = new BashExecutionComponent(command, this.ui);
+        this.chatContainer.addChild(this.bashComponent);
+        this.ui.requestRender();
+        try {
+            const result = await this.executeBashCommand(command, (chunk) => {
+                if (this.bashComponent) {
+                    this.bashComponent.appendOutput(chunk);
+                    this.ui.requestRender();
+                }
+            });
+            if (this.bashComponent) {
+                this.bashComponent.setComplete(result.exitCode, result.cancelled, result.truncationResult, result.fullOutputPath);
+                // Create and save message (even if cancelled, for consistency with LLM aborts)
+                const bashMessage = {
+                    role: "bashExecution",
+                    command,
+                    output: result.truncationResult?.content || this.bashComponent.getOutput(),
+                    exitCode: result.exitCode,
+                    cancelled: result.cancelled,
+                    truncated: result.truncationResult?.truncated || false,
+                    fullOutputPath: result.fullOutputPath,
+                    timestamp: Date.now(),
+                };
+                // Add to agent state
+                this.agent.appendMessage(bashMessage);
+                // Save to session
+                this.sessionManager.saveMessage(bashMessage);
+            }
+        }
+        catch (error) {
+            const errorMessage = error instanceof Error ? error.message : "Unknown error";
+            if (this.bashComponent) {
+                this.bashComponent.setComplete(null, false);
+            }
+            this.showError(`Bash command failed: ${errorMessage}`);
+        }
+        this.bashComponent = null;
+        this.ui.requestRender();
+    }
+    executeBashCommand(command, onChunk) {
+        return new Promise((resolve, reject) => {
+            const { shell, args } = getShellConfig();
+            const child = spawn(shell, [...args, command], {
+                detached: true,
+                stdio: ["ignore", "pipe", "pipe"],
+            });
+            this.bashProcess = child;
+            // Track output for truncation
+            const chunks = [];
+            let chunksBytes = 0;
+            const maxChunksBytes = DEFAULT_MAX_BYTES * 2;
+            // Temp file for large output
+            let tempFilePath;
+            let tempFileStream;
+            let totalBytes = 0;
+            const handleData = (data) => {
+                totalBytes += data.length;
+                // Start writing to temp file if exceeds threshold
+                if (totalBytes > DEFAULT_MAX_BYTES && !tempFilePath) {
+                    const id = randomBytes(8).toString("hex");
+                    tempFilePath = join(tmpdir(), `pi-bash-${id}.log`);
+                    tempFileStream = createWriteStream(tempFilePath);
+                    for (const chunk of chunks) {
+                        tempFileStream.write(chunk);
+                    }
+                }
+                if (tempFileStream) {
+                    tempFileStream.write(data);
+                }
+                // Keep rolling buffer
+                chunks.push(data);
+                chunksBytes += data.length;
+                while (chunksBytes > maxChunksBytes && chunks.length > 1) {
+                    const removed = chunks.shift();
+                    chunksBytes -= removed.length;
+                }
+                // Stream to component (strip ANSI)
+                const text = stripAnsi(data.toString()).replace(/\r/g, "");
+                onChunk(text);
+            };
+            child.stdout?.on("data", handleData);
+            child.stderr?.on("data", handleData);
+            child.on("close", (code) => {
+                if (tempFileStream) {
+                    tempFileStream.end();
+                }
+                this.bashProcess = null;
+                // Combine buffered chunks for truncation
+                const fullBuffer = Buffer.concat(chunks);
+                const fullOutput = stripAnsi(fullBuffer.toString("utf-8")).replace(/\r/g, "");
+                const truncationResult = truncateTail(fullOutput);
+                // code === null means killed (cancelled)
+                const cancelled = code === null;
+                resolve({
+                    exitCode: code,
+                    cancelled,
+                    truncationResult: truncationResult.truncated ? truncationResult : undefined,
+                    fullOutputPath: tempFilePath,
+                });
+            });
+            child.on("error", (err) => {
+                if (tempFileStream) {
+                    tempFileStream.end();
+                }
+                this.bashProcess = null;
+                reject(err);
+            });
+        });
+    }
     compactionAbortController = null;
     /**
      * Shared logic to execute context compaction.