@anton-kochev/pi-extensions 0.1.0 → 0.2.0

package/README.md

@@ -1,32 +1,22 @@
 # pi-extensions
 
-Pi extensions for personal use, installed as a single pi-package from this repository.
+[![npm version](https://img.shields.io/npm/v/@anton-kochev/pi-extensions.svg)](https://www.npmjs.com/package/@anton-kochev/pi-extensions)
 
-## Install via pithos
+Pi extensions for personal use.
 
-In your project's `.pithos`:
-
-```yaml
-pi:
-  extensions:
-    squiggle: "git:https://github.com/anton-kochev/pi-extensions.git#main"
-```
-
-Pithos's entrypoint passes this to `pi install`, which clones the repo, runs `npm install`, and registers the extensions declared in the root `pi.extensions` manifest.
-
-## Install directly with pi
+## Install
 
 ```bash
-pi install git:github.com/anton-kochev/pi-extensions
+pi install npm:@anton-kochev/pi-extensions
 ```
 
-Pin to a tag for reproducibility:
+Or pin to a specific version:
 
 ```bash
-pi install git:github.com/anton-kochev/pi-extensions@v0.1.0
+pi install npm:@anton-kochev/pi-extensions@<version>
 ```
 
-## Extensions in this repo
+## Extensions
 
 - [`squiggle/`](./squiggle) — quietly polish grammar and spelling in user prompts.
 
@@ -39,3 +29,12 @@ pi install -l ./squiggle
 ```
 
 Each subdirectory has its own `package.json` so individual extensions remain installable in isolation.
+
+## Release
+
+```bash
+npm version patch       # or minor/major
+git push --follow-tags
+```
+
+Trusted publishing handles the rest — the workflow at `.github/workflows/publish.yml` fires on tag push and publishes to npm via OIDC.

package/package.json

@@ -1,11 +1,20 @@
 {
   "name": "@anton-kochev/pi-extensions",
-  "version": "0.1.0",
+  "version": "0.2.0",
   "description": "Pi extensions.",
-  "keywords": ["pi-package"],
+  "keywords": [
+    "pi-package"
+  ],
   "license": "MIT",
   "type": "module",
-  "files": ["squiggle", "README.md"],
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/anton-kochev/pi-extensions.git"
+  },
+  "files": [
+    "squiggle",
+    "README.md"
+  ],
   "publishConfig": {
     "access": "public"
   },
@@ -14,6 +23,8 @@
     "@earendil-works/pi-coding-agent": "*"
   },
   "pi": {
-    "extensions": ["./squiggle/extensions"]
+    "extensions": [
+      "./squiggle/extensions"
+    ]
   }
 }

package/squiggle/README.md

@@ -2,7 +2,7 @@
 
 Quietly polish grammar and spelling in your pi prompts.
 
-The extension intercepts user input, corrects spelling and grammar using a configured model, shows a colored diff, and submits the corrected prompt automatically without confirmation. Named after the red squiggle from your favorite spell-checker.
+The extension intercepts user input, shows a `squiggling...` spinner while processing, corrects spelling and grammar using a configured model, shows a colored diff, and submits the corrected prompt automatically without confirmation. Named after the red squiggle from your favorite spell-checker.
 
 ## Install
 
@@ -68,14 +68,17 @@ SQUIGGLE_MODEL=openai-codex/gpt-5.4-mini pi
 SQUIGGLE_MAX_CHARS=1000 pi
 ```
 
-## Status
+## Commands
 
 Inside pi:
 
 ```text
-/squiggle-status
+/squiggle toggle      # switch between on/off
+/squiggle-status      # show status
 ```
 
+The toggle state is saved in the current pi session and overrides `.pi/squiggle.json` and environment configuration for that session.
+
 ## Notes
 
 This package imports pi runtime packages as peer dependencies:

package/squiggle/extensions/squiggle.ts

@@ -4,23 +4,44 @@ import { complete, type UserMessage } from "@earendil-works/pi-ai";
 import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
 
 export default function squiggle(pi: ExtensionAPI) {
+	let runtimeMode: SquiggleConfig["mode"] | undefined;
+
+	pi.on("session_start", async (_event, ctx) => {
+		runtimeMode = restoreRuntimeMode(ctx);
+	});
+
+	pi.registerCommand("squiggle", {
+		description: "Toggle squiggle on/off",
+		handler: async (args, ctx) => {
+			const command = args.trim().toLowerCase();
+			if (command !== "toggle") {
+				ctx.ui.notify("Usage: /squiggle toggle", "warning");
+				return;
+			}
+
+			const config = loadEffectiveConfig(ctx.cwd, runtimeMode);
+			runtimeMode = config.mode === "on" ? "off" : "on";
+			persistRuntimeMode(pi, runtimeMode);
+			ctx.ui.notify(formatStatus(ctx, loadEffectiveConfig(ctx.cwd, runtimeMode)), "info");
+		},
+	});
+
 	pi.registerCommand("squiggle-status", {
 		description: "Show whether squiggle is loaded",
 		handler: async (_args, ctx) => {
-			const config = loadConfig(ctx.cwd);
-			const model = selectCorrectionModel(ctx, config);
-			ctx.ui.notify(`squiggle is loaded (${config.mode}, ${formatModel(model)}).`, "info");
+			ctx.ui.notify(formatStatus(ctx, loadEffectiveConfig(ctx.cwd, runtimeMode)), "info");
 		},
 	});
 
 	pi.on("input", async (event, ctx) => {
 		if (event.source === "extension") return { action: "continue" };
 
-		const config = loadConfig(ctx.cwd);
+		const config = loadEffectiveConfig(ctx.cwd, runtimeMode);
 		if (config.mode === "off") return { action: "continue" };
 		if (!event.text.trim()) return { action: "continue" };
 
-		const corrected = await correctWithModel(event.text, ctx, config);
+		const stopIndicator = startSquiggleIndicator(ctx);
+		const corrected = await correctWithModel(event.text, ctx, config).finally(stopIndicator);
 		if (!corrected || corrected === event.text) return { action: "continue" };
 
 		if (ctx.hasUI) ctx.ui.notify(formatColoredDiff(event.text, corrected), "info");
@@ -100,6 +121,28 @@ function loadConfig(cwd: string): SquiggleConfig {
 	};
 }
 
+function loadEffectiveConfig(cwd: string, runtimeMode: SquiggleConfig["mode"] | undefined): SquiggleConfig {
+	const config = loadConfig(cwd);
+	return { ...config, mode: runtimeMode ?? config.mode };
+}
+
+function restoreRuntimeMode(ctx: ExtensionContext): SquiggleConfig["mode"] | undefined {
+	for (const entry of [...ctx.sessionManager.getEntries()].reverse()) {
+		if (entry.type !== "custom" || entry.customType !== "squiggle-mode") continue;
+		const data = (entry as { data?: { mode?: unknown } }).data;
+		return normalizeMode(data?.mode);
+	}
+	return undefined;
+}
+
+function persistRuntimeMode(pi: ExtensionAPI, mode: SquiggleConfig["mode"]): void {
+	pi.appendEntry("squiggle-mode", { mode });
+}
+
+function formatStatus(ctx: ExtensionContext, config: SquiggleConfig): string {
+	return `squiggle is ${config.mode} (${formatModel(selectCorrectionModel(ctx, config))}).`;
+}
+
 function readConfigFile(cwd: string): Partial<SquiggleConfig> {
 	const path = join(cwd, ".pi", "squiggle.json");
 	if (!existsSync(path)) return {};
@@ -143,27 +186,46 @@ function formatModel(model: ReturnType<typeof selectCorrectionModel>): string {
 	return model ? `${model.provider}/${model.id}` : "no model";
 }
 
+function startSquiggleIndicator(ctx: ExtensionContext): () => void {
+	if (!ctx.hasUI) return () => {};
+
+	const frames = ["⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"];
+	let frame = 0;
+	let timer: ReturnType<typeof setInterval> | undefined;
+
+	const render = () => {
+		const theme = ctx.ui.theme;
+		ctx.ui.setStatus("squiggle", theme.fg("accent", frames[frame]!) + theme.fg("dim", " squiggling..."));
+		frame = (frame + 1) % frames.length;
+	};
+
+	render();
+	timer = setInterval(render, 120);
+
+	return () => {
+		if (timer) clearInterval(timer);
+		ctx.ui.setStatus("squiggle", undefined);
+	};
+}
+
 type DiffOp = {
 	type: "same" | "add" | "remove";
 	text: string;
 };
 
 function formatColoredDiff(before: string, after: string): string {
-	const dim = "\x1b[90;3m";
 	const same = "\x1b[90;3m";
 	const added = "\x1b[32;3m";
 	const removed = "\x1b[31;3m";
 	const reset = "\x1b[0m";
 
-	const rendered = diffChars(before.trim(), after.trim())
+	return diffChars(before.trim(), after.trim())
 		.map((op) => {
 			if (op.type === "add") return `${added}${op.text}${reset}`;
 			if (op.type === "remove") return `${removed}${op.text}${reset}`;
 			return `${same}${op.text}${reset}`;
 		})
 		.join("");
-
-	return `${dim}squiggle:${reset}\n${rendered}`;
 }
 
 function diffChars(before: string, after: string): DiffOp[] {