@huseynsnmz/pi-suite 0.0.0-stage → 0.1.0

package/packages/frame/src/render/paint.ts

@@ -0,0 +1,135 @@
+import type { Theme, ThemeColor } from "@earendil-works/pi-coding-agent";
+
+export type Role =
+	| "path"
+	| "branch"
+	| "ahead"
+	| "behind"
+	| "staged"
+	| "modified"
+	| "untracked"
+	| "conflicted"
+	| "host"
+	| "session"
+	| "board"
+	| "waiting"
+	| "queued"
+	| "version"
+	| "provider"
+	| "model"
+	| "sent"
+	| "received"
+	| "cache"
+	| "speed"
+	| "wait"
+	| "running"
+	| "done"
+	| "cost"
+	| "clock"
+	| "brand";
+
+export const ROLE_TOKENS: Record<Role, ThemeColor> = {
+	path: "accent",
+	branch: "syntaxFunction",
+	ahead: "success",
+	behind: "warning",
+	staged: "success",
+	modified: "warning",
+	untracked: "muted",
+	conflicted: "error",
+	host: "syntaxType",
+	session: "mdHeading",
+	board: "accent",
+	waiting: "warning",
+	queued: "muted",
+	version: "muted",
+	provider: "syntaxKeyword",
+	model: "text",
+	sent: "mdLink",
+	received: "success",
+	cache: "syntaxType",
+	speed: "success",
+	wait: "syntaxNumber",
+	running: "accent",
+	done: "success",
+	cost: "syntaxNumber",
+	clock: "muted",
+	brand: "accent",
+};
+
+const EFFORT_TOKENS: Record<string, ThemeColor> = {
+	off: "thinkingOff",
+	minimal: "thinkingMinimal",
+	low: "thinkingLow",
+	medium: "thinkingMedium",
+	high: "thinkingHigh",
+	xhigh: "thinkingXhigh",
+	max: "thinkingMax",
+};
+
+export const EFFORT_LEVELS = ["minimal", "low", "medium", "high", "xhigh", "max"] as const;
+
+export interface Paint {
+	dim(text: string): string;
+	muted(text: string): string;
+	text(text: string): string;
+	bold(text: string): string;
+	italic(text: string): string;
+	role(role: Role, text: string): string;
+	effort(level: string, text: string): string;
+	stress(percent: number, text: string): string;
+	money(amount: number, text: string): string;
+	frame(text: string): string;
+	/** A theme color a package named in a contract, like a peer's. */
+	raw(color: string, text: string): string;
+}
+
+export function themePaint(theme: Theme, frame?: (text: string) => string): Paint {
+	return {
+		dim: (text) => theme.fg("dim", text),
+		muted: (text) => theme.fg("muted", text),
+		text: (text) => theme.fg("text", text),
+		bold: (text) => theme.bold(text),
+		italic: (text) => theme.italic(text),
+		role: (role, text) => theme.fg(ROLE_TOKENS[role], text),
+		effort: (level, text) => theme.fg(EFFORT_TOKENS[level] ?? "thinkingOff", text),
+		stress: (percent, text) => theme.fg(stressToken(percent), text),
+		money: (amount, text) => theme.fg(moneyToken(amount), text),
+		frame: frame ?? ((text) => theme.fg("border", text)),
+		raw: (color, text) =>
+			theme.fg(
+				theme.colors[color as ThemeColor] === undefined ? "muted" : (color as ThemeColor),
+				text,
+			),
+	};
+}
+
+function stressToken(percent: number): ThemeColor {
+	if (percent >= 90) return "error";
+	if (percent >= 70) return "warning";
+	return "success";
+}
+
+function moneyToken(amount: number): ThemeColor {
+	if (amount >= 10) return "error";
+	if (amount >= 1) return "warning";
+	return ROLE_TOKENS.cost;
+}
+
+export const PLAIN: Paint = {
+	dim: identity,
+	muted: identity,
+	text: identity,
+	bold: identity,
+	italic: identity,
+	role: (_role, text) => text,
+	effort: (_level, text) => text,
+	stress: (_percent, text) => text,
+	money: (_amount, text) => text,
+	frame: identity,
+	raw: (_color, text) => text,
+};
+
+function identity(text: string): string {
+	return text;
+}

package/packages/frame/src/render/safe.ts

@@ -0,0 +1,19 @@
+// Text from a model, a repository or another program must not drive the terminal: no clipboard
+// writes, no moved cursor, no command that reads differently from what runs.
+
+// oxlint-disable-next-line no-control-regex -- escape sequences are what this removes
+const ESCAPE = /\u001b(?:\[[0-?]*[ -/]*[@-~]?|[\]PX^_][^\u0007\u001b\n]*(?:\u0007|\u001b\\)?|[ -/]*[0-~]?)/g;
+// oxlint-disable-next-line no-control-regex -- color codes are escape sequences
+const COLOR = /^\u001b\[[0-9;:]*m$/;
+/** Keeps tabs and line breaks. ESCAPE has already taken every escape character. */
+// oxlint-disable-next-line no-control-regex -- control characters are what this removes
+const CONTROLS = /[\u0000-\u0008\u000b-\u001a\u001c-\u001f\u007f-\u009f‪-‮⁦-⁩]/g;
+
+export function plainText(text: string): string {
+	return text.replace(ESCAPE, "").replace(CONTROLS, "");
+}
+
+/** For lines already painted: keeps the colors and nothing else that talks to the terminal. */
+export function styledText(text: string): string {
+	return text.replace(ESCAPE, (found) => (COLOR.test(found) ? found : "")).replace(CONTROLS, "");
+}