atom-lite 0.5.1 → 0.6.3

package/dist/cli/app.js

@@ -1,13 +1,15 @@
 import { loadConfig } from "../config/load.js";
+import { DEFAULT_SYSTEM_PROMPT } from "../agent/system-prompt.js";
 import { defaultAgentDir, userConfigDir } from "../config/paths.js";
 import { saveProviderSelection, saveTheme, } from "../config/write.js";
 import { resolveApiKey, saveApiKey } from "../config/credentials.js";
 import { join } from "node:path";
 import { writeFileSync, existsSync } from "node:fs";
-import { execSync } from "node:child_process";
+import { execSync, spawn } from "node:child_process";
+import { defaultShell, terminateProcessTree } from "../security/process.js";
 import { openSearchablePicker } from "./picker.js";
 import { openCommandPalette } from "./palette.js";
-import { bareCommandName, createCommandRegistry } from "./commands.js";
+import { bareCommandName, createCommandRegistry, } from "./commands.js";
 import { resolveTheme, themeLabel, themeNames, markdownThemeFor, } from "./theme.js";
 import { SecretInput } from "./secret-input.js";
 import { StatusLine } from "./status-line.js";
@@ -15,7 +17,7 @@ import { Activity } from "./activity.js";
 import { ToolBox } from "./tool-box.js";
 import { ApprovalPrompt } from "./approval-prompt.js";
 import { Banner } from "./banner.js";
-import { formatContextUsage, knownContextWindow, } from "../providers/context-window.js";
+import { formatContextUsage, knownContextWindow, knownVisionModel, } from "../providers/context-window.js";
 import { StreamText } from "./stream-text.js";
 import { Transcript } from "./transcript.js";
 import { AgentRuntime } from "../agent/loop.js";
@@ -27,8 +29,8 @@ import { ProjectTrustPrompt } from "./trust-prompt.js";
 import { formatSessionList, formatSessionSearch, searchSessions, sessionTitleFromEvents, } from "../sessions/search.js";
 import { providerSwitchNotice, sessionHistoryLines, sessionSelectionFromEvents, } from "../sessions/history.js";
 import { blockAtLine, createMouseCapture, handleMouseEvent, parseSgrMouse, shouldEnableMouse, } from "./mouse.js";
-import { readTool, writeTool, editTool, bashTool, webfetchTool, websearchTool } from "../tools/index.js";
-import { McpManager, formatMcpStatus, formatMcpDoctor, createMcpTools } from "../mcp/index.js";
+import { readTool, writeTool, editTool, bashTool, webfetchTool, websearchTool, grepTool, globTool, taskTool, } from "../tools/index.js";
+import { McpManager, formatMcpStatus, formatMcpDoctor, createMcpTools, } from "../mcp/index.js";
 import { createProvider } from "../providers/factory.js";
 import { availableProviderNames, getPreset, listModelsForProvider, materializeProviderConfig, probeApiKey, providerLabel, PROVIDER_PRESETS, } from "../providers/catalog.js";
 import { ContextManager } from "../context/manager.js";
@@ -89,7 +91,8 @@ export async function run(args) {
         }
     }
     if (showVersion) {
-        process.stdout.write("0.3.1\n");
+        const { VERSION } = await import("../version.js");
+        process.stdout.write(VERSION + "\n");
         return;
     }
     if (listSessions || searchSessionsQuery !== undefined) {
@@ -124,12 +127,23 @@ export async function run(args) {
         modelByProvider.set(provider.id, modelOverride);
     const useTools = !positional.includes("__notools__");
     const builtInTools = useTools
-        ? [readTool, writeTool, editTool, bashTool, webfetchTool, websearchTool]
+        ? [
+            readTool,
+            writeTool,
+            editTool,
+            bashTool,
+            webfetchTool,
+            websearchTool,
+            grepTool,
+            globTool,
+            taskTool,
+        ]
         : [];
     const interactive = process.stdin.isTTY === true && !jsonMode;
     // one-shot task from positional or stdin
     let stdinTask = "";
-    if (!process.stdin.isTTY && positional.filter((p) => !p.startsWith("__")).length === 0) {
+    if (!process.stdin.isTTY &&
+        positional.filter((p) => !p.startsWith("__")).length === 0) {
         const chunks = [];
         for await (const chunk of process.stdin)
             chunks.push(chunk);
@@ -139,9 +153,11 @@ export async function run(args) {
     const oneShot = (taskArgs.join(" ").trim() || stdinTask).trim();
     const store = new JsonlSessionStore();
     let session = sessionOverride
-        ? store.resume(sessionOverride) ?? store.create(process.cwd(), provider.id)
+        ? (store.resume(sessionOverride) ??
+            store.create(process.cwd(), provider.id))
         : positional.includes("__continue__")
-            ? store.resumeLatest(process.cwd()) ?? store.create(process.cwd(), provider.id)
+            ? (store.resumeLatest(process.cwd()) ??
+                store.create(process.cwd(), provider.id))
             : store.create(process.cwd(), provider.id);
     const mcpServers = (config.mcpServers ?? {});
     const mcpManager = new McpManager({
@@ -164,7 +180,9 @@ export async function run(args) {
             try {
                 const { execFileSync } = require("node:child_process");
                 const parts = value.slice(1).trim().split(/\s+/);
-                return execFileSync(parts[0], parts.slice(1), { encoding: "utf8" }).trim();
+                return execFileSync(parts[0], parts.slice(1), {
+                    encoding: "utf8",
+                }).trim();
             }
             catch {
                 return undefined;
@@ -202,7 +220,7 @@ export async function run(args) {
     // Bridge Pi-style extension providers into Atom-lite's config so they
     // appear in /provider, /model, and the command palette.
     for (const loaded of extensionLoader.list()) {
-        for (const [name, provider] of (loaded.registries?.providers ?? new Map())) {
+        for (const [name, provider] of loaded.registries?.providers ?? new Map()) {
             const cfg = provider?.config ?? {};
             const firstModel = Array.isArray(cfg.models) && cfg.models.length > 0
                 ? (cfg.models[0].id ?? cfg.models[0].name ?? "default")
@@ -244,7 +262,7 @@ export async function run(args) {
         return results;
     };
     const contextManager = new ContextManager({
-        systemPrompt: "You are a coding assistant.",
+        systemPrompt: DEFAULT_SYSTEM_PROMPT,
         maxInputTokens: config.context.maxInputTokens ?? null,
         reserveOutputTokens: config.context.reserveOutputTokens,
         compactAtPercent: config.context.compactAtPercent,
@@ -256,7 +274,7 @@ export async function run(args) {
     });
     let sessionTrusted = false;
     const approvals = {
-        mode: config.runtime.approvalMode ?? "ask",
+        mode: "ask",
         // Installed below once the interactive UI helpers exist.
         handler: undefined,
         isTrusted: () => sessionTrusted,
@@ -271,12 +289,20 @@ export async function run(args) {
         workspace: process.cwd(),
         tools: allTools,
         approvals,
+        mode: config.runtime.mode,
+        maxTokensPerSession: config.runtime.maxTokensPerSession,
+        maxTokensPerTurn: config.runtime.maxTokensPerTurn,
         contextManager,
         extensionDispatch,
         bashConcurrency: config.runtime.bashConcurrency,
+        checkpointingEnabled: interactive || Boolean(config.runtime.checkpointing),
     });
     mcpManager.onToolsChanged = () => {
-        runtime.setTools([...builtInTools, ...extensionTools, ...createMcpTools(mcpManager)]);
+        runtime.setTools([
+            ...builtInTools,
+            ...extensionTools,
+            ...createMcpTools(mcpManager),
+        ]);
     };
     if (oneShot || !interactive) {
         // non-interactive / one-shot path
@@ -286,7 +312,11 @@ export async function run(args) {
         try {
             if (useTools) {
                 await mcpManager.syncTools();
-                runtime.setTools([...builtInTools, ...extensionTools, ...createMcpTools(mcpManager)]);
+                runtime.setTools([
+                    ...builtInTools,
+                    ...extensionTools,
+                    ...createMcpTools(mcpManager),
+                ]);
             }
             const result = await runtime.runAgentTurn(oneShot || " ", session, controller.signal);
             if (jsonMode) {
@@ -298,7 +328,9 @@ export async function run(args) {
         }
         catch (err) {
             if (jsonMode)
-                process.stderr.write(JSON.stringify({ error: err instanceof Error ? err.message : String(err) }) + "\n");
+                process.stderr.write(JSON.stringify({
+                    error: err instanceof Error ? err.message : String(err),
+                }) + "\n");
             else
                 process.stderr.write(`atom-lite: ${err instanceof Error ? err.message : String(err)}\n`);
             process.exit(1);
@@ -310,14 +342,16 @@ export async function run(args) {
         return;
     }
     // ---- Interactive TUI (first-class) ----
-    const { TuiAltScreen, Text, Input, Editor, Container, Spacer, ScrollView, VStack, Loader, Markdown, Box } = await import("@earendil-works/pi-tui");
+    const { TuiAltScreen, Text, Input, Editor, Container, Spacer, ScrollView, VStack, Loader, Markdown, Box, } = await import("@earendil-works/pi-tui");
     const mouseEnabled = shouldEnableMouse({
         interactive,
         configured: config.ui.mouse,
         stdoutIsTTY: process.stdout.isTTY === true,
     });
     const terminal = new ProcessTerminal();
-    const tui = new TuiAltScreen(terminal, false, undefined, { mouse: mouseEnabled });
+    const tui = new TuiAltScreen(terminal, false, undefined, {
+        mouse: mouseEnabled,
+    });
     const mouseCapture = createMouseCapture((data) => terminal.write(data));
     let effortLevel = "medium";
     runtime.setEffort(effortLevel);
@@ -345,9 +379,17 @@ export async function run(args) {
             displayPath = "~" + cwd.slice(home.length);
         let branch = "";
         try {
-            branch = execSync("git rev-parse --abbrev-ref HEAD", { cwd, encoding: "utf8", stdio: ["ignore", "pipe", "ignore"] }).trim();
+            branch = execSync("git rev-parse --abbrev-ref HEAD", {
+                cwd,
+                encoding: "utf8",
+                stdio: ["ignore", "pipe", "ignore"],
+            }).trim();
             if (branch === "HEAD") {
-                branch = execSync("git rev-parse --short HEAD", { cwd, encoding: "utf8", stdio: ["ignore", "pipe", "ignore"] }).trim();
+                branch = execSync("git rev-parse --short HEAD", {
+                    cwd,
+                    encoding: "utf8",
+                    stdio: ["ignore", "pipe", "ignore"],
+                }).trim();
             }
         }
         catch {
@@ -359,10 +401,12 @@ export async function run(args) {
     const header = new Text(theme.tokens.dim(headerLabel()), 1, 0);
     const statusBar = new StatusLine(theme);
     const activity = new Activity({ theme, onChange: () => updateStatus() });
+    let currentMode = config.runtime.mode;
     statusBar.setParts({
         left: `${provider.id}/${provider.model ?? "default"}`,
         tokens: "tokens: ?",
         effort: "effort: medium",
+        mode: currentMode,
         activity: activity.text(),
     });
     const selectListTheme = {
@@ -395,20 +439,65 @@ export async function run(args) {
         const sig = skillCacheSignature(process.cwd());
         if (sig === registrySkillSignature)
             return;
-        commandRegistry = createCommandRegistry({ skills: cachedSkillRegistry().list() });
+        commandRegistry = createCommandRegistry({
+            skills: cachedSkillRegistry().list(),
+        });
         registrySkillSignature = sig;
-        input.setAutocompleteProvider(new CombinedAutocompleteProvider(commandRegistry.autocomplete(), process.cwd()));
+        input.setAutocompleteProvider(new CombinedAutocompleteProvider(commandRegistry.autocomplete(), process.cwd(), fdPath));
     }
-    input.setAutocompleteProvider(new CombinedAutocompleteProvider(commandRegistry.autocomplete(), process.cwd()));
+    const fdPath = (() => {
+        for (const bin of ["fd", "fdfind"]) {
+            try {
+                const found = execSync(`command -v ${bin}`, {
+                    encoding: "utf8",
+                    stdio: ["ignore", "pipe", "ignore"],
+                }).trim();
+                if (found)
+                    return found;
+            }
+            catch {
+                // ignore
+            }
+        }
+        return null;
+    })();
+    input.setAutocompleteProvider(new CombinedAutocompleteProvider(commandRegistry.autocomplete(), process.cwd(), fdPath));
+    // Skills should autocomplete (fill the input) rather than execute immediately
+    // when selected from the autocomplete list; Enter on a typed command still
+    // runs them. Convert a confirming Enter into a Tab completion for skills.
+    const originalHandleInput = input.handleInput.bind(input);
+    input.handleInput = (data) => {
+        try {
+            if ((data === "\r" || data === "\n") &&
+                input.autocompleteState &&
+                input.autocompleteList) {
+                const selected = input.autocompleteList.getSelectedItem?.();
+                const name = selected?.value ?? selected?.name ?? selected?.label;
+                const cmd = name
+                    ? commandRegistry.get(String(name).replace(/^\//, ""))
+                    : undefined;
+                if (cmd?.bareInvokable) {
+                    originalHandleInput("\t");
+                    return;
+                }
+            }
+        }
+        catch {
+            // Fall through to default handling on any introspection failure.
+        }
+        originalHandleInput(data);
+    };
     const root = new VStack([]);
-    root.addChild(output, { grow: 1 });
-    root.addChild(new Spacer(1));
-    root.addChild(input);
-    root.addChild(statusBar);
+    // Keep fixed chrome (spacer, input, status, header, dropdown) from being
+    // collapsed when the transcript grows; only the output region may shrink.
+    root.addChild(output, { grow: 1, shrink: 1, minSize: 0 });
+    root.addChild(new Spacer(1), { grow: 0, shrink: 0, minSize: 1 });
+    root.addChild(input, { grow: 0, shrink: 0 });
+    root.addChild(statusBar, { grow: 0, shrink: 0 });
     // folder:branch info below the status line, dimmed
-    root.addChild(header);
+    root.addChild(header, { grow: 0, shrink: 0 });
     const dropdownContainer = new Container();
-    root.addChild(dropdownContainer);
+    root.addChild(dropdownContainer, { grow: 0, shrink: 0 });
     tui.setLayoutRoot(root);
     tui.setFocus(input);
     if (!noColor) {
@@ -536,6 +625,7 @@ export async function run(args) {
         streamText.finalize();
     }
     let activeTurnController = null;
+    let shellMode = false;
     const messageQueue = [];
     const steeringQueue = [];
     runtime.setSteeringQueue(steeringQueue);
@@ -547,9 +637,17 @@ export async function run(args) {
         try {
             if (useTools) {
                 await mcpManager.syncTools();
-                runtime.setTools([...builtInTools, ...extensionTools, ...createMcpTools(mcpManager)]);
+                runtime.setTools([
+                    ...builtInTools,
+                    ...extensionTools,
+                    ...createMcpTools(mcpManager),
+                ]);
+            }
+            const turnResult = await runtime.runAgentTurn(prompt, session, controller.signal);
+            if (turnResult === "TOKEN_BUDGET_REACHED") {
+                appendLine("token budget reached; stopping", theme.tokens.warning);
+                tui.requestRender();
             }
-            await runtime.runAgentTurn(prompt, session, controller.signal);
         }
         catch (err) {
             if (err instanceof Error && err.message === "USER_CANCELLED") {
@@ -569,12 +667,94 @@ export async function run(args) {
             }
         }
     }
+    /** Run a shell command directly (shell mode / `!` prefix). */
+    async function runShellCommand(command) {
+        const shell = defaultShell();
+        const flag = process.platform === "win32" ? "/c" : "-c";
+        const start = Date.now();
+        const cap = 30000;
+        let stdout = "";
+        let stderr = "";
+        let truncated = false;
+        const append = (target, chunk) => {
+            const s = chunk.toString("utf8");
+            let accepted = s;
+            if (target === "out") {
+                if (stdout.length + s.length > cap) {
+                    truncated = true;
+                    accepted = s.slice(0, Math.max(0, cap - stdout.length));
+                    stdout += accepted;
+                }
+                else {
+                    stdout += s;
+                }
+            }
+            else {
+                if (stderr.length + s.length > cap) {
+                    truncated = true;
+                    accepted = s.slice(0, Math.max(0, cap - stderr.length));
+                    stderr += accepted;
+                }
+                else {
+                    stderr += s;
+                }
+            }
+            if (accepted) {
+                tui.requestRender();
+            }
+        };
+        try {
+            const child = spawn(shell, [flag, command], {
+                cwd: process.cwd(),
+                env: process.env,
+                detached: process.platform !== "win32",
+            });
+            child.stdout?.on("data", (c) => append("out", c));
+            child.stderr?.on("data", (c) => append("err", c));
+            const exit = await new Promise((resolve) => {
+                const timer = setTimeout(() => {
+                    terminateProcessTree(child, "SIGTERM");
+                    setTimeout(() => terminateProcessTree(child, "SIGKILL"), 1000).unref();
+                    resolve({ code: null, signal: "SIGTERM" });
+                }, 120000);
+                timer.unref();
+                child.on("exit", (code, signal) => {
+                    clearTimeout(timer);
+                    resolve({ code, signal });
+                });
+                child.on("error", (err) => {
+                    clearTimeout(timer);
+                    resolve({ code: null, signal: null });
+                    stderr = `COMMAND_FAILED: ${err instanceof Error ? err.message : String(err)}`;
+                });
+            });
+            const parts = [];
+            parts.push(`exit_code: ${exit.code ?? "null"}`);
+            if (exit.signal)
+                parts.push(`signal: ${exit.signal}`);
+            if (truncated)
+                parts.push("OUTPUT_TRUNCATED");
+            parts.push(`duration_ms: ${Date.now() - start}`);
+            if (stdout)
+                parts.push(`stdout:\n${stdout}`);
+            if (stderr)
+                parts.push(`stderr:\n${stderr}`);
+            const content = parts.join("\n");
+            appendLine(content, exit.code === 0 && !exit.signal && !stderr
+                ? theme.tokens.success
+                : theme.tokens.error);
+        }
+        catch (err) {
+            appendLine(`shell error: ${err instanceof Error ? err.message : String(err)}`, theme.tokens.error);
+        }
+    }
     function updateStatus() {
         const usage = runtime.getContextUsage();
         statusBar.setParts({
             left: `${provider.id}/${currentModelLabel()}`,
             tokens: `tokens: ${formatContextUsage(usage.used, usage.window)}`,
             effort: `effort: ${effortLevel}`,
+            mode: currentMode,
             activity: activity.text(),
         });
         tui.requestRender();
@@ -612,11 +792,21 @@ export async function run(args) {
     function refreshContextWindow() {
         const profile = provider.id;
         const model = currentModelLabel();
-        const configured = profileConfig(profile)?.contextWindow;
-        const window = typeof configured === "number" && configured > 0
-            ? configured
+        const pcfg = profileConfig(profile);
+        const configuredWindow = pcfg?.contextWindow;
+        const window = typeof configuredWindow === "number" && configuredWindow > 0
+            ? configuredWindow
             : knownContextWindow(model);
         runtime.setContextWindow(window);
+        // Config override for vision takes precedence over provider metadata.
+        if (typeof pcfg?.supportsVision === "boolean") {
+            runtime.setSupportsVision(pcfg.supportsVision);
+        }
+        else {
+            const guessed = knownVisionModel(model);
+            if (typeof guessed === "boolean")
+                runtime.setSupportsVision(guessed);
+        }
         updateStatus();
         const getModelInfo = provider.getModelInfo;
         if (typeof getModelInfo !== "function")
@@ -627,10 +817,16 @@ export async function run(args) {
             if (provider.id !== profile || currentModelLabel() !== model)
                 return;
             const resolved = info?.contextWindow;
-            if (typeof resolved === "number" && resolved > 0) {
+            if (typeof resolved === "number" &&
+                resolved > 0 &&
+                typeof configuredWindow !== "number") {
                 runtime.setContextWindow(resolved);
                 updateStatus();
             }
+            if (typeof pcfg?.supportsVision !== "boolean" &&
+                typeof info?.supportsVision === "boolean") {
+                runtime.setSupportsVision(info.supportsVision);
+            }
         })
             .catch(() => {
             // Keep the static/config fallback.
@@ -641,7 +837,7 @@ export async function run(args) {
     const MODEL_CACHE_TTL_MS = 5 * 60 * 1000;
     const modelCache = new Map();
     for (const loaded of extensionLoader.list()) {
-        for (const [name, provider] of (loaded.registries?.providers ?? new Map())) {
+        for (const [name, provider] of loaded.registries?.providers ?? new Map()) {
             const cfg = provider?.config ?? {};
             const models = Array.isArray(cfg.models)
                 ? cfg.models.map((m) => m.id ?? m.name).filter(Boolean)
@@ -877,7 +1073,11 @@ export async function run(args) {
         provider = createProvider(config, preset.id, chosen);
         runtime.switchModel(provider, chosen);
         if (typeof session.appendEvent === "function") {
-            session.appendEvent({ type: "provider_selected", provider: preset.id, model: chosen });
+            session.appendEvent({
+                type: "provider_selected",
+                provider: preset.id,
+                model: chosen,
+            });
         }
         refreshContextWindow();
         appendLine(`provider: ${preset.label} (${preset.id}) · model: ${chosen}`);
@@ -1085,13 +1285,15 @@ export async function run(args) {
             appendLine(`skill not found: ${name}`);
             return false;
         }
+        if (extra) {
+            appendLine(`> ${extra}`, theme.tokens.user);
+        }
         session.appendMessage({ role: "system", content });
         if (typeof session.appendEvent === "function") {
             session.appendEvent({ type: "skill_loaded", name });
         }
-        appendLine(`Loaded skill: ${name}`);
+        appendLine(`Loaded skill: ${name}`, theme.tokens.dim);
         if (extra) {
-            appendLine(`> ${extra}`);
             await runTurn(extra);
         }
         return true;
@@ -1126,7 +1328,9 @@ export async function run(args) {
                 return "No extensions discovered.";
             return report
                 .map((ext) => {
-                const origin = ext.origin ? `${ext.source}:${ext.origin}` : ext.source;
+                const origin = ext.origin
+                    ? `${ext.source}:${ext.origin}`
+                    : ext.source;
                 const version = ext.version ? `v${ext.version}` : "-";
                 const counts = `${ext.toolCount} tools / ${ext.commandCount} commands`;
                 const error = ext.error ? `\t${ext.error}` : "";
@@ -1144,7 +1348,9 @@ export async function run(args) {
             return showThinking;
         },
         setTheme: setThemeTarget,
-        listSkills: () => cachedSkillRegistry().list().map((skill) => ({
+        listSkills: () => cachedSkillRegistry()
+            .list()
+            .map((skill) => ({
             name: skill.name,
             description: skill.description,
         })),
@@ -1211,9 +1417,7 @@ export async function run(args) {
             const summaries = scope === "all" ? store.list() : store.list(process.cwd());
             const toggle = {
                 value: "__scope__",
-                label: scope === "all"
-                    ? "Show this workspace only"
-                    : "Show all workspaces",
+                label: scope === "all" ? "Show this workspace only" : "Show all workspaces",
                 description: `scope: ${scope === "all" ? "all workspaces" : process.cwd()}`,
             };
             return [
@@ -1237,7 +1441,9 @@ export async function run(args) {
                 if (item.value === "__scope__") {
                     scope = scope === "all" ? "workspace" : "all";
                     picker.updateItems(buildItems());
-                    picker.setStatus(scope === "all" ? "showing all workspaces" : "showing this workspace");
+                    picker.setStatus(scope === "all"
+                        ? "showing all workspaces"
+                        : "showing this workspace");
                     return;
                 }
                 picker.close();
@@ -1359,6 +1565,7 @@ export async function run(args) {
         for (const box of toolBoxes) {
             box.setExpanded(expandTools);
         }
+        transcript.invalidate();
         tui.requestRender();
     }
     async function reloadExtensions() {
@@ -1393,7 +1600,8 @@ export async function run(args) {
         }
         extensionProviderNames.clear();
         for (const loaded of extensionLoader.list()) {
-            for (const [name, provider] of (loaded.registries?.providers ?? new Map())) {
+            for (const [name, provider] of loaded.registries?.providers ??
+                new Map()) {
                 const cfg = provider?.config ?? {};
                 const firstModel = Array.isArray(cfg.models) && cfg.models.length > 0
                     ? (cfg.models[0].id ?? cfg.models[0].name ?? "default")
@@ -1482,6 +1690,7 @@ export async function run(args) {
         if (!(child instanceof ToolBox))
             return false;
         child.toggleExpanded();
+        transcript.invalidate();
         return true;
     }
     tui.addInputListener((data) => {
@@ -1516,11 +1725,28 @@ export async function run(args) {
             toggleTools();
             return { consume: true };
         }
-        // Escape cancels the active turn.
-        if (matchesKey(data, Key.escape) && !uiModalOpen && activeTurnController) {
-            activeTurnController.abort();
+        // Tab toggles plan/yolo mode.
+        if (matchesKey(data, Key.tab) && !uiModalOpen) {
+            currentMode = currentMode === "plan" ? "yolo" : "plan";
+            runtime.setMode(currentMode);
+            updateStatus();
+            appendLine(`mode: ${currentMode}`, theme.tokens.dim);
+            tui.requestRender();
             return { consume: true };
         }
+        // Escape cancels the active turn, or exits shell mode.
+        if (matchesKey(data, Key.escape) && !uiModalOpen) {
+            if (activeTurnController) {
+                activeTurnController.abort();
+                return { consume: true };
+            }
+            if (shellMode) {
+                shellMode = false;
+                appendLine("shell mode: off", theme.tokens.dim);
+                tui.requestRender();
+                return { consume: true };
+            }
+        }
         return undefined;
     });
     runtime["onEvent"] = (e) => {
@@ -1560,10 +1786,34 @@ export async function run(args) {
         else if (e.type === "context_usage") {
             updateStatus();
         }
+        else if (e.type === "background_task_complete") {
+            appendLine(`background task ${e.id}: ${e.status} (exit ${e.exitCode ?? "null"})`, theme.tokens.dim);
+            tui.requestRender();
+        }
+        else if (e.type === "checkpoint_created") {
+            appendLine(`checkpoint ${e.id} created`, theme.tokens.dim);
+            tui.requestRender();
+        }
+        else if (e.type === "checkpoint_failed") {
+            appendLine(`checkpoint failed: ${e.reason}`, theme.tokens.warning);
+            tui.requestRender();
+        }
+        else if (e.type === "checkpoint_unavailable") {
+            appendLine(`checkpoint unavailable: ${e.reason ?? "not a git repository"}`, theme.tokens.warning);
+            tui.requestRender();
+        }
+        else if (e.type === "budget_warning") {
+            appendLine(`budget warning (${e.scope}): ${e.used}/${e.limit} tokens`, theme.tokens.warning);
+            tui.requestRender();
+        }
         else if (e.type === "model_switched") {
             appendLine(`switched model to ${e.provider}/${e.model}`);
             if (typeof session.appendEvent === "function") {
-                session.appendEvent({ type: "model_switched", provider: e.provider, model: e.model });
+                session.appendEvent({
+                    type: "model_switched",
+                    provider: e.provider,
+                    model: e.model,
+                });
             }
             updateStatus();
         }
@@ -1581,6 +1831,7 @@ export async function run(args) {
             if (box) {
                 activity.setPhase("working", e.name);
                 box.appendLive(e.text);
+                transcript.invalidate();
                 tui.requestRender();
             }
         }
@@ -1608,7 +1859,11 @@ export async function run(args) {
             lines.push(`  config: FAIL ${e instanceof Error ? e.message : String(e)}`);
         }
         try {
-            const r = await provider.generate({ model: modelOverride ?? "default", messages: [{ role: "user", content: [{ type: "text", text: "ping" }] }], tools: [] });
+            const r = await provider.generate({
+                model: modelOverride ?? "default",
+                messages: [{ role: "user", content: [{ type: "text", text: "ping" }] }],
+                tools: [],
+            });
             lines.push(`  provider generate: ok (${r.finishReason})`);
         }
         catch (e) {
@@ -1633,7 +1888,11 @@ export async function run(args) {
             }
             if (useTools) {
                 await mcpManager.syncTools();
-                runtime.setTools([...builtInTools, ...extensionTools, ...createMcpTools(mcpManager)]);
+                runtime.setTools([
+                    ...builtInTools,
+                    ...extensionTools,
+                    ...createMcpTools(mcpManager),
+                ]);
             }
             appendLine(formatMcpStatus(mcpManager.status()));
             return;
@@ -1741,7 +2000,9 @@ export async function run(args) {
         for (const message of session.messages) {
             if (message.role === "user") {
                 turn++;
-                const snippet = redactText(messageText(message)).split("\\n")[0].slice(0, 80);
+                const snippet = redactText(messageText(message))
+                    .split("\\n")[0]
+                    .slice(0, 80);
                 lines.push(`${turn}. ${snippet || "(empty)"}`);
             }
         }
@@ -1806,6 +2067,27 @@ export async function run(args) {
             tui.requestRender();
             return;
         }
+        // Shell mode: every line is a shell command. `!` alone (or as first char)
+        // enters shell mode; Escape exits it.
+        if (shellMode) {
+            const cmd = trimmed.startsWith("!") ? trimmed.slice(1).trim() : trimmed;
+            appendLine(`$ ${cmd}`, theme.tokens.user);
+            await runShellCommand(cmd);
+            tui.requestRender();
+            return;
+        }
+        // `!` as first character enters shell mode.
+        if (trimmed === "!" || trimmed.startsWith("! ")) {
+            shellMode = true;
+            const rest = trimmed.slice(1).trim();
+            appendLine("shell mode: on (Escape to exit)", theme.tokens.dim);
+            if (rest) {
+                appendLine(`$ ${rest}`, theme.tokens.user);
+                await runShellCommand(rest);
+            }
+            tui.requestRender();
+            return;
+        }
         refreshCommandRegistry();
         if (trimmed === "/steer" || trimmed.startsWith("/steer ")) {
             const text = trimmed.slice("/steer".length).trim();
@@ -1861,7 +2143,7 @@ export async function run(args) {
                 tui.setFocus(input);
                 activity.setPhase("working");
                 if (decision === "trust-session") {
-                    appendLine("trusted this session: shell commands will run without approval");
+                    appendLine("trusted this session: outside-workspace operations will run without approval", theme.tokens.dim);
                 }
                 resolve(decision);
             });