@xzzpig/pi-starline 0.1.0

package/extensions/starline/user-message.ts

@@ -0,0 +1,267 @@
+import { type Theme, type ThemeColor, UserMessageComponent } from "@earendil-works/pi-coding-agent";
+import {
+	Markdown,
+	type MarkdownTheme,
+	truncateToWidth,
+	visibleWidth,
+} from "@earendil-works/pi-tui";
+import type { PolishedTuiConfig } from "./config";
+import { installPrototypePatch } from "./prototype-patch-registry";
+import {
+	EDITOR_ACCENT_FALLBACK,
+	EDITOR_BORDER_FALLBACK,
+	renderStyleForSource,
+	renderStyleForSourceOrFallback,
+} from "./style";
+
+const OSC133_ZONE_START = "\x1b]133;A\x07";
+const OSC133_ZONE_END = "\x1b]133;B\x07";
+const OSC133_ZONE_FINAL = "\x1b]133;C\x07";
+
+type PatchableUserMessagePrototype = {
+	children?: unknown[];
+};
+
+type Cleanup = () => void;
+
+type UserMessageRenderCache = {
+	hasMarkdownText: boolean;
+	text?: string;
+	width?: number;
+	theme?: Theme;
+	configKey?: string;
+	renderedLines?: string[];
+};
+
+const userMessageRenderCache = new WeakMap<object, UserMessageRenderCache>();
+
+function isObject(value: unknown): value is object {
+	return (typeof value === "object" && value !== null) || typeof value === "function";
+}
+
+function isRecord(value: unknown): value is Record<string, unknown> {
+	return typeof value === "object" && value !== null && !Array.isArray(value);
+}
+
+function findMarkdownText(value: unknown): string | undefined {
+	if (!isRecord(value)) return undefined;
+	if (typeof value.text === "string") return value.text;
+
+	const children = value.children;
+	if (!Array.isArray(children)) return undefined;
+
+	for (const child of children) {
+		const text = findMarkdownText(child);
+		if (text !== undefined) return text;
+	}
+
+	return undefined;
+}
+
+function getCachedMarkdownText(instance: object): string | undefined {
+	const cached = userMessageRenderCache.get(instance);
+	if (cached?.hasMarkdownText) return cached.text;
+
+	const text = findMarkdownText(instance);
+	if (text !== undefined) {
+		userMessageRenderCache.set(instance, { ...cached, hasMarkdownText: true, text });
+	}
+	return text;
+}
+
+function getUserMessageConfigKey(config: PolishedTuiConfig): string {
+	return [
+		config.features.copyFriendly ? "copy" : "chrome",
+		config.colorSources.userMessages,
+		config.colors.editorAccent ?? "",
+		config.colors.editorBorder ?? "",
+		config.colors.userMessageBorder ?? "",
+		config.colors.userMessageText ?? "",
+		String(config.userMessagePaddingY),
+		config.icons.rail,
+	].join("\0");
+}
+
+function themeFg(theme: Theme | undefined, color: ThemeColor, text: string): string {
+	if (!theme) return text;
+	try {
+		return theme.fg(color, text);
+	} catch {
+		return text;
+	}
+}
+
+function makeMarkdownTheme(theme: Theme | undefined): MarkdownTheme {
+	return {
+		heading: (text) => themeFg(theme, "mdHeading", text),
+		link: (text) => themeFg(theme, "mdLink", text),
+		linkUrl: (text) => themeFg(theme, "mdLinkUrl", text),
+		code: (text) => themeFg(theme, "mdCode", text),
+		codeBlock: (text) => themeFg(theme, "mdCodeBlock", text),
+		codeBlockBorder: (text) => themeFg(theme, "mdCodeBlockBorder", text),
+		quote: (text) => themeFg(theme, "mdQuote", text),
+		quoteBorder: (text) => themeFg(theme, "mdQuoteBorder", text),
+		hr: (text) => themeFg(theme, "mdHr", text),
+		listBullet: (text) => themeFg(theme, "mdListBullet", text),
+		bold: (text) => (theme ? theme.bold(text) : text),
+		italic: (text) => (theme ? theme.italic(text) : text),
+		underline: (text) => (theme ? theme.underline(text) : text),
+		strikethrough: (text) => (theme ? theme.strikethrough(text) : text),
+	};
+}
+
+function fillLine(content: string, width: number): string {
+	const truncated = truncateToWidth(content, Math.max(0, width), "");
+	const pad = " ".repeat(Math.max(0, width - visibleWidth(truncated)));
+	return `${truncated}${pad}`;
+}
+
+function renderPromptBoxRail(theme: Theme | undefined, config: PolishedTuiConfig): string {
+	if (config.features.copyFriendly) return "";
+	const railGlyph = config.icons.rail;
+
+	return `${
+		theme
+			? renderStyleForSourceOrFallback(
+					theme,
+					config.colorSources.userMessages,
+					config.colors.editorAccent,
+					EDITOR_ACCENT_FALLBACK,
+					railGlyph,
+				)
+			: railGlyph
+	} `;
+}
+
+function renderPromptBoxLine(
+	line: string,
+	width: number,
+	theme: Theme | undefined,
+	config: PolishedTuiConfig,
+): string {
+	if (width <= 0) return "";
+	const rail = renderPromptBoxRail(theme, config);
+	const contentWidth = Math.max(0, width - visibleWidth(rail));
+	const content = config.features.copyFriendly
+		? truncateToWidth(line, contentWidth, "")
+		: fillLine(line, contentWidth);
+	return truncateToWidth(`${rail}${content}`, width, "");
+}
+
+function renderStarlineUserMessage(
+	instance: PatchableUserMessagePrototype,
+	width: number,
+	theme: Theme | undefined,
+	config: PolishedTuiConfig,
+): string[] | undefined {
+	if (!isRecord(instance)) return undefined;
+
+	const text = getCachedMarkdownText(instance);
+	if (text === undefined) return undefined;
+	if (width <= 0) return [""];
+
+	const configKey = getUserMessageConfigKey(config);
+	const cached = userMessageRenderCache.get(instance);
+	if (
+		cached?.hasMarkdownText &&
+		cached.width === width &&
+		cached.theme === theme &&
+		cached.configKey === configKey &&
+		cached.renderedLines
+	) {
+		return cached.renderedLines;
+	}
+
+	const railWidth = visibleWidth(renderPromptBoxRail(theme, config));
+	const contentWidth = Math.max(1, width - railWidth);
+	const renderer = new Markdown(text, 0, 0, makeMarkdownTheme(theme), {
+		color: (content) =>
+			theme && config.colors.userMessageText
+				? renderStyleForSource(
+						theme,
+						config.colorSources.userMessages,
+						config.colors.userMessageText,
+						content,
+					)
+				: themeFg(theme, "userMessageText", content),
+	});
+	const body = renderer.render(contentWidth);
+	const contentLines = body.length > 0 ? body : [""];
+	const border = theme
+		? renderStyleForSourceOrFallback(
+				theme,
+				config.colorSources.userMessages,
+				config.colors.userMessageBorder ?? config.colors.editorBorder,
+				EDITOR_BORDER_FALLBACK,
+				"─".repeat(width),
+			)
+		: "─".repeat(width);
+	const padRows =
+		config.userMessagePaddingY > 0 ? [renderPromptBoxLine("", width, theme, config)] : [];
+	const lines = [
+		truncateToWidth(border, width, ""),
+		...padRows,
+		...contentLines.map((line) => renderPromptBoxLine(line, width, theme, config)),
+		...padRows,
+		truncateToWidth(border, width, ""),
+	];
+
+	userMessageRenderCache.set(instance, {
+		hasMarkdownText: true,
+		text,
+		width,
+		theme,
+		configKey,
+		renderedLines: lines,
+	});
+	return lines;
+}
+
+function withPromptZoneMarkers(lines: string[]): string[] {
+	const markedLines = [...lines];
+	markedLines[0] = OSC133_ZONE_START + markedLines[0];
+	markedLines[markedLines.length - 1] =
+		OSC133_ZONE_END + OSC133_ZONE_FINAL + markedLines[markedLines.length - 1];
+	return markedLines;
+}
+
+export function installUserMessageStyle(
+	getTheme: () => Theme | undefined,
+	getConfig: () => PolishedTuiConfig,
+): Cleanup {
+	const prototype = UserMessageComponent.prototype;
+	const cleanupInvalidate = installPrototypePatch(
+		prototype,
+		"invalidate",
+		"user-message-invalidate",
+		({ predecessor, receiver, args }) => {
+			if (isObject(receiver)) userMessageRenderCache.delete(receiver);
+			return Reflect.apply(predecessor, receiver, args);
+		},
+	);
+	const cleanupRender = installPrototypePatch(
+		prototype,
+		"render",
+		"user-message-render",
+		({ predecessor, receiver, args }) => {
+			const width = args[0];
+			if (typeof width !== "number") return Reflect.apply(predecessor, receiver, args);
+			const lines = renderStarlineUserMessage(
+				receiver as PatchableUserMessagePrototype,
+				width,
+				getTheme(),
+				getConfig(),
+			);
+			if (!lines) return Reflect.apply(predecessor, receiver, args);
+			if (lines.length === 0) return lines;
+			return withPromptZoneMarkers(lines);
+		},
+	);
+	let cleaned = false;
+	return () => {
+		if (cleaned) return;
+		cleaned = true;
+		cleanupRender();
+		cleanupInvalidate();
+	};
+}

package/package.json

@@ -0,0 +1,60 @@
+{
+	"name": "@xzzpig/pi-starline",
+	"version": "0.1.0",
+	"description": "A Starship-inspired statusline and Opencode-style TUI for Pi.",
+	"type": "module",
+	"license": "MIT",
+	"repository": {
+		"type": "git",
+		"url": "git+https://github.com/Andy8647/pi-starline.git"
+	},
+	"homepage": "https://github.com/Andy8647/pi-starline#readme",
+	"bugs": {
+		"url": "https://github.com/Andy8647/pi-starline/issues"
+	},
+	"keywords": [
+		"pi-package",
+		"pi-extension",
+		"pi-theme",
+		"starship",
+		"tui"
+	],
+	"files": [
+		"extensions"
+	],
+	"scripts": {
+		"lint": "biome check .",
+		"typecheck": "tsc --noEmit",
+		"test": "vitest run",
+		"pi:dev": "${PI_BIN:-/opt/homebrew/bin/pi} --no-extensions -e ./extensions/starline/index.ts",
+		"pi:install-local": "${PI_BIN:-/opt/homebrew/bin/pi} install ./ -l",
+		"fmt": "biome format --write .",
+		"fmt:check": "biome format --check .",
+		"verify": "npm run lint && npm run typecheck && npm run test",
+		"pack:check": "npm pack --dry-run",
+		"prepublishOnly": "npm run verify && npm run pack:check"
+	},
+	"peerDependencies": {
+		"@earendil-works/pi-ai": ">=0.80.3",
+		"@earendil-works/pi-coding-agent": ">=0.80.3",
+		"@earendil-works/pi-tui": ">=0.80.3"
+	},
+	"pi": {
+		"extensions": [
+			"./extensions"
+		],
+		"image": "https://raw.githubusercontent.com/Andy8647/pi-starline/main/assets/starline.png"
+	},
+	"publishConfig": {
+		"access": "public"
+	},
+	"devDependencies": {
+		"@biomejs/biome": "^2.5.0",
+		"@earendil-works/pi-ai": "^0.82.1",
+		"@earendil-works/pi-coding-agent": "^0.84.1",
+		"@earendil-works/pi-tui": "^0.84.1",
+		"@types/node": "^26.0.0",
+		"typescript": "^7.0.2",
+		"vitest": "^4.1.8"
+	}
+}