@wierdbytes/pi-facelift 0.6.4 → 0.6.5

package/CHANGELOG.md

@@ -1,5 +1,9 @@
 # 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.

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.4",
+  "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": [