@rosetears/aili-pi 0.1.0 → 0.1.1

package/extensions/zentui/extension-status.ts

@@ -0,0 +1,96 @@
+import { stripVTControlCharacters } from "node:util";
+import type {
+	ExtensionStatusColorMode,
+	ExtensionStatusPlacement,
+	PolishedTuiConfig,
+} from "./config.js";
+import { getExtensionStatusColorMode, getExtensionStatusPlacement } from "./config.js";
+
+export type ExtensionStatusSegment = {
+	key: string;
+	text: string;
+	placement: ExtensionStatusPlacement;
+	colorMode: ExtensionStatusColorMode;
+};
+
+export type ExtensionStatusSegmentsByPlacement = {
+	left: ExtensionStatusSegment[];
+	middle: ExtensionStatusSegment[];
+	right: ExtensionStatusSegment[];
+};
+
+const safeSgrPattern = /\x1b\[[0-9;:]*m/g;
+const sgrPlaceholderPattern = /__ZENTUI_SGR_(\d+)__/g;
+
+function compareKeys(a: ExtensionStatusSegment, b: ExtensionStatusSegment): number {
+	return a.key < b.key ? -1 : a.key > b.key ? 1 : 0;
+}
+
+function normalizeStatusWhitespace(value: string): string {
+	return value
+		.replace(/[\r\n\t\f\v]+/g, " ")
+		.replace(/[\u0000-\u001f\u007f-\u009f]/g, "")
+		.replace(/\s+/g, " ")
+		.trim();
+}
+
+export function sanitizeExtensionStatusText(value: string): string {
+	return normalizeStatusWhitespace(stripVTControlCharacters(value));
+}
+
+function hasVisibleStatusText(value: string): boolean {
+	return sanitizeExtensionStatusText(value).length > 0;
+}
+
+function formatQuotaWindowLabel(key: string, text: string): string {
+	// pi-quota-status uses the compact upstream label `Wk`. Rem Cyberdeck
+	// presents the same weekly Codex window as the more explicit `7d`.
+	return key === "pi-quota-status" ? text.replace(/\bWk\b/g, "7d") : text;
+}
+
+export function sanitizeExtensionStatusOriginalText(value: string): string {
+	const safeSequences: string[] = [];
+	const protectedValue = value.replace(safeSgrPattern, (sequence) => {
+		const index = safeSequences.push(sequence) - 1;
+		return `__ZENTUI_SGR_${index}__`;
+	});
+	const cleaned = normalizeStatusWhitespace(stripVTControlCharacters(protectedValue));
+	const restored = cleaned.replace(sgrPlaceholderPattern, (_match, indexText: string) => {
+		const index = Number.parseInt(indexText, 10);
+		return safeSequences[index] ?? "";
+	});
+
+	return hasVisibleStatusText(restored) ? restored : "";
+}
+
+export function collectExtensionStatusSegments(
+	statuses: ReadonlyMap<string, string>,
+	config: PolishedTuiConfig,
+): ExtensionStatusSegmentsByPlacement {
+	const segments: ExtensionStatusSegmentsByPlacement = {
+		left: [],
+		middle: [],
+		right: [],
+	};
+
+	for (const [key, value] of statuses.entries()) {
+		const placement = getExtensionStatusPlacement(config, key);
+		if (placement === "off") continue;
+
+		const colorMode = getExtensionStatusColorMode(config, key);
+		const text = formatQuotaWindowLabel(
+			key,
+			colorMode === "original"
+				? sanitizeExtensionStatusOriginalText(value)
+				: sanitizeExtensionStatusText(value),
+		);
+		if (!text) continue;
+
+		segments[placement].push({ key, text, placement, colorMode });
+	}
+
+	segments.left.sort(compareKeys);
+	segments.middle.sort(compareKeys);
+	segments.right.sort(compareKeys);
+	return segments;
+}

package/extensions/zentui/fixed-editor/cluster.ts

@@ -0,0 +1,98 @@
+/**
+ * Cluster rendering for the fixed editor.
+ *
+ * Pi-specific cluster discovery and validation live in pi-compat.ts. This module
+ * only renders the already-verified pinned components.
+ *
+ * @internal
+ */
+
+import { CURSOR_MARKER, truncateToWidth, visibleWidth } from "@earendil-works/pi-tui";
+
+import type { PiFixedCluster, PiRenderableCapability } from "./pi-compat.js";
+import type { ClusterRender } from "./types.js";
+
+export type FixedCluster = PiFixedCluster;
+
+function renderComponent(component: PiRenderableCapability | null, width: number): string[] {
+	if (!component) return [];
+	const lines = component.render.call(component.target, width);
+	// Strip only trailing blank lines — internal blank lines (e.g. editor
+	// padding in copy-friendly mode) must be preserved.
+	let end = lines.length;
+	while (end > 0 && visibleWidth(lines[end - 1]) === 0) end--;
+	return lines.slice(0, Math.max(end, 1));
+}
+
+/**
+ * Cap editor lines to `maxLines`, keeping the cursor row visible.
+ * If the cursor marker is found, the window centers on it; otherwise
+ * the last `maxLines` are kept.
+ */
+export function capEditorLines(lines: string[], maxLines: number): string[] {
+	if (maxLines <= 0) return [];
+	if (lines.length <= maxLines) return lines;
+
+	const cursorRow = lines.findIndex((line) => line.includes(CURSOR_MARKER));
+	if (cursorRow !== -1) {
+		const start = Math.max(0, Math.min(cursorRow - maxLines + 1, lines.length - maxLines));
+		return lines.slice(start, start + maxLines);
+	}
+	return lines.slice(lines.length - maxLines);
+}
+
+function sanitizeLines(lines: string[], width: number): string[] {
+	return lines.map((line) =>
+		visibleWidth(line) > width ? truncateToWidth(line, width, "", true) : line,
+	);
+}
+
+/**
+ * Render the full cluster (status + widgets + editor + footer) and extract
+ * the cursor position from the CURSOR_MARKER.
+ */
+export function renderCluster(
+	cluster: FixedCluster,
+	width: number,
+	maxHeight: number,
+): ClusterRender {
+	const w = Math.max(1, width);
+	const maxRows = Math.max(1, maxHeight - 1);
+
+	const statusLines = sanitizeLines(renderComponent(cluster.status, w), w);
+	const aboveLines = sanitizeLines(renderComponent(cluster.aboveWidget, w), w);
+	const editorSource = sanitizeLines(renderComponent(cluster.editor, w), w);
+	const belowLines = sanitizeLines(renderComponent(cluster.belowWidget, w), w);
+	const footerLines = sanitizeLines(renderComponent(cluster.footer, w), w);
+
+	const editorLines = capEditorLines(editorSource, maxRows);
+	let remaining = maxRows - editorLines.length;
+
+	const footer = footerLines.slice(-remaining);
+	remaining -= footer.length;
+
+	const below = belowLines.slice(-remaining);
+	remaining -= below.length;
+
+	const above = aboveLines.slice(-remaining);
+	remaining -= above.length;
+
+	const status = statusLines.slice(-remaining);
+
+	let allLines = [...status, ...above, ...editorLines, ...below, ...footer];
+
+	// Strip leading blank lines (e.g. empty status line above the editor border).
+	let start = 0;
+	while (start < allLines.length - 1 && visibleWidth(allLines[start]) === 0) start++;
+	allLines = allLines.slice(start);
+
+	let cursor: { row: number; col: number } | null = null;
+	const cleaned = allLines.map((line, row) => {
+		const markerIndex = line.indexOf(CURSOR_MARKER);
+		if (markerIndex === -1) return line;
+		cursor ??= { row, col: visibleWidth(line.slice(0, markerIndex)) };
+		return line.slice(0, markerIndex) + line.slice(markerIndex + CURSOR_MARKER.length);
+	});
+
+	return { lines: cleaned, cursor };
+}