@linxiraos/pi-tui 1.1.26 → 1.2.0

package/src/tools/edit.ts

@@ -36,6 +36,7 @@ import {
 	invalidateRenderedStringCache,
 	PREVIEW_LIMITS,
 	previewWindowRows,
+	releaseRenderedStringCache,
 	type RenderedStringCache,
 	replaceTabs,
 	shortenPath,
@@ -1110,41 +1111,49 @@ export const editToolRenderer = {
 			return renderInlineEditRow(uiTheme, { op, rename, rawPath, pending: true });
 		}
 		const callPreviewCaches: RenderedStringCache[] = [];
-		return framedToolCard(uiTheme, ({ width }) => {
-			// No status icon on the head row: it's the head of the framed block,
-			// and native-scrollback commits are prefix-only — an animated glyph
-			// would pin the commit boundary at the top, and the pending hourglass
-			// just adds noise. The liveness cue rides the trailing "(preview)" /
-			// "(streaming)" line instead.
-			const header = renderEditHeader(width, uiTheme, {
-				op,
-				rawPath,
-				rename,
-				extraSuffix: fileCount > 1 ? uiTheme.fg("dim", ` (+${fileCount - 1} more)`) : undefined,
-			});
-			let body = getCallPreview(
-				editArgs,
-				rawPath,
-				width,
-				uiTheme,
-				renderContext,
-				options.expanded,
-				options?.spinnerFrame,
-				callPreviewCaches,
-			);
-			if (applyPatchError) {
-				body += `\n${uiTheme.fg("error", truncateToWidth(replaceTabs(applyPatchError), Math.max(1, width - 2)))}`;
-			}
-			const bodyLines = body ? body.split("\n") : [];
-			while (bodyLines.length > 0 && bodyLines[0].trim() === "") bodyLines.shift();
-			return {
-				header,
-				sections: bodyLines.length > 0 ? [{ content: bodyLines }] : [],
-				phase: applyPatchError ? "error" : "pending",
-				borderColor: applyPatchError ? "error" : "borderMuted",
-				contentPaddingLeft: 0,
-			};
-		});
+		return framedToolCard(
+			uiTheme,
+			({ width }) => {
+				// No status icon on the head row: it's the head of the framed block,
+				// and native-scrollback commits are prefix-only — an animated glyph
+				// would pin the commit boundary at the top, and the pending hourglass
+				// just adds noise. The liveness cue rides the trailing "(preview)" /
+				// "(streaming)" line instead.
+				const header = renderEditHeader(width, uiTheme, {
+					op,
+					rawPath,
+					rename,
+					extraSuffix: fileCount > 1 ? uiTheme.fg("dim", ` (+${fileCount - 1} more)`) : undefined,
+				});
+				let body = getCallPreview(
+					editArgs,
+					rawPath,
+					width,
+					uiTheme,
+					renderContext,
+					options.expanded,
+					options?.spinnerFrame,
+					callPreviewCaches,
+				);
+				if (applyPatchError) {
+					body += `\n${uiTheme.fg("error", truncateToWidth(replaceTabs(applyPatchError), Math.max(1, width - 2)))}`;
+				}
+				const bodyLines = body ? body.split("\n") : [];
+				while (bodyLines.length > 0 && bodyLines[0].trim() === "") bodyLines.shift();
+				return {
+					header,
+					sections: bodyLines.length > 0 ? [{ content: bodyLines }] : [],
+					phase: applyPatchError ? "error" : "pending",
+					borderColor: applyPatchError ? "error" : "borderMuted",
+					contentPaddingLeft: 0,
+				};
+			},
+			{
+				onReleaseRenderCaches: () => {
+					for (const cache of callPreviewCaches) releaseRenderedStringCache(cache);
+				},
+			},
+		);
 	},
 
 	describeCall(
@@ -1452,74 +1461,54 @@ function renderSingleFileResult(
 	const renderedDiffCache = createRenderedStringCache();
 	const statsSuffixCache = createRenderedStringCache();
 
-	return framedToolCard(uiTheme, ({ width }) => {
-		const { expanded, renderContext } = options;
-		// A finalized result is authoritative: its `details` describe exactly
-		// what happened. The shared streaming `editDiffPreview` is a call-phase
-		// artifact (in a batch it reflects only the first file), so consulting it
-		// for an empty-diff delete/move/no-op result mislabels the card. Fall
-		// back to the preview only when no details exist yet.
-		const editDiffPreview = details ? undefined : renderContext?.editDiffPreview;
-		const renderDiffFn = renderContext?.renderDiff ?? renderFallbackDiff;
-
-		if (diffSectionRenderDiffFn !== renderDiffFn) {
-			diffSectionRenderDiffFn = renderDiffFn;
-			invalidateRenderedStringCache(diffSectionCache);
-			invalidateRenderedStringCache(renderedDiffCache);
-		}
-		const firstChangedLine =
-			(editDiffPreview && "firstChangedLine" in editDiffPreview ? editDiffPreview.firstChangedLine : undefined) ||
-			(details && !isError ? details.firstChangedLine : undefined);
-		const linkPath = details && "path" in details ? details.path : undefined;
-
-		// Change stats ride inline on the header bar next to the path.
-		const previewDiff = editDiffPreview && !("error" in editDiffPreview) ? editDiffPreview.diff : undefined;
-		const headerDiff = isError ? undefined : details?.diff || previewDiff;
-		const statsSuffix = headerDiff
-			? cachedRenderedString(statsSuffixCache, uiTheme, false, "", headerDiff, () =>
-					formatDiffStatsSuffix(headerDiff, uiTheme),
-				)
-			: "";
-		const header = renderEditHeader(width, uiTheme, {
-			icon: isError ? "error" : "success",
-			iconOverride: !isError && !options.isPartial ? uiTheme.styledSymbol("tool.edit", "accent") : undefined,
-			op,
-			rawPath,
-			rename,
-			firstChangedLine,
-			linkPath,
-			statsSuffix,
-		});
-		const innerWidth = Math.max(1, width - 2);
-
-		let body = "";
-		if (isError) {
-			if (errorText) body = uiTheme.fg("error", replaceTabs(errorText));
-		} else if (details?.diff) {
-			body = renderDiffSection(
-				details.diff,
-				rawPath,
-				expanded,
-				innerWidth,
-				uiTheme,
-				renderDiffFn,
-				renderedDiffCache,
-				diffSectionCache,
-			);
-		} else if (details) {
-			// Authoritative result with no textual diff: a delete, a move-only
-			// rename, or a genuine no-op. The header already names the op
-			// (Delete / `src → dst`); only a true no-op needs an explanatory
-			// body so an empty card isn't mistaken for a stalled edit.
-			if (op !== "delete" && op !== "create" && !rename) {
-				const noChangePath = linkPath ? shortenPath(linkPath) : rawPath ? shortenPath(rawPath) : "";
-				body = uiTheme.fg("dim", `No changes were made${noChangePath ? ` to ${noChangePath}` : ""}.`);
+	return framedToolCard(
+		uiTheme,
+		({ width }) => {
+			const { expanded, renderContext } = options;
+			// A finalized result is authoritative: its `details` describe exactly
+			// what happened. The shared streaming `editDiffPreview` is a call-phase
+			// artifact (in a batch it reflects only the first file), so consulting it
+			// for an empty-diff delete/move/no-op result mislabels the card. Fall
+			// back to the preview only when no details exist yet.
+			const editDiffPreview = details ? undefined : renderContext?.editDiffPreview;
+			const renderDiffFn = renderContext?.renderDiff ?? renderFallbackDiff;
+
+			if (diffSectionRenderDiffFn !== renderDiffFn) {
+				diffSectionRenderDiffFn = renderDiffFn;
+				invalidateRenderedStringCache(diffSectionCache);
+				invalidateRenderedStringCache(renderedDiffCache);
 			}
-		} else if (editDiffPreview) {
-			if ("error" in editDiffPreview) body = uiTheme.fg("error", replaceTabs(editDiffPreview.error));
-			else if (editDiffPreview.diff)
+			const firstChangedLine =
+				(editDiffPreview && "firstChangedLine" in editDiffPreview ? editDiffPreview.firstChangedLine : undefined) ||
+				(details && !isError ? details.firstChangedLine : undefined);
+			const linkPath = details && "path" in details ? details.path : undefined;
+
+			// Change stats ride inline on the header bar next to the path.
+			const previewDiff = editDiffPreview && !("error" in editDiffPreview) ? editDiffPreview.diff : undefined;
+			const headerDiff = isError ? undefined : details?.diff || previewDiff;
+			const statsSuffix = headerDiff
+				? cachedRenderedString(statsSuffixCache, uiTheme, false, "", headerDiff, () =>
+						formatDiffStatsSuffix(headerDiff, uiTheme),
+					)
+				: "";
+			const header = renderEditHeader(width, uiTheme, {
+				icon: isError ? "error" : "success",
+				iconOverride: !isError && !options.isPartial ? uiTheme.styledSymbol("tool.edit", "accent") : undefined,
+				op,
+				rawPath,
+				rename,
+				firstChangedLine,
+				linkPath,
+				statsSuffix,
+			});
+			const innerWidth = Math.max(1, width - 2);
+
+			let body = "";
+			if (isError) {
+				if (errorText) body = uiTheme.fg("error", replaceTabs(errorText));
+			} else if (details?.diff) {
 				body = renderDiffSection(
-					editDiffPreview.diff,
+					details.diff,
 					rawPath,
 					expanded,
 					innerWidth,
@@ -1528,27 +1517,56 @@ function renderSingleFileResult(
 					renderedDiffCache,
 					diffSectionCache,
 				);
-		}
-		if (details?.diagnostics) {
-			body += formatDiagnostics(details.diagnostics, expanded, uiTheme, (fp: string) =>
-				uiTheme.getLangIcon(getLanguageFromPath(fp)),
-			);
-		}
+			} else if (details) {
+				// Authoritative result with no textual diff: a delete, a move-only
+				// rename, or a genuine no-op. The header already names the op
+				// (Delete / `src → dst`); only a true no-op needs an explanatory
+				// body so an empty card isn't mistaken for a stalled edit.
+				if (op !== "delete" && op !== "create" && !rename) {
+					const noChangePath = linkPath ? shortenPath(linkPath) : rawPath ? shortenPath(rawPath) : "";
+					body = uiTheme.fg("dim", `No changes were made${noChangePath ? ` to ${noChangePath}` : ""}.`);
+				}
+			} else if (editDiffPreview) {
+				if ("error" in editDiffPreview) body = uiTheme.fg("error", replaceTabs(editDiffPreview.error));
+				else if (editDiffPreview.diff)
+					body = renderDiffSection(
+						editDiffPreview.diff,
+						rawPath,
+						expanded,
+						innerWidth,
+						uiTheme,
+						renderDiffFn,
+						renderedDiffCache,
+						diffSectionCache,
+					);
+			}
+			if (details?.diagnostics) {
+				body += formatDiagnostics(details.diagnostics, expanded, uiTheme, (fp: string) =>
+					uiTheme.getLangIcon(getLanguageFromPath(fp)),
+				);
+			}
 
-		// Diff lines self-wrap with a continuation gutter; pre-wrap to the frame's
-		// inner width so renderOutputBlock's generic wrap is a no-op. Edit frames
-		// use a flush left border because code-frame gutters already provide padding.
-		const bodyLines = body.length > 0 ? body.split("\n").flatMap(line => wrapEditRendererLine(line, innerWidth)) : [];
-		while (bodyLines.length > 0 && bodyLines[0].trim() === "") bodyLines.shift();
+			// Diff lines self-wrap with a continuation gutter; pre-wrap to the frame's
+			// inner width so renderOutputBlock's generic wrap is a no-op. Edit frames
+			// use a flush left border because code-frame gutters already provide padding.
+			const bodyLines =
+				body.length > 0 ? body.split("\n").flatMap(line => wrapEditRendererLine(line, innerWidth)) : [];
+			while (bodyLines.length > 0 && bodyLines[0].trim() === "") bodyLines.shift();
 
-		return {
-			header,
-			sections: bodyLines.length > 0 ? [{ content: bodyLines }] : [],
-			phase: isError ? "error" : options.isPartial ? "partial" : "success",
-			borderColor: isError ? "error" : "borderMuted",
-			contentPaddingLeft: 0,
-		};
-	});
+			return {
+				header,
+				sections: bodyLines.length > 0 ? [{ content: bodyLines }] : [],
+				phase: isError ? "error" : options.isPartial ? "partial" : "success",
+				borderColor: isError ? "error" : "borderMuted",
+				contentPaddingLeft: 0,
+			};
+		},
+		{
+			// Only the width-keyed diff section goes; the highlighted diff and stats suffix do not depend on width, and
+			// re-highlighting every committed card on each resize replay costs seconds.
+			onReleaseRenderCaches: () => releaseRenderedStringCache(diffSectionCache),
+		},
+	);
 }
 
 function renderMultiFileResult(
@@ -1603,5 +1621,9 @@ function renderMultiFileResult(
 			cached = undefined;
 			for (const c of fileComponents) c.invalidate?.();
 		},
+		releaseRenderCaches() {
+			cached = undefined;
+			for (const c of fileComponents) c.releaseRenderCaches?.();
+		},
 	};
 }

package/src/tools/eval-format/index.ts

@@ -1,3 +1,4 @@
+import { LRUCache } from "@linxiraos/pi-utils/lru";
 import type { EvalLanguage } from "../eval";
 import { formatJavaScriptForDisplay } from "./javascript";
 import { formatPythonForDisplay } from "./python";
@@ -5,13 +6,38 @@ import { formatPythonForDisplay } from "./python";
 export * from "./javascript";
 export * from "./python";
 
+function createDisplayFormatMemo(): LRUCache<string, string> {
+	return new LRUCache<string, string>({
+		max: 64,
+		maxSize: 4_000_000,
+		// LRUCache rejects sizes <= 0; empty sources (cells not yet streamed) still cost one.
+		sizeCalculation: (formatted, source) => formatted.length + source.length + 1,
+	});
+}
+
+/**
+ * Formatted display text keyed by source, one memo per formatter. Renderers
+ * reformat every cell on each streamed argument delta and result update; the
+ * memo keeps that linear in the changed cell instead of the whole call.
+ */
+const jsDisplayMemo = createDisplayFormatMemo();
+const pythonDisplayMemo = createDisplayFormatMemo();
+
 /** Formats an arbitrary eval-code prefix for display without changing the executed source. */
 export function formatEvalCodeForDisplay(source: string, language: EvalLanguage): string {
 	if (/^\s*%/.test(source)) return source;
 	switch (language) {
 		case "js":
-			return formatJavaScriptForDisplay(source);
+			return memoizedFormat(jsDisplayMemo, source, formatJavaScriptForDisplay);
 		case "python":
-			return formatPythonForDisplay(source);
+			return memoizedFormat(pythonDisplayMemo, source, formatPythonForDisplay);
 	}
 }
+
+function memoizedFormat(memo: LRUCache<string, string>, source: string, format: (source: string) => string): string {
+	const cached = memo.get(source);
+	if (cached !== undefined) return cached;
+	const formatted = format(source);
+	memo.set(source, formatted);
+	return formatted;
+}

package/src/tools/eval-format/javascript.ts

@@ -249,7 +249,9 @@ interface DisplayBraceNode {
 
 function classifySourceBraces(source: string): boolean[] {
 	const objects: boolean[] = [];
-	const statementWords: string[] = [];
+	// Whether the current statement has seen `class`/`function`/`switch`, whose
+	// next `{` opens a block. A flag instead of a word list keeps each brace O(1).
+	let statementDeclaresBlock = false;
 	let previousToken = "";
 	let previousWord = "";
 	let regexAllowed = true;
@@ -287,7 +289,7 @@ function classifySourceBraces(source: string): boolean[] {
 		if (isIdentifierStart(char)) {
 			const end = scanIdentifier(source, index);
 			const word = source.slice(index, end);
-			statementWords.push(word);
+			if (word === "class" || word === "function" || word === "switch") statementDeclaresBlock = true;
 			previousWord = word;
 			previousToken = word;
 			regexAllowed = REGEX_PREFIX_WORDS[word] === true;
@@ -307,7 +309,7 @@ function classifySourceBraces(source: string): boolean[] {
 				previousToken === ")" ||
 				previousToken === "}" ||
 				BLOCK_BRACE_WORDS[previousWord] === true ||
-				statementWords.some(word => word === "class" || word === "function" || word === "switch");
+				statementDeclaresBlock;
 			const isObject =
 				!isBlock &&
 				(previousToken === "=" ||
@@ -319,7 +321,7 @@ function classifySourceBraces(source: string): boolean[] {
 					previousToken === "" ||
 					OBJECT_PREFIX_WORDS[previousWord] === true);
 			objects.push(isObject);
-			if (isBlock) statementWords.length = 0;
+			if (isBlock) statementDeclaresBlock = false;
 			previousToken = "{";
 			previousWord = "";
 			regexAllowed = true;
@@ -334,7 +336,7 @@ function classifySourceBraces(source: string): boolean[] {
 			continue;
 		}
 		if (char === ";") {
-			statementWords.length = 0;
+			statementDeclaresBlock = false;
 			previousToken = ";";
 			previousWord = "";
 			regexAllowed = true;

package/src/tools/eval.d.ts

@@ -24,6 +24,8 @@ export interface EvalCellResult {
     durationMs?: number;
     exitCode?: number;
     statusEvents?: EvalStatusEvent[];
+    /** Discrete events dropped from the front of `statusEvents` (see {@link recordStatusEvent}). */
+    statusEventsElided?: number;
     hasMarkdown?: boolean;
 }
 /** Tool result detail object surfaced to the UI/transcript. */
@@ -32,6 +34,8 @@ export interface EvalToolDetails {
     jsonOutputs?: unknown[];
     images?: ImageContent[];
     statusEvents?: EvalStatusEvent[];
+    /** Discrete events dropped from the front of `statusEvents` (see {@link recordStatusEvent}). */
+    statusEventsElided?: number;
     isError?: boolean;
     meta?: OutputMeta;
     /** First backend that produced cells. Kept for transcript compatibility. */
@@ -74,13 +78,28 @@ export declare function statusEventKey(event: {
     op: string;
     [key: string]: unknown;
 }): string | undefined;
+/**
+ * Discrete status events a log keeps. A cell polling a helper in a loop emits
+ * one event per call (hundreds of thousands), and every live update and the
+ * persisted tool result would otherwise carry them all.
+ */
+export declare const MAX_STATUS_EVENTS = 200;
+/** A status event list plus the count of discrete events dropped from its front. */
+export type StatusEventLog = Pick<EvalCellResult, "statusEvents" | "statusEventsElided">;
 /**
  * Append or replace a status event. Progress snapshots (see
  * {@link statusEventKey}) coalesce in place, preserving first-seen order; every
- * other op is a discrete action and simply appends. Keeps the persisted event
- * list bounded even when a subagent or batch emits hundreds of progress ticks.
+ * other op is a discrete action and simply appends.
  */
 export declare function upsertStatusEvent(events: EvalStatusEvent[], event: EvalStatusEvent): void;
+/**
+ * {@link upsertStatusEvent} into a bounded log. Past {@link MAX_STATUS_EVENTS},
+ * the oldest discrete event is dropped and counted. Snapshots stay, since agent
+ * cards read them, and so does the newest committed `todo` result, which the
+ * session todo panel refreshes from. Renderers already show only the newest
+ * events behind an "… N earlier" row.
+ */
+export declare function recordStatusEvent(log: StatusEventLog, event: EvalStatusEvent): void;
 /** Render eval code cells, structured display output, and progress events. */
 export declare const evalToolRenderer: {
     animatedPendingPreview: true;
@@ -95,7 +114,7 @@ export declare const evalToolRenderer: {
     }, options: RenderResultOptions & {
         renderContext?: EvalRenderContext;
     }, uiTheme: Theme, _args?: EvalRenderArgs): Component;
-    describeCall(args: EvalRenderArgs, _options: RenderResultOptions): NativeToolView;
+    describeCall(args: EvalRenderArgs, options: RenderResultOptions): NativeToolView;
     describeResult(result: {
         content: Array<{
             type: string;