@underactive/pi-topping-moa-fusion 0.4.4 → 0.5.0

package/src/activityMeter.ts

@@ -1,193 +0,0 @@
-/**
- * Output-token activity meter.
- *
- * Vendored from pi-topping (`src/activity-meter.ts` for the meter/rate
- * tracker, `src/format.ts` for StreamingWordCounter) rather than depended on,
- * so the two extensions stay independently installable. Behavior is meant to
- * stay identical — glyph ramp, thresholds, EMA coefficient, direction and
- * colouring are all load-bearing for visual parity, so port changes upstream
- * rather than tuning them here. test/activity-meter.test.mjs mirrors
- * pi-topping's tests/activity-meter.test.ts to catch drift.
- */
-
-import type { ThemeColor } from "@earendil-works/pi-coding-agent";
-
-export const ActivityMeterLevel = {
-	IDLE: 0,
-	LOW: 1,
-	MEDIUM: 2,
-	HIGH: 3,
-	PEAK_1: 4,
-	PEAK_2: 5,
-	PEAK_3: 6,
-	FULL: 7,
-} as const;
-
-export type ActivityMeterLevel = (typeof ActivityMeterLevel)[keyof typeof ActivityMeterLevel];
-
-const EMA_ALPHA = 0.4;
-const BRAILLE: Record<ActivityMeterLevel, string> = {
-	[ActivityMeterLevel.IDLE]: "⢀",
-	[ActivityMeterLevel.LOW]: "⣀",
-	[ActivityMeterLevel.MEDIUM]: "⣠",
-	[ActivityMeterLevel.HIGH]: "⣤",
-	[ActivityMeterLevel.PEAK_1]: "⣴",
-	[ActivityMeterLevel.PEAK_2]: "⣶",
-	[ActivityMeterLevel.PEAK_3]: "⣾",
-	[ActivityMeterLevel.FULL]: "⣿",
-};
-export const ACTIVITY_METER_WIDTH = 8;
-type CellColorizer = (level: ActivityMeterLevel, char: string) => string;
-
-/** Convert an estimated output-token rate to a display level. */
-export function rateToLevel(tokensPerSecond: number): ActivityMeterLevel {
-	const THRESHOLDS = [0, 5, 10, 15, 22, 30, 40];
-	for (let i = THRESHOLDS.length - 1; i >= 0; i--) {
-		if (tokensPerSecond > THRESHOLDS[i]) return (i + 1) as ActivityMeterLevel;
-	}
-	return 0;
-}
-
-/** EMA-smoothed rate tracker for a cumulative output-token estimate. */
-export class TokRateTracker {
-	#lastTotal = 0;
-	#lastTime = 0;
-	#rate = 0;
-	#hasSample = false;
-	#pendingTokens = 0;
-
-	sample(totalTokens: number, now: number): number {
-		if (!this.#hasSample) {
-			this.#lastTotal = totalTokens;
-			this.#hasSample = true;
-			this.#lastTime = now;
-			return this.#rate;
-		}
-
-		const elapsedSeconds = (now - this.#lastTime) / 1_000;
-		if (elapsedSeconds <= 0) {
-			this.#pendingTokens += Math.max(0, totalTokens - this.#lastTotal);
-			this.#lastTotal = totalTokens;
-			return this.#rate;
-		}
-
-		const totalDelta = this.#pendingTokens + Math.max(0, totalTokens - this.#lastTotal);
-		const instantRate = totalDelta / elapsedSeconds;
-		this.#rate = EMA_ALPHA * instantRate + (1 - EMA_ALPHA) * this.#rate;
-		this.#lastTotal = totalTokens;
-		this.#lastTime = now;
-		this.#pendingTokens = 0;
-		return this.#rate;
-	}
-
-	reset(): void {
-		this.#lastTotal = 0;
-		this.#lastTime = 0;
-		this.#rate = 0;
-		this.#hasSample = false;
-		this.#pendingTokens = 0;
-	}
-}
-
-/** Eight-column scrolling activity meter, scrolling either left-to-right or right-to-left. */
-export class ActivityMeter {
-	#levels: ActivityMeterLevel[] = Array<ActivityMeterLevel>(ACTIVITY_METER_WIDTH).fill(0);
-	#direction: "ltr" | "rtl";
-
-	constructor(direction: "ltr" | "rtl" = "ltr") {
-		this.#direction = direction;
-	}
-
-	setDirection(direction: "ltr" | "rtl"): void {
-		if (direction !== this.#direction) {
-			this.#direction = direction;
-			// Reverse the existing data so the visual flow instantly flips.
-			this.#levels.reverse();
-		}
-	}
-
-	push(level: ActivityMeterLevel): void {
-		if (this.#direction === "rtl") {
-			this.#levels.shift();
-			this.#levels.push(level);
-		} else {
-			this.#levels.pop();
-			this.#levels.unshift(level);
-		}
-	}
-
-	render(colorize?: CellColorizer): string {
-		return this.#levels
-			.map((level) => {
-				const char = BRAILLE[level];
-				return colorize ? colorize(level, char) : char;
-			})
-			.join("");
-	}
-
-	reset(): void {
-		this.#levels.fill(0);
-	}
-
-	/** Colorize a meter cell using the default theme mapping (dim at IDLE, `color` otherwise). */
-	static colorizeCell(
-		level: ActivityMeterLevel,
-		char: string,
-		theme: { fg: (style: ThemeColor, s: string) => string },
-		color: ThemeColor = "accent",
-		dimmed?: boolean,
-	): string {
-		if (level === ActivityMeterLevel.IDLE) return theme.fg("dim", char);
-		const colored = theme.fg(color, char);
-		// ANSI dim (SGR 2) reduces color brightness; SGR 22 resets dim/bold.
-		// This is necessary because theme.fg("dim", ...) sets a gray color instead
-		// of applying the ANSI dim attribute, so nesting would overwrite the accent color.
-		return dimmed ? `\x1b[2m${colored}\x1b[22m` : colored;
-	}
-}
-
-/**
- * Counts words across a split stream without re-scanning earlier chunks: each
- * stream remembers whether it ended mid-word, so a word broken across two
- * deltas is counted once. Used as the output-token estimate for providers that
- * never report usage.
- */
-export class StreamingWordCounter {
-	#inWordByStream = new Map<string, boolean>();
-
-	count(text: string, stream = "default"): number {
-		let inWord = this.#inWordByStream.get(stream) ?? false;
-		let count = 0;
-		for (let i = 0; i < text.length; i++) {
-			const code = text.charCodeAt(i);
-			if (isWhitespace(code)) {
-				inWord = false;
-			} else if (!inWord) {
-				count++;
-				inWord = true;
-			}
-		}
-		this.#inWordByStream.set(stream, inWord);
-		return count;
-	}
-
-	reset(): void {
-		this.#inWordByStream.clear();
-	}
-}
-
-function isWhitespace(code: number): boolean {
-	return (
-		code === 32 ||
-		(code >= 9 && code <= 13) ||
-		code === 160 ||
-		code === 0x1680 ||
-		(code >= 0x2000 && code <= 0x200a) ||
-		code === 0x2028 ||
-		code === 0x2029 ||
-		code === 0x202f ||
-		code === 0x205f ||
-		code === 0x3000 ||
-		code === 0xfeff
-	);
-}

package/src/config/modelCatalogue.ts

@@ -1,54 +0,0 @@
-/**
- * Cached view of the models pi reports as usable, with the thinking levels the
- * registry declares for each.
- *
- * The registry is the only authority the pickers consult: it decides both which
- * models may be selected and which thinking levels each one supports, so a
- * confirmed selection needs no further verification. Whether a model actually
- * answers is decided when it runs, not here.
- */
-
-import { getSupportedThinkingLevels, type Api, type Model } from "@earendil-works/pi-ai";
-import { isThinkingLevel, modelRefLabel, parseRef, type ModelRef, type ThinkingLevel } from "../shared/modelRefs.ts";
-
-export interface ModelCatalogue {
-	/** Selectable models, sorted by `provider/id`. */
-	availableRefs(): ModelRef[];
-	/** Levels the registry declares for a model, in pi's canonical order. */
-	thinkingLevelsFor(ref: ModelRef): ThinkingLevel[];
-}
-
-/** The slice of pi's `ModelRegistry` a catalogue is built from. */
-export interface CatalogueRegistry {
-	getAvailable(): Model<Api>[];
-}
-
-// Lives for as long as the registry object itself: pi hands extensions a stable
-// registry per session, and dropping the registry drops its catalogue with it.
-const CATALOGUES = new WeakMap<CatalogueRegistry, ModelCatalogue>();
-
-export function getModelCatalogue(registry: CatalogueRegistry): ModelCatalogue {
-	const cached = CATALOGUES.get(registry);
-	if (cached) return cached;
-	const built = buildCatalogue(registry);
-	CATALOGUES.set(registry, built);
-	return built;
-}
-
-function buildCatalogue(registry: CatalogueRegistry): ModelCatalogue {
-	const levels = new Map<string, ThinkingLevel[]>();
-	for (const model of registry.getAvailable()) {
-		const key = `${model.provider}/${model.id}`;
-		if (levels.has(key)) continue;
-		levels.set(key, getSupportedThinkingLevels(model).filter(isThinkingLevel));
-	}
-
-	const refs = [...levels.keys()]
-		.sort((a, b) => a.localeCompare(b))
-		.map((key) => parseRef(key));
-
-	return {
-		availableRefs: () => [...refs],
-		thinkingLevelsFor: (ref) => levels.get(modelRefLabel(ref)) ?? [],
-	};
-}

package/src/ui/modelLabel.ts

@@ -1,61 +0,0 @@
-import { truncateToWidth, visibleWidth, type SelectListLayoutOptions } from "@earendil-works/pi-tui";
-
-/**
- * Truncates a slash-namespaced model ref (`provider/accounts/.../model-name`)
- * so the parts that matter to identify the model survive first. Priority:
- * the model name (last segment) is mandatory, then the first segment (the
- * namespace root), then the segment right before the model name, then
- * whatever else still fits. Skipped runs collapse to a single ellipsis.
- */
-export function smartTruncateModelLabel(text: string, maxWidth: number): string {
-	const ELLIPSIS = "…";
-	const ANSI_RE = /\x1b\[[0-9;]*m/g;
-	if (visibleWidth(text) <= maxWidth) return text;
-	const segments = text.split("/");
-	const n = segments.length;
-	if (n === 1) return truncateToWidth(text, maxWidth, ELLIPSIS).replace(ANSI_RE, "");
-
-	const lastIndex = n - 1;
-	const kept = new Set<number>([lastIndex]);
-	const priority: number[] = [];
-	priority.push(0);
-	if (n - 2 > 0) priority.push(n - 2);
-	for (let i = 1; i <= n - 3; i++) priority.push(i);
-
-	const render = () => {
-		const parts: string[] = [];
-		let i = 0;
-		while (i < n) {
-			if (kept.has(i)) {
-				parts.push(segments[i]!);
-				i++;
-			} else {
-				while (i < n && !kept.has(i)) i++;
-				parts.push(ELLIPSIS);
-			}
-		}
-		return parts.join("/");
-	};
-
-	let current = render();
-	if (visibleWidth(current) > maxWidth) {
-		const skeleton = `${ELLIPSIS}/`;
-		const budget = Math.max(0, maxWidth - visibleWidth(skeleton));
-		return (skeleton + truncateToWidth(segments[lastIndex]!, budget, ELLIPSIS)).replace(ANSI_RE, "");
-	}
-
-	for (const index of priority) {
-		kept.add(index);
-		const candidate = render();
-		if (visibleWidth(candidate) > maxWidth) {
-			kept.delete(index);
-			continue;
-		}
-		current = candidate;
-	}
-	return current;
-}
-
-export const MODEL_LIST_LAYOUT: SelectListLayoutOptions = {
-	truncatePrimary: ({ text, maxWidth }) => smartTruncateModelLabel(text, maxWidth),
-};

package/src/ui/shimmer.ts

@@ -1,77 +0,0 @@
-/**
- * Codex-style light-sweep text shimmer.
- *
- * Vendored from pi-topping-persona-audit (`src/shimmer.ts`) rather than
- * depended on, so the extensions stay independently installable. The sweep
- * geometry — band half-width, padding, speed scaling — is load-bearing for
- * visual parity, so port changes upstream rather than tuning them here.
- * test/shimmer.test.mjs mirrors the upstream tests to catch drift.
- */
-
-import type { ThemeColor } from "@earendil-works/pi-coding-agent";
-
-const SHIMMER_SWEEP_S = 2.0;
-const SHIMMER_BAND_HALF = 5.0;
-const SHIMMER_PADDING = 10;
-
-/** The slice of pi's `Theme` shimmerString needs. Structural so callers can supply a stub. */
-export interface ShimmerTheme {
-	getFgAnsi(color: ThemeColor): string;
-	fg(color: ThemeColor, text: string): string;
-}
-
-/**
- * With `invert=true`, the text color is used at rest and the sweep moves toward
- * the theme's dim color.
- */
-export function shimmerString(
-	text: string,
-	elapsedMs: number,
-	theme: ShimmerTheme,
-	direction: "ltr" | "rtl" = "ltr",
-	speed: "slow" | "normal" | "fast" = "normal",
-	invert = false,
-): string {
-	const chars = [...text];
-	if (chars.length === 0) return "";
-	const shimmerBase = ansiToRgb(theme.getFgAnsi(invert ? "text" : "dim"));
-	const shimmerHighlight = ansiToRgb(theme.getFgAnsi(invert ? "dim" : "text"));
-	if (!shimmerBase || !shimmerHighlight) return theme.fg("text", text);
-	const period = chars.length + SHIMMER_PADDING * 2;
-	const unitsPerS = period / SHIMMER_SWEEP_S;
-	// The band crosses padding at either end while every character is still dim. Scaling only
-	// the stretch where it actually overlaps the text keeps that dark pause identical at every
-	// speed, so `speed` changes the sweep alone rather than the whole cycle.
-	const litEnter = SHIMMER_PADDING - SHIMMER_BAND_HALF;
-	const litExit = SHIMMER_PADDING + chars.length - 1 + SHIMMER_BAND_HALF;
-	const enterS = litEnter / unitsPerS;
-	const litS = (litExit - litEnter) / unitsPerS / (speed === "slow" ? 0.5 : speed === "fast" ? 2 : 1);
-	const phase = (elapsedMs / 1000) % (enterS + litS + (period - litExit) / unitsPerS);
-	const linear =
-		phase < enterS
-			? phase * unitsPerS
-			: phase < enterS + litS
-				? litEnter + ((phase - enterS) / litS) * (litExit - litEnter)
-				: litExit + (phase - enterS - litS) * unitsPerS;
-	const pos = direction === "rtl" ? period - linear : linear;
-
-	let out = "";
-	for (let i = 0; i < chars.length; i++) {
-		const ch = chars[i]!;
-		const dist = Math.abs(i + SHIMMER_PADDING - pos);
-		const t = dist <= SHIMMER_BAND_HALF ? 0.5 * (1 + Math.cos((Math.PI * dist) / SHIMMER_BAND_HALF)) : 0;
-		const alpha = t * 0.9;
-		const r = Math.round(shimmerHighlight[0] * alpha + shimmerBase[0] * (1 - alpha));
-		const g = Math.round(shimmerHighlight[1] * alpha + shimmerBase[1] * (1 - alpha));
-		const b = Math.round(shimmerHighlight[2] * alpha + shimmerBase[2] * (1 - alpha));
-		const bold = !invert && t > 0.2 ? "\x1b[1m" : "";
-		out += `${bold}\x1b[38;2;${r};${g};${b}m${ch}\x1b[22m`;
-	}
-	return out + "\x1b[0m";
-}
-
-function ansiToRgb(ansi: string): [number, number, number] | null {
-	const match = ansi.match(/^\x1b\[38;2;(\d+);(\d+);(\d+)m$/);
-	if (!match) return null;
-	return [Number(match[1]), Number(match[2]), Number(match[3])];
-}

package/src/ui/toolActivity.ts

@@ -1,35 +0,0 @@
-/**
- * Shared renderer for one-line tool-activity labels (`formatToolActivity`
- * output), so the MoA progress table and the cancel overlay color a tool
- * name and its argument identically, following pi's own read/grep/edit/write
- * renderers (bold `toolTitle` name, `accent` argument).
- *
- * Splits only at the first run of whitespace — the boundary
- * `formatToolActivity` itself produces between a tool name and its argument
- * — and never rewrites the text on either side, so activity strings stay
- * byte-identical for the runtime's equality-based loop detection.
- */
-
-import type { ThemeColor } from "@earendil-works/pi-coding-agent";
-
-/** The slice of pi's `Theme` this renderer needs. */
-export interface ActivityTheme {
-	fg(color: ThemeColor, text: string): string;
-	bold(text: string): string;
-}
-
-const ACTIVITY_SPLIT = /^(\s*)(\S+)(\s*)([\s\S]*)$/;
-
-/**
- * Highlight a plain activity string as a bold `toolTitle` tool name followed
- * by its `accent` argument, preserving every character — including leading
- * and repeated whitespace — exactly. Falls back to the raw string when no
- * tool token can be found at all (empty or whitespace-only input).
- */
-export function highlightActivity(theme: ActivityTheme, activity: string): string {
-	const match = activity.match(ACTIVITY_SPLIT);
-	if (!match) return activity;
-	const [, leading, tool, sep, detail] = match;
-	const styledTool = theme.fg("toolTitle", theme.bold(tool));
-	return detail ? `${leading}${styledTool}${sep}${theme.fg("accent", detail)}` : `${leading}${styledTool}${sep}`;
-}