@agentuity/coder 1.0.37

package/src/footer.ts

@@ -0,0 +1,305 @@
+/**
+ * Coder footer for the Pi TUI.
+ *
+ * Uses transparent backgrounds with foreground-only ANSI true-color text.
+ * Includes a braille spinner animation when an agent is actively working.
+ *
+ * Layout:
+ *   [brand] [branch] > [model/agent]        [hub] | [N] label  token-stats
+ *
+ * Observer awareness (ASCII only):
+ *   [3] SwiftRaven — 3 observers watching, session label "SwiftRaven"
+ */
+
+import type { ExtensionContext, ReadonlyFooterDataProvider } from '@mariozechner/pi-coding-agent';
+
+const RESET = '\x1b[0m';
+const SEP = '>';
+
+// ──────────────────────────────────────────────
+// ANSI true-color helper (foreground only)
+// ──────────────────────────────────────────────
+
+type RGB = [number, number, number];
+
+function fg(color: RGB, text: string): string {
+	return `\x1b[38;2;${color[0]};${color[1]};${color[2]}m${text}`;
+}
+
+// ──────────────────────────────────────────────
+// Color palette (foreground only, no backgrounds)
+// ──────────────────────────────────────────────
+
+const FG_BRAND: RGB = [100, 200, 255];
+const FG_MODEL: RGB = [215, 135, 175];
+const FG_AGENT: RGB = [130, 200, 130];
+const FG_BRANCH: RGB = [150, 180, 150];
+const FG_HUB_OK: RGB = [80, 200, 120];
+const FG_HUB_WARN: RGB = [245, 179, 66];
+const FG_HUB_ERR: RGB = [220, 80, 80];
+const FG_DIM: RGB = [100, 110, 120];
+
+type HubStatus = 'connected' | 'reconnecting' | 'offline';
+
+/** Observer state provided by the extension's presence tracking. */
+export interface ObserverState {
+	/** Number of observers watching this session (excludes lead + sub-agents). */
+	count: number;
+	/** Human-readable session label (e.g. "SwiftRaven"). */
+	label: string;
+}
+
+const FG_OBSERVER: RGB = [140, 180, 220];
+
+// ──────────────────────────────────────────────
+// Braille spinner
+// ──────────────────────────────────────────────
+
+const SPINNER_FRAMES = [
+	'\u280B',
+	'\u2819',
+	'\u2839',
+	'\u2838',
+	'\u283C',
+	'\u2834',
+	'\u2826',
+	'\u2827',
+	'\u2807',
+	'\u280F',
+];
+
+// ──────────────────────────────────────────────
+// Footer builder (transparent bg, foreground only)
+// ──────────────────────────────────────────────
+
+/** Strip ANSI escape sequences to get visible character count. */
+function visibleLength(str: string): number {
+	// eslint-disable-next-line no-control-regex
+	return str.replace(/\x1b\[[0-9;]*m/g, '').length;
+}
+
+/** Truncate an ANSI-colored string to a maximum visible width. */
+function truncateAnsi(str: string, maxWidth: number): string {
+	let visible = 0;
+	let i = 0;
+	while (i < str.length && visible < maxWidth) {
+		if (str[i] === '\x1b') {
+			// Skip entire ANSI escape sequence
+			const end = str.indexOf('m', i);
+			if (end !== -1) {
+				i = end + 1;
+			} else {
+				i++;
+			}
+		} else {
+			visible++;
+			i++;
+		}
+	}
+	return str.slice(0, i) + RESET;
+}
+
+function buildFooter(left: string, rightText: string, width: number): string {
+	// Safety margin for Unicode characters that may be double-width
+	const safeWidth = width - 4;
+	const leftLen = visibleLength(left);
+	const rightLen = visibleLength(rightText);
+	const total = leftLen + 1 + rightLen;
+
+	if (total > safeWidth) {
+		const maxLeft = safeWidth - rightLen - 1;
+		if (maxLeft > 0) {
+			return truncateAnsi(left, maxLeft) + ' ' + rightText;
+		}
+		return truncateAnsi(left + ' ' + rightText, safeWidth);
+	}
+
+	const gap = safeWidth - leftLen - rightLen;
+	return left + ' '.repeat(gap) + rightText;
+}
+
+// ──────────────────────────────────────────────
+// Minimal component
+// ──────────────────────────────────────────────
+
+class FooterComponent {
+	private getText: (width: number) => string;
+	private _unsubscribeBranch?: () => void;
+
+	constructor(
+		getText: (width: number) => string,
+		footerData: ReadonlyFooterDataProvider,
+		cleanupSpinner: () => void
+	) {
+		this.getText = getText;
+		this._cleanupSpinner = cleanupSpinner;
+		this._unsubscribeBranch = footerData.onBranchChange(() => {
+			// Triggers TUI refresh
+		});
+	}
+
+	private _cleanupSpinner: () => void;
+
+	render(width: number): string[] {
+		const text = this.getText(width);
+		// Final safety: ensure line never exceeds terminal width
+		if (visibleLength(text) > width) {
+			return [truncateAnsi(text, width)];
+		}
+		return [text];
+	}
+
+	invalidate(): void {
+		// no-op
+	}
+
+	dispose(): void {
+		this._unsubscribeBranch?.();
+		this._cleanupSpinner();
+	}
+}
+
+// ──────────────────────────────────────────────
+// Token stat formatters
+// ──────────────────────────────────────────────
+
+function formatTokens(n: number): string {
+	if (n >= 1_000_000) return `${(n / 1_000_000).toFixed(1)}M`;
+	if (n >= 1_000) return `${(n / 1_000).toFixed(1)}K`;
+	return String(n);
+}
+
+function formatCost(n: number): string {
+	if (n === 0) return '$0.00';
+	if (n < 0.01) return `$${n.toFixed(4)}`;
+	return `$${n.toFixed(2)}`;
+}
+
+// ──────────────────────────────────────────────
+// Public API
+// ──────────────────────────────────────────────
+
+/**
+ * Set up the Coder footer (transparent bg, foreground-colored text).
+ * Call this once with the extension context to replace Pi's default footer.
+ *
+ * Includes a braille spinner animation when an agent is actively working.
+ *
+ * @param ctx  Extension context with UI access
+ * @param getHubStatus  Callback that returns current Hub connection status
+ * @param getObserverState  Optional callback that returns observer count + session label
+ */
+export function setupCoderFooter(
+	ctx: ExtensionContext,
+	getHubStatus: () => HubStatus,
+	getObserverState?: () => ObserverState
+): void {
+	if (!ctx.hasUI) return;
+
+	ctx.ui.setFooter((tui, _theme, footerData) => {
+		// Spinner state
+		let spinnerTimer: ReturnType<typeof setInterval> | null = null;
+		let spinnerFrame = 0;
+
+		const getText = (width: number): string => {
+			// Detect active agent
+			const activeAgent = footerData.getExtensionStatuses().get('active_agent');
+
+			// Start/stop spinner based on agent activity
+			if (activeAgent && !spinnerTimer) {
+				spinnerTimer = setInterval(() => {
+					spinnerFrame = (spinnerFrame + 1) % SPINNER_FRAMES.length;
+					tui.requestRender();
+				}, 80);
+			} else if (!activeAgent && spinnerTimer) {
+				clearInterval(spinnerTimer);
+				spinnerTimer = null;
+				spinnerFrame = 0;
+			}
+
+			// Token stats from session messages
+			let inputTokens = 0;
+			let outputTokens = 0;
+			let totalCost = 0;
+			for (const entry of ctx.sessionManager.getBranch()) {
+				if (entry.type === 'message') {
+					const msg = entry.message as {
+						role?: string;
+						usage?: { input: number; output: number; cost: { total: number } };
+					};
+					if (msg.role === 'assistant' && msg.usage) {
+						inputTokens += msg.usage.input;
+						outputTokens += msg.usage.output;
+						totalCost += msg.usage.cost.total;
+					}
+				}
+			}
+			const tokenStr = `\u2191${formatTokens(inputTokens)} \u2193${formatTokens(outputTokens)} ${formatCost(totalCost)}`;
+
+			// LEFT side: brand > branch > model/agent
+			const leftParts: string[] = [];
+
+			// Brand (with spinner)
+			const brandChar = spinnerTimer ? SPINNER_FRAMES[spinnerFrame]! : '\u2A3A';
+			leftParts.push(fg(FG_BRAND, ` ${brandChar}`));
+
+			// Branch
+			const branch = footerData.getGitBranch();
+			if (branch) {
+				leftParts.push('  ');
+				leftParts.push(fg(FG_BRANCH, branch));
+			}
+
+			// Model or active agent
+			leftParts.push(fg(FG_DIM, ` ${SEP} `));
+			if (activeAgent) {
+				leftParts.push(fg(FG_AGENT, activeAgent));
+			} else {
+				const modelId = ctx.model ? String((ctx.model as { id?: string }).id ?? '?') : '?';
+				leftParts.push(fg(FG_MODEL, modelId));
+			}
+
+			const left = leftParts.join('');
+
+			// RIGHT side: hub status + observer info + token stats
+			const rightParts: string[] = [];
+			const hubStatus = getHubStatus();
+			if (hubStatus === 'connected') {
+				rightParts.push(fg(FG_HUB_OK, 'Hub'));
+			} else if (hubStatus === 'reconnecting') {
+				rightParts.push(fg(FG_HUB_WARN, 'Hub...'));
+			} else {
+				rightParts.push(fg(FG_HUB_ERR, 'Hub Off'));
+			}
+
+			// Observer awareness (ASCII only, no emojis)
+			if (getObserverState && hubStatus === 'connected') {
+				const obs = getObserverState();
+				if (obs.count > 0 || obs.label) {
+					rightParts.push(fg(FG_DIM, ' | '));
+					if (obs.count > 0) {
+						rightParts.push(fg(FG_OBSERVER, `[${obs.count}]`));
+						if (obs.label) rightParts.push(' ');
+					}
+					if (obs.label) {
+						rightParts.push(fg(FG_OBSERVER, obs.label));
+					}
+				}
+			}
+
+			rightParts.push(fg(FG_DIM, `  ${tokenStr}`) + RESET);
+			const rightText = rightParts.join('');
+
+			return buildFooter(left, rightText, width);
+		};
+
+		const cleanupSpinner = (): void => {
+			if (spinnerTimer) {
+				clearInterval(spinnerTimer);
+				spinnerTimer = null;
+			}
+		};
+
+		return new FooterComponent(getText, footerData, cleanupSpinner);
+	});
+}

package/src/handlers.ts

@@ -0,0 +1,113 @@
+import type { HubAction } from './protocol.ts';
+
+export interface ActionResult {
+	block?: { block: true; reason: string };
+	returnValue?: unknown;
+	systemPrompt?: string;
+	systemPromptMode?: 'replace' | 'prefix' | 'suffix';
+	// undefined means ACK (proceed normally)
+}
+
+/** Minimal UI surface used by action handlers — avoids a hard dep on pi-coding-agent. */
+interface ActionContext {
+	ui?: {
+		notify(message: string, level?: 'info' | 'warning' | 'error'): void;
+		confirm(title: string, message: string): Promise<boolean>;
+		setStatus(key: string, text?: string): void;
+	};
+	sendUserMessage?: (message: string, options?: { deliverAs?: 'followUp' }) => void;
+}
+
+export async function processActions(
+	actions: HubAction[],
+	ctx: ActionContext
+): Promise<ActionResult> {
+	let result: ActionResult = {};
+
+	for (const action of actions) {
+		switch (action.action) {
+			case 'ACK':
+				// Terminal: proceed normally
+				result = {};
+				break;
+
+			case 'BLOCK':
+				// Terminal: block
+				result = { block: { block: true, reason: action.reason } };
+				break;
+
+			case 'RETURN':
+				// Terminal: return a specific result
+				result = { returnValue: action.result };
+				break;
+
+			case 'NOTIFY':
+				// Side effect: show notification, continue
+				if (ctx?.ui) {
+					ctx.ui.notify(action.message, action.level ?? 'info');
+				}
+				break;
+
+			case 'STATUS':
+				// Side effect: set status, continue
+				if (ctx?.ui) {
+					ctx.ui.setStatus(action.key, action.text);
+				}
+				break;
+
+			case 'CONFIRM': {
+				// Gate: if user denies, stop and block
+				if (ctx?.ui) {
+					const confirmed = await ctx.ui.confirm(action.title, action.message);
+					if (!confirmed) {
+						return {
+							block: {
+								block: true,
+								reason: action.deny_reason ?? 'Denied by user',
+							},
+						};
+					}
+				} else {
+					// No UI available — block by default for safety
+					result = {
+						block: {
+							block: true,
+							reason: action.deny_reason ?? 'Confirmation required but no UI available',
+						},
+					};
+				}
+				break;
+			}
+
+			case 'SYSTEM_PROMPT':
+				// System prompt injection — store for before_agent_start handler
+				result = {
+					...result,
+					systemPrompt: action.systemPrompt,
+					systemPromptMode: action.mode,
+				};
+				break;
+
+			case 'INJECT_MESSAGE': {
+				const content = action.message?.content?.trim();
+				if (!content) break;
+
+				if (action.message?.role === 'user') {
+					if (ctx.sendUserMessage) {
+						ctx.sendUserMessage(content, { deliverAs: 'followUp' });
+					} else if (ctx.ui) {
+						ctx.ui.notify(content, 'info');
+					}
+					break;
+				}
+
+				if (ctx.ui) {
+					ctx.ui.notify(content, 'info');
+				}
+				break;
+			}
+		}
+	}
+
+	return result;
+}