pi-cc-extensions 0.8.59 → 0.8.61

package/README.en.md

@@ -56,7 +56,8 @@ Run `/reload` after installation.
   "diffViewMode": "auto",                  // diff layout: auto / split / unified
   "diffIndicatorMode": "bars",             // diff change indicators: bars / classic / none
   "diffSplitMinWidth": 120,                // min terminal width before auto layout uses side-by-side columns
-  "diffCollapsedLines": 24,                // diff body lines shown when collapsed; beyond that shows the expand hint (Ctrl+O / click)
+  "diffCollapsedLines": 24,                // edit/diff body lines shown when collapsed; beyond that shows the expand hint (Ctrl+O / click)
+  "writeDiffCollapsedLines": 0,            // write collapsed body lines; 0 = ↳ created • expand hint (stats stay on the title)
   "diffWordWrap": true,                    // whether long diff lines wrap (otherwise truncated)
   "expandedPreviewMaxLines": 40,           // max body lines for expanded output/diff
   "useSummaryTitlesAsThinkingTitle": true, // use the latest provider summary as the active thinking title

package/README.md

@@ -56,7 +56,8 @@ pi install git:github.com/minuque/pi-cc-extensions
   "diffViewMode": "auto",                  // diff 布局：auto / split / unified
   "diffIndicatorMode": "bars",             // diff 变更指示：bars / classic / none
   "diffSplitMinWidth": 120,                // 自动布局下使用左右分栏的最小终端宽度
-  "diffCollapsedLines": 24,                // 折叠时展示的 diff 行数，超出显示展开提示（Ctrl+O / 点击）
+  "diffCollapsedLines": 24,                // 折叠时展示的 edit/diff 行数，超出显示展开提示（Ctrl+O / 点击）
+  "writeDiffCollapsedLines": 0,            // write 折叠行数；0 = ↳ created • 展开提示（统计在标题）
   "diffWordWrap": true,                    // 长 diff 行是否换行（否则截断）
   "expandedPreviewMaxLines": 40,           // 展开后输出/diff 正文的最大行数
   "useSummaryTitlesAsThinkingTitle": true, // 用最新 provider 摘要作为思考标题

package/extensions/config/config.ts

@@ -16,6 +16,8 @@ export interface ToolDisplayConfig {
 	diffIndicatorMode: DiffIndicatorMode;
 	diffSplitMinWidth: number;
 	diffCollapsedLines: number;
+	/** Write-only collapsed body lines. 0 = `↳ created • click to show more`. */
+	writeDiffCollapsedLines: number;
 	diffWordWrap: boolean;
 	expandedPreviewMaxLines: number;
 }
@@ -26,6 +28,11 @@ export const DEFAULT_TOOL_DISPLAY_CONFIG: ToolDisplayConfig = {
 	diffSplitMinWidth: 120,
 	/** Collapsed tool/diff body: ~half a typical terminal after chrome. */
 	diffCollapsedLines: 24,
+	/**
+	 * Write create/overwrite collapsed body.
+	 * 0 = `↳ created • click to show more` (stats stay on the title).
+	 */
+	writeDiffCollapsedLines: 0,
 	diffWordWrap: true,
 	/**
 	 * Expanded tool/diff body cap. 40 ≈ one screen of content after title,
@@ -42,6 +49,7 @@ export type Config = {
 	diffIndicatorMode: DiffIndicatorMode;
 	diffSplitMinWidth: number;
 	diffCollapsedLines: number;
+	writeDiffCollapsedLines: number;
 	diffWordWrap: boolean;
 	expandedPreviewMaxLines: number;
 	useSummaryTitlesAsThinkingTitle: boolean;
@@ -64,6 +72,8 @@ export const DIFF_VIEW_MODES: DiffViewMode[] = ["auto", "split", "unified"];
 export const DIFF_INDICATOR_MODES: DiffIndicatorMode[] = ["bars", "classic", "none"];
 export const DIFF_SPLIT_MIN_WIDTH_VALUES = ["80", "100", "120", "140", "160", "180"];
 export const DIFF_COLLAPSED_LINES_VALUES = ["12", "24", "36", "48", "80", "120"];
+/** Write collapsed presets. 0 = stats only (`+N -0` + expand hint). */
+export const WRITE_DIFF_COLLAPSED_LINES_VALUES = ["0", "4", "8", "12", "24", "36"];
 /** Presets for expanded body height — keep low options first so cycling stays TUI-friendly. */
 export const EXPANDED_PREVIEW_MAX_LINES_VALUES = ["40", "60", "80", "120", "200", "500", "2000"];
 export const THINKING_PREVIEW_LINES_VALUES = ["0", "1", "3", "5", "10"];
@@ -90,6 +100,7 @@ export const DEFAULT_CONFIG: Config = {
 	diffIndicatorMode: DEFAULT_TOOL_DISPLAY_CONFIG.diffIndicatorMode,
 	diffSplitMinWidth: DEFAULT_TOOL_DISPLAY_CONFIG.diffSplitMinWidth,
 	diffCollapsedLines: DEFAULT_TOOL_DISPLAY_CONFIG.diffCollapsedLines,
+	writeDiffCollapsedLines: DEFAULT_TOOL_DISPLAY_CONFIG.writeDiffCollapsedLines,
 	diffWordWrap: DEFAULT_TOOL_DISPLAY_CONFIG.diffWordWrap,
 	expandedPreviewMaxLines: DEFAULT_TOOL_DISPLAY_CONFIG.expandedPreviewMaxLines,
 	useSummaryTitlesAsThinkingTitle: true,
@@ -164,6 +175,12 @@ export function normalizeConfig(input: unknown): Config {
 			1,
 			500,
 		),
+		writeDiffCollapsedLines: pickPositiveInt(
+			source.writeDiffCollapsedLines,
+			DEFAULT_CONFIG.writeDiffCollapsedLines,
+			0,
+			500,
+		),
 		diffWordWrap: source.diffWordWrap !== false,
 		expandedPreviewMaxLines: pickPositiveInt(
 			source.expandedPreviewMaxLines,
@@ -207,6 +224,7 @@ export function getToolDisplayConfig(source: Config = config): ToolDisplayConfig
 		diffIndicatorMode: source.diffIndicatorMode,
 		diffSplitMinWidth: source.diffSplitMinWidth,
 		diffCollapsedLines: source.diffCollapsedLines,
+		writeDiffCollapsedLines: source.writeDiffCollapsedLines,
 		diffWordWrap: source.diffWordWrap,
 		expandedPreviewMaxLines: source.expandedPreviewMaxLines,
 	};
@@ -224,6 +242,7 @@ export function formatConfigStatus(source: Config = config): string {
 		`diffIndicator=${source.diffIndicatorMode}`,
 		`diffSplitMin=${source.diffSplitMinWidth}`,
 		`diffCollapsed=${source.diffCollapsedLines}`,
+		`writeCollapsed=${source.writeDiffCollapsedLines}`,
 		`diffWordWrap=${source.diffWordWrap ? "on" : "off"}`,
 		`expandedMax=${source.expandedPreviewMaxLines}`,
 		`thinkingTitle=${source.useSummaryTitlesAsThinkingTitle ? "summary" : "default"}`,

package/extensions/config/panel.ts

@@ -25,6 +25,7 @@ import {
 	SCROLL_STEP_LINES_VALUES,
 	THINKING_ANIMATION_INTERVAL_VALUES,
 	THINKING_PREVIEW_LINES_VALUES,
+	WRITE_DIFF_COLLAPSED_LINES_VALUES,
 	updateConfig,
 	type CompactStyleMode,
 	type Config,
@@ -40,7 +41,7 @@ export type CcstylePanelHooks = {
 
 function modeSettingDescription(mode: CompactStyleMode): string {
 	if (mode === "compact") {
-		return "(Experimental) One summary line per assistant round; expanded edit/write rows show rich diffs.";
+		return "(Experimental) One summary line per assistant round; edit/write reuse the same Diff preview settings as on.";
 	}
 	if (mode === "off") {
 		return "Pi native tool rendering. Diff options below still apply independently.";
@@ -288,14 +289,24 @@ export async function showCcstylePanel(
 		};
 		const diffCollapsedSetting = {
 			id: "diffCollapsedLines",
-			label: "Collapsed lines",
+			label: "Edit collapsed lines",
 			description:
-				"How many diff body lines to show before the expand hint (Ctrl+O / click). Enter to type a custom value.",
+				"How many edit/diff body lines to show before the expand hint (Ctrl+O / click). Write uses its own setting below.",
 			currentValue: String(config.diffCollapsedLines),
 			values: [...DIFF_COLLAPSED_LINES_VALUES],
 			submenu: (_current: string, closeSubmenu: (selected?: string) => void) =>
 				buildNumberInputSubmenu(theme, diffCollapsedSetting, closeSubmenu),
 		};
+		const writeDiffCollapsedSetting = {
+			id: "writeDiffCollapsedLines",
+			label: "Write collapsed lines",
+			description:
+				"Write-only collapsed body lines. 0 shows ↳ created + expand hint (stats stay on the title). Enter to type a custom value.",
+			currentValue: String(config.writeDiffCollapsedLines),
+			values: [...WRITE_DIFF_COLLAPSED_LINES_VALUES],
+			submenu: (_current: string, closeSubmenu: (selected?: string) => void) =>
+				buildNumberInputSubmenu(theme, writeDiffCollapsedSetting, closeSubmenu),
+		};
 		const diffWordWrapSetting = {
 			id: "diffWordWrap",
 			label: "Diff word wrap",
@@ -453,6 +464,17 @@ export async function showCcstylePanel(
 					});
 					diffCollapsedSetting.currentValue = String(config.diffCollapsedLines);
 					break;
+				case "writeDiffCollapsedLines":
+					updateConfig({
+						writeDiffCollapsedLines: pickPositiveInt(
+							value,
+							DEFAULT_CONFIG.writeDiffCollapsedLines,
+							0,
+							500,
+						),
+					});
+					writeDiffCollapsedSetting.currentValue = String(config.writeDiffCollapsedLines);
+					break;
 				case "diffWordWrap":
 					updateConfig({ diffWordWrap: value === "on" });
 					diffWordWrapSetting.description = config.diffWordWrap
@@ -519,6 +541,7 @@ export async function showCcstylePanel(
 					diffIndicatorSetting,
 					diffSplitSetting,
 					diffCollapsedSetting,
+					writeDiffCollapsedSetting,
 					diffWordWrapSetting,
 					expandedMaxSetting,
 				],

package/extensions/feature/compact-thinking.ts

@@ -11,8 +11,14 @@ import {
 	type ExtensionAPI,
 	type Theme,
 } from "@earendil-works/pi-coding-agent";
-import { getKeybindings } from "@earendil-works/pi-tui";
-import { Markdown, Spacer, Text, type Component } from "@earendil-works/pi-tui";
+import {
+	getKeybindings,
+	truncateToWidth,
+	visibleWidth,
+	wrapTextWithAnsi,
+} from "@earendil-works/pi-tui";
+import { Box, Markdown, Spacer, Text, type Component } from "@earendil-works/pi-tui";
+import { isToolTuiFullscreen } from "../renderer/tool/show-more-hint.ts";
 import {
 	animateCompactThinkingText,
 	formatThoughtDuration,
@@ -163,27 +169,206 @@ function getThinkingToggleHint() {
 	return keys.length > 0 ? `${keys.join("/")} to expand` : undefined;
 }
 
-class StrictThinkingPreview implements Component {
+function thinkingExpandAction(): string | undefined {
+	return isToolTuiFullscreen() ? "click to show more" : getThinkingToggleHint();
+}
+
+// 只 wrap 尾部窗口；缓存未着色折行，着色放到取出时做，避免主题切换命中旧 ANSI。
+const previewCache = new Map<string, { lines: string[]; hiddenBefore: number }>();
+const PREVIEW_CACHE_MAX = 500;
+const PREVIEW_BUDGET_MIN_CHARS = 2_000;
+const PREVIEW_BUDGET_SLACK_LINES = 2;
+
+/** 按可见宽度估算折行数，不走 Text 全量 wrap。无换行长段也能计到隐藏行。 */
+function countWrappedLines(text: string, contentWidth: number): number {
+	if (!text) return 0;
+	const width = Math.max(1, contentWidth);
+	let lines = 0;
+	for (const raw of text.replace(/\t/g, "   ").split(/\r\n|\r|\n/)) {
+		const w = visibleWidth(raw);
+		lines += w === 0 ? 1 : Math.ceil(w / width);
+	}
+	return lines;
+}
+
+function padPreviewLine(line: string, width: number, padding: number): string {
+	const left = padding > 0 ? " ".repeat(padding) : "";
+	const right = padding > 0 ? " ".repeat(padding) : "";
+	const withMargins = left + line + right;
+	return withMargins + " ".repeat(Math.max(0, width - visibleWidth(withMargins)));
+}
+
+function layoutThinkingPreview(
+	text: string,
+	width: number,
+	padding: number,
+): { visible: string[]; hiddenLines: number } {
+	if (!text || config.previewLines <= 0) return { visible: [], hiddenLines: 0 };
+	const contentWidth = Math.max(1, width - padding * 2);
+	const budget = Math.max(
+		width * (config.previewLines + PREVIEW_BUDGET_SLACK_LINES),
+		PREVIEW_BUDGET_MIN_CHARS,
+	);
+	let source = text;
+	let cut = 0;
+	if (source.length > budget) {
+		const start = source.length - budget;
+		const newline = source.indexOf("\n", start);
+		cut = newline === -1 ? start : newline + 1;
+		if (cut < source.length) source = source.slice(cut);
+		else cut = 0;
+	}
+	const cacheKey = `${width}:${padding}:${cut}:${source}`;
+	let entry = previewCache.get(cacheKey);
+	if (!entry) {
+		entry = {
+			lines: wrapTextWithAnsi(source.replace(/\t/g, "   "), contentWidth),
+			hiddenBefore: cut > 0 ? countWrappedLines(text.slice(0, cut), contentWidth) : 0,
+		};
+		if (previewCache.size >= PREVIEW_CACHE_MAX) previewCache.clear();
+		previewCache.set(cacheKey, entry);
+	}
+	const hiddenLines = entry.hiddenBefore + Math.max(0, entry.lines.length - config.previewLines);
+	const visible = hiddenLines > 0 ? entry.lines.slice(-config.previewLines) : entry.lines;
+	return { visible, hiddenLines };
+}
+
+function hiddenPreviewHint(
+	hiddenLines: number,
+	forceExpandHint = false,
+): { prefix: string; action: string; suffix: string } | undefined {
+	const action = thinkingExpandAction() ?? "";
+	if (hiddenLines > 0) {
+		const noun = hiddenLines === 1 ? "line" : "lines";
+		return {
+			prefix: ` • (${hiddenLines} more ${noun}${action ? ", " : ""}`,
+			action,
+			suffix: ")",
+		};
+	}
+	if (forceExpandHint && isToolTuiFullscreen()) {
+		return { prefix: " • ", action: "click to show more", suffix: "" };
+	}
+	return undefined;
+}
+
+const expandedThinking = new Set<number>();
+
+/** 折叠预览 + 展开全文。fullscreen 点击 hint 展开、再点整块收起，对齐工具卡。 */
+export class ThinkingPreviewBlock implements Component {
+	private heading: string;
 	private text: string;
 	private padding: number;
+	private messageTimestamp: number;
 	private style: (text: string) => string;
-
-	constructor(text: string, padding: number, style: (text: string) => string) {
+	private theme: Theme | undefined;
+	private _expanded: boolean;
+	private hintHovered = false;
+	private expandedBody: { width: number; lines: string[] } | undefined;
+
+	constructor(
+		heading: string,
+		text: string,
+		padding: number,
+		messageTimestamp: number,
+		style: (text: string) => string,
+		theme?: Theme,
+	) {
+		this.heading = heading;
 		this.text = text;
 		this.padding = padding;
+		this.messageTimestamp = messageTimestamp;
 		this.style = style;
+		this.theme = theme;
+		this._expanded = expandedThinking.has(messageTimestamp);
 	}
 
-	render(width: number) {
-		const lines = new Text(this.style(this.text), this.padding, 0).render(width);
-		if (lines.length <= config.previewLines) return lines;
+	private paint(color: string, text: string): string {
+		return this.theme && typeof this.theme.fg === "function"
+			? this.theme.fg(color as never, text)
+			: text;
+	}
 
-		const hiddenLines = lines.length - config.previewLines;
-		const noun = hiddenLines === 1 ? "line" : "lines";
-		const toggleHint = getThinkingToggleHint();
-		const hint = `... (${hiddenLines} more ${noun}${toggleHint ? `, ${toggleHint}` : ""})`;
-		const hintLines = new Text(this.style(hint), this.padding, 0).render(width);
-		return [...hintLines, ...lines.slice(-config.previewLines)];
+	get expanded(): boolean {
+		return this._expanded;
+	}
+
+	setExpanded(expanded: boolean): void {
+		this._expanded = expanded;
+		if (expanded) expandedThinking.add(this.messageTimestamp);
+		else {
+			expandedThinking.delete(this.messageTimestamp);
+			this.expandedBody = undefined;
+		}
+	}
+
+	setHintHovered(hovered: boolean): void {
+		this.hintHovered = hovered;
+	}
+
+	private headingLines(width: number, hiddenLines: number, padding: number): string[] {
+		const hint = this._expanded
+			? undefined
+			: hiddenPreviewHint(hiddenLines, Boolean(this.text) && config.previewLines <= 0);
+		if (!hint) return new Text(this.heading, padding, 0).render(width);
+		const rawHint = hint.prefix + hint.action + hint.suffix;
+		const contentWidth = Math.max(1, width - padding * 2);
+		const headingBudget = Math.max(0, contentWidth - visibleWidth(rawHint));
+		const clipped =
+			visibleWidth(this.heading) > headingBudget
+				? truncateToWidth(this.heading, headingBudget, "")
+				: this.heading;
+		const action = this.paint(this.hintHovered ? "text" : "dim", hint.action);
+		return new Text(
+			clipped + this.paint("dim", hint.prefix) + action + this.paint("dim", hint.suffix),
+			padding,
+			0,
+		).render(width);
+	}
+
+	private bodyLines(width: number, padding: number): { lines: string[]; hiddenLines: number } {
+		if (!this.text || (config.previewLines <= 0 && !this._expanded)) {
+			return { lines: [], hiddenLines: 0 };
+		}
+		if (this._expanded) {
+			if (this.expandedBody?.width !== width) {
+				this.expandedBody = {
+					width,
+					lines: wrapTextWithAnsi(this.text.replace(/\t/g, "   "), Math.max(1, width)),
+				};
+			}
+			return { lines: this.expandedBody.lines, hiddenLines: 0 };
+		}
+		const preview = layoutThinkingPreview(this.text, width, padding);
+		return { lines: preview.visible, hiddenLines: preview.hiddenLines };
+	}
+
+	render(width: number) {
+		if (this._expanded) {
+			const innerWidth = Math.max(1, width - 2);
+			const body = this.bodyLines(innerWidth, 0);
+			const heading = this.headingLines(innerWidth, 0, 0);
+			const inner = body.lines.length
+				? [...heading, ...body.lines.map((line) => this.style(line))]
+				: heading;
+			const bgFn =
+				this.theme && typeof this.theme.bg === "function"
+					? (text: string) => this.theme!.bg("userMessageBg" as never, text)
+					: undefined;
+			const box = new Box(1, 1, bgFn);
+			box.addChild({
+				render: () => inner,
+				invalidate() {},
+			});
+			return box.render(width);
+		}
+		const body = this.bodyLines(width, this.padding);
+		const heading = this.headingLines(width, body.hiddenLines, this.padding);
+		if (!body.lines.length) return heading;
+		return [
+			...heading,
+			...body.lines.map((line) => padPreviewLine(this.style(line), width, this.padding)),
+		];
 	}
 
 	invalidate() {}
@@ -461,14 +646,17 @@ function compactThinking(pi: ExtensionAPI) {
 					durationText ? `Thought for ${durationText}` : (latestSummary?.title ?? "Thought"),
 				);
 			}
-			self.contentContainer.addChild(new Text(heading, self.outputPad, 0));
-
-			const previewSource = latestSummary?.body ?? thinkingBlocks.join("\n\n");
-			if (config.previewLines > 0 && previewSource.trim()) {
-				self.contentContainer.addChild(
-					new StrictThinkingPreview(previewSource.trim(), self.outputPad, thinkingStyle),
-				);
-			}
+			const previewSource = (latestSummary?.body ?? thinkingBlocks.join("\n\n")).trim();
+			self.contentContainer.addChild(
+				new ThinkingPreviewBlock(
+					heading,
+					previewSource,
+					self.outputPad,
+					message.timestamp,
+					thinkingStyle,
+					activeTheme,
+				),
+			);
 
 			const hasVisibleContentAfter = message.content
 				.slice(i + 1)
@@ -733,6 +921,7 @@ function compactThinking(pi: ExtensionAPI) {
 		latestComponentTimestamp = undefined;
 		completedDurations.clear();
 		streamingComponents.clear();
+		expandedThinking.clear();
 		if (ctx.mode === "tui") ctx.ui.setWidget(WIDGET_ID, undefined);
 
 		if (patchInstalled) {