@quandev104/pi-style 0.2.0 → 0.2.1

package/extension-src/pi-style/app/index.ts

@@ -96,6 +96,7 @@ export interface PiStyleApp {
 	update(
 		values: import("../domain/status.js").StatusSnapshot,
 		kind?: import("./render-scheduler.js").UpdateClass,
+		options?: import("./runtime.js").RuntimeUpdateOptions,
 	): void;
 }
 
@@ -268,10 +269,10 @@ export function createPiStyleApp(
 		sessionShutdown() {
 			runtime.stop();
 		},
-		update(values, kind = "coalesced") {
+		update(values, kind = "coalesced", options) {
 			const active = runtime.current;
 			if (!active) return;
-			active.update(values);
+			if (!active.update(values, options)) return;
 			active.scheduler.schedule(kind);
 		},
 		reload() {

package/extension-src/pi-style/app/runtime.ts

@@ -2,7 +2,7 @@ import type { ExtensionContext, ExtensionUIContext } from "@earendil-works/pi-co
 import { diffConfig } from "../domain/config-diff.js";
 import type { NormalizedPiStyleConfig } from "../domain/config-types.js";
 import type { GitCommandRunner } from "../domain/providers.js";
-import type { StatusSnapshot } from "../domain/status.js";
+import type { ContextSnapshot, StatusSnapshot } from "../domain/status.js";
 import { normalizeThinkingLevel } from "../domain/status.js";
 import { installEditor } from "../features/editor/index.js";
 import {
@@ -24,6 +24,11 @@ export interface RuntimeInstallationState {
 	readonly startup: "installed" | "disabled" | "failed";
 }
 
+export interface RuntimeUpdateOptions {
+	readonly refreshContextUsage?: boolean;
+	readonly refreshExtensionStatuses?: boolean;
+}
+
 export interface PiStyleRuntime {
 	generation: number;
 	readonly providerIdentity: { git: object; context: object; usage: object };
@@ -34,8 +39,8 @@ export interface PiStyleRuntime {
 	snapshot: UiSnapshot;
 	disposables: DisposableStore;
 	scheduler: RenderScheduler;
-	update(values: StatusSnapshot): void;
-	updateStartupResources(resources: StartupResources): void;
+	update(values: StatusSnapshot, options?: RuntimeUpdateOptions): boolean;
+	updateStartupResources(resources: StartupResources): boolean;
 	dismissStartup(): void;
 	invalidateGit(): void;
 	configure(config: NormalizedPiStyleConfig): void;
@@ -64,7 +69,7 @@ export interface RuntimeHost {
 export function createPiStyleRuntime(
 	host: RuntimeHost,
 	generation: number,
-	requestRender: () => void = () => {},
+	requestRender: () => void = host.requestRender ?? (() => {}),
 ): PiStyleRuntime {
 	let disposed = false;
 	const extensionStatuses = (): readonly import("../domain/status.js").ExtensionStatus[] | undefined => {
@@ -76,13 +81,22 @@ export function createPiStyleRuntime(
 			return [];
 		}
 	};
+	const contextSnapshot = (): ContextSnapshot | undefined => {
+		const usage = host.getContextUsage?.();
+		if (!usage) return undefined;
+		return {
+			...(usage.tokens !== null ? { currentTokens: usage.tokens } : {}),
+			windowTokens: usage.contextWindow,
+			...(usage.percent !== null ? { percent: usage.percent } : {}),
+		};
+	};
 	let currentConfig = host.config;
 	const disposables = new DisposableStore();
 	const scheduler = new RenderScheduler({ requestRender }, generation, () => !disposed);
 	const git = new CachedGitProvider(host.gitRunner);
 	const contextProvider = new InMemoryContextProvider();
 	const usageProvider = new InMemoryUsageProvider();
-	const usage = host.getContextUsage?.();
+	const initialContext = contextSnapshot();
 	const initialStatuses = extensionStatuses();
 	const initialValues = {
 		...(initialStatuses ? { extensionStatuses: initialStatuses } : {}),
@@ -91,15 +105,7 @@ export function createPiStyleRuntime(
 		...(host.model?.reasoning !== undefined ? { reasoning: host.model.reasoning } : {}),
 		...(host.thinkingLevel ? { thinkingLevel: normalizeThinkingLevel(host.thinkingLevel) } : {}),
 		...(host.cwd ? { cwd: host.cwd } : {}),
-		...(usage
-			? {
-					context: {
-						...(usage.tokens !== null ? { currentTokens: usage.tokens } : {}),
-						windowTokens: usage.contextWindow,
-						...(usage.percent !== null ? { percent: usage.percent } : {}),
-					},
-				}
-			: {}),
+		...(initialContext ? { context: initialContext } : {}),
 	};
 	let currentSnapshot = createSnapshot(generation, 0, initialValues);
 	let statusLine: ReturnType<typeof installStatusLine> | undefined;
@@ -110,14 +116,41 @@ export function createPiStyleRuntime(
 		editor: "disabled",
 		startup: "disabled",
 	};
-	const startupSnapshot = (config: NormalizedPiStyleConfig): StartupSnapshot => ({
+	const snapshotValues = (snapshot: UiSnapshot): StatusSnapshot => {
+		const { generation: _generation, revision: _revision, ...values } = snapshot;
+		return values;
+	};
+	const withSnapshotPatch = (
+		patch: Partial<Record<keyof StatusSnapshot, StatusSnapshot[keyof StatusSnapshot] | undefined>>,
+	): StatusSnapshot => {
+		const next = { ...snapshotValues(currentSnapshot) } as Record<string, unknown>;
+		for (const [key, value] of Object.entries(patch)) {
+			if (value === undefined) delete next[key];
+			else next[key] = value;
+		}
+		return next as StatusSnapshot;
+	};
+	const createStartupSnapshot = (
+		config: NormalizedPiStyleConfig,
+		resources: StartupResources | undefined = host.resources,
+	): StartupSnapshot => ({
 		...currentSnapshot,
 		reason: host.startupReason ?? "startup",
 		...(host.provider ? { startupProvider: host.provider } : {}),
 		...(host.cwd ? { project: host.cwd.split(/[\\/]/).filter(Boolean).at(-1) } : {}),
 		preset: config.preset,
-		...(host.resources ? { resources: host.resources } : {}),
+		...(resources ? { resources } : {}),
 	});
+	const applySnapshot = (nextSnapshot: UiSnapshot): boolean => {
+		if (nextSnapshot === currentSnapshot) return false;
+		currentSnapshot = nextSnapshot;
+		statusLine?.update(currentSnapshot);
+		editor?.update(currentSnapshot);
+		startup?.update(createStartupSnapshot(currentConfig));
+		return true;
+	};
+	const updateSnapshot = (values: StatusSnapshot): boolean =>
+		applySnapshot(replaceSnapshot(currentSnapshot, generation, values));
 	const installStatus = () => {
 		if (!host.hasUI || host.mode !== "tui" || !host.ui || !currentConfig.enabled || !currentConfig.statusLine.enabled) {
 			installationState = { ...installationState, status: "disabled" };
@@ -186,7 +219,7 @@ export function createPiStyleRuntime(
 			startup = installStartup({
 				host: { ...(host.ui as unknown as StartupHost), mode: host.mode, hasUI: host.hasUI },
 				config: currentConfig,
-				snapshot: startupSnapshot(currentConfig),
+				snapshot: createStartupSnapshot(currentConfig),
 				generation,
 				requestRender,
 				timeoutMs: 3000,
@@ -235,27 +268,10 @@ export function createPiStyleRuntime(
 	if (host.cwd && currentConfig.enabled && currentConfig.statusLine.enabled) {
 		void git.get(host.cwd).then((value) => {
 			if (disposed) return;
-			currentSnapshot = replaceSnapshot(currentSnapshot, generation, { ...currentSnapshot, git: value });
-			statusLine?.update(currentSnapshot);
-			editor?.update(currentSnapshot);
-			startup?.update({
-				...currentSnapshot,
-				reason: host.startupReason ?? "startup",
-				...(host.provider ? { startupProvider: host.provider } : {}),
-				...(host.cwd ? { project: host.cwd.split(/[\\/]/).filter(Boolean).at(-1) } : {}),
-				preset: currentConfig.preset,
-				...(host.resources ? { resources: host.resources } : {}),
-			});
-			requestRender();
-		});
-	}
-	if (usage) {
-		contextProvider.set("active", {
-			...(usage.tokens !== null ? { currentTokens: usage.tokens } : {}),
-			windowTokens: usage.contextWindow,
-			...(usage.percent !== null ? { percent: usage.percent } : {}),
+			if (updateSnapshot(withSnapshotPatch({ git: value }))) requestRender();
 		});
 	}
+	if (initialContext) contextProvider.set("active", initialContext);
 	return {
 		generation,
 		providerIdentity: { git, context: contextProvider, usage: usageProvider },
@@ -264,51 +280,33 @@ export function createPiStyleRuntime(
 		},
 		mode: host.mode,
 		hasUI: host.hasUI,
-		snapshot: currentSnapshot,
+		get snapshot() {
+			return currentSnapshot;
+		},
 		disposables,
 		scheduler,
 		updateStartupResources(resources) {
-			if (disposed) return;
-			startup?.update({
-				...currentSnapshot,
-				reason: host.startupReason ?? "startup",
-				...(host.provider ? { startupProvider: host.provider } : {}),
-				...(host.cwd ? { project: host.cwd.split(/[\\/]/).filter(Boolean).at(-1) } : {}),
-				preset: currentConfig.preset,
-				resources,
-			});
+			if (disposed || !startup) return false;
+			startup.update(createStartupSnapshot(currentConfig, resources));
 			requestRender();
+			return true;
 		},
 		dismissStartup() {
 			startup?.dismiss();
 		},
-		update(values) {
-			if (disposed) return;
-			const liveUsage = host.getContextUsage?.();
-			const context = liveUsage
-				? {
-						...(liveUsage.tokens !== null ? { currentTokens: liveUsage.tokens } : {}),
-						windowTokens: liveUsage.contextWindow,
-						...(liveUsage.percent !== null ? { percent: liveUsage.percent } : {}),
-					}
-				: undefined;
-			const statuses = extensionStatuses();
-			currentSnapshot = replaceSnapshot(currentSnapshot, generation, {
-				...currentSnapshot,
-				...values,
-				...(context ? { context } : {}),
-				...(statuses ? { extensionStatuses: statuses } : {}),
-			});
-			statusLine?.update(currentSnapshot);
-			editor?.update(currentSnapshot);
-			startup?.update({
-				...currentSnapshot,
-				reason: host.startupReason ?? "startup",
-				...(host.provider ? { startupProvider: host.provider } : {}),
-				...(host.cwd ? { project: host.cwd.split(/[\\/]/).filter(Boolean).at(-1) } : {}),
-				preset: currentConfig.preset,
-				...(host.resources ? { resources: host.resources } : {}),
-			});
+		update(values, options = {}) {
+			if (disposed) return false;
+			const patch: Partial<Record<keyof StatusSnapshot, StatusSnapshot[keyof StatusSnapshot] | undefined>> = {
+				...(values as Partial<Record<keyof StatusSnapshot, StatusSnapshot[keyof StatusSnapshot] | undefined>>),
+			};
+			if (options.refreshContextUsage) {
+				const context = contextSnapshot();
+				patch.context = context;
+				if (context) contextProvider.set("active", context);
+				else contextProvider.clear();
+			}
+			if (options.refreshExtensionStatuses) patch.extensionStatuses = extensionStatuses();
+			return updateSnapshot(withSnapshotPatch(patch));
 		},
 		configure(nextConfig) {
 			if (disposed) return;
@@ -350,17 +348,7 @@ export function createPiStyleRuntime(
 			git.invalidate(host.cwd);
 			void git.get(host.cwd).then((value) => {
 				if (disposed) return;
-				currentSnapshot = replaceSnapshot(currentSnapshot, generation, { ...currentSnapshot, git: value });
-				statusLine?.update(currentSnapshot);
-				editor?.update(currentSnapshot);
-				startup?.update({
-					...currentSnapshot,
-					reason: host.startupReason ?? "startup",
-					...(host.provider ? { startupProvider: host.provider } : {}),
-					...(host.cwd ? { project: host.cwd.split(/[\\/]/).filter(Boolean).at(-1) } : {}),
-					preset: currentConfig.preset,
-				});
-				requestRender();
+				if (updateSnapshot(withSnapshotPatch({ git: value }))) requestRender();
 			});
 		},
 		get disposed() {

package/extension-src/pi-style/app/snapshot.ts

@@ -6,10 +6,49 @@ export interface UiSnapshot extends StatusSnapshot {
 }
 
 export function createSnapshot(generation: number, revision = 0, values: StatusSnapshot = {}): UiSnapshot {
-	return Object.freeze({ generation, revision, ...values });
+	return Object.freeze({ ...values, generation, revision });
 }
 
 export function replaceSnapshot(current: UiSnapshot, generation: number, values: StatusSnapshot): UiSnapshot {
 	if (current.generation !== generation) return current;
-	return createSnapshot(generation, current.revision + 1, values);
+	return equalStatusSnapshot(current, values) ? current : createSnapshot(generation, current.revision + 1, values);
+}
+
+function equalStatusSnapshot(current: StatusSnapshot, next: StatusSnapshot): boolean {
+	const currentEntries = Object.entries(current).filter(([key]) => key !== "generation" && key !== "revision");
+	const nextEntries = Object.entries(next).filter(([key]) => key !== "generation" && key !== "revision");
+	if (currentEntries.length !== nextEntries.length) return false;
+	for (const [key, value] of nextEntries) {
+		if (!hasOwn(current, key)) return false;
+		if (!equalValue((current as Record<string, unknown>)[key], value)) return false;
+	}
+	return true;
+}
+
+function equalValue(left: unknown, right: unknown): boolean {
+	if (Object.is(left, right)) return true;
+	if (Array.isArray(left) && Array.isArray(right)) {
+		if (left.length !== right.length) return false;
+		for (let index = 0; index < left.length; index++) {
+			if (!equalValue(left[index], right[index])) return false;
+		}
+		return true;
+	}
+	if (!isPlainRecord(left) || !isPlainRecord(right)) return false;
+	const leftKeys = Object.keys(left);
+	const rightKeys = Object.keys(right);
+	if (leftKeys.length !== rightKeys.length) return false;
+	for (const key of rightKeys) {
+		if (!hasOwn(left, key)) return false;
+		if (!equalValue(left[key], right[key])) return false;
+	}
+	return true;
+}
+
+function hasOwn(value: object, key: string): boolean {
+	return Object.hasOwn(value, key);
+}
+
+function isPlainRecord(value: unknown): value is Record<string, unknown> {
+	return typeof value === "object" && value !== null && !Array.isArray(value);
 }

package/extension-src/pi-style/features/editor/index.ts

@@ -39,6 +39,19 @@ interface EditorOptions {
 	onSnapshot: (snapshot: StatusSnapshot) => void;
 }
 
+interface RenderPlan {
+	readonly style: "compact" | "boxed" | "dock" | "native";
+	readonly kind: "compact" | "boxed" | "outline" | "rounded" | "native";
+	readonly prompt: string;
+	readonly promptWidth: number;
+	readonly padding: number;
+	readonly sideReserve: number;
+	readonly renderWidth: number;
+	readonly innerWidth: number;
+	readonly prefix: string;
+	readonly continuation: string;
+}
+
 const widthOf = visibleWidth;
 
 function widthSafe(value: string, width: number): string {
@@ -130,6 +143,7 @@ export class StyledEditor extends CustomEditor implements EditorComponent {
 	private readonly onSnapshot: (snapshot: StatusSnapshot) => void;
 	private semantic: ResolvedTheme;
 	private disposed = false;
+	private renderPlanCache: { key: string; plan: RenderPlan } | undefined;
 
 	constructor(tui: Tui, theme: PiEditorTheme, keybindings: Keybindings, options: EditorOptions) {
 		super(tui, theme, keybindings);
@@ -164,88 +178,37 @@ export class StyledEditor extends CustomEditor implements EditorComponent {
 	override invalidate(): void {
 		super.invalidate();
 		this.semantic = semanticTheme(this.piTheme, this.config);
+		this.renderPlanCache = undefined;
 		this.tui.requestRender();
 	}
 
 	override render(width: number): string[] {
 		if (width <= 0) return [];
-		const nativeLines = super.render(width);
-		const style = this.styleFor(width);
-		// Autocomplete (slash menu / @-mentions) restructure: Pi draws the
-		// suggestions after its own bottom border, which pushes the below-editor
-		// widgets (status line) down. Re-frame the native output so the dropdown
-		// lives INSIDE the input box, keeping the footer directly below the input.
-		// Native layout: [top border, text lines, bottom border, dropdown lines...].
-		if ((this as unknown as { autocompleteState?: unknown }).autocompleteState) {
-			const prompt = this.prompt();
-			const padding = this.paddingFor(width, style);
-			const promptWidth = widthOf(prompt) + 1;
-			const prefix = `${" ".repeat(padding)}${prompt} `;
-			const continuation = " ".repeat(padding + promptWidth);
-			const borderIndex = nativeLines.slice(1).findIndex((line) => isNativeBorderLine(line));
-			const split = borderIndex >= 0 ? borderIndex + 1 : nativeLines.length;
-			const body = nativeLines.slice(1, split);
-			const dropdown = nativeLines.slice(split);
-			const border = this.borderFor();
-			const kind = this.frameKind(style);
-			const renderWidth = width - (kind === "rounded" ? 2 : 0);
-			const sideColor = kind === "rounded" ? this.borderColorFor() : undefined;
-			const wrap = (line: string) =>
-				kind === "rounded" && sideColor ? `${sideColor("│")}${line}${sideColor("│")}` : line;
-			const bashHidden = this.bashHiddenCount();
-			const renderedBody = body.map((line, index) => {
-				const source = index === 0 && bashHidden > 0 ? stripLeadingVisibleChars(line, bashHidden) : line;
-				return wrap(widthSafe(`${index === 0 ? prefix : continuation}${source}`, renderWidth));
-			});
-			const dropdownLines = dropdown.map((line) => wrap(widthSafe(line, renderWidth)));
-			if (kind === "rounded") {
-				return [
-					border(`╭${"─".repeat(Math.max(0, width - 2))}╮`),
-					...renderedBody,
-					...dropdownLines,
-					border(`╰${"─".repeat(Math.max(0, width - 2))}╯`),
-				];
-			}
-			return [border("─".repeat(width)), ...renderedBody, ...dropdownLines, border("─".repeat(width))];
-		}
-		if (style === "native") return nativeLines.map((line) => widthSafe(line, width));
+		const plan = this.renderPlan(width);
+		const autocompleteState = (this as unknown as { autocompleteState?: unknown }).autocompleteState;
+		if (plan.style === "native") return super.render(width).map((line) => widthSafe(line, width));
+		if (autocompleteState) return this.renderAutocompleteFrame(width, plan);
 
-		const prompt = this.prompt();
-		const promptWidth = widthOf(prompt) + 1;
-		const padding = this.paddingFor(width, style);
-		const kind = this.frameKind(style);
-		const sideReserve = kind === "rounded" ? 2 : 0;
-		const renderWidth = Math.max(1, width - sideReserve);
-		const innerWidth = Math.max(1, renderWidth - promptWidth - padding * 2);
-		const innerLines = super.render(innerWidth);
+		const innerLines = super.render(plan.innerWidth);
 		if (innerLines.length === 0) return [];
-
 		const body = innerLines.slice(1, -1);
-		const prefix = `${" ".repeat(padding)}${prompt} `;
-		const continuation = " ".repeat(padding + promptWidth);
 		const hint = this.config.editor.hint;
 		const showHint = hint !== "" && this.getText() === "";
 		const bashHidden = this.bashHiddenCount();
 		const renderedBody = body.map((line, index) => {
-			const lead = index === 0 ? prefix : continuation;
+			const lead = index === 0 ? plan.prefix : plan.continuation;
 			const source = index === 0 && bashHidden > 0 ? stripLeadingVisibleChars(line, bashHidden) : line;
 			let content = `${lead}${source}`;
-			// Empty-input hint: the cursor block (first cell of the native empty
-			// line) stays at the input position, the dim hint trails it. The native
-			// line is pre-padded to renderWidth with literal spaces; drop them from
-			// the raw end (safe: no ANSI follows the padding) before appending the
-			// hint, or the hint is truncated away by widthSafe. Typing any character
-			// makes the text non-empty and the hint disappears.
 			if (showHint && index === 0 && line) {
 				let end = content.length;
 				while (end > 0 && content[end - 1] === " ") end--;
 				if (end < content.length) content = content.slice(0, end);
 				content += this.semantic.apply("hint", hint);
 			}
-			return widthSafe(content, renderWidth);
+			return widthSafe(content, plan.renderWidth);
 		});
-		const metadata = this.metadata(width, style);
-		const framed = this.frame(width, style, renderedBody, metadata);
+		const metadata = this.metadata(width, plan.style);
+		const framed = this.frame(width, plan.style, renderedBody, metadata);
 		return framed.map((line) => widthSafe(line, width));
 	}
 
@@ -255,6 +218,62 @@ export class StyledEditor extends CustomEditor implements EditorComponent {
 		this.invalidate();
 	}
 
+	private renderPlan(width: number): RenderPlan {
+		const style = this.styleFor(width);
+		const prompt = this.prompt();
+		const kind = this.frameKind(style);
+		const key = `${width}:${style}:${kind}:${prompt}`;
+		if (this.renderPlanCache?.key === key) return this.renderPlanCache.plan;
+		const promptWidth = widthOf(prompt) + 1;
+		const padding = this.paddingFor(width, style);
+		const sideReserve = kind === "rounded" ? 2 : 0;
+		const renderWidth = Math.max(1, width - sideReserve);
+		const innerWidth = Math.max(1, renderWidth - promptWidth - padding * 2);
+		const prefix = `${" ".repeat(padding)}${prompt} `;
+		const continuation = " ".repeat(padding + promptWidth);
+		const plan = {
+			style,
+			kind,
+			prompt,
+			promptWidth,
+			padding,
+			sideReserve,
+			renderWidth,
+			innerWidth,
+			prefix,
+			continuation,
+		};
+		this.renderPlanCache = { key, plan };
+		return plan;
+	}
+
+	private renderAutocompleteFrame(width: number, plan: RenderPlan): string[] {
+		const nativeLines = super.render(width);
+		const borderIndex = nativeLines.slice(1).findIndex((line) => isNativeBorderLine(line));
+		const split = borderIndex >= 0 ? borderIndex + 1 : nativeLines.length;
+		const body = nativeLines.slice(1, split);
+		const dropdown = nativeLines.slice(split);
+		const border = this.borderFor();
+		const sideColor = plan.kind === "rounded" ? this.borderColorFor() : undefined;
+		const wrap = (line: string) =>
+			plan.kind === "rounded" && sideColor ? `${sideColor("│")}${line}${sideColor("│")}` : line;
+		const bashHidden = this.bashHiddenCount();
+		const renderedBody = body.map((line, index) => {
+			const source = index === 0 && bashHidden > 0 ? stripLeadingVisibleChars(line, bashHidden) : line;
+			return wrap(widthSafe(`${index === 0 ? plan.prefix : plan.continuation}${source}`, plan.renderWidth));
+		});
+		const dropdownLines = dropdown.map((line) => wrap(widthSafe(line, plan.renderWidth)));
+		if (plan.kind === "rounded") {
+			return [
+				border(`╭${"─".repeat(Math.max(0, width - 2))}╮`),
+				...renderedBody,
+				...dropdownLines,
+				border(`╰${"─".repeat(Math.max(0, width - 2))}╯`),
+			];
+		}
+		return [border("─".repeat(width)), ...renderedBody, ...dropdownLines, border("─".repeat(width))];
+	}
+
 	private prompt(): string {
 		if (this.isBashMode()) {
 			// Bash mode (`!` prefix): the prompt glyph becomes the bash icon and the