@mariozechner/pi-coding-agent 0.13.2 → 0.14.1

package/dist/main.js

@@ -1,19 +1,25 @@
 import { Agent, ProviderTransport } from "@mariozechner/pi-agent-core";
 import { ProcessTerminal, TUI } from "@mariozechner/pi-tui";
 import chalk from "chalk";
-import { existsSync, readFileSync, statSync } from "fs";
-import { homedir } from "os";
+import { spawn } from "child_process";
+import { randomBytes } from "crypto";
+import { createWriteStream, existsSync, readFileSync, statSync } from "fs";
+import { homedir, tmpdir } from "os";
 import { extname, join, resolve } from "path";
+import stripAnsi from "strip-ansi";
 import { getChangelogPath, getNewEntries, parseChangelog } from "./changelog.js";
 import { calculateContextTokens, compact, shouldCompact } from "./compaction.js";
 import { APP_NAME, CONFIG_DIR_NAME, ENV_AGENT_DIR, getAgentDir, getModelsPath, getReadmePath, VERSION, } from "./config.js";
 import { exportFromFile } from "./export-html.js";
+import { messageTransformer } from "./messages.js";
 import { findModel, getApiKeyForModel, getAvailableModels } from "./model-config.js";
 import { loadSessionFromEntries, SessionManager } from "./session-manager.js";
 import { SettingsManager } from "./settings-manager.js";
+import { getShellConfig } from "./shell.js";
 import { expandSlashCommand, loadSlashCommands } from "./slash-commands.js";
 import { initTheme } from "./theme/theme.js";
 import { allTools, codingTools } from "./tools/index.js";
+import { DEFAULT_MAX_BYTES, truncateTail } from "./tools/truncate.js";
 import { ensureTool } from "./tools-manager.js";
 import { SessionSelectorComponent } from "./tui/session-selector.js";
 import { TuiRenderer } from "./tui/tui-renderer.js";
@@ -88,11 +94,16 @@ function parseArgs(args) {
         }
         else if (arg === "--thinking" && i + 1 < args.length) {
             const level = args[++i];
-            if (level === "off" || level === "minimal" || level === "low" || level === "medium" || level === "high") {
+            if (level === "off" ||
+                level === "minimal" ||
+                level === "low" ||
+                level === "medium" ||
+                level === "high" ||
+                level === "xhigh") {
                 result.thinking = level;
             }
             else {
-                console.error(chalk.yellow(`Warning: Invalid thinking level "${level}". Valid values: off, minimal, low, medium, high`));
+                console.error(chalk.yellow(`Warning: Invalid thinking level "${level}". Valid values: off, minimal, low, medium, high, xhigh`));
             }
         }
         else if (arg === "--print" || arg === "-p") {
@@ -212,7 +223,7 @@ ${chalk.bold("Options:")}
   --models <patterns>            Comma-separated model patterns for quick cycling with Ctrl+P
   --tools <tools>                Comma-separated list of tools to enable (default: read,bash,edit,write)
                                  Available: read, bash, edit, write, grep, find, ls
-  --thinking <level>             Set thinking level: off, minimal, low, medium, high
+  --thinking <level>             Set thinking level: off, minimal, low, medium, high, xhigh
   --export <file>                Export session file to HTML and exit
   --help, -h                     Show this help
 
@@ -509,7 +520,12 @@ async function resolveModelScope(patterns) {
         let thinkingLevel = "off";
         if (parts.length > 1) {
             const level = parts[1];
-            if (level === "off" || level === "minimal" || level === "low" || level === "medium" || level === "high") {
+            if (level === "off" ||
+                level === "minimal" ||
+                level === "low" ||
+                level === "medium" ||
+                level === "high" ||
+                level === "xhigh") {
                 thinkingLevel = level;
             }
             else {
@@ -599,8 +615,8 @@ async function selectSession(sessionManager) {
         ui.start();
     });
 }
-async function runInteractiveMode(agent, sessionManager, settingsManager, version, changelogMarkdown = null, modelFallbackMessage = null, versionCheckPromise, scopedModels = [], initialMessages = [], initialMessage, initialAttachments, fdPath = null) {
-    const renderer = new TuiRenderer(agent, sessionManager, settingsManager, version, changelogMarkdown, scopedModels, fdPath);
+async function runInteractiveMode(agent, sessionManager, settingsManager, version, changelogMarkdown = null, collapseChangelog = false, modelFallbackMessage = null, versionCheckPromise, scopedModels = [], initialMessages = [], initialMessage, initialAttachments, fdPath = null) {
+    const renderer = new TuiRenderer(agent, sessionManager, settingsManager, version, changelogMarkdown, collapseChangelog, scopedModels, fdPath);
     // Initialize TUI (subscribes to agent events internally)
     await renderer.init();
     // Handle version check result when it completes (don't block)
@@ -687,6 +703,70 @@ async function runSingleShotMode(agent, _sessionManager, messages, mode, initial
         }
     }
 }
+/**
+ * Execute a bash command for RPC mode.
+ * Similar to tui-renderer's executeBashCommand but without streaming callbacks.
+ */
+async function executeRpcBashCommand(command) {
+    return new Promise((resolve, reject) => {
+        const { shell, args } = getShellConfig();
+        const child = spawn(shell, [...args, command], {
+            detached: true,
+            stdio: ["ignore", "pipe", "pipe"],
+        });
+        const chunks = [];
+        let chunksBytes = 0;
+        const maxChunksBytes = DEFAULT_MAX_BYTES * 2;
+        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;
+            }
+        };
+        child.stdout?.on("data", handleData);
+        child.stderr?.on("data", handleData);
+        child.on("close", (code) => {
+            if (tempFileStream) {
+                tempFileStream.end();
+            }
+            // Combine buffered chunks
+            const fullBuffer = Buffer.concat(chunks);
+            const fullOutput = stripAnsi(fullBuffer.toString("utf-8")).replace(/\r/g, "");
+            const truncationResult = truncateTail(fullOutput);
+            resolve({
+                output: fullOutput,
+                exitCode: code,
+                truncationResult: truncationResult.truncated ? truncationResult : undefined,
+                fullOutputPath: tempFilePath,
+            });
+        });
+        child.on("error", (err) => {
+            if (tempFileStream) {
+                tempFileStream.end();
+            }
+            reject(err);
+        });
+    });
+}
 async function runRpcMode(agent, sessionManager, settingsManager) {
     // Track if auto-compaction is in progress
     let autoCompactionInProgress = false;
@@ -796,6 +876,35 @@ async function runRpcMode(agent, sessionManager, settingsManager) {
                     console.log(JSON.stringify({ type: "error", error: `Compaction failed: ${error.message}` }));
                 }
             }
+            else if (input.type === "bash" && input.command) {
+                // Execute bash command and add to context
+                try {
+                    const result = await executeRpcBashCommand(input.command);
+                    // Create bash execution message
+                    const bashMessage = {
+                        role: "bashExecution",
+                        command: input.command,
+                        output: result.truncationResult?.content || result.output,
+                        exitCode: result.exitCode,
+                        cancelled: false,
+                        truncated: result.truncationResult?.truncated || false,
+                        fullOutputPath: result.fullOutputPath,
+                        timestamp: Date.now(),
+                    };
+                    // Add to agent state and save to session
+                    agent.appendMessage(bashMessage);
+                    sessionManager.saveMessage(bashMessage);
+                    // Initialize session if needed (same logic as message_end handler)
+                    if (sessionManager.shouldInitializeSession(agent.state.messages)) {
+                        sessionManager.startSession(agent.state);
+                    }
+                    // Emit bash_end event with the message
+                    console.log(JSON.stringify({ type: "bash_end", message: bashMessage }));
+                }
+                catch (error) {
+                    console.log(JSON.stringify({ type: "error", error: `Bash command failed: ${error.message}` }));
+                }
+            }
         }
         catch (error) {
             // Output error as JSON
@@ -1049,6 +1158,7 @@ export async function main(args) {
             thinkingLevel: initialThinking,
             tools: selectedTools,
         },
+        messageTransformer,
         queueMode: settingsManager.getQueueMode(),
         transport: new ProviderTransport({
             // Dynamic API key lookup based on current model's provider
@@ -1163,7 +1273,8 @@ export async function main(args) {
         // Ensure fd tool is available for file autocomplete
         const fdPath = await ensureTool("fd");
         // Interactive mode - use TUI (may have initial messages from CLI args)
-        await runInteractiveMode(agent, sessionManager, settingsManager, VERSION, changelogMarkdown, modelFallbackMessage, versionCheckPromise, scopedModels, parsed.messages, initialMessage, initialAttachments, fdPath);
+        const collapseChangelog = settingsManager.getCollapseChangelog();
+        await runInteractiveMode(agent, sessionManager, settingsManager, VERSION, changelogMarkdown, collapseChangelog, modelFallbackMessage, versionCheckPromise, scopedModels, parsed.messages, initialMessage, initialAttachments, fdPath);
     }
     else {
         // Non-interactive mode (--print flag or --mode flag)