@xynogen/pix-core 0.2.4 → 0.3.1

package/src/commands/models/models.ts

@@ -1,367 +0,0 @@
-/**
- * models.ts — enhanced /models command with benchlm rank + score
- *
- * Replaces (or supplements) the built-in /model selector by registering
- * /models (plural). Each row shows:
- *   <name>  <provider> · <ctx> · <cost> · 🏅 #rank score
- *
- * Sorted by benchlm rank when available (best first), then alphabetical.
- */
-
-import type {
-	ExtensionAPI,
-	ExtensionContext,
-} from "@earendil-works/pi-coding-agent";
-import { DynamicBorder } from "@earendil-works/pi-coding-agent";
-import {
-	Container,
-	fuzzyFilter,
-	Input,
-	matchesKey,
-	type SelectItem,
-	SelectList,
-	Text,
-	visibleWidth,
-} from "@earendil-works/pi-tui";
-import { lookupBenchmark, lookupModelsDev } from "../../lib/data";
-import { patchOutBuiltinModelCommand } from "./patch-builtin";
-
-// ─── Pure logic (exported for tests) ─────────────────────────────────────────
-
-export function fmtCtx(n: number): string {
-	if (!n || n < 1_000) return `${n}`;
-	if (n >= 1_000_000) {
-		const m = n / 1_000_000;
-		return `${Number.isInteger(m) ? m.toFixed(0) : m.toFixed(1).replace(/\.0$/, "")}M`;
-	}
-	return `${Math.round(n / 1_000)}k`;
-}
-
-export function fmtCost(
-	entry: { cost?: { input?: number; output?: number } } | undefined,
-): string {
-	if (!entry?.cost) return "\u2014";
-	const i = entry.cost.input ?? 0;
-	const o = entry.cost.output ?? 0;
-	if (i === 0 && o === 0) return "free";
-	return `${i.toFixed(2)}/${o.toFixed(2)}`;
-}
-
-export function benchStars(score: number | null | undefined): {
-	filled: number;
-	empty: number;
-} {
-	const total = 5;
-	let filled = 1;
-	if (typeof score === "number") {
-		if (score >= 90) filled = 5;
-		else if (score >= 80) filled = 4;
-		else if (score >= 70) filled = 3;
-		else if (score >= 50) filled = 2;
-	}
-	return { filled, empty: total - filled };
-}
-
-export type SortableModel = {
-	provider: string;
-	id: string;
-	name?: string;
-	score?: number | null;
-};
-
-export function sortModels<T extends SortableModel>(models: T[]): T[] {
-	return [...models].sort((a, b) => {
-		const sa = a.score ?? -1;
-		const sb = b.score ?? -1;
-		if (sa !== sb) return sb - sa;
-		return (a.name ?? a.id).localeCompare(b.name ?? b.id);
-	});
-}
-
-export async function showEnhancedPicker(
-	pi: ExtensionAPI,
-	ctx: ExtensionContext,
-): Promise<void> {
-	// Mirror the built-in /model selector, which calls refresh() then awaits
-	// getAvailable() (see model-selector.js). Without refresh(), this extension
-	// reads whatever `this.models` was last loaded into — which, depending on
-	// extension load order vs oauth/auth resolution, can omit oauth providers
-	// whose models were registered as built-ins but resolved after the last
-	// load (notably `openai-codex`). 9router survives because it's registered
-	// as a custom provider with an env-key apiKey. refresh() rebuilds the model
-	// list (resetOAuthProviders → loadModels → re-apply registered providers)
-	// so oauth-backed codex models reappear, exactly as the built-in does.
-	//
-	// The public ExtensionContext type narrows modelRegistry to a sync
-	// getAvailable() only; at runtime ctx.modelRegistry is the full
-	// ModelRegistry instance (verified in runner.js) with refresh() and an
-	// async-capable getAvailable(). Reach through the narrowed type.
-	type AvailableModels = ReturnType<typeof ctx.modelRegistry.getAvailable>;
-	const registry = ctx.modelRegistry as unknown as {
-		refresh?: () => void;
-		getAvailable(): AvailableModels | Promise<AvailableModels>;
-	};
-	registry.refresh?.();
-	const available = await registry.getAvailable();
-	if (available.length === 0) {
-		ctx.ui.notify("No models with configured auth.", "warning");
-		return;
-	}
-
-	const current = ctx.model;
-
-	// Build items with benchmark data
-	type Row = {
-		m: (typeof available)[number];
-		dev: ReturnType<typeof lookupModelsDev>;
-		bench: ReturnType<typeof lookupBenchmark>;
-	};
-	const rows: Row[] = available.map((m) => ({
-		m,
-		dev: lookupModelsDev(m.provider, m.id),
-		bench: lookupBenchmark(m.name ?? m.id),
-	}));
-
-	// Sort: by score desc (highest first), unscored last alphabetical
-	rows.sort((a, b) => {
-		const sa = a.bench?.overallScore ?? -1;
-		const sb = b.bench?.overallScore ?? -1;
-		if (sa !== sb) return sb - sa;
-		return (a.m.name ?? a.m.id).localeCompare(b.m.name ?? b.m.id);
-	});
-
-	// Show all models (no deduplication)
-	const dedupedRows = rows;
-
-	// items built inside the custom() factory so we have theme access for colors
-
-	const result = await ctx.ui.custom<string | null>(
-		(_tui, theme, _kb, done) => {
-			const container = new Container();
-			const accent = "accent";
-
-			// Find max rank width across all benchmarked rows for # padding
-			const maxRankWidth = Math.max(
-				...dedupedRows.map((r) => (r.bench ? String(r.bench.rank).length : 0)),
-				1,
-			);
-
-			// Mute low-info parts (separators, padding, #, ☆) so the actual values pop.
-			const mute = (s: string) => theme.fg("muted", s);
-			const sep = mute(" · ");
-
-			// Track rank per item value so fuzzy results can prioritize ranked models.
-			const rankByValue = new Map<string, number>();
-			for (const { m, bench } of dedupedRows) {
-				if (bench) rankByValue.set(`${m.provider}/${m.id}`, bench.rank);
-			}
-
-			const items: SelectItem[] = dedupedRows.map(({ m, dev, bench }) => {
-				const isCurrent =
-					current && m.provider === current.provider && m.id === current.id;
-
-				// Label: marker + muted '#' + bright rank + accent-colored model name
-				const marker = isCurrent ? theme.fg(accent, "▶") : " ";
-				let rankPrefix: string;
-				if (bench) {
-					const rankStr = String(bench.rank).padEnd(maxRankWidth);
-					rankPrefix = mute("#") + theme.fg("warning", rankStr);
-				} else {
-					rankPrefix = " ".repeat(maxRankWidth + 1);
-				}
-				// Display model id only; m.provider is routing provider, not part of id.
-				const idColored = theme.fg(accent, m.id);
-				const label = `${marker} ${rankPrefix} ${idColored}`;
-
-				// Description: ctx · cost · score stars
-				// Colors: ctx muted · cost success (free muted) · score+stars warning
-				const ctxRaw = fmtCtx(dev?.limit?.context ?? 0);
-				const ctxStr = mute(ctxRaw.padStart(4));
-				const rawCost = fmtCost(dev);
-				let costSeg: string;
-				if (rawCost === "—") {
-					costSeg = theme.fg("dim", "—".padEnd(10));
-				} else if (rawCost === "free") {
-					costSeg = mute("free".padEnd(10));
-				} else {
-					costSeg = theme.fg("success", rawCost.padEnd(10));
-				}
-				let benchSeg = "";
-				if (bench) {
-					const score = bench.overallScore ?? "?";
-					const s = bench.overallScore;
-					let filled = 1;
-					if (typeof s === "number") {
-						if (s >= 90) filled = 5;
-						else if (s >= 80) filled = 4;
-						else if (s >= 70) filled = 3;
-						else if (s >= 50) filled = 2;
-					}
-					const starBar =
-						theme.fg("warning", "★".repeat(filled)) +
-						mute("☆".repeat(5 - filled));
-					benchSeg = `⚡${theme.fg("warning", String(score))} ${starBar}`;
-				}
-				const desc = [ctxStr, costSeg, benchSeg].filter(Boolean).join(sep);
-
-				return {
-					value: `${m.provider}/${m.id}`,
-					label,
-					description: desc,
-				};
-			});
-
-			const currentIdx = current
-				? items.findIndex(
-						(it) => it.value === `${current.provider}/${current.id}`,
-					)
-				: 0;
-
-			container.addChild(new DynamicBorder((s) => theme.fg(accent, s)));
-			container.addChild(
-				new Text(theme.fg(accent, theme.bold("󰚩  Select model"))),
-			);
-			container.addChild(
-				new Text(
-					theme.fg(
-						"dim",
-						"context & pricing from models.dev · ranks from benchlm.ai",
-					),
-				),
-			);
-
-			// Widest label (visible width, ANSI-stripped) so the model name
-			// column never truncates to "…". Add gap headroom.
-			const widestLabel = items.reduce(
-				(w, it) => Math.max(w, visibleWidth(it.label)),
-				0,
-			);
-
-			const search = new Input();
-			const list = new SelectList(
-				items,
-				Math.min(items.length, 14),
-				{
-					selectedPrefix: (t) => theme.fg(accent, t),
-					selectedText: (t) => theme.fg(accent, t),
-					description: (t) => t, // raw — per-segment colors set in items.map
-					scrollInfo: (t) => theme.fg("dim", t),
-					noMatch: (t) => theme.fg("warning", t),
-				},
-				{
-					minPrimaryColumnWidth: widestLabel + 2,
-					maxPrimaryColumnWidth: widestLabel + 2,
-				},
-			);
-			if (currentIdx >= 0) list.setSelectedIndex(currentIdx);
-
-			list.onSelect = (item) => done(item.value);
-			list.onCancel = () => done(null);
-			search.onEscape = () => done(null);
-
-			const applyFuzzy = (query: string) => {
-				const internal = list as unknown as {
-					items: SelectItem[];
-					filteredItems: SelectItem[];
-					selectedIndex: number;
-					invalidate(): void;
-				};
-				const q = query.trim();
-				let next: SelectItem[];
-				if (q.length === 0) {
-					next = internal.items;
-				} else if (/^\d+$/.test(q)) {
-					// Pure number → match by benchlm rank, not name.
-					const wanted = Number(q);
-					next = internal.items.filter(
-						(it) => rankByValue.get(it.value) === wanted,
-					);
-				} else {
-					next = fuzzyFilter(
-						internal.items,
-						q,
-						(it) => `${it.label} ${it.description ?? ""}`,
-					);
-					// Stable sort: ranked models (by rank asc) before unranked.
-					next = next
-						.map((it, i) => ({ it, i }))
-						.sort((a, b) => {
-							const ra = rankByValue.get(a.it.value) ?? Infinity;
-							const rb = rankByValue.get(b.it.value) ?? Infinity;
-							if (ra !== rb) return ra - rb;
-							return a.i - b.i;
-						})
-						.map(({ it }) => it);
-				}
-				internal.filteredItems = next;
-				internal.selectedIndex = 0;
-				internal.invalidate();
-			};
-
-			container.addChild(new Text(theme.fg("muted", "Search:")));
-			container.addChild(search);
-			container.addChild(list);
-			container.addChild(
-				new Text(
-					theme.fg(
-						"dim",
-						"fuzzy search · ↑↓ navigate · enter select · esc cancel",
-					),
-				),
-			);
-			container.addChild(new DynamicBorder((s) => theme.fg(accent, s)));
-
-			return {
-				render(w: number) {
-					return container.render(w);
-				},
-				invalidate() {
-					container.invalidate();
-				},
-				handleInput(data: string) {
-					// Detect keys via pi-tui's own parser — the same recognition
-					// SelectList uses. Arrows arrive as named keys ("up"/"down"),
-					// not raw escape sequences, so string-equality checks fail.
-					const isNav = matchesKey(data, "up") || matchesKey(data, "down");
-					if (isNav || matchesKey(data, "enter")) {
-						list.handleInput?.(data);
-					} else if (matchesKey(data, "escape")) {
-						done(null);
-					} else {
-						search.handleInput?.(data);
-						applyFuzzy(search.getValue?.() ?? "");
-					}
-					container.invalidate();
-				},
-			};
-		},
-	);
-
-	if (!result) return;
-
-	// Apply selection
-	const [provider, ...rest] = result.split("/");
-	const id = rest.join("/");
-	const picked = available.find((m) => m.provider === provider && m.id === id);
-	if (!picked) {
-		ctx.ui.notify(`Model not found: ${result}`, "error");
-		return;
-	}
-	const ok = await pi.setModel(picked);
-	if (ok) ctx.ui.notify(`Switched to ${picked.name ?? picked.id}`, "info");
-	else ctx.ui.notify(`Failed to switch to ${picked.id}`, "error");
-}
-
-export default function modelPickerExtension(pi: ExtensionAPI) {
-	// Remove Pi's built-in /model so only the enhanced /models picker remains.
-	// Self-healing: re-applies on every load, so a Pi upgrade can't restore it.
-	patchOutBuiltinModelCommand();
-
-	const handler = async (_args: unknown, ctx: ExtensionContext) => {
-		await showEnhancedPicker(pi, ctx);
-	};
-	pi.registerCommand("models", {
-		description: "Enhanced model picker — shows benchlm rank + score",
-		handler,
-	});
-}

package/src/commands/models/patch-builtin.test.ts

@@ -1,66 +0,0 @@
-import { describe, expect, it } from "bun:test";
-import { mkdtempSync, readFileSync, writeFileSync } from "node:fs";
-import { tmpdir } from "node:os";
-import { join } from "node:path";
-
-// Pure replacement tested in isolation (the exported fn resolves the host
-// package, which isn't present in the test sandbox).
-const MODEL_COMMAND_LINE =
-	'{ name: "model", description: "Select model (opens selector UI)" },';
-
-function escapeRegExp(text: string): string {
-	return text.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
-}
-
-function patchSource(source: string): string {
-	if (!source.includes(MODEL_COMMAND_LINE)) return source;
-	return source.replace(
-		new RegExp(`[ \\t]*${escapeRegExp(MODEL_COMMAND_LINE)}\\n?`),
-		"",
-	);
-}
-
-const UNPATCHED = `export const BUILTIN_SLASH_COMMANDS = [
-    { name: "settings", description: "Open settings menu" },
-    { name: "model", description: "Select model (opens selector UI)" },
-    { name: "login", description: "Configure provider authentication" },
-];
-`;
-
-describe("patch-builtin /model removal", () => {
-	it("removes the built-in /model line and keeps neighbors", () => {
-		const out = patchSource(UNPATCHED);
-		expect(out).not.toContain('name: "model"');
-		expect(out).toContain('name: "settings"');
-		expect(out).toContain('name: "login"');
-	});
-
-	it("is idempotent — second pass is a no-op", () => {
-		const once = patchSource(UNPATCHED);
-		const twice = patchSource(once);
-		expect(twice).toBe(once);
-	});
-
-	it("leaves an already-clean file untouched", () => {
-		const clean = `export const X = [\n    { name: "login" },\n];\n`;
-		expect(patchSource(clean)).toBe(clean);
-	});
-
-	it("does not strip the plural /models entry", () => {
-		const withPlural = `[
-    { name: "models", description: "Enhanced picker" },
-    { name: "model", description: "Select model (opens selector UI)" },
-]`;
-		const out = patchSource(withPlural);
-		expect(out).toContain('name: "models"');
-		expect(out).not.toContain('{ name: "model", description');
-	});
-
-	it("round-trips through disk", () => {
-		const dir = mkdtempSync(join(tmpdir(), "pix-patch-"));
-		const file = join(dir, "slash-commands.js");
-		writeFileSync(file, UNPATCHED, "utf8");
-		writeFileSync(file, patchSource(readFileSync(file, "utf8")), "utf8");
-		expect(readFileSync(file, "utf8")).not.toContain('name: "model"');
-	});
-});

package/src/commands/models/patch-builtin.ts

@@ -1,120 +0,0 @@
-/**
- * patch-builtin.ts — strip Pi's built-in /model slash command at load time.
- *
- * Built-in commands can't be removed via the extension API, so we edit Pi's
- * compiled slash-commands.js directly. Done on every load: idempotent and
- * self-healing across Pi upgrades, so no manual repatch is ever needed.
- *
- * Resolution strategy (in order):
- *   1. Locate the `pi` binary via PATH → infer package root from its realpath.
- *      The binary is always at <pkg>/dist/cli.js so ../../ is the package root.
- *   2. Probe well-known global install locations (bun, npm).
- *   3. Fall back to createRequire against the extension's own node_modules
- *      (works when pi and the extension share the same install tree).
- */
-
-import { execSync } from "node:child_process";
-import { existsSync, readFileSync, writeFileSync } from "node:fs";
-import { createRequire } from "node:module";
-import { homedir } from "node:os";
-import { dirname, join, resolve } from "node:path";
-
-const MODEL_COMMAND_LINE =
-	'{ name: "model", description: "Select model (opens selector UI)" },';
-
-/** Candidate slash-commands.js paths, most-specific first. */
-function candidatePaths(): string[] {
-	const paths: string[] = [];
-
-	// 1. Resolve via the running `pi` binary → its realpath gives the dist dir.
-	try {
-		const piReal = execSync("realpath $(which pi)", {
-			encoding: "utf8",
-			stdio: ["pipe", "pipe", "pipe"],
-		}).trim();
-		if (piReal) {
-			// piReal = /.../pi-coding-agent/dist/cli.js → dist/ → ../dist/core/
-			const distCore = resolve(dirname(piReal), "core");
-			paths.push(join(distCore, "slash-commands.js"));
-		}
-	} catch {
-		// `pi` not on PATH or `which`/`realpath` unavailable — skip
-	}
-
-	// 2. Well-known global install locations.
-	const home = homedir();
-	const globalRoots = [
-		join(home, ".bun", "install", "global", "node_modules"),
-		join(home, ".npm-global", "lib", "node_modules"),
-		"/usr/local/lib/node_modules",
-		"/usr/lib/node_modules",
-	];
-	for (const root of globalRoots) {
-		paths.push(
-			join(
-				root,
-				"@earendil-works",
-				"pi-coding-agent",
-				"dist",
-				"core",
-				"slash-commands.js",
-			),
-		);
-	}
-
-	// 3. Fallback: createRequire from this file (works when extension is co-installed).
-	try {
-		const require = createRequire(import.meta.url);
-		const entry = require.resolve("@earendil-works/pi-coding-agent");
-		paths.push(resolve(dirname(entry), "core", "slash-commands.js"));
-	} catch {
-		// local resolution failed — skip
-	}
-
-	return paths;
-}
-
-/** Locate the host's compiled slash-commands.js, or null if not found. */
-function findSlashCommandsFile(): string | null {
-	for (const p of candidatePaths()) {
-		if (existsSync(p)) return p;
-	}
-	return null;
-}
-
-/**
- * Remove the built-in /model command line from Pi's slash-commands.js.
- * Idempotent: returns silently if the file is missing or already patched.
- */
-export function patchOutBuiltinModelCommand(): void {
-	const file = findSlashCommandsFile();
-	if (!file) return;
-
-	let source: string;
-	try {
-		source = readFileSync(file, "utf8");
-	} catch {
-		return;
-	}
-
-	if (!source.includes(MODEL_COMMAND_LINE)) return; // already patched
-
-	const patched = source.replace(
-		new RegExp(`[ \\t]*${escapeRegExp(MODEL_COMMAND_LINE)}\\n?`),
-		"",
-	);
-	if (patched === source) return;
-
-	try {
-		writeFileSync(file, patched, "utf8");
-	} catch {
-		// Read-only install — leave /model in place rather than crash.
-	}
-}
-
-function escapeRegExp(text: string): string {
-	return text.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
-}
-
-// Export for tests
-export { candidatePaths, findSlashCommandsFile };

package/src/commands/tools.test.ts

@@ -1,15 +0,0 @@
-/**
- * Smoke tests for the command extensions merged from pix-tools.
- * Each default export must be a registrable (pi) => void function.
- */
-
-import { describe, expect, it } from "bun:test";
-
-describe("merged pix-tools commands", () => {
-	for (const name of ["diff"]) {
-		it(`${name} exports a register function`, async () => {
-			const mod = await import(`./${name}/${name}.ts`);
-			expect(mod.default).toBeFunction();
-		});
-	}
-});

package/src/commands/update/update.test.ts

@@ -1,112 +0,0 @@
-import { describe, expect, it } from "bun:test";
-import {
-	commandFor,
-	formatUpdateSummary,
-	type InstallMethod,
-	isTransient,
-	PACKAGE_NAME,
-} from "./update.ts";
-
-describe("isTransient", () => {
-	it("matches network errors", () => {
-		expect(isTransient("ETIMEDOUT")).toBe(true);
-		expect(isTransient("ECONNRESET")).toBe(true);
-		expect(isTransient("ECONNREFUSED")).toBe(true);
-		expect(isTransient("socket hang up")).toBe(true);
-		expect(isTransient("network error occurred")).toBe(true);
-	});
-
-	it("matches HTTP status codes", () => {
-		expect(isTransient("Error 429: Too many requests")).toBe(true);
-		expect(isTransient("502 Bad Gateway")).toBe(true);
-		expect(isTransient("503 Service Unavailable")).toBe(true);
-		expect(isTransient("504 Gateway Timeout")).toBe(true);
-	});
-
-	it("matches timeout/temporary", () => {
-		expect(isTransient("Request timeout after 30s")).toBe(true);
-		expect(isTransient("temporary failure")).toBe(true);
-		expect(isTransient("EAI_AGAIN")).toBe(true);
-	});
-
-	it("returns false for permanent errors", () => {
-		expect(isTransient("permission denied")).toBe(false);
-		expect(isTransient("command not found")).toBe(false);
-		expect(isTransient("syntax error")).toBe(false);
-		expect(isTransient("")).toBe(false);
-	});
-
-	it("is case-insensitive", () => {
-		expect(isTransient("NETWORK FAILURE")).toBe(true);
-		expect(isTransient("Timeout after 30s")).toBe(true);
-	});
-});
-
-describe("commandFor", () => {
-	it("returns correct command for each method", () => {
-		const methods: InstallMethod[] = ["vp", "bun", "npm", "brew"];
-		for (const m of methods) {
-			const spec = commandFor(m);
-			expect(spec).toBeDefined();
-			expect(spec?.command).toBeTruthy();
-			expect(spec?.label).toBeTruthy();
-		}
-	});
-
-	it("vp uses vp add -g", () => {
-		const spec = commandFor("vp")!;
-		expect(spec.command).toBe("vp");
-		expect(spec.args).toContain(`${PACKAGE_NAME}@latest`);
-	});
-
-	it("bun uses bun add -g", () => {
-		const spec = commandFor("bun")!;
-		expect(spec.command).toBe("bun");
-		expect(spec.args).toContain(`${PACKAGE_NAME}@latest`);
-	});
-
-	it("npm uses npm install -g", () => {
-		const spec = commandFor("npm")!;
-		expect(spec.command).toBe("npm");
-		expect(spec.args).toContain("-g");
-		expect(spec.args).toContain(`${PACKAGE_NAME}@latest`);
-	});
-
-	it("brew uses sh -lc", () => {
-		const spec = commandFor("brew")!;
-		expect(spec.command).toBe("/bin/sh");
-		expect(spec.label).toContain("brew upgrade");
-	});
-
-	it("native returns undefined", () => {
-		expect(commandFor("native")).toBeUndefined();
-	});
-});
-
-describe("formatUpdateSummary", () => {
-	it("shows updated message when version changed", () => {
-		const msg = formatUpdateSummary("0.75.0", "0.76.0", 1);
-		expect(msg).toContain("0.75.0 → 0.76.0");
-	});
-
-	it("shows up-to-date when version unchanged", () => {
-		const msg = formatUpdateSummary("0.76.0", "0.76.0", 1);
-		expect(msg).toContain("up to date");
-		expect(msg).toContain("0.76.0");
-	});
-
-	it("shows retry count when attempts > 1", () => {
-		const msg = formatUpdateSummary("0.75.0", "0.76.0", 3);
-		expect(msg).toContain("Retried 2 transient failure");
-	});
-
-	it("no retry mention when attempts = 1", () => {
-		const msg = formatUpdateSummary("0.75.0", "0.76.0", 1);
-		expect(msg).not.toContain("Retried");
-	});
-
-	it("handles unknown versions gracefully", () => {
-		const msg = formatUpdateSummary("unknown", "unknown", 1);
-		expect(msg).toContain("up to date");
-	});
-});