Something went wrong. Try again.
more up to date at codeberg https://codeberg.org/cowie/tau.git
Something went wrong. Try again.
6.8 kB · 218 lines
TypeScript
at main
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219/** * Component for displaying bash command execution with streaming output. */
import { Container, Loader, Spacer, Text, type TUI } from "@earendil-works/pi-tui";import { DEFAULT_MAX_BYTES, DEFAULT_MAX_LINES, type TruncationResult, truncateTail,} from "../../../core/tools/truncate.ts";import { stripAnsi } from "../../../utils/ansi.ts";import { theme } from "../theme/theme.ts";import { DynamicBorder } from "./dynamic-border.ts";import { keyHint, keyText } from "./keybinding-hints.ts";import { truncateToVisualLines } from "./visual-truncate.ts";
// Preview line limit when not expanded (matches tool execution behavior)const PREVIEW_LINES = 20;
export class BashExecutionComponent extends Container { private command: string; private outputLines: string[] = []; private status: "running" | "complete" | "cancelled" | "error" = "running"; private exitCode: number | undefined = undefined; private loader: Loader; private truncationResult?: TruncationResult; private fullOutputPath?: string; private expanded = false; private contentContainer: Container;
constructor(command: string, ui: TUI, excludeFromContext = false) { super(); this.command = command;
// Use dim border for excluded-from-context commands (!! prefix) const colorKey = excludeFromContext ? "dim" : "bashMode"; const borderColor = (str: string) => theme.fg(colorKey, str);
// Add spacer this.addChild(new Spacer(1));
// Top border this.addChild(new DynamicBorder(borderColor));
// Content container (holds dynamic content between borders) this.contentContainer = new Container(); this.addChild(this.contentContainer);
// Command header const header = new Text(theme.fg(colorKey, theme.bold(`$ ${command}`)), 1, 0); this.contentContainer.addChild(header);
// Loader this.loader = new Loader( ui, (spinner) => theme.fg(colorKey, spinner), (text) => theme.fg("muted", text), `Running... (${keyText("tui.select.cancel")} to cancel)`, // Plain text for loader ); this.contentContainer.addChild(this.loader);
// Bottom border this.addChild(new DynamicBorder(borderColor)); }
/** * Set whether the output is expanded (shows full output) or collapsed (preview only). */ setExpanded(expanded: boolean): void { this.expanded = expanded; this.updateDisplay(); }
override invalidate(): void { super.invalidate(); this.updateDisplay(); }
appendOutput(chunk: string): void { // Strip ANSI codes and normalize line endings // Note: binary data is already sanitized in tui-renderer.ts executeBashCommand const clean = stripAnsi(chunk).replace(/\r\n/g, "\n").replace(/\r/g, "\n");
// Append to output lines const newLines = clean.split("\n"); if (this.outputLines.length > 0 && newLines.length > 0) { // Append first chunk to last line (incomplete line continuation) this.outputLines[this.outputLines.length - 1] += newLines[0]; this.outputLines.push(...newLines.slice(1)); } else { this.outputLines.push(...newLines); }
this.updateDisplay(); }
setComplete( exitCode: number | undefined, cancelled: boolean, truncationResult?: TruncationResult, fullOutputPath?: string, ): void { this.exitCode = exitCode; this.status = cancelled ? "cancelled" : exitCode !== 0 && exitCode !== undefined && exitCode !== null ? "error" : "complete"; this.truncationResult = truncationResult; this.fullOutputPath = fullOutputPath;
// Stop loader this.loader.stop();
this.updateDisplay(); }
private updateDisplay(): void { // Apply truncation for LLM context limits (same limits as bash tool) const fullOutput = this.outputLines.join("\n"); const contextTruncation = truncateTail(fullOutput, { maxLines: DEFAULT_MAX_LINES, maxBytes: DEFAULT_MAX_BYTES, });
// Get the lines to potentially display (after context truncation) const availableLines = contextTruncation.content ? contextTruncation.content.split("\n") : [];
// Apply preview truncation based on expanded state const previewLogicalLines = availableLines.slice(-PREVIEW_LINES); const hiddenLineCount = availableLines.length - previewLogicalLines.length;
// Rebuild content container this.contentContainer.clear();
// Command header const header = new Text(theme.fg("bashMode", theme.bold(`$ ${this.command}`)), 1, 0); this.contentContainer.addChild(header);
// Output if (availableLines.length > 0) { if (this.expanded) { // Show all lines const displayText = availableLines.map((line) => theme.fg("muted", line)).join("\n"); this.contentContainer.addChild(new Text(`\n${displayText}`, 1, 0)); } else { // Use shared visual truncation utility with width-aware caching const styledOutput = previewLogicalLines.map((line) => theme.fg("muted", line)).join("\n"); const styledInput = `\n${styledOutput}`; let cachedWidth: number | undefined; let cachedLines: string[] | undefined; this.contentContainer.addChild({ render: (width: number) => { if (cachedLines === undefined || cachedWidth !== width) { const result = truncateToVisualLines(styledInput, PREVIEW_LINES, width, 1); cachedLines = result.visualLines; cachedWidth = width; } return cachedLines ?? []; }, invalidate: () => { cachedWidth = undefined; cachedLines = undefined; }, }); } }
// Loader or status if (this.status === "running") { this.contentContainer.addChild(this.loader); } else { const statusParts: string[] = [];
// Show how many lines are hidden (collapsed preview) if (hiddenLineCount > 0) { if (this.expanded) { statusParts.push(`(${keyHint("app.tools.expand", "to collapse")})`); } else { statusParts.push( `${theme.fg("muted", `... ${hiddenLineCount} more lines`)} (${keyHint("app.tools.expand", "to expand")})`, ); } }
if (this.status === "cancelled") { statusParts.push(theme.fg("warning", "(cancelled)")); } else if (this.status === "error") { statusParts.push(theme.fg("error", `(exit ${this.exitCode})`)); }
// Add truncation warning (context truncation, not preview truncation) const wasTruncated = this.truncationResult?.truncated || contextTruncation.truncated; if (wasTruncated && this.fullOutputPath) { statusParts.push(theme.fg("warning", `Output truncated. Full output: ${this.fullOutputPath}`)); }
if (statusParts.length > 0) { this.contentContainer.addChild(new Text(`\n${statusParts.join("\n")}`, 1, 0)); } } }
/** * Get the raw output for creating BashExecutionMessage. */ getOutput(): string { return this.outputLines.join("\n"); }
/** * Get the command that was executed. */ getCommand(): string { return this.command; }}