@xynogen/pix-optimizer 1.1.6 → 1.1.7

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@xynogen/pix-optimizer",
-	"version": "1.1.6",
+	"version": "1.1.7",
 	"description": "Performance optimization suite for Pi Coding Agent - caveman mode + RTK tool rewriting + jq/TOON JSON compression + ponytail lazy-dev mode",
 	"type": "module",
 	"main": "src/index.ts",

package/src/caveman.ts

@@ -10,6 +10,7 @@ import type {
 	ExtensionCommandContext,
 	ExtensionContext,
 } from "@earendil-works/pi-coding-agent";
+import { loadOptValue, saveOptValue } from "./persist.ts";
 import type { OptimizerHandle, OptimizerStatus } from "./status.ts";
 
 // ── Levels ────────────────────────────────────────────────────────────────────
@@ -155,11 +156,16 @@ export function caveman(
 	// -- Restore live level from the session log on load --
 
 	pi.on("session_start", async (_event, ctx) => {
+		// Session log first (survives in-session branch nav), then disk (survives
+		// a full quit/restart). Disk wins when present so a new session restores
+		// the last chosen level instead of defaulting to off.
 		for (const entry of ctx.sessionManager.getEntries()) {
 			if (entry.type === "custom" && entry.customType === "caveman-level") {
 				level = (entry.data as { level: Level })?.level ?? level;
 			}
 		}
+		const saved = loadOptValue("caveman");
+		if (saved && LEVELS.includes(saved as Level)) level = saved as Level;
 		syncStatus(ctx);
 	});
 
@@ -182,6 +188,7 @@ export function caveman(
 		level = resolved;
 
 		pi.appendEntry("caveman-level", { level });
+		saveOptValue("caveman", level);
 		syncStatus(ctx);
 
 		ctx.ui.notify(

package/src/json.ts

@@ -22,6 +22,7 @@ import type {
 	ExtensionCommandContext,
 	ExtensionContext,
 } from "@earendil-works/pi-coding-agent";
+import { loadOptValue, saveOptValue } from "./persist.ts";
 import type { OptimizerHandle, OptimizerStatus } from "./status.ts";
 
 // ── System prompt ───────────────────────────────────────────────────────────
@@ -225,6 +226,9 @@ export function json(
 	}
 
 	pi.on("session_start", async (_event, ctx) => {
+		// Restore the user's on/off choice from disk (survives quit/restart).
+		const saved = loadOptValue("toon");
+		if (saved === "on" || saved === "off") enabled = saved === "on";
 		syncStatus(ctx);
 	});
 	pi.on("agent_start", async (_event, ctx) => {
@@ -277,6 +281,7 @@ export function json(
 		ctx: ExtensionCommandContext,
 	): Promise<void> {
 		enabled = value === "on";
+		saveOptValue("toon", enabled ? "on" : "off");
 
 		syncStatus(ctx);
 		ctx.ui.notify(`JSON/TOON guidance ${enabled ? "on" : "off"}.`, "info");

package/src/persist.test.ts

@@ -0,0 +1,56 @@
+/**
+ * Regression test for the cross-session reset bug: optimizer tool states
+ * (caveman/ponytail/rtk/toon) were lost on a full quit/restart because they
+ * only persisted to the session log. persist.ts adds disk persistence so a
+ * value written in one session is readable in the next.
+ */
+
+import { afterAll, beforeAll, describe, expect, test } from "bun:test";
+import { mkdtempSync, rmSync } from "node:fs";
+import { tmpdir } from "node:os";
+import { join } from "node:path";
+import { loadOptValue, saveOptValue } from "./persist.ts";
+
+let tmpAgentDir: string;
+
+beforeAll(() => {
+	tmpAgentDir = mkdtempSync(join(tmpdir(), "optimizer-persist-test-"));
+	process.env.PI_CODING_AGENT_DIR = tmpAgentDir;
+});
+
+afterAll(() => {
+	delete process.env.PI_CODING_AGENT_DIR;
+	try {
+		rmSync(tmpAgentDir, { recursive: true });
+	} catch {
+		// temp dir may already be gone — safe to ignore
+	}
+});
+
+describe("optimizer persistence", () => {
+	test("returns undefined before anything is saved", () => {
+		expect(loadOptValue("caveman")).toBeUndefined();
+	});
+
+	test("round-trips a single tool value across save/load (new-session sim)", () => {
+		saveOptValue("caveman", "lite");
+		// A fresh load (as a new session would do) sees the persisted value.
+		expect(loadOptValue("caveman")).toBe("lite");
+	});
+
+	test("persists each tool independently in one shared file", () => {
+		saveOptValue("ponytail", "full");
+		saveOptValue("rtk", "off");
+		saveOptValue("toon", "on");
+		expect(loadOptValue("caveman")).toBe("lite");
+		expect(loadOptValue("ponytail")).toBe("full");
+		expect(loadOptValue("rtk")).toBe("off");
+		expect(loadOptValue("toon")).toBe("on");
+	});
+
+	test("overwriting one tool leaves the others intact", () => {
+		saveOptValue("caveman", "ultra");
+		expect(loadOptValue("caveman")).toBe("ultra");
+		expect(loadOptValue("ponytail")).toBe("full");
+	});
+});

package/src/persist.ts

@@ -0,0 +1,50 @@
+/**
+ * persist.ts — disk-backed persistence for the /optimizer tool states.
+ *
+ * caveman/ponytail previously saved only to the session log (lost on a fresh
+ * session); rtk/toon never persisted at all. This stores every tool's current
+ * value in one file under the agent dir so the picker survives a full quit and
+ * restart. Each tool reads its value on session_start and writes on run().
+ *
+ *   ~/.pi/agent/optimizer.json  →  { "caveman": "lite", "rtk": "on", ... }
+ */
+
+import { existsSync, mkdirSync, readFileSync, writeFileSync } from "node:fs";
+import { dirname, join } from "node:path";
+import { getAgentDir } from "@earendil-works/pi-coding-agent";
+import type { OptimizerTool } from "./status.ts";
+
+type OptimizerConfig = Partial<Record<OptimizerTool, string>>;
+
+function getStatePath(): string {
+	return join(getAgentDir(), "optimizer.json");
+}
+
+function readAll(): OptimizerConfig {
+	try {
+		const sp = getStatePath();
+		if (!existsSync(sp)) return {};
+		const raw = JSON.parse(readFileSync(sp, "utf-8")) as OptimizerConfig;
+		return raw && typeof raw === "object" ? raw : {};
+	} catch {
+		// Missing, corrupt, or no getAgentDir (tests/headless) — start clean.
+		return {};
+	}
+}
+
+/** Read a single tool's persisted value, or undefined if none. */
+export function loadOptValue(tool: OptimizerTool): string | undefined {
+	return readAll()[tool];
+}
+
+/** Persist a single tool's value, merging into the shared config file. */
+export function saveOptValue(tool: OptimizerTool, value: string): void {
+	try {
+		const sp = getStatePath();
+		mkdirSync(dirname(sp), { recursive: true });
+		const next = { ...readAll(), [tool]: value };
+		writeFileSync(sp, JSON.stringify(next, null, 2), "utf-8");
+	} catch (err) {
+		console.warn(`optimizer: persist ${tool} failed:`, err);
+	}
+}

package/src/ponytail.ts

@@ -14,6 +14,7 @@ import type {
 	ExtensionCommandContext,
 	ExtensionContext,
 } from "@earendil-works/pi-coding-agent";
+import { loadOptValue, saveOptValue } from "./persist.ts";
 import type { OptimizerHandle, OptimizerStatus } from "./status.ts";
 
 // ── Levels ────────────────────────────────────────────────────────────────────
@@ -159,11 +160,15 @@ export function ponytail(
 	// -- Restore live level from the session log on load --
 
 	pi.on("session_start", async (_event, ctx) => {
+		// Session log first (in-session branch nav), then disk (survives a full
+		// quit/restart). Disk wins when present.
 		for (const entry of ctx.sessionManager.getEntries()) {
 			if (entry.type === "custom" && entry.customType === "ponytail-level") {
 				level = (entry.data as { level: Level })?.level ?? level;
 			}
 		}
+		const saved = loadOptValue("ponytail");
+		if (saved && LEVELS.includes(saved as Level)) level = saved as Level;
 		syncStatus(ctx);
 	});
 
@@ -186,6 +191,7 @@ export function ponytail(
 		level = resolved;
 
 		pi.appendEntry("ponytail-level", { level });
+		saveOptValue("ponytail", level);
 		syncStatus(ctx);
 
 		ctx.ui.notify(

package/src/rtk.ts

@@ -11,6 +11,7 @@ import type {
 	ExtensionCommandContext,
 	ExtensionContext,
 } from "@earendil-works/pi-coding-agent";
+import { loadOptValue, saveOptValue } from "./persist.ts";
 import type { OptimizerHandle, OptimizerStatus } from "./status.ts";
 
 /**
@@ -283,6 +284,9 @@ export function rtk(
 	// Keep the status indicator in sync across the agent lifecycle. Probe
 	// availability on session start so the icon reflects reality immediately.
 	pi.on("session_start", async (_event, ctx) => {
+		// Restore the user's on/off choice from disk (survives quit/restart).
+		const saved = loadOptValue("rtk");
+		if (saved === "on" || saved === "off") enabled = saved === "on";
 		const probe = await checkRtkAvailability();
 		if (!probe.available && !warnedMissing) {
 			ctx.ui.notify(
@@ -307,6 +311,7 @@ export function rtk(
 		ctx: ExtensionCommandContext,
 	): Promise<void> {
 		enabled = value === "on";
+		saveOptValue("rtk", enabled ? "on" : "off");
 
 		await checkRtkAvailability();
 		syncStatus(ctx);