@herbertgao/pi-extensions 2026.10.5 → 2026.10.6

package/node_modules/@herbertgao/pi-cc-extensions/extensions/renderer/compact-mode.ts

@@ -42,7 +42,7 @@ import { toolLoadingIcon } from "../utils/tool-loading-icon.ts";
 import { sanitizeToolResultText } from "../utils/tool-result-sanitize.ts";
 import { refreshTranscriptComponent } from "./transcript-refresh.ts";
 import { getMessageDisplayTheme } from "./tool/message-display.ts";
-import { showMoreHintText } from "./tool/show-more-hint.ts";
+import { showMoreHintText } from "./tool/names.ts";
 import {
 	countEditDiffStats,
 	countWriteDiffStats,
@@ -53,13 +53,11 @@ import { displayConfigCacheKey } from "./tool/diff/diff-component.ts";
 import type { WriteExecutionMetadataStore } from "./tool/diff/write-execution.ts";
 import { isToolCallHovered } from "./mouse/hover.ts";
 import { getToolMouseTui } from "./mouse/scroll.ts";
-import {
-	insetComponent,
-	isLiveToolCall,
-	renderExpandedToolResult,
-	scheduleAnimation,
-} from "./tool/result.ts";
+import { insetComponent } from "./tool/result.ts";
+import { isLiveToolCall, scheduleAnimation } from "./tool/tool-state.ts";
+import { renderExpandedToolResult } from "./tool/tool-io.ts";
 import { paddedBackgroundRow } from "./tool/grouping.ts";
+import { styleChangeCounts } from "./tool/ccstyle-tool.ts";
 import { formatDisplayPath } from "./tool/names.ts";
 import {
 	hasVisibleText,
@@ -594,6 +592,11 @@ export type CompactModeHooks = {
 	assertOwnership(): void;
 	/** 重新渲染包含指定 toolCallId 的 assistant 消息（思考收尾时刷新）。 */
 	refreshToolCallMessage(toolCallId: string | undefined): void;
+	/**
+	 * 收尾 resume 重放留下的活动回合。重放是同步的，回合之间靠下一条消息收口，
+	 * 只有最后一条没有后续消息，会停在 Running。
+	 */
+	settleReplayedRound(): void;
 	shutdown(): void;
 };
 
@@ -620,17 +623,6 @@ const trackedAssistantComponents = new Set<any>();
 const trackedToolComponents = new Set<any>();
 let hoveredAssistantComponent: any;
 
-/**
- * 活动回合摘要 getter：常驻状态行（pi working row）镜像用。
- * installCompactMode 注册、卸载时清空，直接读 activeRound，不持有回合闭包。
- */
-let compactRunStatusGetter: (() => string | undefined) | undefined;
-
-/** 活动回合的摘要文本（`Running... · 9s, bash×1`）；无活动回合返回 undefined。 */
-export function getCompactRunStatusText(): string | undefined {
-	return compactRunStatusGetter?.();
-}
-
 /**
  * 回合尾行记录：摘要行（live 时连槽位卡）挂成 transcript 容器里回合末尾的
  * 兄弟组件，而不是 anchor 内部。anchor 内部的可变内容一旦被工具/diff 顶出视口
@@ -663,31 +655,37 @@ export function markCompactRoundToolExpanded(tool: any): void {
 	roundToolExpansion?.markUserExpanded(tool);
 }
 
-/** 面板展开前的视口位置：收起时还原，避免面板撑高后收起导致视口错位。 */
-const panelViewportBeforeExpand = new WeakMap<
-	object,
-	{ scrollTop: number; followingEnd: boolean }
->();
+/**
+ * 展开面板前的分隔空行与面板绑成一体：挂在同一尾行上，才不会被拆到
+ * anchor 正文之后（面板挂回合尾行，空行却留在消息里）。
+ * 带 childAtRow 返回 null：拆分判据认它为面板成员，行命中落空时归外层面板。
+ */
+function panelBoundSpacer(): any {
+	return {
+		render: () => [""],
+		childAtRow: () => null,
+		invalidate() {},
+	};
+}
+
+/** 面板展开前的视口标记：收起时需要重钉视口的卡。 */
+const panelViewportBeforeExpand = new WeakSet<object>();
 
 function capturePanelViewport(card: any): void {
-	if (!card || panelViewportBeforeExpand.has(card)) return;
-	const view = getToolMouseTui()?.getPrimaryScrollView?.();
-	if (!view || typeof view.scrollTop !== "number" || typeof view.scrollTo !== "function") return;
-	panelViewportBeforeExpand.set(card, {
-		scrollTop: view.scrollTop,
-		followingEnd: view.isFollowingEnd !== false,
-	});
+	if (!card) return;
+	panelViewportBeforeExpand.add(card);
 }
 
 function restorePanelViewport(card: any): void {
-	const snapshot = panelViewportBeforeExpand.get(card);
-	if (!snapshot) return;
-	panelViewportBeforeExpand.delete(card);
+	if (!panelViewportBeforeExpand.delete(card)) return;
 	const view = getToolMouseTui()?.getPrimaryScrollView?.();
 	if (!view) return;
-	// 展开前就在底部：交回官方 follow；否则回到展开前的滚动偏移。
-	if (snapshot.followingEnd && typeof view.scrollToEnd === "function") view.scrollToEnd();
-	else if (typeof view.scrollTo === "function") view.scrollTo(snapshot.scrollTop);
+	// 不用展开前的快照：展开期间用户可能滚进面板里，回跳旧位置才是"跳"。
+	// 面板在摘要行之下，收起只删它以下的行，保持当前 scrollTop 摘要行原地不动；
+	// clamp 后落在新底时 scrollTo 会自动恢复 followingEnd（并清 followSuppressedAtEnd）。
+	if (typeof view.scrollTo === "function" && typeof view.scrollTop === "number") {
+		view.scrollTo(view.scrollTop);
+	}
 }
 
 function compactEditWriteLine(
@@ -724,8 +722,13 @@ function compactEditWriteLine(
 						)
 					: undefined;
 		if (stats) {
-			statsText = ` (+${stats.added} -${stats.removed})`;
-			statsStyled = ` ${theme.fg("dim", "(")}${theme.fg("success", `+${stats.added}`)} ${theme.fg("error", `-${stats.removed}`)}${theme.fg("dim", ")")}`;
+			const counts = styleChangeCounts(
+				(color, text) => theme.fg(color, text),
+				stats.added,
+				stats.removed,
+			);
+			statsText = counts.text;
+			statsStyled = counts.styled;
 		}
 	}
 	// 展开卡 Box(1,1) 已 pad；折叠行自己留 1 格前导空格
@@ -1187,6 +1190,8 @@ export function installCompactMode(deps: CompactModeInstallDeps): CompactModeHoo
 		liveSlotToolIds: Set<string>;
 		/** 收尾后的静默期截止时间戳；期间仍按围观态渲染，到点才收拢。 */
 		settleUntil?: number;
+		/** 消息已有 toolCall、工具组件还没到：等它一 updateDisplay 就重画槽位。 */
+		pendingLiveTool?: boolean;
 		foldTimer?: ReturnType<typeof setTimeout>;
 		/** 回合挂钟起点，保证 Running 时长连续递增。 */
 		startedAt: number;
@@ -1348,12 +1353,6 @@ export function installCompactMode(deps: CompactModeInstallDeps): CompactModeHoo
 		...round.detachedMessages,
 	];
 
-	/** 镜像给常驻状态行的文案：只认当前活动回合（摘要行被顶出视口时由 working-message 取用）。 */
-	compactRunStatusGetter = () => {
-		const round = activeRound;
-		return round?.active ? summarize(roundMessages(round), true, round) || undefined : undefined;
-	};
-
 	const roundToolCallIds = (round: CompactRound): Set<string> => {
 		const ids = new Set<string>();
 		for (const message of roundMessages(round)) {
@@ -1364,6 +1363,26 @@ export function installCompactMode(deps: CompactModeInstallDeps): CompactModeHoo
 		return ids;
 	};
 
+	/**
+	 * 摘要行有东西可汇总才挂：普通工具（edit/write/ask 之外）或非空思考。
+	 * 只剩外层卡的回合不挂——那些卡各自能展开，空摘要点开只会跳位。
+	 */
+	const roundHasSummaryContent = (round: CompactRound): boolean => {
+		for (const message of roundMessages(round)) {
+			for (const item of Array.isArray(message?.content) ? message.content : []) {
+				if (item?.type === "thinking" && String(item.thinking ?? "").trim()) return true;
+				if (
+					item?.type === "toolCall" &&
+					typeof item.name === "string" &&
+					!OUTER_TOOLS.has(item.name)
+				) {
+					return true;
+				}
+			}
+		}
+		return false;
+	};
+
 	/**
 	 * anchor 所在的 transcript 容器（pi chatContainer）。
 	 * 优先走分组补丁在 addChild 时挂的父指针；没有则沿 tui 组件树找直属父容器。
@@ -1572,10 +1591,12 @@ export function installCompactMode(deps: CompactModeInstallDeps): CompactModeHoo
 	const syncTailForTool = (toolCallId: string): void => {
 		if (!toolCallId) return;
 		for (const round of [activeRound, ...pendingFoldRounds]) {
-			if (round?.tail && roundToolCallIds(round).has(toolCallId)) {
-				syncTailPosition(round);
-				return;
-			}
+			if (!round || !roundToolCallIds(round).has(toolCallId)) continue;
+			// 槽位是在消息更新时画的，当时工具组件还不存在：组件到了就补上。
+			// 参数流式更新也走这里，槽位卡跟着换内容。
+			if (round.pendingLiveTool || roundLive(round)) renderRound(round);
+			else if (round.tail) syncTailPosition(round);
+			return;
 		}
 	};
 
@@ -1719,6 +1740,7 @@ export function installCompactMode(deps: CompactModeInstallDeps): CompactModeHoo
 		}
 		let runningTool: any;
 		let lastTool: any;
+		let missingTool = false;
 		for (const [, message] of round.messages) {
 			for (const item of Array.isArray(message?.content) ? message.content : []) {
 				if (item?.type !== "toolCall" || typeof item.id !== "string") continue;
@@ -1727,11 +1749,17 @@ export function installCompactMode(deps: CompactModeInstallDeps): CompactModeHoo
 				if (OUTER_TOOLS.has(String(tool?.toolName ?? item.name ?? ""))) continue;
 				round.liveSlotToolIds.add(item.id);
 				liveSlotToolIds.add(item.id);
-				if (!tool) continue;
+				// pi 的 message_update 先 updateContent 再建工具组件：这一帧计数已经算上，
+				// 组件要等 updateDisplay 才能进槽位。记下来，组件一到就补画。
+				if (!tool) {
+					missingTool = true;
+					continue;
+				}
 				lastTool = tool;
 				if (isRunningTool(tool)) runningTool = tool;
 			}
 		}
+		round.pendingLiveTool = missingTool;
 		// 槽位单块：活动思考最新；否则本回合最近的工具（运行中原位增长，完成也留着）。
 		const slotTool = thinkingKids.length > 0 ? undefined : (runningTool ?? lastTool);
 		const cardItems: Array<{ child?: any; tool?: any }> = slotTool
@@ -1786,6 +1814,8 @@ export function installCompactMode(deps: CompactModeInstallDeps): CompactModeHoo
 		}
 		for (const id of round.liveSlotToolIds) liveSlotToolIds.delete(id);
 		round.liveSlotToolIds.clear();
+		// pendingLiveTool 留给 renderRoundLive 重算：这里清掉的话，组件刚到的那次
+		// 补画会把自己的触发条件抹掉，槽位仍是空的。
 
 		for (const [component, message] of round.messages) {
 			component.lastMessage = message;
@@ -1807,6 +1837,13 @@ export function installCompactMode(deps: CompactModeInstallDeps): CompactModeHoo
 			};
 		}
 
+		if (round.anchor.expanded === true) {
+			// 没有可汇总内容的回合不进展开态：摘要行不存在，展开只会把正文搬回回合开头。
+			if (!roundHasSummaryContent(round)) {
+				for (const member of round.messages.keys()) member.expanded = false;
+				round.anchor.expanded = false;
+			}
+		}
 		if (round.anchor.expanded === true) {
 			// 展开内容留在摘要行那个位置：回合最后一个成员（含 write/edit）之后。
 			// 放进 anchor 内部会跳到回合起点，盖到中间的 write/edit 上面。
@@ -1828,7 +1865,8 @@ export function installCompactMode(deps: CompactModeInstallDeps): CompactModeHoo
 				} else {
 					// 面板上沿留 1 行卡外空行：面板底色与 userMessage 同色，
 					// 贴在一起会看成一个块；卡内顶部再加 1 行 padding 保持上下对称。
-					anchorChildren.push(new Spacer(1));
+					// 空行必须与面板同进退（childAtRow 参与面板拆分），用 panelBoundSpacer。
+					anchorChildren.push(panelBoundSpacer());
 					anchorChildren.push(
 						compactRoundCard(
 							cardItems,
@@ -1949,8 +1987,9 @@ export function installCompactMode(deps: CompactModeInstallDeps): CompactModeHoo
 		for (const [component, message] of round.messages) {
 			if (component === round.anchor) {
 				renderAssistantWithoutThinking(component, message);
-				// 空摘要不挂行（getter 在 Running 启动瞬间也可能短暂为空）
-				if (summary || round.active) {
+				// 空摘要不挂行（getter 在 Running 启动瞬间也可能短暂为空）。
+				// 只剩外层卡的回合同样不挂：摘要行没有可汇总的内容。
+				if ((summary || round.active) && (round.active || roundHasSummaryContent(round))) {
 					mountRoundTail(round, (identity) => [
 						compactAssistantLineComponent(identity, getSummary, deps.query),
 					]);
@@ -2142,8 +2181,13 @@ export function installCompactMode(deps: CompactModeInstallDeps): CompactModeHoo
 
 	patch.toolInstalledUpdateDisplay = function (this: any) {
 		const id = String(this.toolCallId ?? "");
-		if (patch.active && config.mode === "compact" && expandedRoundToolIds.has(id)) {
-			// 用户点开的保留展开；其余（含全局展开被收回的）继续强制折叠。
+		if (
+			patch.active &&
+			config.mode === "compact" &&
+			(expandedRoundToolIds.has(id) || liveSlotToolIds.has(id))
+		) {
+			// 用户点开的保留展开；其余（含全局展开、live 槽位接管的）继续强制折叠。
+			// 槽位卡在消息更新时就接管了 id，组件后到时 setExpanded 会把它翻开，外层多出一张卡。
 			if (this.expanded !== true) explicitRoundToolIds.delete(id);
 			else if (!explicitRoundToolIds.has(id)) this.expanded = false;
 		}
@@ -2192,7 +2236,6 @@ export function installCompactMode(deps: CompactModeInstallDeps): CompactModeHoo
 		trackedToolComponents.clear();
 		hoveredAssistantComponent = undefined;
 		resetRounds();
-		compactRunStatusGetter = undefined;
 	};
 
 	assistantPrototype.updateContent = patch.assistantInstalled;
@@ -2249,6 +2292,13 @@ export function installCompactMode(deps: CompactModeInstallDeps): CompactModeHoo
 				}
 			}
 		},
+		settleReplayedRound() {
+			if (!patch.active || config.mode !== "compact") return;
+			const round = activeRound;
+			if (!round) return;
+			finalizeRound(round); // clears activeRound — capture it first
+			renderRound(round);
+		},
 		shutdown() {
 			patch.dispose();
 		},