@diegopetrucci/pi-extensions 0.1.21 → 0.1.24

package/README.md

@@ -5,7 +5,8 @@ A collection of [pi](https://github.com/earendil-works/pi-mono) agent extensions
 - [`confirm-destructive`](./extensions/confirm-destructive): Confirms before destructive session actions like clear, switch, and fork.
 - [`context-cap`](./extensions/context-cap): Caps effective model context windows at 200k tokens by default so pi avoids the `dumb zone`; toggle temporarily with `/context-cap`.
 - [`context-inspector`](./extensions/context-inspector): Adds `/context`, a local self-contained HTML dashboard that breaks down where the current session context is going, with category overview, top offenders, and drilldown search.
-- [`librarian`](./extensions/librarian): Adds a GitHub research scout that asks whether to use an opt-in local repo checkout cache under the OS user cache directory, with cached repos expiring after 30 days of non-use.
+- [`dirty-repo-guard`](./extensions/dirty-repo-guard): Prompts before new sessions, session switches, or forks when the current git repo has uncommitted changes.
+- [`librarian`](./extensions/librarian): Adds a GitHub research scout with a local repo checkout cache enabled by default under the OS user cache directory, toggleable with `/librarian-cache`, with cached repos expiring after 30 days of non-use.
 - [`minimal-footer`](./extensions/minimal-footer): Replaces pi's built-in footer with a minimal configurable two-line layout: branch/repo on the first line, context/model on the second, optional `DUMB ZONE`, plus OpenAI Codex 5-hour and 7-day usage when available.
 - [`notify`](./extensions/notify): Sends configurable terminal, desktop, bell, and sound notifications when pi finishes and is ready for input.
 - [`openai-fast`](./extensions/openai-fast): Adds `/fast` to enable OpenAI Codex Fast mode for ChatGPT-auth GPT-5.4 and GPT-5.5 by injecting the priority service tier.
@@ -26,7 +27,7 @@ pi install npm:@diegopetrucci/pi-extensions
 Or pin the GitHub package to this release:
 
 ```bash
-pi install git:github.com/diegopetrucci/pi-extensions@v0.1.21
+pi install git:github.com/diegopetrucci/pi-extensions@v0.1.24
 ```
 
 Or a specific extension:

package/extensions/dirty-repo-guard/README.md

@@ -0,0 +1,47 @@
+# dirty-repo-guard
+
+A small pi extension that prompts before session changes when the current git repo has uncommitted changes.
+
+This is copied from the original `dirty-repo-guard.ts` example in [`earendil-works/pi`](https://github.com/earendil-works/pi/blob/main/packages/coding-agent/examples/extensions/dirty-repo-guard.ts) and kept basically the same.
+
+## What it checks
+
+Before creating a new session, switching sessions, or forking, the extension runs:
+
+```bash
+git status --porcelain
+```
+
+If the command reports changed files, pi asks whether to proceed anyway. If pi is running without an interactive UI, matching actions are cancelled by default.
+
+## Install
+
+### Standalone npm package
+
+```bash
+pi install npm:@diegopetrucci/pi-dirty-repo-guard
+```
+
+### Collection package
+
+```bash
+pi install npm:@diegopetrucci/pi-extensions
+```
+
+### GitHub package
+
+```bash
+pi install git:github.com/diegopetrucci/pi-extensions
+```
+
+Then reload pi:
+
+```text
+/reload
+```
+
+## Notes
+
+- Hooks `session_before_switch` and `session_before_fork`.
+- Allows session changes outside git repos.
+- Cancels the action when the user declines.

package/extensions/dirty-repo-guard/index.ts

@@ -0,0 +1,56 @@
+/**
+ * Dirty Repo Guard Extension
+ *
+ * Prevents session changes when there are uncommitted git changes.
+ * Useful to ensure work is committed before switching context.
+ */
+
+import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
+
+async function checkDirtyRepo(
+	pi: ExtensionAPI,
+	ctx: ExtensionContext,
+	action: string,
+): Promise<{ cancel: boolean } | undefined> {
+	// Check for uncommitted changes
+	const { stdout, code } = await pi.exec("git", ["status", "--porcelain"]);
+
+	if (code !== 0) {
+		// Not a git repo, allow the action
+		return;
+	}
+
+	const hasChanges = stdout.trim().length > 0;
+	if (!hasChanges) {
+		return;
+	}
+
+	if (!ctx.hasUI) {
+		// In non-interactive mode, block by default
+		return { cancel: true };
+	}
+
+	// Count changed files
+	const changedFiles = stdout.trim().split("\n").filter(Boolean).length;
+
+	const choice = await ctx.ui.select(`You have ${changedFiles} uncommitted file(s). ${action} anyway?`, [
+		"Yes, proceed anyway",
+		"No, let me commit first",
+	]);
+
+	if (choice !== "Yes, proceed anyway") {
+		ctx.ui.notify("Commit your changes first", "warning");
+		return { cancel: true };
+	}
+}
+
+export default function (pi: ExtensionAPI) {
+	pi.on("session_before_switch", async (event, ctx) => {
+		const action = event.reason === "new" ? "new session" : "switch session";
+		return checkDirtyRepo(pi, ctx, action);
+	});
+
+	pi.on("session_before_fork", async (_event, ctx) => {
+		return checkDirtyRepo(pi, ctx, "fork");
+	});
+}

package/extensions/dirty-repo-guard/package.json

@@ -0,0 +1,27 @@
+{
+  "name": "@diegopetrucci/pi-dirty-repo-guard",
+  "version": "0.1.0",
+  "description": "A pi extension that prompts before session changes when the current git repo has uncommitted changes.",
+  "keywords": ["pi-package", "pi", "git", "safety"],
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/diegopetrucci/pi-extensions.git",
+    "directory": "extensions/dirty-repo-guard"
+  },
+  "files": [
+    "index.ts",
+    "README.md"
+  ],
+  "publishConfig": {
+    "access": "public"
+  },
+  "pi": {
+    "extensions": [
+      "index.ts"
+    ]
+  },
+  "peerDependencies": {
+    "@earendil-works/pi-coding-agent": "*"
+  }
+}

package/extensions/librarian/README.md

@@ -1,8 +1,8 @@
 # librarian
 
-A pi GitHub research scout inspired by `pi-librarian`, with an opt-in local checkout cache.
+A pi GitHub research scout inspired by `pi-librarian`, with a local checkout cache enabled by default.
 
-When the `librarian` tool runs, it asks whether to cache/reuse repository checkouts locally. If you say no, cancel, time out, or run without UI, it uses GitHub API/search and temporary fetched files only.
+When the `librarian` tool runs, it can cache/reuse repository checkouts locally. Use `/librarian-cache off` to force GitHub API/search and temporary fetched files only, or `/librarian-cache on` to re-enable cached local checkouts.
 
 ## Install
 
@@ -35,10 +35,21 @@ Then reload pi:
 - Tool name: `librarian`
 - Uses a restricted subagent with `bash` and `read`
 - Uses `gh` for GitHub search/API access
-- Asks on each call whether to use cached local checkouts
-- Defaults to no checkout/cache
+- Uses cached local checkouts by default
+- Toggle cache behavior for future calls with `/librarian-cache on | off | toggle | status`
 - Cached repos are removed lazily after 30 days without use
 
+## Commands
+
+```text
+/librarian-cache status
+/librarian-cache off
+/librarian-cache on
+/librarian-cache toggle
+```
+
+The command works in interactive mode, RPC mode, and print/JSON mode. It writes a global preference to `~/.pi/agent/extensions/librarian.json`, so separate non-UI invocations use the same setting. In non-UI modes, command feedback is written to stderr so stdout remains usable for normal output or JSON events.
+
 ## Cache location
 
 macOS:

package/extensions/librarian/index.ts

@@ -25,11 +25,14 @@ const CACHE_TTL_DAYS = 30;
 const CACHE_TTL_MS = CACHE_TTL_DAYS * 24 * 60 * 60 * 1000;
 const CACHE_METADATA_FILE = ".pi-librarian-cache.json";
 const CACHE_MARKER_FILE = ".pi-librarian-cache-used";
+const CACHE_CONFIG_FILE = "librarian.json";
 
 type LibrarianStatus = "running" | "done" | "error" | "aborted";
 
 type CacheMode = "disabled" | "enabled";
 
+const DEFAULT_CACHE_MODE: CacheMode = "enabled";
+
 type ToolCall = {
 	id: string;
 	name: string;
@@ -252,30 +255,72 @@ async function cleanupExpiredCache(cacheRoot: string): Promise<{ deleted: number
 	}
 }
 
-async function askForCache(ctx: ExtensionContext, cacheRoot: string): Promise<{ enabled: boolean; reason: string }> {
-	if (!ctx.hasUI) return { enabled: false, reason: "no UI available; using GitHub API/temp files only" };
+function getCacheConfigPath(): string {
+	return path.join(getAgentDir(), "extensions", CACHE_CONFIG_FILE);
+}
+
+function parseCacheMode(value: unknown): CacheMode | undefined {
+	if (typeof value === "string") {
+		const normalized = value.trim().toLowerCase();
+		if (normalized === "enabled" || normalized === "on" || normalized === "true") return "enabled";
+		if (normalized === "disabled" || normalized === "off" || normalized === "false") return "disabled";
+	}
+	if (value === true) return "enabled";
+	if (value === false) return "disabled";
+	return undefined;
+}
 
+async function readCachePreference(): Promise<CacheMode> {
 	try {
-		const enabled = await ctx.ui.confirm(
-			"Librarian repo cache",
-			[
-				"Cache/reuse local GitHub repo checkouts for this Librarian call?",
-				"",
-				`Cache directory: ${cacheRoot}`,
-				`Repos unused for ${CACHE_TTL_DAYS} days are removed lazily on future Librarian calls.`,
-				"",
-				"Choose No to use GitHub API and temporary fetched files only.",
-			].join("\n"),
-			{ timeout: 30_000 },
+		const raw = await fs.readFile(getCacheConfigPath(), "utf8");
+		const parsed = JSON.parse(raw) as {
+			cacheMode?: unknown;
+			cacheEnabled?: unknown;
+			cache?: { mode?: unknown; enabled?: unknown };
+		};
+		return (
+			parseCacheMode(parsed.cacheMode) ??
+			parseCacheMode(parsed.cache?.mode) ??
+			parseCacheMode(parsed.cacheEnabled) ??
+			parseCacheMode(parsed.cache?.enabled) ??
+			DEFAULT_CACHE_MODE
 		);
-
-		return enabled
-			? { enabled: true, reason: "user opted into cached local checkouts" }
-			: { enabled: false, reason: "user declined or prompt timed out" };
 	} catch (error) {
-		const message = error instanceof Error ? error.message : String(error);
-		return { enabled: false, reason: `cache prompt failed (${message}); using GitHub API/temp files only` };
+		return (error as NodeJS.ErrnoException).code === "ENOENT" ? DEFAULT_CACHE_MODE : "disabled";
+	}
+}
+
+async function writeCachePreference(preference: CacheMode): Promise<void> {
+	const configPath = getCacheConfigPath();
+	const config = {
+		cacheMode: preference,
+		cacheEnabled: preference === "enabled",
+		updatedAt: new Date().toISOString(),
+	};
+
+	await fs.mkdir(path.dirname(configPath), { recursive: true });
+	await fs.writeFile(configPath, `${JSON.stringify(config, null, 2)}\n`, "utf8");
+}
+
+function resolveCacheDecision(preference: CacheMode): { enabled: boolean; reason: string } {
+	if (preference === "enabled") {
+		return { enabled: true, reason: "cache preference enabled; using cached local checkouts" };
+	}
+
+	return { enabled: false, reason: "cache preference disabled; using GitHub API/temp files only" };
+}
+
+function formatCachePreference(preference: CacheMode): string {
+	return preference === "enabled" ? "on" : "off";
+}
+
+function notifyCommand(ctx: ExtensionContext, message: string, type: "info" | "warning" | "error" = "info"): void {
+	if (ctx.hasUI) {
+		ctx.ui.notify(message, type);
+		return;
 	}
+
+	console.error(message);
 }
 
 function resolveToolPath(cwd: string, rawPath: string): string {
@@ -284,7 +329,12 @@ function resolveToolPath(cwd: string, rawPath: string): string {
 }
 
 function getBlockedBashReason(command: string, options: { workspace: string; cacheRoot: string; cacheEnabled: boolean }): string | undefined {
-	const scan = command.split(options.workspace).join("<WORKSPACE>").split(options.cacheRoot).join("<CACHE>");
+	if (!options.cacheEnabled && command.includes(options.cacheRoot)) {
+		return "Local repo checkout cache is disabled for this Librarian call.";
+	}
+
+	let scan = command.split(options.workspace).join("<WORKSPACE>");
+	if (options.cacheEnabled) scan = scan.split(options.cacheRoot).join("<CACHE>");
 	if (/(^|[\n;&|()])\s*\//.test(scan)) return "Librarian bash blocks absolute-path executables.";
 
 	const destructiveLocal = /(^|[\n;&|()])\s*(?:sudo|su|rm|rmdir|mv|cp|chmod|chown|dd|truncate|killall|pkill|launchctl|osascript|pbcopy|pbpaste|eval|exec|xargs)(?=$|[\s;&|()])/;
@@ -438,13 +488,81 @@ function isAbortLikeError(error: unknown): boolean {
 }
 
 export default function librarianExtension(pi: ExtensionAPI) {
+	let cachePreference: CacheMode = DEFAULT_CACHE_MODE;
+
+	pi.on("session_start", async () => {
+		cachePreference = await readCachePreference();
+	});
+
+	pi.registerCommand("librarian-cache", {
+		description: "Toggle Librarian local checkout cache for future librarian calls",
+		getArgumentCompletions: (prefix) => {
+			const commands = ["on", "off", "toggle", "status"];
+			const query = prefix.trim().toLowerCase();
+			const matches = commands.filter((command) => command.startsWith(query));
+			return matches.length > 0 ? matches.map((value) => ({ value, label: value })) : null;
+		},
+		handler: async (args, ctx) => {
+			const action = args.trim().toLowerCase() || "status";
+			const cacheRoot = getUserCacheReposRoot();
+			const configPath = getCacheConfigPath();
+
+			const setPreference = async (mode: CacheMode): Promise<string | undefined> => {
+				cachePreference = mode;
+				try {
+					await writeCachePreference(mode);
+					return undefined;
+				} catch (error) {
+					const message = error instanceof Error ? error.message : String(error);
+					return `Preference changed for this process, but could not save ${configPath}: ${message}`;
+				}
+			};
+
+			const formatSetMessage = (mode: CacheMode, warning?: string): string => {
+				const main = mode === "enabled"
+					? `Librarian cache enabled. Future librarian calls will reuse local checkouts under ${cacheRoot}.`
+					: "Librarian cache disabled. Future librarian calls will use GitHub API/search and temporary fetched files only.";
+				return warning ? `${main} ${warning}` : main;
+			};
+
+			if (action === "on" || action === "enable") {
+				const warning = await setPreference("enabled");
+				notifyCommand(ctx, formatSetMessage("enabled", warning), warning ? "warning" : "info");
+				return;
+			}
+
+			if (action === "off" || action === "disable") {
+				const warning = await setPreference("disabled");
+				notifyCommand(ctx, formatSetMessage("disabled", warning), warning ? "warning" : "info");
+				return;
+			}
+
+			if (action === "toggle") {
+				const next = cachePreference === "enabled" ? "disabled" : "enabled";
+				const warning = await setPreference(next);
+				notifyCommand(ctx, formatSetMessage(next, warning), warning ? "warning" : "info");
+				return;
+			}
+
+			if (action === "status") {
+				notifyCommand(
+					ctx,
+					`Librarian cache is ${formatCachePreference(cachePreference)}. Cache directory: ${cacheRoot}. Config: ${configPath}. Repos unused for ${CACHE_TTL_DAYS} days are removed lazily.`,
+				);
+				return;
+			}
+
+			notifyCommand(ctx, "Usage: /librarian-cache on | off | toggle | status", "warning");
+		},
+	});
+
 	pi.registerTool({
 		name: "librarian",
 		label: "Librarian",
 		description:
-			"GitHub research scout for coding and personal-assistant tasks. Use when the answer likely lives in GitHub repos, exact repo/path locations are unknown, or you'd otherwise do exploratory gh search/tree probes plus local rg/read inspection. Librarian asks whether to use an optional 30-day local checkout cache, otherwise it behaves like API-only pi-librarian.",
+			"GitHub research scout for coding and personal-assistant tasks. Use when the answer likely lives in GitHub repos, exact repo/path locations are unknown, or you'd otherwise do exploratory gh search/tree probes plus local rg/read inspection. Librarian uses an optional 30-day local checkout cache by default; toggle it with /librarian-cache.",
 		promptSnippet:
-			"Research GitHub repositories with evidence-first path and line citations; optionally ask the user to cache local repo checkouts.",
+			"Research GitHub repositories with evidence-first path and line citations; local checkout cache is enabled by default and user-toggleable with /librarian-cache.",
 		promptGuidelines: [
 			"Use librarian when the answer likely requires exploratory GitHub repository search or line-cited evidence from external repos.",
 			"Do not use librarian for files already present in the current workspace unless the user asks for external GitHub research.",
@@ -472,9 +590,21 @@ export default function librarianExtension(pi: ExtensionAPI) {
 			await fs.mkdir(path.join(workspace, "repos"), { recursive: true });
 
 			const cacheRoot = getUserCacheReposRoot();
-			const cacheDecision = await askForCache(ctx, cacheRoot);
-			if (cacheDecision.enabled) await fs.mkdir(cacheRoot, { recursive: true });
-			const cleanup = cacheDecision.enabled ? await cleanupExpiredCache(cacheRoot) : { deleted: 0, errors: [] };
+			let cacheDecision = resolveCacheDecision(cachePreference);
+			let cleanup: { deleted: number; errors: string[] } = { deleted: 0, errors: [] };
+			if (cacheDecision.enabled) {
+				try {
+					await fs.mkdir(cacheRoot, { recursive: true });
+					cleanup = await cleanupExpiredCache(cacheRoot);
+				} catch (error) {
+					const message = error instanceof Error ? error.message : String(error);
+					cacheDecision = {
+						enabled: false,
+						reason: `cache setup failed (${message}); using GitHub API/temp files only`,
+					};
+					cleanup = { deleted: 0, errors: [`cache setup: ${message}`] };
+				}
+			}
 
 			const details: LibrarianDetails = {
 				status: "running",

package/extensions/librarian/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@diegopetrucci/pi-librarian",
-  "version": "0.1.0",
-  "description": "A pi GitHub research scout that can optionally cache local repo checkouts under the user's OS cache directory.",
+  "version": "0.1.1",
+  "description": "A pi GitHub research scout with a toggleable local repo checkout cache under the user's OS cache directory.",
   "keywords": ["pi-package", "pi", "github", "research", "subagent", "cache"],
   "license": "MIT",
   "repository": {

package/extensions/openai-fast/README.md

@@ -25,6 +25,7 @@ Fast mode is only injected when all of these are true:
 ## Commands
 
 ```text
+/fast
 /fast status
 /fast on
 /fast off
@@ -32,7 +33,7 @@ Fast mode is only injected when all of these are true:
 /fast toggle
 ```
 
-`/fast on` and `/fast off` are temporary session/runtime overrides. Use `/fast auto` to reload and follow config defaults again.
+Run `/fast` without arguments to pick an action from a menu. `/fast on` and `/fast off` are temporary session/runtime overrides. Use `/fast auto` to reload and follow config defaults again.
 
 The extension defaults to off so installing the full collection does not accidentally spend Fast-mode credits.
 

package/extensions/openai-fast/index.ts

@@ -249,7 +249,17 @@ export default function openAIFastExtension(pi: ExtensionAPI) {
 		},
 		handler: async (args, ctx) => {
 			const state = getState(ctx);
-			const action = args.trim().toLowerCase() || "status";
+			let action = args.trim().toLowerCase();
+
+			if (!action) {
+				if (!ctx.hasUI) {
+					action = "status";
+				} else {
+					const selection = await ctx.ui.select("OpenAI Fast mode", COMMANDS);
+					if (!selection) return;
+					action = selection;
+				}
+			}
 
 			if (action === "on" || action === "enable") {
 				state.override = "on";

package/extensions/openai-fast/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@diegopetrucci/pi-openai-fast",
-  "version": "0.1.0",
+  "version": "0.1.1",
   "description": "A pi extension that enables OpenAI Codex Fast mode for ChatGPT-auth GPT-5.4 and GPT-5.5 by injecting the priority service tier.",
   "keywords": ["pi-package", "pi", "openai", "codex", "fast"],
   "license": "MIT",

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@diegopetrucci/pi-extensions",
-  "version": "0.1.21",
-  "description": "A collection of pi extensions, including a GitHub librarian with opt-in local repo checkout caching, a minimal custom footer, an Amp-style oracle, a 200k context cap for auto-compaction, a local HTML context inspector, OpenAI Codex Fast mode controls, quiet one-line collapsed invocation previews, a permission gate for dangerous bash commands, confirm-before-destructive session actions, and terminal notifications when pi is ready for input.",
+  "version": "0.1.24",
+  "description": "A collection of pi extensions, including a GitHub librarian with toggleable local repo checkout caching, a minimal custom footer, an Amp-style oracle, a 200k context cap for auto-compaction, a local HTML context inspector, a dirty repository guard for session changes, OpenAI Codex Fast mode controls, quiet one-line collapsed invocation previews, a permission gate for dangerous bash commands, confirm-before-destructive session actions, and terminal notifications when pi is ready for input.",
   "keywords": ["pi-package", "pi", "terminal", "agent"],
   "license": "MIT",
   "repository": {
@@ -32,6 +32,7 @@
       "./extensions/oracle/index.ts",
       "./extensions/context-cap/index.ts",
       "./extensions/context-inspector/index.ts",
+      "./extensions/dirty-repo-guard/index.ts",
       "./extensions/librarian/index.ts",
       "./extensions/quiet-tools/index.ts",
       "./extensions/permission-gate/index.ts",