@wierdbytes/pi-facelift 0.6.3 → 0.6.5

package/CHANGELOG.md

@@ -1,5 +1,16 @@
 # Changelog
 
+## 0.6.5
+
+- **Fix:** pi crashing with "extension ctx is stale" when the session was replaced or reloaded while the working timer was ticking.
+
+## 0.6.4
+
+- **Fix:** declare `shiki` as a runtime dependency instead of a dev-only one.
+  `index.ts` imports `bundledThemes` from `shiki` at runtime, so installs
+  without dev dependencies crashed with `Cannot find module 'shiki'` and
+  the extension failed to load.
+
 ## 0.6.3
 
 - **New:** `tps` (output tokens/sec) next to the working timer — live `~`-prefixed estimate from streamed deltas and exact final from `usage.output`.

package/index.ts

@@ -1180,11 +1180,24 @@ export default function piFaceliftExtension(pi: PiFaceliftApi, deps?: PiFacelift
 	};
 
 	const renderWorkingMessage = (ctx: SessionContextLike): void => {
-		if (ctx.hasUI) {
-			ctx.ui.setWorkingMessage(workingMessageText(workingTracker.elapsedMs(), workingTracker.liveTps()));
+		// The ticker captures `ctx`; after session replacement / reload any
+		// access to it throws (stale ctx). Inside a timer that would be an
+		// uncaughtException and kill pi, so swallow and stop ticking.
+		try {
+			if (ctx.hasUI) {
+				ctx.ui.setWorkingMessage(workingMessageText(workingTracker.elapsedMs(), workingTracker.liveTps()));
+			}
+		} catch {
+			stopWorkingTick();
 		}
 	};
 
+	// Session replaced / reloaded / exiting mid-stream: message_end and
+	// agent_settled may never arrive, so kill the ticker here.
+	pi.on("session_shutdown", () => {
+		stopWorkingTick();
+	});
+
 	// `before_agent_start` fires right after the user submits the prompt,
 	// so it anchors the wall-clock `total` (retries + tools + overhead).
 	pi.on("before_agent_start", () => {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@wierdbytes/pi-facelift",
-  "version": "0.6.3",
+  "version": "0.6.5",
   "description": "Cosmetic facelift for built-in pi tool output — open-right rounded frames, syntax-highlighted reads, status-aware bash summaries, find/grep.",
   "type": "module",
   "files": [
@@ -53,7 +53,8 @@
   },
   "dependencies": {
     "@shikijs/cli": "^4.0.2",
-    "@wierdbytes/pi-common": "^0.4.0"
+    "@wierdbytes/pi-common": "^0.4.0",
+    "shiki": "^4.0.2"
   },
   "peerDependencies": {
     "@earendil-works/pi-ai": ">=0.74.0",
@@ -64,7 +65,6 @@
     "@earendil-works/pi-ai": "^0.74.0",
     "@earendil-works/pi-coding-agent": "^0.74.0",
     "@earendil-works/pi-tui": "^0.74.0",
-    "shiki": "^4.0.2",
     "vitest": "^4.0.18"
   }
 }