pi-firecode 0.8.1 → 1.0.0

package/tools/grouping.ts

@@ -1,233 +1,158 @@
-/**
- * 连续工具行的视觉合并与第三方工具通用单行兜底：
- * 1. 原生内置或声明了 self 的工具：保留自渲染，去除相邻空行形成轨道；
- * 2. 未知第三方或默认宽框工具：自动收口为通用 ToolLine 单行紧凑轨道与全宽背景。
- */
-import type { ExtensionUIContext, Theme } from "@earendil-works/pi-coding-agent";
-import { ToolLine } from "./line.js";
-import { genericArgsParts } from "./parts.js";
-
-const GROUP_PATCH = Symbol.for("pi.firecode.group-patch");
-const REQUEST_RENDER_PATCH = Symbol.for("pi.firecode.request-render-patch");
-const GLOBAL_STATE = Symbol.for("pi.firecode.tools-state");
-const THEME_KEY = Symbol.for("@earendil-works/pi-coding-agent:theme");
-
-function getTheme(): Theme {
-	const globalTheme = (globalThis as any)[THEME_KEY];
-	if (globalTheme && typeof globalTheme.fg === "function") return globalTheme;
-	return {
-		fg: (_color: string, text: string) => text,
-		bg: (_color: string, text: string) => text,
-		bold: (text: string) => text,
-	} as unknown as Theme;
-}
-
-type RenderFunction = (width: number) => string[];
-type RequestRenderFunction = (force?: boolean) => unknown;
-type ToolRow = { toolName: unknown; setExpanded: unknown; render: RenderFunction };
-type ToolRowPatch = { original: RenderFunction; patched: RenderFunction };
-type ToolRowPrototype = ToolRow & { [GROUP_PATCH]?: ToolRowPatch };
-type RequestRenderPatch = { original: RequestRenderFunction; patched: RequestRenderFunction };
-type ContainerLike = { children: unknown[] };
-type TuiLike = {
-	requestRender: RequestRenderFunction;
-	children?: unknown[];
-	[REQUEST_RENDER_PATCH]?: RequestRenderPatch;
-};
-type GlobalState = {
-	tui?: TuiLike;
-	chat?: ContainerLike;
-	indexedChildren?: unknown[];
-	indexedLength?: number;
-	joinedRows?: WeakSet<object>;
-	patchedPrototype?: ToolRowPrototype;
-	patchErrorReported?: boolean;
-	decorated: Set<string>;
-};
-
-// 扩展热重载会重新执行模块，patch 状态必须挂在全局才能正确卸载。
-const runtimeGlobal = globalThis as typeof globalThis & { [GLOBAL_STATE]?: GlobalState };
-const globalState = (runtimeGlobal[GLOBAL_STATE] ??= { decorated: new Set<string>() });
-
-function isToolRow(value: unknown): value is ToolRow {
-	if (!value || typeof value !== "object") return false;
-	const row = value as ToolRow;
-	return typeof row.render === "function" && typeof row.setExpanded === "function" && "toolName" in row;
-}
-
-function isDecoratedToolRow(value: unknown): value is ToolRow {
-	return isToolRow(value);
+/** 过程分组的安装：原始聊天树不变，渲染与鼠标命中共用同一份投影；宿主私有细节全部经 host.ts。 */
+import { AssistantMessageComponent, CustomMessageComponent, ToolExecutionComponent, UserMessageComponent, type ExtensionUIContext } from "@earendil-works/pi-coding-agent";
+import { Container, type Component, type TUI } from "@earendil-works/pi-tui";
+import { onFrame } from "../flame.js";
+import { ClickAnchor } from "./click-anchor.js";
+import { isMachineMessage, projectProcessGroups, toggleToolDetails, type ProjectionEnv } from "./group-view.js";
+import { assistantFacts, captureTui, findChat, HostShapeError, isCardOpened, patchMethod, rowUiOf, scrollContentHeight, scrollViewOf, toolFacts } from "./host.js";
+import type { TurnClock } from "./turn-clock.js";
+
+const OWNER = Symbol.for("pi.firecode.tool-groups");
+const runtime = globalThis as typeof globalThis & { [OWNER]?: () => void };
+
+export interface GroupOptions {
+	clock: TurnClock;
 }
 
-export function renderFallbackToolRow(row: any, width: number, customTheme?: Theme): string[] {
-	const isError = row.result?.isError ?? false;
-	const isPartial = row.isPartial ?? false;
-	const toolName = typeof row.toolName === "string" ? row.toolName : "tool";
-	const activeTheme = customTheme ?? getTheme();
-	const toolLine = new ToolLine({
-		label: toolName,
-		value: genericArgsParts(row.args),
-		clip: "end",
-		theme: activeTheme,
-		ctx: {
-			state: row.rendererState ?? {},
-			cwd: row.cwd ?? process.cwd(),
-			toolCallId: row.toolCallId ?? "",
-			isPartial,
-			isError,
-			expanded: row.expanded ?? false,
-		},
+export function installGroupPatch(ui: ExtensionUIContext, options: GroupOptions): () => void {
+	runtime[OWNER]?.();
+	let detach = () => {};
+	captureTui(ui, (tui) => {
+		detach = attach(tui, ui, options);
 	});
-	const lines = ["", ...toolLine.render(width)];
-	if (row.expanded && row.result) {
-		const textContent = Array.isArray(row.result.content)
-			? row.result.content
-					.map((c: any) => (c?.type === "text" && typeof c.text === "string" ? c.text : ""))
-					.filter(Boolean)
-					.join("\n")
-			: "";
-		if (textContent) {
-			const outputLines = textContent.split("\n").map((l: string) => activeTheme.fg("toolOutput", l));
-			lines.push("", ...outputLines);
-		}
-	}
-	return lines;
-}
-
-function findChatContainer(value: unknown, seen = new Set<unknown>()): ContainerLike | undefined {
-	if (!value || typeof value !== "object" || seen.has(value)) return undefined;
-	seen.add(value);
-	const children = (value as ContainerLike).children;
-	if (!Array.isArray(children)) return undefined;
-	if (children.some(isToolRow)) return value as ContainerLike;
-	for (const child of children) {
-		const found = findChatContainer(child, seen);
-		if (found) return found;
-	}
-	return undefined;
-}
-
-/** 增量标记"紧跟另一条装饰行"的行，聊天记录只增不改，可以从上次位置续算。 */
-function indexJoinedRows(children: unknown[]): void {
-	const canAppend =
-		globalState.indexedChildren === children &&
-		globalState.indexedLength !== undefined &&
-		globalState.indexedLength <= children.length &&
-		globalState.joinedRows !== undefined;
-	const joinedRows =
-		canAppend && globalState.joinedRows ? globalState.joinedRows : new WeakSet<object>();
-	const start = canAppend ? Math.max(1, globalState.indexedLength ?? 1) : 1;
-	for (let index = start; index < children.length; index++) {
-		const current = children[index];
-		if (isDecoratedToolRow(current) && isDecoratedToolRow(children[index - 1]))
-			joinedRows.add(current);
-	}
-	globalState.indexedChildren = children;
-	globalState.indexedLength = children.length;
-	globalState.joinedRows = joinedRows;
+	const dispose = () => {
+		if (runtime[OWNER] !== dispose) return;
+		detach();
+		delete runtime[OWNER];
+	};
+	runtime[OWNER] = dispose;
+	return dispose;
 }
 
-function followsToolRow(row: ToolRow): boolean {
-	if (!globalState.chat) globalState.chat = findChatContainer(globalState.tui);
-	const children = globalState.chat?.children;
-	if (!children) return false;
-	if (globalState.indexedChildren !== children || globalState.indexedLength !== children.length)
-		indexJoinedRows(children);
-	return globalState.joinedRows?.has(row) ?? false;
+/** 首个真实实例的形状自检：私有字段改名或换型就不安装。 */
+function checkShape(child: Component): void {
+	if (child instanceof ToolExecutionComponent) toolFacts(child);
+	if (child instanceof AssistantMessageComponent) assistantFacts(child);
 }
 
-/** 走 pi 原生渲染（会消费工具自带 renderCall）的工具；其余走通用兜底行。
- * subagents 自带中文 ToolLine 渲染（master/index.ts），不列入会被兜底行遮掉。 */
-export const FIRECODE_TOOLS = new Set(["read", "bash", "edit", "write", "find", "grep", "ls", "subagents"]);
-
-function patchToolRows(): void {
-	if (globalState.patchedPrototype) return;
-	const chat = globalState.chat ?? findChatContainer(globalState.tui);
-	if (!chat) return;
-	globalState.chat = chat;
-	const row = chat.children.find(isToolRow);
-	if (!row) return;
-	const prototype = Object.getPrototypeOf(row) as ToolRowPrototype;
-	if (prototype[GROUP_PATCH]) {
-		globalState.patchedPrototype = prototype;
-		return;
-	}
-	const original = prototype.render;
-	const patched: RenderFunction = function (this: ToolRow, width: number): string[] {
-		const toolName = typeof this.toolName === "string" ? this.toolName : "";
-		const isFirecodeTool = FIRECODE_TOOLS.has(toolName);
-		const isExpanded = Boolean((this as any).expanded);
-		const lines = isFirecodeTool || isExpanded
-			? original.call(this, width)
-			: renderFallbackToolRow(this, width);
-		return followsToolRow(this) && lines[0] === "" ? lines.slice(1) : lines;
+function attach(tui: TUI, ui: ExtensionUIContext, options: GroupOptions): () => void {
+	const originalExpand = ToolExecutionComponent.prototype.setExpanded;
+	const originalMessageExpand = CustomMessageComponent.prototype.setExpanded;
+	const originalAdd = Container.prototype.addChild;
+	const restores: (() => void)[] = [];
+	let stopFrames: (() => void) | undefined;
+	let attached = false;
+	const detach = () => {
+		stopFrames?.();
+		stopFrames = undefined;
+		for (const restore of restores.splice(0).reverse()) restore();
 	};
-	Object.defineProperty(prototype, GROUP_PATCH, {
-		configurable: true,
-		value: { original, patched } satisfies ToolRowPatch,
-	});
-	prototype.render = patched;
-	globalState.patchedPrototype = prototype;
-}
-
-export function installGroupPatch(ui: ExtensionUIContext, decorated: Set<string>): void {
-	globalState.decorated = decorated;
-	globalState.patchErrorReported = false;
-	// 借一次 widget 生命周期拿到 tui 实例，拿到即注销。
-	ui.setWidget("firecode-tui-capture", (tui) => {
-		const patchableTui = tui as TuiLike;
-		globalState.tui = patchableTui;
-		globalState.chat = undefined;
-		globalState.indexedChildren = undefined;
-		globalState.joinedRows = undefined;
-		if (!patchableTui[REQUEST_RENDER_PATCH]) {
-			const original = patchableTui.requestRender;
-			const patched: RequestRenderFunction = (force) => {
-				try {
-					patchToolRows();
-				} catch (error) {
-					if (!globalState.patchErrorReported) {
-						globalState.patchErrorReported = true;
-						ui.notify(
-							`工具轨道连接失败：${error instanceof Error ? error.message : String(error)}`,
-							"warning",
-						);
-					}
-				}
-				return original.call(patchableTui, force);
-			};
-			Object.defineProperty(patchableTui, REQUEST_RENDER_PATCH, {
-				configurable: true,
-				value: { original, patched } satisfies RequestRenderPatch,
-			});
-			patchableTui.requestRender = patched;
+	let abandoned = false;
+	/** 自检、渲染、点击或全局展开时发现宿主形状不符：整体退回原生显示并明确提示一次，不悄悄画错。 */
+	const abandon = (error: HostShapeError) => {
+		detach();
+		if (abandoned) return;
+		abandoned = true;
+		ui.notify(error.message, "warning");
+	};
+	/** 宿主调进来的每个入口都经这里：形状不符就退回原生，再按原生行为完成这次调用。 */
+	const guarded = <T>(run: () => T, native: () => T): T => {
+		try {
+			return run();
+		} catch (error) {
+			if (!(error instanceof HostShapeError)) throw error;
+			abandon(error);
+			return native();
 		}
-		patchToolRows();
-		return { render: () => [], invalidate: () => {} };
+	};
+	// 归属只比对 TUI 引用，不校验别的 TUI 的工具行形状。
+	const belongsHere = (row: ToolExecutionComponent) => rowUiOf(row) === tui;
+	// 全局展开只控制组摘要/列表；单工具正文通过下方独立的鼠标入口调用原方法。
+	restores.push(patchMethod(ToolExecutionComponent.prototype, "setExpanded", function (this: ToolExecutionComponent, value) {
+		originalExpand.call(this, belongsHere(this) ? false : value);
+	}));
+	// 机器消息的原生卡片只由投影点开（host.ts 的 openCard）；全局展开不平铺它（其余 CustomMessage 照旧跟随全局）。
+	restores.push(patchMethod(CustomMessageComponent.prototype, "setExpanded", function (this: CustomMessageComponent, value) {
+		const target = guarded(() => (isMachineMessage(this) && !isCardOpened(this) ? false : value), () => value);
+		originalMessageExpand.call(this, target);
+	}));
+
+	const install = (chat: Container) => {
+		const render = chat.render;
+		const mouse = chat.handleMouse;
+		const projection = new Container();
+		const overrides = new Set<object>();
+		let lastExpanded = ui.getToolsExpanded();
+		const anchor = new ClickAnchor(() => {
+			const view = scrollViewOf(tui, chat);
+			return view && {
+				top: view.scrollTop,
+				following: view.isFollowingEnd,
+				viewport: view.viewportHeight,
+				contentHeight: scrollContentHeight(view),
+				holdAt: (top: number) => view.scrollTo(top, { disableFollow: true }),
+				follow: () => view.scrollToEnd(),
+			};
+		});
+		const env: ProjectionEnv = {
+			ui, clock: options.clock, headless: {},
+			toggleRow: (row) => {
+				toggleToolDetails(row, originalExpand);
+				tui.requestRender();
+			},
+			isOpen: (key) => overrides.has(key),
+			toggleOpen: (key) => {
+				if (!overrides.delete(key)) overrides.add(key);
+				tui.requestRender();
+			},
+		};
+		chat.render = (width) => {
+			// ctrl+o 永远是全部展开/全部折叠：全局档位一变，逐轮覆盖（含被点开的机器消息卡）一并复位。
+			if (ui.getToolsExpanded() !== lastExpanded) {
+				lastExpanded = ui.getToolsExpanded();
+				overrides.clear();
+			}
+			return guarded(() => {
+				const { nodes, animating } = projectProcessGroups(chat.children, env);
+				projection.children = nodes;
+				// 动效只经全局时钟：有活的摘要才订阅，静止即取消。
+				if (animating && !stopFrames) stopFrames = onFrame(() => tui.requestRender());
+				else if (!animating && stopFrames) { stopFrames(); stopFrames = undefined; }
+				const lines = projection.render(width);
+				anchor.layout(lines.length);
+				return lines;
+			}, () => render.call(chat, width));
+		};
+		chat.handleMouse = (event) => guarded(() => {
+			const result = projection.handleMouse(event);
+			if (result && event.type === "click") anchor.click(event.y);
+			return result;
+		}, () => mouse.call(chat, event));
+		restores.push(() => {
+			chat.render = render;
+			chat.handleMouse = mouse;
+		});
+	};
+	const discover = (trigger?: Component) => {
+		if (attached) return;
+		const chat = findChat(tui);
+		if (!chat) return;
+		attached = true;
+		removeHook();
+		const healthy = guarded(() => {
+			for (const child of trigger ? [trigger, ...chat.children] : chat.children) checkShape(child);
+			return true;
+		}, () => false);
+		if (healthy) install(chat);
+	};
+	// 宿主没有聊天容器句柄；首条用户消息、助手或工具插入时定位（首条用户消息一出现就带竖条），随后立即卸掉发现钩子。
+	const removeHook = patchMethod(Container.prototype, "addChild", function (this: Container, child: Component) {
+		originalAdd.call(this, child);
+		if (child instanceof AssistantMessageComponent || child instanceof UserMessageComponent
+			|| (child instanceof ToolExecutionComponent && rowUiOf(child) === tui)) discover(child);
 	});
-	ui.setWidget("firecode-tui-capture", undefined);
+	restores.push(removeHook);
+	discover();
+	return detach;
 }
 
-export function uninstallGroupPatch(): void {
-	const tui = globalState.tui;
-	const requestPatch = tui?.[REQUEST_RENDER_PATCH];
-	if (tui && requestPatch && tui.requestRender === requestPatch.patched) {
-		tui.requestRender = requestPatch.original;
-		delete tui[REQUEST_RENDER_PATCH];
-	}
-
-	const prototype = globalState.patchedPrototype;
-	const rowPatch = prototype?.[GROUP_PATCH];
-	if (prototype && rowPatch && prototype.render === rowPatch.patched) {
-		prototype.render = rowPatch.original;
-		delete prototype[GROUP_PATCH];
-	}
-
-	globalState.tui = undefined;
-	globalState.chat = undefined;
-	globalState.indexedChildren = undefined;
-	globalState.indexedLength = undefined;
-	globalState.joinedRows = undefined;
-	globalState.patchedPrototype = undefined;
-	globalState.patchErrorReported = false;
-}