@quandev104/pi-style 0.1.3 → 0.1.5

package/extension-src/pi-style/features/tools/boxed/grep.ts

@@ -13,7 +13,7 @@
 
 import type { Component } from "@earendil-works/pi-tui";
 import { stripAnsi } from "../../../shared/ansi.js";
-import { type BoxTheme, getTextOutput, shortenPath } from "../../../shared/box.js";
+import { type BoxTheme, dimLine, getTextOutput, shortenPath } from "../../../shared/box.js";
 import { safeTruncateToWidth } from "../../../shared/render-budget.js";
 import {
 	type GrepMatch,
@@ -111,7 +111,7 @@ function renderErrorLines(theme: BoxTheme, errorText: string, width: number): st
 		.map((line) => line.trim())
 		.filter((line) => line.length > 0);
 	if (raw.length === 0) return [];
-	const prefix = `${TREE_INDENT}${theme.fg("borderMuted", "└─")} `;
+	const prefix = `${TREE_INDENT}${dimLine("└─")} `;
 	const out = raw
 		.slice(0, GREP_ERROR_LINES)
 		.map((line) => safeTruncateToWidth(`${prefix}${theme.fg("error", line)}`, Math.max(1, width), "…"));

package/extension-src/pi-style/features/tools/boxed/output-tree.ts

@@ -20,6 +20,7 @@
 //   as one visual family.
 
 import type { BoxTheme } from "../../../shared/box.js";
+import { dimLine } from "../../../shared/box.js";
 import { safeTruncateToWidth } from "../../../shared/render-budget.js";
 
 /** Indent for top-level tree rows; matches the quiet-tool batch panel. */
@@ -251,11 +252,11 @@ export function renderOutputTree(
 	const lastIndex = visible.length - 1;
 	for (let i = 0; i < visible.length; i++) {
 		const branch = i < lastIndex || more > 0 ? "├─" : "└─";
-		const line = `${indent}${theme.fg("borderMuted", branch)} ${theme.fg(entryColor, label(visible[i] ?? ""))}`;
+		const line = `${indent}${dimLine(branch)} ${theme.fg(entryColor, label(visible[i] ?? ""))}`;
 		out.push(safeTruncateToWidth(line, safeWidth, "…"));
 	}
 	if (more > 0) {
-		const line = `${indent}${theme.fg("borderMuted", "└─")} ${theme.fg("dim", `… ${more} more ${pluralForm(moreUnit, more)}`)}`;
+		const line = `${indent}${dimLine("└─")} ${theme.fg("dim", `… ${more} more ${pluralForm(moreUnit, more)}`)}`;
 		out.push(safeTruncateToWidth(line, safeWidth, "…"));
 	}
 	return out;
@@ -274,7 +275,7 @@ function formatMatchRow(theme: BoxTheme, match: GrepMatch): string {
 	// Match rows render in the output text color (not primary) so they read like
 	// the matched code; only the file nodes carry the primary color.
 	const label = theme.fg("toolOutput", `*${match.line}`);
-	const sep = theme.fg("borderMuted", "│");
+	const sep = dimLine("│");
 	return `${label}${sep} ${theme.fg("toolOutput", match.content)}`;
 }
 
@@ -313,7 +314,7 @@ export function renderGrepTree(
 	if (singleFile) {
 		budget.forEach((match, index) => {
 			const isLast = index === totalVisible - 1 && !truncated;
-			push(`${indent}${theme.fg("borderMuted", isLast ? "└─" : "├─")} ${formatMatchRow(theme, match)}`);
+			push(`${indent}${dimLine(isLast ? "└─" : "├─")} ${formatMatchRow(theme, match)}`);
 		});
 	} else {
 		// Walk the budget, tracking position within each file group so the file
@@ -323,7 +324,7 @@ export function renderGrepTree(
 			const group = groups[gi];
 			if (!group) continue;
 			const isLastGroup = gi === groups.length - 1;
-			const trunk = isLastGroup ? " " : theme.fg("borderMuted", "│");
+			const trunk = isLastGroup ? " " : dimLine("│");
 
 			const visibleHere: GrepMatch[] = [];
 			for (const match of group.matches) {
@@ -336,22 +337,20 @@ export function renderGrepTree(
 			const groupIsLastRendered = shown >= totalVisible && !truncated;
 			const fileLabel = options.withIcons ? `${fileIcon(group.file)} ${group.file}` : group.file;
 			// File nodes use the primary (accent) color, matching read/ls/find paths.
-			push(`${indent}${theme.fg("borderMuted", groupIsLastRendered ? "└─" : "├─")} ${theme.fg("accent", fileLabel)}`);
+			push(`${indent}${dimLine(groupIsLastRendered ? "└─" : "├─")} ${theme.fg("accent", fileLabel)}`);
 
 			visibleHere.forEach((match, index) => {
 				const isLastInGroup = index === visibleHere.length - 1;
 				const isLastOverall = groupIsLastRendered && isLastInGroup;
 				push(
-					`${indent}${trunk}${TREE_CHILD_INDENT}${theme.fg("borderMuted", isLastOverall ? "└─" : "├─")} ${formatMatchRow(theme, match)}`,
+					`${indent}${trunk}${TREE_CHILD_INDENT}${dimLine(isLastOverall ? "└─" : "├─")} ${formatMatchRow(theme, match)}`,
 				);
 			});
 		}
 	}
 
 	if (truncated) {
-		push(
-			`${indent}${theme.fg("borderMuted", "└─")} ${theme.fg("dim", `… ${remaining} more ${pluralForm("match", remaining)}`)}`,
-		);
+		push(`${indent}${dimLine("└─")} ${theme.fg("dim", `… ${remaining} more ${pluralForm("match", remaining)}`)}`);
 	}
 	return out;
 }

package/extension-src/pi-style/features/tools/boxed/quick-edit.ts

@@ -4,19 +4,35 @@ import { getLanguageFromPath } from "@earendil-works/pi-coding-agent";
 import { stripAnsi } from "../../../shared/ansi.js";
 import {
 	type BoxTheme,
-	boxInnerWidth,
+	formatBoxedRunningStatus,
 	getTextOutput,
 	renderBoxedToolCall,
 	renderBoxedToolResult,
 } from "../../../shared/box.js";
 import { formatElapsedMs, getElapsedMs } from "../../../shared/elapsed.js";
 import { AdaptiveDiffComponent, buildSplitRows, countDiffStats } from "../../../shared/split-diff.js";
-import { getStateElapsedMs } from "./session-config.js";
-import { type BoxedToolContext, type BoxedToolDefinition, displayPath, noteExecutionStart } from "./shared.js";
+import { getStateElapsedMs, isResultSeen } from "./session-config.js";
+import {
+	type BoxedToolContext,
+	type BoxedToolDefinition,
+	displayPath,
+	noteBoxedCallState,
+	noteBoxedResultPhase,
+	noteExecutionStart,
+	stateElapsedMs,
+} from "./shared.js";
 
 const MAX_HIGHLIGHT_DIFF_CHARS = 12000;
 const MAX_HIGHLIGHT_DIFF_ROWS = 120;
 
+/** First-partial-pass result: the pending/running call card stands alone. */
+const EMPTY_QUICK_EDIT_RESULT = Object.freeze({
+	invalidate() {},
+	render() {
+		return [];
+	},
+});
+
 interface QuickEditToolConfig {
 	toolLabel: string;
 	applyingLabel: string;
@@ -131,10 +147,12 @@ function renderQuickEditResult(
 	config: QuickEditToolConfig,
 ) {
 	if (options.isPartial) {
+		const firstResultPass = noteBoxedResultPhase(context, options.isPartial);
+		if (firstResultPass) return EMPTY_QUICK_EDIT_RESULT;
 		return renderBoxedToolResult(
 			theme,
 			() => [`${theme.fg("dim", "↳")} ${theme.fg("muted", `Applying ${config.applyingLabel}...`)}`],
-			{ isPartial: true },
+			{ showDivider: false, footerLines: [formatBoxedRunningStatus(theme, stateElapsedMs(context))], isPartial: true },
 		);
 	}
 
@@ -196,12 +214,15 @@ export function quickEditTool(config: QuickEditToolConfig): BoxedToolDefinition
 	return {
 		call(args, theme, context) {
 			noteExecutionStart(context);
+			noteBoxedCallState(context);
 			const detail = displayPath(String(args?.path ?? ""), context);
 			return renderBoxedToolCall(theme, config.toolLabel, [], {
 				headerDetail: detail,
 				isError: Boolean(context.isError),
 				isPartial: Boolean(context.isPartial),
 				isPending: Boolean(context.isPartial),
+				running: Boolean(context.executionStarted),
+				resultSeen: isResultSeen(context.state),
 			});
 		},
 		result(result, options, theme, context) {

package/extension-src/pi-style/features/tools/boxed/read.ts

@@ -1,9 +1,9 @@
 // Boxed read tool renderer
 // (renderCall/renderResult only; no tool re-registration).
 //
-// Read calls render as a boxless tree panel — a lone read is a batch of one,
-// consecutive reads group into one panel (see batch.ts). There is no boxed
-// single-call special case.
+// Read calls render boxless: a lone read is a single inline line
+// (`➔ Read <path>`), consecutive reads group into one tree panel (see
+// batch.ts).
 
 import { stripAnsi } from "../../../shared/ansi.js";
 import { getTextOutput } from "../../../shared/box.js";

package/extension-src/pi-style/features/tools/boxed/session-config.ts

@@ -33,19 +33,79 @@ export function getToolsRenderConfig(): ToolsRenderConfig {
 
 // Wall-clock elapsed tracking through the renderer context state (no tool
 // re-registration, so result.details has no execution timing).
+//
+// Elapsed is computed live from the recorded start on every read, so a running
+// tool keeps growing its displayed time. The value only freezes once the
+// execution end is recorded (terminal result), which keeps the completed footer
+// stable across later re-renders (expand toggles, terminal resizes).
 
 const STARTED_AT_KEY = "__piStyleStartedAt";
-const ELAPSED_MS_KEY = "__piStyleElapsedMs";
+const ENDED_AT_KEY = "__piStyleEndedAt";
+const RESULT_SEEN_KEY = "__piStyleResultSeen";
+const TICKER_KEY = "__piStyleElapsedTicker";
 
 export function recordExecutionStarted(state: Record<string, unknown> | undefined, executionStarted: boolean): void {
 	if (!executionStarted || !state || typeof state !== "object") return;
 	if (typeof state[STARTED_AT_KEY] !== "number") state[STARTED_AT_KEY] = performance.now();
 }
 
+/** Freeze the elapsed at the first terminal render (idempotent). */
+export function recordExecutionEnded(state: Record<string, unknown> | undefined): void {
+	if (!state || typeof state !== "object") return;
+	if (typeof state[ENDED_AT_KEY] !== "number") state[ENDED_AT_KEY] = performance.now();
+}
+
 export function getStateElapsedMs(state: Record<string, unknown> | undefined): number | undefined {
 	if (!state || typeof state !== "object") return undefined;
-	if (typeof state[ELAPSED_MS_KEY] !== "number" && typeof state[STARTED_AT_KEY] === "number") {
-		state[ELAPSED_MS_KEY] = performance.now() - state[STARTED_AT_KEY];
+	const started = state[STARTED_AT_KEY];
+	if (typeof started !== "number") return undefined;
+	const ended = state[ENDED_AT_KEY];
+	if (typeof ended === "number") return Math.max(0, ended - started);
+	return Math.max(0, performance.now() - started);
+}
+
+/** Whether a result renderer already produced a continuation for this call. */
+export function isResultSeen(state: Record<string, unknown> | undefined): boolean {
+	return Boolean(state && typeof state === "object" && state[RESULT_SEEN_KEY] === true);
+}
+
+/** Record that a result renderer ran for this call (streaming or final). */
+export function markResultSeen(state: Record<string, unknown> | undefined): void {
+	if (!state || typeof state !== "object") return;
+	state[RESULT_SEEN_KEY] = true;
+}
+
+type TickerHandle = ReturnType<typeof setInterval>;
+
+/** States that currently own a 1s elapsed-render interval, for session cleanup. */
+const tickerStates = new Set<Record<string, unknown>>();
+
+/**
+ * While a tool is running, re-render once per second so live elapsed labels
+ * tick without any output events. Idempotent per state.
+ */
+export function startElapsedTicker(state: Record<string, unknown> | undefined, invalidate: () => void): void {
+	if (!state || typeof state !== "object") return;
+	if (state[TICKER_KEY] !== undefined) return;
+	state[TICKER_KEY] = setInterval(() => invalidate(), 1000) as unknown as TickerHandle;
+	tickerStates.add(state);
+}
+
+/** Stop a running tool's elapsed ticker (terminal result, error, session end). */
+export function stopElapsedTicker(state: Record<string, unknown> | undefined): void {
+	if (!state || typeof state !== "object") return;
+	const handle = state[TICKER_KEY] as TickerHandle | undefined;
+	if (handle !== undefined) clearInterval(handle);
+	delete state[TICKER_KEY];
+	tickerStates.delete(state);
+}
+
+/** Stop every elapsed ticker (session start/shutdown). */
+export function stopAllElapsedTickers(): void {
+	for (const state of tickerStates) {
+		const handle = state[TICKER_KEY] as TickerHandle | undefined;
+		if (handle !== undefined) clearInterval(handle);
+		delete state[TICKER_KEY];
 	}
-	return typeof state[ELAPSED_MS_KEY] === "number" ? state[ELAPSED_MS_KEY] : undefined;
+	tickerStates.clear();
 }

package/extension-src/pi-style/features/tools/boxed/shared.ts

@@ -13,12 +13,21 @@ import {
 	resolveRelativePath,
 	shortenPath,
 } from "../../../shared/box.js";
-import { getStateElapsedMs, recordExecutionStarted } from "./session-config.js";
+import {
+	getStateElapsedMs,
+	isResultSeen,
+	markResultSeen,
+	recordExecutionEnded,
+	recordExecutionStarted,
+	startElapsedTicker,
+	stopElapsedTicker,
+} from "./session-config.js";
 
 /** Renderer context delivered by Pi's ToolExecutionComponent (getRenderContext). */
 export interface BoxedToolContext {
 	readonly args: Record<string, unknown>;
 	readonly toolCallId: string;
+	readonly invalidate: () => void;
 	readonly state: Record<string, unknown>;
 	readonly cwd: string;
 	readonly executionStarted: boolean;
@@ -88,6 +97,7 @@ export function compactCall(
 		isError: Boolean(options.context.isError),
 		isPartial: Boolean(options.context.isPartial),
 		isPending: pendingFlag(options.context),
+		running: Boolean(options.context.executionStarted),
 	});
 }
 
@@ -96,6 +106,36 @@ export function noteExecutionStart(context: BoxedToolContext): void {
 	recordExecutionStarted(context.state, context.executionStarted);
 }
 
+/**
+ * Keep running/ended execution state in sync from a call renderer pass. While
+ * the tool runs, a 1s re-render ticker keeps live elapsed labels current; once
+ * the call renders in its terminal form the elapsed freezes.
+ */
+export function noteBoxedCallState(context: BoxedToolContext): void {
+	if (!context.executionStarted) return;
+	if (context.isPartial) startElapsedTicker(context.state, context.invalidate);
+	else {
+		recordExecutionEnded(context.state);
+		stopElapsedTicker(context.state);
+	}
+}
+
+/**
+ * Record a result renderer pass and keep the ticker/ended state in sync.
+ * Returns whether this is the first result pass for the call, so renderers can
+ * render nothing while the pending/running call card stands alone.
+ */
+export function noteBoxedResultPhase(context: BoxedToolContext, isPartial: boolean): boolean {
+	const firstResultPass = !isResultSeen(context.state);
+	markResultSeen(context.state);
+	if (isPartial) startElapsedTicker(context.state, context.invalidate);
+	else {
+		recordExecutionEnded(context.state);
+		stopElapsedTicker(context.state);
+	}
+	return firstResultPass;
+}
+
 export function stateElapsedMs(context: BoxedToolContext): number | undefined {
 	return getStateElapsedMs(context.state);
 }

package/extension-src/pi-style/features/tools/boxed/write.ts

@@ -15,6 +15,7 @@ import { stripAnsi } from "../../../shared/ansi.js";
 import {
 	type BoxTheme,
 	boxedToolWidthKey,
+	dimLine,
 	getTextOutput,
 	renderBoxedToolResult,
 	renderCompactBoxedToolCall,
@@ -34,6 +35,14 @@ import {
 /** Right-side bottom-border hint shown when the compact preview is truncated. */
 const WRITE_EXPAND_HINT = "Ctrl+O for more";
 
+/** Partial-pass result: the compact call keeps its `◌ Running` card. */
+const EMPTY_WRITE_RESULT: Component = Object.freeze({
+	invalidate() {},
+	render() {
+		return [];
+	},
+});
+
 type NumberedLine = { number: string; content: string };
 
 /**
@@ -54,7 +63,7 @@ function numberedPreviewLines(content: string): NumberedLine[] {
 
 /** One boxed preview row: dim gutter + toolOutput content. */
 function formatNumberedLine(theme: BoxTheme, line: NumberedLine): string {
-	return `${theme.fg("borderMuted", `${line.number} `)}${theme.fg("toolOutput", line.content)}`;
+	return `${dimLine(`${line.number} `)}${theme.fg("toolOutput", line.content)}`;
 }
 
 /** Compact write box: path header, numbered content preview, metrics footer. */
@@ -66,6 +75,7 @@ function renderWritePreviewBox(
 		state?: Record<string, unknown>;
 		isError: boolean;
 		isPending: boolean;
+		running?: boolean;
 		expanded: boolean;
 	},
 ): Component {
@@ -78,6 +88,7 @@ function renderWritePreviewBox(
 		...(options.state ? { state: options.state } : {}),
 		isError: options.isError,
 		isPending: options.isPending,
+		running: Boolean(options.running),
 		bodyLines: () => {
 			if (preview.length === 0) return [];
 			const shown = preview.slice(0, budget).map((line) => formatNumberedLine(theme, line));
@@ -107,10 +118,11 @@ export const writeTool: BoxedToolDefinition = {
 			state: context.state,
 			isError: Boolean(context.isError),
 			isPending: Boolean(context.isPartial),
+			running: Boolean(context.executionStarted),
 			expanded: Boolean(context.expanded),
 		});
 	},
-	result(result, _options, theme, context) {
+	result(result, options, theme, context) {
 		clearFooterState(context);
 		const output = getTextOutput(result);
 		const detail = displayPath(String(context?.args?.path ?? context?.args?.file_path ?? ""), context);
@@ -124,6 +136,11 @@ export const writeTool: BoxedToolDefinition = {
 			});
 		}
 
+		// While the result is still streaming, don't stamp a metrics footer into
+		// the shared state: the compact call keeps its `◌ Running` card and only
+		// closes with `elapsed · words` once the tool settles.
+		if (options.isPartial) return EMPTY_WRITE_RESULT;
+
 		// Success (compact and expanded): the preview box closes with the metrics
 		// footer stored into the shared renderer state; the result adds nothing.
 		return compactFooterWithState(theme, result, context);

package/extension-src/pi-style/pi/compatibility-coordinator.ts

@@ -125,6 +125,21 @@ export function createCompatibilityCoordinator(dispose = disposePiCompatibilityP
 					config,
 				});
 			const messagesEnabled = (assistantEnabled || specialBlocksEnabled) && config.messages.enabled;
+			// The hidden-thinking collapse is an assistant-message surface patch: it needs
+			// the assistant flag and `messages.hideThinkingLabel`, independent of the
+			// assistant prefix feature.
+			const thinkingCollapseEnabled = Boolean(
+				authorization.assistant &&
+					config.messages.enabled &&
+					config.messages.hideThinkingLabel &&
+					isTierCAuthorized({
+						certifiedHost,
+						coreFlag: authorization.core,
+						surfaceFlag: true,
+						surface: "messages",
+						config,
+					}),
+			);
 			const toolsEnabled =
 				authorization.tools &&
 				isTierCAuthorized({ certifiedHost, coreFlag: authorization.core, surfaceFlag: true, surface: "tools", config });
@@ -137,6 +152,7 @@ export function createCompatibilityCoordinator(dispose = disposePiCompatibilityP
 						...config.messages,
 						enabled: messagesEnabled,
 						assistantPrefix: assistantEnabled,
+						hideThinkingLabel: thinkingCollapseEnabled,
 						specialBlocks: messagesEnabled && config.messages.specialBlocks && specialBlocksEnabled,
 					},
 					tools: {
@@ -147,6 +163,7 @@ export function createCompatibilityCoordinator(dispose = disposePiCompatibilityP
 				messageSnapshot: {
 					assistantPrefix: authorization.ascii ? "[assistant] " : "│ ",
 					assistantEnabled,
+					collapseHiddenThinking: thinkingCollapseEnabled,
 				},
 				toolSnapshot: {
 					callMarker: authorization.ascii ? "[tool] " : "[tool] ",

package/extension-src/pi-style/pi/compatibility-probe.ts

@@ -3,14 +3,20 @@ import { dirname, join } from "node:path";
 import { fileURLToPath } from "node:url";
 import {
 	AssistantMessageComponent,
+	BashExecutionComponent,
 	BranchSummaryMessageComponent,
 	CompactionSummaryMessageComponent,
 	CustomMessageComponent,
 	SkillInvocationMessageComponent,
 	ToolExecutionComponent,
 } from "@earendil-works/pi-coding-agent";
-import { decorateMessageRender, type MessageDecorationSnapshot } from "../features/messages/index.js";
+import {
+	decorateMessageRender,
+	decorateMessageUpdate,
+	type MessageDecorationSnapshot,
+} from "../features/messages/index.js";
 import { renderSpecialMessageBlock, type SpecialBlockSubtype } from "../features/messages/special-blocks.js";
+import { renderBashExecutionBox } from "../features/tools/bash-execution.js";
 import { createToolDecorationOwner } from "../features/tools/index.js";
 import {
 	type CompatibilityRecord,
@@ -31,12 +37,14 @@ const reportStates = new WeakMap<
 // fail closed on every unrecorded Pi build and never use module-load capture as trust.
 export const TRUSTED_NATIVE_FINGERPRINTS: Readonly<Record<string, string>> = Object.freeze({
 	"native-assistant-message:render": "2a39243f",
+	"native-assistant-message:updateContent": "4a2f15ff",
 	"native-compaction-message:updateDisplay": "f8c44e78",
 	"native-branch-message:updateDisplay": "415d57b7",
 	"native-skill-message:updateDisplay": "48099ea6",
 	"native-custom-message:rebuild": "76ae2e3a",
 	"tool-call-renderer:getCallRenderer": "951ea0e0",
 	"tool-result-renderer:getResultRenderer": "8a25cd71",
+	"native-bash-execution:render": "a5b5abca",
 });
 
 export const CERTIFICATION_TABLE = Object.freeze({
@@ -80,6 +88,19 @@ export const CERTIFICATION_TABLE = Object.freeze({
 			adapterId: "tool-renderer-component-v1",
 			status: "certified" as const,
 		}),
+		"native-assistant-message:updateContent": Object.freeze({
+			feature: "messages",
+			subtype: "native-assistant-message",
+			target: AssistantMessageComponent.prototype,
+			method: "updateContent",
+			writable: true,
+			configurable: true,
+			name: "updateContent",
+			arity: 1,
+			fingerprint: TRUSTED_NATIVE_FINGERPRINTS["native-assistant-message:updateContent"],
+			adapterId: "message-thinking-collapse-v1",
+			status: "certified" as const,
+		}),
 		"native-compaction-message:updateDisplay": Object.freeze({
 			feature: "messages",
 			subtype: "native-compaction-message",
@@ -120,6 +141,23 @@ export const CERTIFICATION_TABLE = Object.freeze({
 			adapterId: "message-block-boxed-v1",
 			status: "certified" as const,
 		}),
+		"native-bash-execution:render": Object.freeze({
+			feature: "tools",
+			subtype: "native-bash-execution",
+			target: BashExecutionComponent.prototype,
+			method: "render",
+			writable: true,
+			configurable: true,
+			// The additive render patch is certified by the class constructor identity
+			// (name/arity/source fingerprint): the class defines no own `render`, so
+			// the installed own method is the only one and the inherited Container
+			// render is the native fallback.
+			name: "BashExecutionComponent",
+			arity: 2,
+			fingerprint: TRUSTED_NATIVE_FINGERPRINTS["native-bash-execution:render"],
+			adapterId: "bash-execution-box-v1",
+			status: "certified" as const,
+		}),
 	}),
 });
 
@@ -187,6 +225,12 @@ export interface TargetSpec {
 	adapterId: string | undefined;
 	status: "certified" | "native-fallback";
 	fallbackReason?: string;
+	/** "add-method" installs a new own method; the class constructor fingerprint certifies the target. */
+	kind?: "method" | "add-method";
+	/** Function name to verify (defaults to `method`; additive patches verify the class constructor name). */
+	identityName?: string;
+	/** Expected arity (defaults to the standard per-method rule; additive patches verify the constructor arity). */
+	arity?: number;
 }
 
 export function fingerprint(value: unknown): string | undefined {
@@ -201,15 +245,33 @@ export function fingerprint(value: unknown): string | undefined {
 
 function trustedNativeIdentity(spec: TargetSpec, piVersion: string | undefined): unknown {
 	if (piVersion !== TRUSTED_PI_VERSION) return undefined;
+	if (spec.kind === "add-method") {
+		// Additive install: the prototype must not already own the method (it is
+		// inherited), the class constructor identity must match the recorded build,
+		// and the inherited method becomes the native fallback for the delegate.
+		if (Object.getOwnPropertyDescriptor(spec.target, spec.method)) return undefined;
+		const ctor = Object.getOwnPropertyDescriptor(spec.target, "constructor")?.value;
+		const key = `${spec.subtype}:${spec.method}`;
+		if (
+			typeof ctor !== "function" ||
+			ctor.name !== (spec.identityName ?? spec.method) ||
+			ctor.length !== (spec.arity ?? 0) ||
+			fingerprint(ctor) !== TRUSTED_NATIVE_FINGERPRINTS[key]
+		)
+			return undefined;
+		const inherited = Object.getOwnPropertyDescriptor(Object.getPrototypeOf(spec.target), spec.method)?.value;
+		return typeof inherited === "function" ? inherited : undefined;
+	}
 	const descriptor = Object.getOwnPropertyDescriptor(spec.target, spec.method);
 	const value = descriptor?.value;
 	const key = `${spec.subtype}:${spec.method}`;
+	const expectedArity = spec.arity ?? (spec.method === "render" || spec.method === "updateContent" ? 1 : 0);
 	if (
 		descriptor?.writable !== true ||
 		descriptor.configurable !== true ||
 		typeof value !== "function" ||
-		value.name !== spec.method ||
-		value.length !== (spec.method === "render" ? 1 : 0) ||
+		value.name !== (spec.identityName ?? spec.method) ||
+		value.length !== expectedArity ||
 		fingerprint(value) !== TRUSTED_NATIVE_FINGERPRINTS[key]
 	)
 		return undefined;
@@ -225,6 +287,14 @@ export const targetSpecs: readonly TargetSpec[] = [
 		adapterId: "message-prefix-osc133-v1",
 		status: "certified",
 	},
+	{
+		feature: "messages",
+		subtype: "native-assistant-message",
+		target: AssistantMessageComponent.prototype,
+		method: "updateContent",
+		adapterId: "message-thinking-collapse-v1",
+		status: "certified",
+	},
 	{
 		feature: "messages",
 		subtype: "native-compaction-message",
@@ -273,6 +343,17 @@ export const targetSpecs: readonly TargetSpec[] = [
 		adapterId: "tool-renderer-component-v1",
 		status: "certified",
 	},
+	{
+		feature: "tools",
+		subtype: "native-bash-execution",
+		target: BashExecutionComponent.prototype,
+		method: "render",
+		kind: "add-method",
+		identityName: "BashExecutionComponent",
+		arity: 2,
+		adapterId: "bash-execution-box-v1",
+		status: "certified",
+	},
 ];
 
 export interface PiVersionResolution {
@@ -373,15 +454,18 @@ function versionInRange(version: string | undefined): boolean {
 	return version === TRUSTED_PI_VERSION;
 }
 
-function shape(target: object, method: string): boolean {
-	const descriptor = Object.getOwnPropertyDescriptor(target, method);
+function shape(spec: TargetSpec): boolean {
+	const descriptor = Object.getOwnPropertyDescriptor(spec.target, spec.method);
+	// Additive installs need an unowned slot (the method is inherited); every
+	// other patch requires the native own writable/configurable method.
+	if (spec.kind === "add-method") return descriptor === undefined;
 	return typeof descriptor?.value === "function" && descriptor.writable === true && descriptor.configurable === true;
 }
 
 export interface CompatibilityProbeOptions {
 	markers?: Set<string>;
 	config?: Readonly<{
-		messages: { enabled: boolean; assistantPrefix: boolean; specialBlocks: boolean };
+		messages: { enabled: boolean; assistantPrefix: boolean; specialBlocks: boolean; hideThinkingLabel: boolean };
 		tools: { enabled: boolean; style: string; maxCollapsedLines: number; maxExpandedLines: number; dimOutput: boolean };
 		preset: string;
 	}>;
@@ -418,7 +502,7 @@ function createFallbackRecord(
 
 function probeDiagnostic(spec: TargetSpec, piVersion: string | undefined, identity: unknown): string {
 	if (!versionInRange(piVersion)) return "Pi version is unknown or outside the recorded 0.83.0 support build";
-	if (!shape(spec.target, spec.method)) return "target method shape is not an own writable/configurable function";
+	if (!shape(spec)) return "target method shape is not an own writable/configurable function";
 	if (identity === undefined) return "recorded 0.83.0 native fingerprint, name, or arity did not match";
 	return "exact native identity verified; certified guarded decoration enabled";
 }
@@ -427,7 +511,9 @@ function surfaceDisabled(spec: TargetSpec, config: CompatibilityProbeOptions["co
 	if (!config) return false;
 	if (spec.feature === "tools") return !config.tools.enabled;
 	if (!config.messages.enabled) return true;
-	if (spec.subtype === "native-assistant-message") return !config.messages.assistantPrefix;
+	if (spec.subtype === "native-assistant-message" && spec.method === "render") return !config.messages.assistantPrefix;
+	if (spec.subtype === "native-assistant-message" && spec.method === "updateContent")
+		return !config.messages.hideThinkingLabel;
 	if (isSpecialBlock(spec)) return !config.messages.specialBlocks;
 	return true;
 }
@@ -456,13 +542,19 @@ function probeSpec(options: {
 		method: spec.method,
 		piVersion: piVersion ?? "unknown",
 		versionRange: PI_VERSION_RANGE,
-		shape: identity !== undefined && versionInRange(piVersion) && shape(spec.target, spec.method),
+		shape: identity !== undefined && versionInRange(piVersion) && shape(spec),
 		generation,
 		expectedIdentity: identity,
 		hasExpectedIdentity: true,
 		diagnostic,
+		...(spec.kind ? { kind: spec.kind } : {}),
 		delegate: (original, target, args) => {
 			markers.add(`${spec.subtype}:delegated`);
+			if (spec.subtype === "native-bash-execution")
+				return (
+					renderBashExecutionBox(target, args) ??
+					Reflect.apply(original as (...values: unknown[]) => unknown, target, args)
+				);
 			if (spec.feature === "tools")
 				return (
 					toolOwner?.decorateToolRendererSelection(
@@ -472,8 +564,10 @@ function probeSpec(options: {
 						args,
 					) ?? Reflect.apply(original as (...values: unknown[]) => unknown, target, args)
 				);
-			if (spec.subtype === "native-assistant-message")
+			if (spec.subtype === "native-assistant-message") {
+				if (spec.method === "updateContent") return decorateMessageUpdate(original, target, args, messageSnapshot);
 				return decorateMessageRender(original, target, args, messageSnapshot);
+			}
 			return renderSpecialMessageBlock(spec.subtype as SpecialBlockSubtype, original, target, args);
 		},
 	});