@ceo.paludetto/pi-starship 0.0.0

package/package.json

@@ -0,0 +1,61 @@
+{
+	"name": "@ceo.paludetto/pi-starship",
+	"type": "module",
+	"version": "0.0.0",
+	"description": "Starship-style status bar for pi with mode, model, tokens and path segments",
+	"pi": {
+		"extensions": [
+			"./dist/index.mjs"
+		]
+	},
+	"author": {
+		"name": "Carlos Paludetto",
+		"email": "ceo.paludetto@gmail.com"
+	},
+	"license": "MIT",
+	"repository": {
+		"type": "git",
+		"url": "https://github.com/ceopaludetto/pi",
+		"directory": "packages/pi-starship"
+	},
+	"sideEffects": false,
+	"publishConfig": {
+		"access": "public",
+		"registry": "https://registry.npmjs.org/"
+	},
+	"exports": {
+		".": {
+			"import": "./dist/index.mjs",
+			"require": "./dist/index.cjs"
+		},
+		"./package.json": "./package.json"
+	},
+	"main": "./dist/index.cjs",
+	"module": "./dist/index.mjs",
+	"types": "./dist/index.d.cts",
+	"scripts": {
+		"build": "tsdown",
+		"test": "bun test"
+	},
+	"peerDependencies": {
+		"@earendil-works/pi-ai": ">=0.79.0",
+		"@earendil-works/pi-coding-agent": ">=0.79.0",
+		"@earendil-works/pi-tui": ">=0.79.0",
+		"@gotgenes/pi-permission-system": ">=16.0.0",
+		"pi-mcp-adapter": ">=2.0.0"
+	},
+	"peerDependenciesMeta": {
+		"@gotgenes/pi-permission-system": {
+		"optional": true
+	},
+	"pi-mcp-adapter": {
+	"optional": true
+}
+},
+"devDependencies": {
+	"@ceo.paludetto/pi-utilities": "0.0.0",
+	"@earendil-works/pi-ai": "^0.80.2",
+	"@earendil-works/pi-coding-agent": "^0.80.2",
+	"@earendil-works/pi-tui": "^0.80.2"
+}
+}

package/src/index.ts

@@ -0,0 +1,102 @@
+import type { Module, ModuleContext } from "./utilities/types";
+import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
+
+import { dim } from "@ceo.paludetto/pi-utilities";
+import { truncateToWidth, visibleWidth } from "@earendil-works/pi-tui";
+
+import { McpModule } from "./modules/mcp.module";
+import { createPhaseTracker, ModeModule } from "./modules/mode.module";
+import { ModelModule } from "./modules/model.module";
+import { StarshipModule } from "./modules/starship.module";
+import { TokensModule } from "./modules/tokens.module";
+import { YoloModule } from "./modules/yolo.module";
+
+const MODULE_SEPARATOR = dim(" ⋅ ");
+const HORIZONTAL_PADDING = " ";
+
+export default function StarshipExtension(pi: ExtensionAPI) {
+	let thinkingLevel = "off";
+	let lastRenderWidth = 120;
+	let requestRender: (() => void) | undefined;
+	let refreshAll: (() => Promise<void>) | undefined;
+	let extensionStatuses: ReadonlyMap<string, string> = new Map();
+
+	const phaseTracker = createPhaseTracker(pi, () => requestRender?.());
+
+	pi.on("session_start", async (_, context) => {
+		thinkingLevel = pi.getThinkingLevel();
+
+		const moduleContext: ModuleContext = {
+			...context,
+			getStatuses: () => extensionStatuses,
+			getThinkingLevel: () => thinkingLevel,
+			getWidth: () => lastRenderWidth,
+		};
+
+		const leftModules: Module[] = [
+			new StarshipModule(moduleContext),
+			new McpModule(moduleContext),
+		];
+
+		const rightModules: Module[] = [
+			new YoloModule(moduleContext),
+			new ModeModule(moduleContext, phaseTracker),
+			new ModelModule(moduleContext),
+			new TokensModule(moduleContext),
+		];
+
+		const allModules = [...leftModules, ...rightModules];
+
+		refreshAll = async () => {
+			await Promise.all(allModules.map((item) => item.refresh()));
+			requestRender?.();
+		};
+
+		await refreshAll();
+
+		context.ui.setFooter((tui, _, data) => {
+			requestRender = () => tui.requestRender();
+			const unsubscribe = data.onBranchChange(() => refreshAll?.());
+
+			return {
+				dispose: unsubscribe,
+				invalidate() {},
+
+				render(width: number): string[] {
+					extensionStatuses = data.getExtensionStatuses();
+					phaseTracker.query();
+
+					if (width !== lastRenderWidth) {
+						lastRenderWidth = width;
+						for (const item of leftModules)
+							item.refresh();
+					}
+
+					const contentWidth = width - 2;
+
+					const left = leftModules
+						.map((item) => item.render())
+						.filter((value): value is string => value !== null)
+						.join(MODULE_SEPARATOR);
+
+					const right = rightModules
+						.map((item) => item.render())
+						.filter((value): value is string => value !== null)
+						.join(MODULE_SEPARATOR);
+
+					const gap = " ".repeat(Math.max(1, contentWidth - visibleWidth(left) - visibleWidth(right)));
+					return [HORIZONTAL_PADDING + truncateToWidth(left + gap + right, contentWidth) + HORIZONTAL_PADDING];
+				},
+			};
+		});
+	});
+
+	pi.on("agent_end", () => {
+		refreshAll?.();
+	});
+
+	pi.on("thinking_level_select", (event, _) => {
+		thinkingLevel = event.level;
+		requestRender?.();
+	});
+}

package/src/modules/mcp.module.ts

@@ -0,0 +1,23 @@
+import { cyan, dim } from "@ceo.paludetto/pi-utilities";
+
+import { Module } from "~/utilities/types";
+
+/** Status key published by pi-mcp-adapter (e.g. "MCP: 2/3 servers"). */
+const MCP_STATUS_KEY = "mcp";
+const SERVER_RATIO_PATTERN = /(\d+)\/(\d+)/;
+
+export class McpModule extends Module {
+	public override render(): string | null {
+		const status = this.getStatuses().get(MCP_STATUS_KEY);
+
+		if (!status)
+			return null;
+
+		const match = SERVER_RATIO_PATTERN.exec(status);
+
+		if (!match)
+			return null;
+
+		return `${cyan("MCP")} ${dim("(")}${match[1]}/${match[2]}${dim(")")}`;
+	}
+}

package/src/modules/mode.module.ts

@@ -0,0 +1,67 @@
+import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
+import type { ModuleContext } from "~/utilities/types";
+
+import { green, yellow } from "@ceo.paludetto/pi-utilities";
+
+import { Module } from "~/utilities/types";
+
+const REQUEST_CHANNEL = "plannotator:request";
+
+type PlannotatorPhase = "idle" | "planning" | "executing";
+
+export type PhaseTracker = {
+	query: () => void;
+	getPhase: () => PlannotatorPhase | null;
+};
+
+export function createPhaseTracker(pi: ExtensionAPI, onPhaseChange: () => void): PhaseTracker {
+	let currentPhase: PlannotatorPhase | null = null;
+
+	function applyPhaseFromResponse(response: any) {
+		if (response?.status !== "handled" || !response.result?.phase)
+			return;
+		if (currentPhase === response.result.phase)
+			return;
+
+		currentPhase = response.result.phase;
+		onPhaseChange();
+	}
+
+	pi.events.on(REQUEST_CHANNEL, (data: any) => {
+		if (data?.action !== "plan-mode" || typeof data?.respond !== "function")
+			return;
+
+		const originalRespond = data.respond;
+
+		data.respond = (response: any) => {
+			originalRespond(response);
+			applyPhaseFromResponse(response);
+		};
+	});
+
+	function query() {
+		pi.events.emit(REQUEST_CHANNEL, {
+			requestId: Math.random().toString(36).slice(2),
+			action: "plan-mode",
+			payload: { mode: "status" },
+			respond: (response: any) => applyPhaseFromResponse(response),
+		});
+	}
+
+	return { query, getPhase: () => currentPhase };
+}
+
+export class ModeModule extends Module {
+	public constructor(context: ModuleContext, private readonly phaseTracker: PhaseTracker) {
+		super(context);
+	}
+
+	public override render(): string | null {
+		const phase = this.phaseTracker.getPhase();
+
+		if (phase === null)
+			return null;
+
+		return phase !== "idle" ? yellow("Plan") : green("Build");
+	}
+}

package/src/modules/model.module.ts

@@ -0,0 +1,21 @@
+import { bold, capitalizeFirst, cyan, dim, formatProviderName, magenta } from "@ceo.paludetto/pi-utilities";
+
+import { Module } from "~/utilities/types";
+
+export class ModelModule extends Module {
+	public override render(): string | null {
+		const model = this.context.model;
+
+		if (!model)
+			return null;
+
+		const provider = formatProviderName(model.provider);
+		const thinkingLevel = this.context.getThinkingLevel();
+
+		const parenthesisContent = thinkingLevel !== "off"
+			? `${provider}${dim("/")}${cyan(capitalizeFirst(thinkingLevel))}`
+			: provider;
+
+		return `${bold(magenta(model.name))}${dim(" (")}${parenthesisContent}${dim(")")}`;
+	}
+}

package/src/modules/starship.module.ts

@@ -0,0 +1,63 @@
+import { execFile } from "node:child_process";
+import { promisify } from "node:util";
+
+import { Module } from "~/utilities/types";
+
+const execFileAsync = promisify(execFile);
+
+const STARSHIP_ARGUMENTS = [
+	"--status=0",
+	"--keymap=",
+	"--pipestatus=0",
+	"--cmd-duration=0",
+	"--jobs=0",
+];
+
+const ESCAPE_CHARACTER = "\x1B";
+const TRAILING_SGR_PATTERN = new RegExp(`${ESCAPE_CHARACTER}\\[[0-9;]*m+$`, "g");
+
+export class StarshipModule extends Module {
+	private prompt: string | null = null;
+
+	public override async refresh(): Promise<void> {
+		this.prompt = await this.fetchStarshipPrompt();
+	}
+
+	public override render(): string | null {
+		return this.prompt;
+	}
+
+	private async fetchStarshipPrompt(): Promise<string | null> {
+		try {
+			const { stdout } = await execFileAsync(
+				"starship",
+				["prompt", `--terminal-width=${this.context.getWidth()}`, ...STARSHIP_ARGUMENTS],
+				{
+					cwd: this.context.cwd,
+					timeout: 3000,
+					env: { ...process.env, PWD: this.context.cwd, STARSHIP_SHELL: "bash" },
+				},
+			);
+
+			return this.cleanStarshipOutput(stdout);
+		} catch {
+			return null;
+		}
+	}
+
+	private cleanStarshipOutput(stdout: string): string | null {
+		const firstLine = stdout.split("\n")[0] ?? "";
+
+		const withAnsiEscapes = firstLine
+			.replace(/\\\[/g, "")
+			.replace(/\\\]/g, "")
+			.replace(/%\{/g, ESCAPE_CHARACTER)
+			.replace(/%\}/g, "");
+
+		const trimmed = withAnsiEscapes
+			.replace(TRAILING_SGR_PATTERN, "")
+			.trimEnd();
+
+		return trimmed || null;
+	}
+}

package/src/modules/tokens.module.ts

@@ -0,0 +1,38 @@
+import type { AssistantMessage } from "@earendil-works/pi-ai";
+
+import { cyan, dim, formatTokenCount, green, yellow } from "@ceo.paludetto/pi-utilities";
+
+import { Module } from "~/utilities/types";
+
+export class TokensModule extends Module {
+	public override render(): string | null {
+		const { inputTokens, outputTokens, totalCost } = this.aggregateUsage(this.context.sessionManager.getBranch());
+
+		if (inputTokens === 0 && outputTokens === 0)
+			return null;
+
+		return [
+			cyan(`↑${formatTokenCount(inputTokens)}`),
+			green(`↓${formatTokenCount(outputTokens)}`),
+			yellow(`$${totalCost.toFixed(3)}`),
+		].join(dim("/"));
+	}
+
+	private aggregateUsage(branch: any[]): { inputTokens: number; outputTokens: number; totalCost: number } {
+		let inputTokens = 0;
+		let outputTokens = 0;
+		let totalCost = 0;
+
+		for (const entry of branch) {
+			if (entry.type !== "message" || entry.message.role !== "assistant")
+				continue;
+
+			const message = entry.message as AssistantMessage;
+			inputTokens += message.usage.input;
+			outputTokens += message.usage.output;
+			totalCost += message.usage.cost.total;
+		}
+
+		return { inputTokens, outputTokens, totalCost };
+	}
+}

package/src/modules/yolo.module.ts

@@ -0,0 +1,16 @@
+import { bold, rainbow } from "@ceo.paludetto/pi-utilities";
+
+import { Module } from "~/utilities/types";
+
+/** Published by @gotgenes/pi-permission-system via ctx.ui.setStatus(). */
+const PERMISSION_SYSTEM_STATUS_KEY = "pi-permission-system";
+const PERMISSION_SYSTEM_YOLO_STATUS_VALUE = "yolo";
+
+export class YoloModule extends Module {
+	public override render(): string | null {
+		if (this.getStatuses().get(PERMISSION_SYSTEM_STATUS_KEY) !== PERMISSION_SYSTEM_YOLO_STATUS_VALUE)
+			return null;
+
+		return bold(rainbow("Yolo"));
+	}
+}

package/src/utilities/types.ts

@@ -0,0 +1,21 @@
+import type { ExtensionContext } from "@earendil-works/pi-coding-agent";
+
+export type ModuleContext = {
+	getStatuses: () => ReadonlyMap<string, string>;
+	getThinkingLevel: () => string;
+	getWidth: () => number;
+} & ExtensionContext;
+
+export abstract class Module {
+	public constructor(
+		protected readonly context: ModuleContext,
+	) {}
+
+	protected getStatuses(): ReadonlyMap<string, string> {
+		return this.context.getStatuses();
+	}
+
+	public abstract render(): string | null;
+
+	public async refresh(): Promise<void> {}
+}

package/tsconfig.json

@@ -0,0 +1,12 @@
+{
+	"extends": "@total-typescript/tsconfig/bundler/no-dom",
+	"compilerOptions": {
+		"paths": {
+			"~/*": ["./src/*"]
+		},
+
+		"types": ["node"]
+	},
+	"include": ["**/*.ts"],
+	"exclude": ["node_modules", "dist"]
+}

package/tsdown.config.ts

@@ -0,0 +1,13 @@
+import { defineConfig } from "tsdown";
+
+// @keep-sorted
+export default defineConfig({
+	clean: true,
+	deps: { alwaysBundle: ["@ceo.paludetto/pi-utilities"] },
+	dts: { sourcemap: true },
+	entry: ["./src/index.ts"],
+	exports: { legacy: true },
+	format: ["cjs", "esm"],
+	publint: true,
+	sourcemap: true,
+});