pi-crew 0.10.6 → 0.11.0

package/src/extension/crew-vibes/footer.ts

@@ -1,313 +0,0 @@
-import { isAbsolute, relative, resolve, sep } from "node:path";
-import type { ExtensionContext } from "@earendil-works/pi-coding-agent";
-import { getFooterDockProvider } from "../../ui/dock-footer.ts";
-import { requestRenderTarget } from "../../ui/pi-ui-compat.ts";
-import { asCrewTheme, type CrewTheme } from "../../ui/theme-adapter.ts";
-import { colorWidgetLine, renderLines } from "../../ui/widget/widget-renderer.ts";
-import { truncateToWidth, visibleWidth } from "../../utils/visual.ts";
-import type { CrewVibesConfig } from "./config.ts";
-import type { ProviderUsage } from "./provider-usage.ts";
-import { formatCount, getCapacityUsage, renderCapacity, renderProviderUsage } from "./render.ts";
-
-/**
- * Custom footer replacement for crew-vibes.
- *
- * Why this exists: pi joins ALL `setStatus()` entries onto ONE line and
- * right-truncates when the line overflows. The provider quota lives at the
- * tail, so it gets chopped to "W..." whenever the `pi-crew` widget status grows
- * during sub-agent runs. Five prior attempts (right-align padding, dynamic
- * padding, separate keys) all failed because an extension using `setStatus`
- * cannot see the widget's width, other extensions' widths, or pi's real render
- * width. `setFooter` sidesteps all of that: `render(width)` receives the REAL
- * width and we own the line layout, so the meters get a dedicated line that is
- * never truncated by the join.
- */
-
-/** Live values the footer reads on each render (owned by the extension). */
-export interface CrewVibesFooterSource {
-	getConfig(): CrewVibesConfig;
-	getQuotaUsage(): ProviderUsage | null;
-	getThinkingLevel(): string | undefined;
-}
-
-export interface CrewVibesFooterDeps {
-	tui: unknown;
-	theme: unknown;
-	footerData: unknown;
-	ctx: ExtensionContext;
-	source: CrewVibesFooterSource;
-}
-
-interface FooterData {
-	getGitBranch(): string | null;
-	getExtensionStatuses(): ReadonlyMap<string, string>;
-	getAvailableProviderCount(): number;
-	onBranchChange(cb: () => void): () => void;
-}
-
-interface FooterComponent {
-	render(width: number): string[];
-	invalidate(): void;
-	dispose(): void;
-}
-
-function num(value: unknown): number {
-	return typeof value === "number" && Number.isFinite(value) ? value : 0;
-}
-
-/** Mirror pi's footer cwd formatting: collapse the home prefix to `~`. */
-function formatCwdForFooter(cwd: string, home: string | undefined): string {
-	if (!home) return cwd;
-	const resolvedCwd = resolve(cwd);
-	const resolvedHome = resolve(home);
-	const rel = relative(resolvedHome, resolvedCwd);
-	const inside = rel === "" || (rel !== ".." && !rel.startsWith(`..${sep}`) && !isAbsolute(rel));
-	if (!inside) return cwd;
-	return rel === "" ? "~" : `~${sep}${rel}`;
-}
-
-/** Mirror pi's status sanitizer (collapse control chars + runs of spaces). */
-function sanitizeStatusText(text: string): string {
-	return text
-		.replace(/[\r\n\t]/g, " ")
-		.replace(/ +/g, " ")
-		.trim();
-}
-
-function asFooterData(value: unknown): FooterData | undefined {
-	if (!value || typeof value !== "object") return undefined;
-	const record = value as Record<string, unknown>;
-	if (
-		typeof record.getGitBranch !== "function" ||
-		typeof record.getExtensionStatuses !== "function" ||
-		typeof record.getAvailableProviderCount !== "function" ||
-		typeof record.onBranchChange !== "function"
-	) {
-		return undefined;
-	}
-	return value as FooterData;
-}
-
-interface SessionTotals {
-	input: number;
-	output: number;
-	cacheRead: number;
-	cacheWrite: number;
-	cost: number;
-}
-
-function computeTotals(entries: unknown[]): SessionTotals {
-	const totals: SessionTotals = { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, cost: 0 };
-	for (const entry of entries) {
-		const rec = entry as {
-			type?: string;
-			message?: { role?: string; usage?: Record<string, unknown> & { cost?: { total?: unknown } } };
-		};
-		if (rec?.type !== "message" || rec.message?.role !== "assistant") continue;
-		const usage = rec.message.usage;
-		if (!usage) continue;
-		totals.input += num(usage.input);
-		totals.output += num(usage.output);
-		totals.cacheRead += num(usage.cacheRead);
-		totals.cacheWrite += num(usage.cacheWrite);
-		totals.cost += num(usage.cost?.total);
-	}
-	return totals;
-}
-
-class CrewVibesFooter implements FooterComponent {
-	private readonly theme: CrewTheme;
-	private readonly footerData: FooterData | undefined;
-	private readonly ctx: ExtensionContext;
-	private readonly source: CrewVibesFooterSource;
-	private readonly tui: unknown;
-	private readonly unsubscribeBranch: () => void;
-
-	constructor(deps: CrewVibesFooterDeps) {
-		this.theme = asCrewTheme(deps.theme);
-		this.footerData = asFooterData(deps.footerData);
-		this.ctx = deps.ctx;
-		this.source = deps.source;
-		this.tui = deps.tui;
-		this.unsubscribeBranch = this.footerData ? this.footerData.onBranchChange(() => requestRenderTarget(this.tui)) : () => undefined;
-	}
-
-	invalidate(): void {
-		/* no-op */
-	}
-
-	dispose(): void {
-		this.unsubscribeBranch();
-	}
-
-	private buildPwdLine(width: number): string {
-		const sm = this.ctx.sessionManager;
-		let pwd = formatCwdForFooter(sm.getCwd(), process.env.HOME || process.env.USERPROFILE);
-		const branch = this.footerData?.getGitBranch();
-		if (branch) pwd = `${pwd} (${branch})`;
-		const sessionName = sm.getSessionName?.();
-		if (sessionName) pwd = `${pwd} • ${sessionName}`;
-		return truncateToWidth(this.theme.fg("dim", pwd), width, this.theme.fg("dim", "..."));
-	}
-
-	private buildStatsLine(width: number): string {
-		const theme = this.theme;
-		const model = this.ctx.model as { id?: string; provider?: string; reasoning?: unknown; contextWindow?: number } | undefined;
-		const totals = computeTotals(this.ctx.sessionManager.getEntries() as unknown[]);
-		const contextUsage = this.ctx.getContextUsage?.();
-		const contextWindow = contextUsage?.contextWindow ?? model?.contextWindow ?? 0;
-		const percentValue = contextUsage?.percent ?? 0;
-		const percentKnown = contextUsage?.percent !== null && contextUsage?.percent !== undefined;
-
-		const statsParts: string[] = [];
-		if (totals.input) statsParts.push(`↑${formatCount(totals.input)}`);
-		if (totals.output) statsParts.push(`↓${formatCount(totals.output)}`);
-		if (totals.cacheRead) statsParts.push(`R${formatCount(totals.cacheRead)}`);
-		if (totals.cacheWrite) statsParts.push(`W${formatCount(totals.cacheWrite)}`);
-
-		const usingSubscription = !!(model && this.ctx.modelRegistry?.isUsingOAuth?.(this.ctx.model as never));
-		if (totals.cost || usingSubscription) {
-			statsParts.push(`$${totals.cost.toFixed(3)}${usingSubscription ? " (sub)" : ""}`);
-		}
-
-		// User choice: always show the "(auto)" compaction indicator (pi's default);
-		// the toggle state is not observable by extensions.
-		const autoIndicator = " (auto)";
-		const contextDisplay = percentKnown
-			? `${percentValue.toFixed(1)}%/${formatCount(contextWindow)}${autoIndicator}`
-			: `?/${formatCount(contextWindow)}${autoIndicator}`;
-		const contextColored =
-			percentValue > 90
-				? theme.fg("error", contextDisplay)
-				: percentValue > 70
-					? theme.fg("warning", contextDisplay)
-					: contextDisplay;
-		statsParts.push(contextColored);
-
-		let statsLeft = statsParts.join(" ");
-		let statsLeftWidth = visibleWidth(statsLeft);
-		if (statsLeftWidth > width) {
-			statsLeft = truncateToWidth(statsLeft, width, "...");
-			statsLeftWidth = visibleWidth(statsLeft);
-		}
-
-		const minPadding = 2;
-		const modelName = model?.id || "no-model";
-		let rightSideWithoutProvider = modelName;
-		if (model?.reasoning) {
-			// Prefer the authoritative thinking level from session state
-			// (buildSessionContext), which reflects the current effective level
-			// including the model default — not just the last thinking_level_select
-			// event (which only fires on manual switch).
-			let level = this.source.getThinkingLevel();
-			try {
-				const ctx = this.ctx.sessionManager as { buildSessionContext?: () => { thinkingLevel?: string } };
-				if (typeof ctx.buildSessionContext === "function") {
-					const resolved = ctx.buildSessionContext()?.thinkingLevel;
-					if (resolved) level = resolved;
-				}
-			} catch {
-				// buildSessionContext may throw on empty/early sessions — fall back.
-			}
-			const finalLevel = level || "off";
-			rightSideWithoutProvider = finalLevel === "off" ? `${modelName} • thinking off` : `${modelName} • ${finalLevel}`;
-		}
-
-		let rightSide = rightSideWithoutProvider;
-		const providerCount = this.footerData?.getAvailableProviderCount() ?? 0;
-		if (providerCount > 1 && model?.provider) {
-			rightSide = `(${model.provider}) ${rightSideWithoutProvider}`;
-			if (statsLeftWidth + minPadding + visibleWidth(rightSide) > width) rightSide = rightSideWithoutProvider;
-		}
-
-		const rightSideWidth = visibleWidth(rightSide);
-		let statsLine: string;
-		if (statsLeftWidth + minPadding + rightSideWidth <= width) {
-			statsLine = statsLeft + " ".repeat(width - statsLeftWidth - rightSideWidth) + rightSide;
-		} else {
-			const availableForRight = width - statsLeftWidth - minPadding;
-			if (availableForRight > 0) {
-				const truncatedRight = truncateToWidth(rightSide, availableForRight, "");
-				const padding = " ".repeat(Math.max(0, width - statsLeftWidth - visibleWidth(truncatedRight)));
-				statsLine = statsLeft + padding + truncatedRight;
-			} else {
-				statsLine = statsLeft;
-			}
-		}
-
-		const dimStatsLeft = theme.fg("dim", statsLeft);
-		const dimRemainder = theme.fg("dim", statsLine.slice(statsLeft.length));
-		return dimStatsLeft + dimRemainder;
-	}
-
-	private buildStatusLine(width: number): string | undefined {
-		if (!this.footerData) return undefined;
-		const statuses = this.footerData.getExtensionStatuses();
-		if (statuses.size === 0) return undefined;
-		const joined = Array.from(statuses.entries())
-			.sort(([a], [b]) => a.localeCompare(b))
-			.map(([, text]) => sanitizeStatusText(text))
-			.join(" ");
-		if (!joined) return undefined;
-		return truncateToWidth(joined, width, this.theme.fg("dim", "..."));
-	}
-
-	private rightAlign(text: string, width: number): string {
-		const w = visibleWidth(text);
-		if (w >= width) return truncateToWidth(text, width, "…");
-		return " ".repeat(width - w) + text;
-	}
-
-	/** Capacity + provider quota. Uses the REAL render width, so the quota is
-	 * never chopped. When both do not fit on one line, wrap to two lines
-	 * (capacity above, quota right-aligned below) per the chosen behavior. */
-	private buildMeterLines(width: number): string[] {
-		const config = this.source.getConfig();
-		if (!config.enabled) return [];
-		const capText = config.capacity.enabled ? renderCapacity(this.theme, config.capacity, getCapacityUsage(this.ctx)) : undefined;
-		const quotaText = config.capacity.providerUsage ? renderProviderUsage(this.theme, this.source.getQuotaUsage()) : undefined;
-
-		if (!capText && !quotaText) return [];
-		if (capText && !quotaText) return [truncateToWidth(capText, width, "…")];
-		if (!capText && quotaText) return [this.rightAlign(quotaText, width)];
-
-		const cap = capText as string;
-		const quota = quotaText as string;
-		const capWidth = visibleWidth(cap);
-		const quotaWidth = visibleWidth(quota);
-		if (capWidth + 1 + quotaWidth <= width) {
-			const pad = Math.max(1, width - capWidth - quotaWidth);
-			return [cap + " ".repeat(pad) + quota];
-		}
-		return [truncateToWidth(cap, width, "…"), this.rightAlign(quota, width)];
-	}
-
-	/** Dock lines registered by the crew widget (`widgetPlacement: "bottom"`):
-	 *  painted as the very last block of the footer, below the quota/meters.
-	 *  The provider returns RAW lines; we color them with this footer's theme
-	 *  (same per-line palette the widget would have used) and truncate to the
-	 *  real render width. */
-	private buildDockLines(width: number): string[] {
-		const provider = getFooterDockProvider();
-		if (!provider) return [];
-		const raw = provider(width);
-		if (raw.length === 0) return [];
-		return renderLines(
-			raw.map((line, index) => colorWidgetLine(line, index, this.theme)),
-			width,
-		).map((line) => truncateToWidth(line, width, "…"));
-	}
-
-	render(width: number): string[] {
-		const lines = [this.buildPwdLine(width), this.buildStatsLine(width)];
-		const statusLine = this.buildStatusLine(width);
-		if (statusLine) lines.push(statusLine);
-		lines.push(...this.buildMeterLines(width));
-		lines.push(...this.buildDockLines(width));
-		return lines;
-	}
-}
-
-export function createCrewVibesFooter(deps: CrewVibesFooterDeps): FooterComponent {
-	return new CrewVibesFooter(deps);
-}

package/src/extension/crew-vibes/speed.ts

@@ -1,286 +0,0 @@
-import type { SpeedConfig } from "./config.ts";
-
-/**
- * Token-speed engine ported from pi-speeed (MIT, attribution in NOTICE.md).
- * Tracks output tok/s with a sliding window and suppresses unreliable
- * burst-only readings. No stats/font handling — just the live + session avg.
- */
-
-export type TokenEvent = { time: number; tokens: number };
-
-const COMPACTION_THRESHOLD = 5000;
-
-export type EngineConfig = Pick<SpeedConfig, "slidingWindowMs" | "minReliableDurationMs" | "maxDisplayTokS">;
-
-export function estimateTokensFromDelta(text: string): number {
-	if (!text) return 0;
-	const matches = text.match(/\w+|[^\s\w]/g);
-	return matches ? matches.length : 0;
-}
-
-export class TokenSpeedEngine {
-	private _isStreaming = false;
-	private _tokenCount = 0;
-	private _startTime = 0;
-	private _endTime = 0;
-	private _events: TokenEvent[] = [];
-	private _windowStartIndex = 0;
-	private _lastStableTokS = 0;
-	private _lastUsageOutput = 0;
-	private _config: EngineConfig;
-
-	constructor(config: EngineConfig) {
-		this._config = config;
-	}
-
-	updateConfig(config: EngineConfig): void {
-		this._config = config;
-	}
-
-	get isStreaming(): boolean {
-		return this._isStreaming;
-	}
-
-	get tokenCount(): number {
-		return this._tokenCount;
-	}
-
-	get elapsedMs(): number {
-		if (this._startTime === 0) return 0;
-		return this._isStreaming ? Date.now() - this._startTime : this._endTime - this._startTime;
-	}
-
-	get avgTokS(): number {
-		const elapsedSec = this.elapsedMs / 1000;
-		if (elapsedSec <= 0) return 0;
-		return this._tokenCount / elapsedSec;
-	}
-
-	sanitizeTokS(value: number | null, durationMs = this.elapsedMs): number | null {
-		if (value === null || !Number.isFinite(value) || value <= 0) return null;
-		if (durationMs < this._config.minReliableDurationMs) return null;
-		if (value > this._config.maxDisplayTokS) return null;
-		return value;
-	}
-
-	get tokS(): number {
-		const candidate = this.rawTokS;
-		const stable = this.sanitizeTokS(candidate);
-		if (stable !== null) this._lastStableTokS = stable;
-		return this._lastStableTokS;
-	}
-
-	get rawTokS(): number {
-		if (this.elapsedMs < this._config.slidingWindowMs) return this.avgTokS;
-		if (!this._isStreaming) return this.avgTokS;
-
-		const now = Date.now();
-		const windowStart = now - this._config.slidingWindowMs;
-
-		while (this._windowStartIndex < this._events.length && this._events[this._windowStartIndex].time < windowStart) {
-			this._windowStartIndex++;
-		}
-
-		if (this._windowStartIndex >= this._events.length) return this.avgTokS;
-
-		let windowTokenCount = 0;
-		for (let i = this._windowStartIndex; i < this._events.length; i++) {
-			windowTokenCount += this._events[i].tokens;
-		}
-		if (windowTokenCount === 0) return this.avgTokS;
-
-		const windowDuration = (now - this._events[this._windowStartIndex].time) / 1000;
-		if (windowDuration <= 0) return 0;
-
-		return windowTokenCount / windowDuration;
-	}
-
-	start(): void {
-		this._tokenCount = 0;
-		this._isStreaming = true;
-		this._startTime = Date.now();
-		this._endTime = this._startTime;
-		this._events = [];
-		this._windowStartIndex = 0;
-		this._lastStableTokS = 0;
-		this._lastUsageOutput = 0;
-	}
-
-	stop(): void {
-		this._isStreaming = false;
-		this._endTime = Date.now();
-		this._events = [];
-		this._windowStartIndex = 0;
-	}
-
-	recordDelta(delta: string, usageOutput?: number): void {
-		if (!this._isStreaming) return;
-		if (usageOutput !== undefined && usageOutput > 0) {
-			// usageOutput may be cumulative across message_update events. Track the
-			// delta to avoid inflating _tokenCount by N× the true total.
-			const increment = usageOutput - this._lastUsageOutput;
-			this._lastUsageOutput = usageOutput;
-			this.recordTokens(Math.max(0, increment));
-			return;
-		}
-		this.recordTokens(estimateTokensFromDelta(delta));
-	}
-
-	reconcileTotal(tokens: number): void {
-		if (tokens > 0) this._tokenCount = tokens;
-	}
-
-	recordTokens(tokens: number): void {
-		if (!this._isStreaming || tokens <= 0) return;
-		this._tokenCount += tokens;
-		this._events.push({ time: Date.now(), tokens });
-		if (this._windowStartIndex >= COMPACTION_THRESHOLD) this.compact();
-	}
-
-	private compact(): void {
-		if (this._windowStartIndex === 0) return;
-		this._events = this._events.slice(this._windowStartIndex);
-		this._windowStartIndex = 0;
-	}
-}
-
-export type CompletedMessageSpeed = {
-	outputTokens: number;
-	durationMs: number;
-	tokS: number | null;
-};
-
-function isSuccessfulStop(stopReason: string | undefined): boolean {
-	return stopReason !== "error" && stopReason !== "aborted";
-}
-
-export class SpeedTracker {
-	private readonly engine: TokenSpeedEngine;
-	private lastStableTokS: number | null = null;
-	private sessionOutputTokens = 0;
-	private sessionDurationMs = 0;
-
-	constructor(config: SpeedConfig) {
-		this.engine = new TokenSpeedEngine(config);
-	}
-
-	updateConfig(config: SpeedConfig): void {
-		this.engine.updateConfig(config);
-	}
-
-	get isStreaming(): boolean {
-		return this.engine.isStreaming;
-	}
-
-	get lastTokS(): number | null {
-		return this.lastStableTokS;
-	}
-
-	resetSession(): void {
-		this.sessionOutputTokens = 0;
-		this.sessionDurationMs = 0;
-	}
-
-	startMessage(): void {
-		this.engine.start();
-	}
-
-	recordDelta(delta: string, usageOutput?: number): void {
-		this.engine.recordDelta(delta, usageOutput);
-	}
-
-	stopMessage(): void {
-		if (this.engine.isStreaming) this.engine.stop();
-	}
-
-	liveTokS(): number | null {
-		const speed = this.engine.tokS;
-		return speed > 0 ? speed : this.lastStableTokS;
-	}
-
-	sessionAvgTokS(): number | null {
-		return this.sessionDurationMs > 0 ? this.sessionOutputTokens / (this.sessionDurationMs / 1000) : null;
-	}
-
-	finishMessage(outputTokens: number, stopReason: string | undefined): CompletedMessageSpeed | null {
-		if (!this.engine.isStreaming) return null;
-
-		this.engine.reconcileTotal(outputTokens);
-		const durationMs = this.engine.elapsedMs;
-		const tokens = this.engine.tokenCount;
-		const rawAvgTokS = durationMs > 0 ? tokens / (durationMs / 1000) : null;
-		const tokS = this.engine.sanitizeTokS(rawAvgTokS, durationMs);
-		this.lastStableTokS = tokS;
-		this.engine.stop();
-
-		if (tokS !== null && isSuccessfulStop(stopReason)) {
-			this.sessionOutputTokens += tokens;
-			this.sessionDurationMs += durationMs;
-		}
-
-		return { outputTokens: tokens, durationMs, tokS };
-	}
-}
-
-export class SpeedAnimator {
-	private from: number | null = null;
-	private target: number | null = null;
-	private startedAt = 0;
-	private durationMs: number;
-
-	constructor(durationMs: number) {
-		this.durationMs = durationMs;
-	}
-
-	updateDuration(durationMs: number): void {
-		this.durationMs = durationMs;
-	}
-
-	reset(value: number | null = null, now = Date.now()): void {
-		this.from = value;
-		this.target = value;
-		this.startedAt = now;
-	}
-
-	setTarget(target: number | null, now = Date.now()): number | null {
-		if (target === null) {
-			this.reset(null, now);
-			return null;
-		}
-
-		const current = this.value(now);
-		if (current === null) {
-			this.reset(target, now);
-			return target;
-		}
-
-		if (this.target !== null && Math.abs(target - this.target) < 0.05) return current;
-
-		this.from = current;
-		this.target = target;
-		this.startedAt = now;
-		return current;
-	}
-
-	value(now = Date.now()): number | null {
-		if (this.target === null) return null;
-		if (this.from === null || this.durationMs <= 0) return this.target;
-
-		const progress = Math.max(0, Math.min(1, (now - this.startedAt) / this.durationMs));
-		if (progress >= 1) {
-			this.from = this.target;
-			return this.target;
-		}
-
-		return this.from + (this.target - this.from) * progress;
-	}
-
-	isAnimating(now = Date.now()): boolean {
-		return (
-			this.target !== null &&
-			this.from !== null &&
-			Math.abs(this.target - this.from) >= 0.05 &&
-			now - this.startedAt < this.durationMs
-		);
-	}
-}

package/src/ui/dock-footer.ts

@@ -1,49 +0,0 @@
-/**
- * dock-footer.ts — dependency-free bridge between the crew dock and the
- * crew-vibes footer.
- *
- * When `ui.widgetPlacement` is `"bottom"`, the dock's rendered lines are
- * produced by the crew widget but PAINTED by the crew-vibes footer, below the
- * quota/meter lines — the very bottom of the terminal. The two halves are
- * mounted by different modules (widget/index.ts owns the widget lifecycle,
- * crew-vibes/footer.ts owns the footer), so the connection lives here as a
- * tiny observable registry with no imports:
- *   - the widget registers a per-render line provider;
- *   - crew-vibes flips the "sink" flag when its footer is (un)installed;
- *   - the footer pulls the provider's lines on every render.
- *
- * Safety: the widget falls back to pi's `belowEditor` widget slot whenever no
- * footer sink is active (crew-vibes disabled), so `"bottom"` never silently
- * hides the dock.
- */
-
-export type FooterDockLinesProvider = (width: number) => string[];
-
-let dockProvider: FooterDockLinesProvider | undefined;
-let sinkActive = false;
-
-/** Register the dock line provider (widget side). Pass `undefined` to detach. */
-export function setFooterDockProvider(provider: FooterDockLinesProvider | undefined): void {
-	dockProvider = provider;
-}
-
-/** Current dock line provider, or undefined when the dock is not in the footer. */
-export function getFooterDockProvider(): FooterDockLinesProvider | undefined {
-	return dockProvider;
-}
-
-/** Mark whether a footer sink (crew-vibes) is currently installed. */
-export function setFooterDockSinkActive(active: boolean): void {
-	sinkActive = active;
-}
-
-/** True while crew-vibes' custom footer is installed and can host the dock. */
-export function isFooterDockSinkActive(): boolean {
-	return sinkActive;
-}
-
-/** Test isolation. */
-export function resetFooterDockRegistry(): void {
-	dockProvider = undefined;
-	sinkActive = false;
-}