@herbertgao/pi-extensions 2026.9.19 → 2026.10.0

package/node_modules/@herbertgao/pi-cc-extensions/extensions/feature/shell/startup-header.ts

@@ -1,8 +1,14 @@
-import { VERSION, type AppKeybinding } from "@earendil-works/pi-coding-agent";
+import { InteractiveMode, VERSION, type AppKeybinding } from "@earendil-works/pi-coding-agent";
 import { getKeybindings } from "@earendil-works/pi-tui";
 import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
 import { config } from "../../config/config.ts";
-
+import { ansi16ToRgb, ansi256ToRgb } from "../../utils/ansi-color.ts";
+import { stripAnsi } from "../../utils/ansi-text.ts";
+import {
+	EARLY_STARTUP_HEADER_PATCH,
+	EARLY_STARTUP_HEADER_SWAP_KEY,
+	patchRegistry,
+} from "../../utils/patch-keys.ts";
 type Rgb = [number, number, number];
 type StyledPart = {
 	raw: string;
@@ -10,8 +16,6 @@ type StyledPart = {
 };
 
 const ANSI_RESET = "\x1b[0m";
-const ANSI_PATTERN =
-	/[\u001B\u009B][[\]()#;?]*(?:(?:(?:[a-zA-Z\d]*(?:;[a-zA-Z\d]*)*)?\u0007)|(?:(?:\d{1,4}(?:;\d{0,4})*)?[\dA-PR-TZcf-nq-uy=><~]))/g;
 
 // 官方 install.sh 静态 logo（4 行原样，短行补尾随空格统一到 8 列）+ 底部空行补到 5 行，
 // 与右侧 tips 行数等高；着色保持现状（accent 渐变）
@@ -40,29 +44,6 @@ const PALETTE_SPAN = 0.25;
 // 行间相位偏移：小步累加 → 整体左上暗→右下亮的对角渐变
 const LOGO_ROW_PHASE_STEP = 0.08;
 
-const ANSI_16_RGB_TABLE: Rgb[] = [
-	[0, 0, 0],
-	[128, 0, 0],
-	[0, 128, 0],
-	[128, 128, 0],
-	[0, 0, 128],
-	[128, 0, 128],
-	[0, 128, 128],
-	[192, 192, 192],
-	[128, 128, 128],
-	[255, 0, 0],
-	[0, 255, 0],
-	[255, 255, 0],
-	[0, 0, 255],
-	[255, 0, 255],
-	[0, 255, 255],
-	[255, 255, 255],
-];
-
-function stripAnsi(text: string): string {
-	return text.replace(ANSI_PATTERN, "");
-}
-
 function getVisibleLength(text: string): number {
 	return [...stripAnsi(text)].length;
 }
@@ -104,27 +85,6 @@ function applyTruecolor(rgb: Rgb, text: string): string {
 	return `\x1b[38;2;${red};${green};${blue}m${text}${ANSI_RESET}`;
 }
 
-function ansi16ToRgb(index: number): Rgb {
-	return ANSI_16_RGB_TABLE[index] ?? [255, 255, 255];
-}
-
-function ansi256ToRgb(index: number): Rgb {
-	if (index < 16) return ansi16ToRgb(index);
-
-	if (index >= 232) {
-		const gray = 8 + (index - 232) * 10;
-		return [gray, gray, gray];
-	}
-
-	const cubeIndex = index - 16;
-	const redIndex = Math.floor(cubeIndex / 36);
-	const greenIndex = Math.floor((cubeIndex % 36) / 6);
-	const blueIndex = cubeIndex % 6;
-	const values = [0, 95, 135, 175, 215, 255];
-
-	return [values[redIndex]!, values[greenIndex]!, values[blueIndex]!];
-}
-
 function parseTruecolorAnsi(ansi: string): Rgb | undefined {
 	const match = ansi.match(/38;2;(\d+);(\d+);(\d+)/);
 	if (!match) return undefined;
@@ -324,23 +284,94 @@ export function renderHeaderLines(
 	);
 }
 
-/**
- * 按配置应用启动头：on → 自定义 header；off → 恢复官方默认 header。
- * 导出供 /ccstyle 面板在切换开关时实时重应用。
- */
-export function applyStartupHeader(ctx: any): void {
-	if (!ctx?.hasUI || typeof ctx.ui?.setHeader !== "function") return;
-	if (!config.showStartupHeader) {
-		// 恢复官方内置 header（logo + 快捷键提示 + onboarding）。
-		ctx.ui.setHeader(undefined);
-		return;
-	}
-	ctx.ui.setHeader((_tui, theme) => ({
+/** 官方 setHeader 的组件工厂：theme 由宿主注入。 */
+function headerFactory(): (
+	tui: unknown,
+	theme: any,
+) => {
+	render(width: number): string[];
+	invalidate(): void;
+} {
+	return (_tui: unknown, theme: any) => ({
 		render(width: number): string[] {
 			return renderHeaderLines(width, theme);
 		},
 		invalidate() {},
-	}));
+	});
+}
+
+/** 按配置安装启动头。禁用时不碰槽位，避免清掉其他扩展的 header。 */
+export function applyStartupHeader(ctx: any): void {
+	if (!ctx?.hasUI || typeof ctx.ui?.setHeader !== "function" || !config.showStartupHeader) {
+		return;
+	}
+	ctx.ui.setHeader(headerFactory());
+}
+
+type EarlyHeaderPatch = {
+	active: boolean;
+	original: (...args: any[]) => unknown;
+	installed: (...args: any[]) => unknown;
+};
+
+type HeaderContainerSwap = {
+	original: (child: any) => unknown;
+	installed: (child: any) => unknown;
+};
+
+/**
+ * Pi 在 session_start 之前就把原生启动头画上去了：init() 先建 header 再 requestRender，
+ * 之后才 await 工具检查、加载扩展，所以扩展最早只能到 session_start 换头，启动时会闪一下原生头。
+ * 这里补 init，在原生 header 刚进容器的瞬间换成 ccstyle 的，首次绘制就已经是我们的。
+ * 判定用 child === mode.builtInHeader，不依赖类名；配置关掉时不碰槽位，保留原生头。
+ */
+export function installEarlyStartupHeader(): void {
+	const prototype = (InteractiveMode as any)?.prototype;
+	if (!prototype || typeof prototype.init !== "function") return;
+	const previous = patchRegistry.get<EarlyHeaderPatch>(EARLY_STARTUP_HEADER_PATCH);
+	if (previous) previous.active = false;
+	const original =
+		previous && prototype.init === previous.installed ? previous.original : prototype.init;
+	const patch: EarlyHeaderPatch = {
+		active: true,
+		original,
+		installed: async function (this: any, ...args: any[]) {
+			installHeaderContainerSwap(this, patch);
+			return original.apply(this, args);
+		},
+	};
+	prototype.init = patch.installed;
+	patchRegistry.install(EARLY_STARTUP_HEADER_PATCH, patch);
+}
+
+/** 只包一次 headerContainer.addChild；/reload 后按所有权接管旧包装。 */
+function installHeaderContainerSwap(mode: any, patch: EarlyHeaderPatch): void {
+	const container = mode?.headerContainer;
+	if (!container || typeof container.addChild !== "function") return;
+	const previous = container[EARLY_STARTUP_HEADER_SWAP_KEY] as HeaderContainerSwap | undefined;
+	const original =
+		previous && container.addChild === previous.installed
+			? previous.original
+			: container.addChild.bind(container);
+	const installed = function (child: any) {
+		const result = original(child);
+		if (!patch.active || !config.showStartupHeader || child !== mode.builtInHeader) return result;
+		if (typeof mode.setExtensionHeader !== "function") return result;
+		try {
+			mode.setExtensionHeader(headerFactory());
+		} catch {
+			// 换头失败就退回原生 header，不影响启动
+		}
+		return result;
+	};
+	container.addChild = installed;
+	container[EARLY_STARTUP_HEADER_SWAP_KEY] = { original, installed };
+}
+
+/** 用户从 /ccstyle 关掉本扩展启动头时，恢复官方 header。 */
+export function clearStartupHeader(ctx: any): void {
+	if (!ctx?.hasUI || typeof ctx.ui?.setHeader !== "function") return;
+	ctx.ui.setHeader(undefined);
 }
 
 export default function piStartupHeader(pi: ExtensionAPI) {

package/node_modules/@herbertgao/pi-cc-extensions/extensions/feature/shell/working-message.ts

@@ -1,38 +1,38 @@
 import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
+import { getCompactRunStatusText } from "../../renderer/compact-mode.ts";
+import {
+	fullscreenLazyTui,
+	getToolMouseTui,
+	isFullscreenAtBottom,
+} from "../../renderer/mouse/scroll.ts";
+import { formatDuration } from "../../utils/format.ts";
 
 const REFRESH_INTERVAL_MS = 1_000;
 /** Elapsed time is only shown once the turn has run this long. */
 const SHOW_TIMER_AFTER_MS = 3_000;
 
-function formatDuration(ms: number): string {
-	const totalSec = Math.floor(ms / 1000);
-	const hours = Math.floor(totalSec / 3600);
-	const minutes = Math.floor((totalSec % 3600) / 60);
-	const seconds = totalSec % 60;
-	if (hours > 0) return `${hours}h ${minutes}m ${seconds}s`;
-	if (minutes > 0) return `${minutes}m ${seconds}s`;
-	return `${seconds}s`;
-}
-
 function formatCount(value: number): string {
 	return new Intl.NumberFormat("en-US").format(value);
 }
 
-function estimateTextLength(message: any): number {
-	if (!Array.isArray(message?.content)) return 0;
-	return message.content.reduce((sum: number, block: any) => {
-		if (block?.type === "text" && typeof block.text === "string") return sum + block.text.length;
-		if (block?.type === "thinking" && block.thinkingSignature?.body)
-			return sum + (block.thinkingSignature.body as string).length;
-		return sum;
-	}, 0);
-}
-
-function textBlockLengths(message: any): number[] {
-	if (!Array.isArray(message?.content)) return [];
+type ContentBlock = {
+	type?: unknown;
+	text?: unknown;
+	thinkingSignature?: { body?: unknown };
+};
+
+type StreamMessage = {
+	content?: unknown;
+	usage?: { output?: unknown };
+};
+
+/** 每个 content index 的可见文本/思考长度；无对应块的 index 保持稀疏洞。 */
+function textBlockLengths(message: StreamMessage): number[] {
+	const content = message.content;
+	if (!Array.isArray(content)) return [];
 	const lengths: number[] = [];
-	for (let index = 0; index < message.content.length; index++) {
-		const block = message.content[index];
+	for (let index = 0; index < content.length; index++) {
+		const block = content[index] as ContentBlock;
 		if (block?.type === "text" && typeof block.text === "string") {
 			lengths[index] = block.text.length;
 		} else if (block?.type === "thinking" && block.thinkingSignature?.body) {
@@ -42,11 +42,15 @@ function textBlockLengths(message: any): number[] {
 	return lengths;
 }
 
-function outputUsage(message: any): number {
+function outputUsage(message: StreamMessage): number {
 	const value = Number(message?.usage?.output);
 	return Number.isFinite(value) && value > 0 ? Math.round(value) : 0;
 }
 
+type WorkingUi = {
+	setWorkingMessage(message?: string): void;
+};
+
 /**
  * Extend Pi's footer working row while preserving its spinner and "Working...":
  * `⠋ Working... (↓ 1,234 tokens · 12s)`
@@ -63,7 +67,7 @@ export default function (pi: ExtensionAPI): void {
 	let providerOutputTokens = 0;
 	let refreshTimer: ReturnType<typeof setTimeout> | null = null;
 	let lastMessage: string | null = null;
-	let activeCtx: { ui: any; hasUI: boolean } | null = null;
+	let activeCtx: { ui: WorkingUi | undefined; hasUI: boolean } | null = null;
 
 	function tokenCount(): number {
 		return providerOutputTokens || Math.max(0, Math.round(responseLength / 4));
@@ -75,26 +79,41 @@ export default function (pi: ExtensionAPI): void {
 		responseLength = Math.max(0, responseLength + responseTextBlockLengths[index] - previous);
 	}
 
-	function resetResponseTracking(message?: any): void {
+	function resetResponseTracking(message?: StreamMessage): void {
 		responseTextBlockLengths = message ? textBlockLengths(message) : [];
-		responseLength = message ? estimateTextLength(message) : 0;
+		responseLength = responseTextBlockLengths.reduce((sum, length) => sum + length, 0);
 		providerOutputTokens = message ? outputUsage(message) : 0;
 	}
 
-	function updateProviderUsage(message: any): void {
+	function updateProviderUsage(message: StreamMessage): void {
 		const output = outputUsage(message);
 		if (output > 0) providerOutputTokens = output;
 	}
 
+	/**
+	 * 摘要行滚出视口才镜像：仅 fullscreen 且已离开 transcript 底部。
+	 * regular 没有“离开底部”信号（transcript 在终端回滚区），保持 Pi 默认文案。
+	 */
+	function compactMirrorText(): string | undefined {
+		const tui = getToolMouseTui();
+		if (!tui || !fullscreenLazyTui(tui) || isFullscreenAtBottom(tui)) return undefined;
+		return getCompactRunStatusText();
+	}
+
 	function buildWorkingMessage(): string {
+		// turn_end 后迟到的 provider 事件不得再驱动 footer（否则渲染出 epoch 级计时）。
 		if (!turnActive) return "";
-		const startedAt = agentStartTime || turnStartTime;
-		if (!startedAt) return "";
-		const elapsed = Date.now() - startedAt;
-		const tokens = tokenCount();
 		const parts: string[] = [];
+		const tokens = tokenCount();
 		if (tokens > 0) parts.push(`↓ ${formatCount(tokens)} tokens`);
-		if (elapsed >= SHOW_TIMER_AFTER_MS || tokens > 0) parts.push(formatDuration(elapsed));
+		// compact 活动回合且已滚出摘要行：直接用摘要行文案（自带回合时长，不叠 agent 计时）。
+		const compactStatus = compactMirrorText();
+		if (compactStatus) return [compactStatus, ...parts].join(" · ");
+		const elapsed = Date.now() - (agentStartTime || turnStartTime);
+		if (elapsed >= SHOW_TIMER_AFTER_MS || tokens > 0) {
+			// formatDuration 低于 1 秒返回 ""，此处回退 "0s" 保持计时器连续跳动。
+			parts.push(formatDuration(elapsed) || "0s");
+		}
 		return parts.length ? `Working... (${parts.join(" · ")})` : "";
 	}
 
@@ -102,7 +121,7 @@ export default function (pi: ExtensionAPI): void {
 		try {
 			return activeCtx?.hasUI === true;
 		} catch {
-			// A stale ctx after session replacement/reload throws from its getter; stop driving the footer.
+			// 会话替换/reload 后捕获的 ctx 失效，getter 抛错；停止驱动 footer。
 			turnActive = false;
 			activeCtx = null;
 			stopRefreshLoop();
@@ -114,7 +133,7 @@ export default function (pi: ExtensionAPI): void {
 		lastMessage = null;
 		if (!workingUiAvailable()) return;
 		try {
-			activeCtx?.ui.setWorkingMessage();
+			activeCtx?.ui?.setWorkingMessage();
 		} catch {
 			// Noop when the TUI is unavailable.
 		}
@@ -130,7 +149,7 @@ export default function (pi: ExtensionAPI): void {
 		if (!force && next === lastMessage) return;
 		lastMessage = next;
 		try {
-			activeCtx?.ui.setWorkingMessage(next);
+			activeCtx?.ui?.setWorkingMessage(next);
 		} catch {
 			// Noop when the TUI is unavailable.
 		}
@@ -143,7 +162,7 @@ export default function (pi: ExtensionAPI): void {
 			try {
 				syncWorkingMessage();
 			} catch {
-				// An exception here would become an uncaughtException and kill Pi; stop the decorative refresh.
+				// 定时器内的异常会成为 uncaughtException 终止 Pi；装饰性刷新直接停止。
 				turnActive = false;
 				return;
 			}
@@ -203,6 +222,7 @@ export default function (pi: ExtensionAPI): void {
 		} else if (evt.type === "done") {
 			resetResponseTracking(evt.message);
 		} else if (evt.type === "error") {
+			// provider 错误事件的 usage.output 不可信，不得计入 token 统计。
 			resetResponseTracking();
 		} else {
 			updateProviderUsage(evt.partial);

package/node_modules/@herbertgao/pi-cc-extensions/extensions/index.ts

@@ -3,7 +3,9 @@ import { config } from "./config/config.ts";
 
 // shell
 import piAliases from "./feature/shell/aliases.ts";
-import piStartupHeader from "./feature/shell/startup-header.ts";
+import { installFlushDockedBash } from "./feature/shell/flush-docked-bash.ts";
+import customFooter from "./feature/shell/footer.ts";
+import piStartupHeader, { installEarlyStartupHeader } from "./feature/shell/startup-header.ts";
 import workingMessage from "./feature/shell/working-message.ts";
 
 // feature
@@ -12,22 +14,36 @@ import agentSummary from "./feature/agent-summary/index.ts";
 import context from "./feature/context.ts";
 import sessionReference from "./feature/reference/index.ts";
 import {
-	installCompactThinking,
 	type CompactThinkingController,
+	installCompactThinking,
 } from "./feature/compact-thinking.ts";
 
 // renderer
 import claudeCodeStyle, { getCompactThinkingConfig } from "./renderer/index.ts";
 import markdownEnhance from "./renderer/markdown-enhance.ts";
+import {
+	COMPACT_MODE_PATCH_KEY,
+	COMPACT_THINKING_OWNER,
+	patchRegistry,
+} from "./utils/patch-keys.ts";
 
 export default function (pi: ExtensionAPI): void {
+	// /reload 或会话切换接手：先释放旧一代 compact 外层，再停旧 thinking
+	// owner（顺序不可反，否则新一代会把旧 wrapper 误采样为 native 原型，
+	// 补丁因此在 shutdown 后残留或叠加）。
+	patchRegistry.get<{ dispose(): void }>(COMPACT_MODE_PATCH_KEY)?.dispose();
+	patchRegistry.get<{ stop(): void }>(COMPACT_THINKING_OWNER)?.stop();
+
 	// shell chrome
 	if (config.enableAliases) piAliases(pi);
+	installFlushDockedBash();
+	installEarlyStartupHeader();
 	piStartupHeader(pi);
 	if (config.enableWorkingMessage) workingMessage(pi);
+	customFooter(pi);
 
 	// Renderer 必须先注册 lifecycle handler，shutdown 时才能先解开外层 compact
-	// patch；bridge 让稍后安装的 thinking controller 仍可直接供 renderer query。
+	// patch，thinking 随后才能恢复原生原型；bridge 转发给稍后安装的 controller。
 	let compactThinking: CompactThinkingController | undefined;
 	const compactThinkingQuery: CompactThinkingController = {
 		updateConfig: (next) => compactThinking?.updateConfig(next),