@xynogen/pix-core 0.1.6 → 0.1.7

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@xynogen/pix-core",
-	"version": "0.1.6",
+	"version": "0.1.7",
 	"description": "Pi extension — core UI/UX bundle (welcome banner, footer, model picker, self-update)",
 	"type": "module",
 	"main": "src/index.ts",

package/src/commands/models/models.ts

@@ -24,6 +24,7 @@ import {
 	visibleWidth,
 } from "@earendil-works/pi-tui";
 import { lookupBenchmark, lookupModelsDev } from "../../lib/data";
+import { patchOutBuiltinModelCommand } from "./patch-builtin";
 
 // ─── Pure logic (exported for tests) ─────────────────────────────────────────
 
@@ -352,6 +353,10 @@ export async function showEnhancedPicker(
 }
 
 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);
 	};

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

@@ -0,0 +1,66 @@
+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

@@ -0,0 +1,60 @@
+/**
+ * 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.
+ */
+
+import { readFileSync, writeFileSync } from "node:fs";
+import { createRequire } from "node:module";
+import { dirname, resolve } from "node:path";
+
+const HOST_PACKAGE = "@earendil-works/pi-coding-agent";
+const MODEL_COMMAND_LINE =
+	'{ name: "model", description: "Select model (opens selector UI)" },';
+
+/** Locate the host's compiled slash-commands.js, or null if it can't be found. */
+function findSlashCommandsFile(): string | null {
+	try {
+		const require = createRequire(import.meta.url);
+		const entry = require.resolve(HOST_PACKAGE);
+		return resolve(dirname(entry), "core", "slash-commands.js");
+	} catch {
+		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; // file not present (different Pi layout) — nothing to do
+	}
+
+	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, "\\$&");
+}