pi-insomnia 0.1.2 → 0.1.3

package/CHANGELOG.md

@@ -1,11 +1,17 @@
 # Changelog
 
+## 0.1.3 - 2026-07-17
+
 All notable changes to this project will be documented in this file.
 
 This project follows the spirit of [Keep a Changelog](https://keepachangelog.com/en/1.1.0/) and uses semantic versioning for releases.
 
 ## [Unreleased]
 
+### Fixed
+
+- Keep macOS sleep inhibition active across automatic retries, compaction retries, and queued follow-up work until Pi emits `agent_settled`.
+
 ## [0.1.2] - 2026-07-01
 
 ### Changed

package/README.md

@@ -8,7 +8,7 @@ A source-distributed [Pi](https://pi.dev) package that prevents macOS idle sleep
 ## Features
 
 - Automatically starts a macOS sleep assertion when a Pi agent run starts.
-- Automatically releases the assertion when the agent run ends.
+- Automatically releases the assertion when Pi settles after all retries, compaction, and queued follow-up work.
 - Cleans up on Pi session shutdown, reloads, and quits.
 - Uses the macOS built-in `/usr/bin/caffeinate` command; no third-party runtime dependency is required.
 - Shows a small `☕ sleep inhibited` footer status while active in UI modes.
@@ -24,7 +24,7 @@ On macOS, the extension launches:
 /usr/bin/caffeinate -i -w <pi-pid>
 ```
 
-`-i` prevents idle system sleep while the process is running. `-w <pi-pid>` also ties the assertion to the Pi process so the helper exits if Pi exits unexpectedly. The extension still explicitly terminates the helper when the agent becomes idle.
+`-i` prevents idle system sleep while the process is running. `-w <pi-pid>` also ties the assertion to the Pi process so the helper exits if Pi exits unexpectedly. The extension still explicitly terminates the helper when Pi becomes fully idle after any automatic retries, compaction retries, or queued follow-up work.
 
 ## Installation
 

package/extensions/index.ts

@@ -34,16 +34,17 @@ function stopSpinner(ctx: ExtensionContext): void {
   setStatus(ctx, undefined);
 }
 
-export default function piInsomnia(pi: ExtensionAPI) {
+export default function piInsomnia(
+  pi: ExtensionAPI,
+  inhibitor = new MacSleepInhibitor(),
+) {
   reportInstallTelemetry();
 
-  const inhibitor = new MacSleepInhibitor();
-
   pi.on("agent_start", async (_event, ctx) => {
     if (inhibitor.acquire()) startSpinner(ctx);
   });
 
-  pi.on("agent_end", async (_event, ctx) => {
+  pi.on("agent_settled", async (_event, ctx) => {
     inhibitor.release();
     if (!inhibitor.isInhibiting) stopSpinner(ctx);
   });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-insomnia",
-  "version": "0.1.2",
+  "version": "0.1.3",
   "description": "Prevent macOS idle sleep while Pi agents are working.",
   "type": "module",
   "license": "MIT",
@@ -45,6 +45,7 @@
   "scripts": {
     "check": "tsc --noEmit",
     "typecheck": "tsc --noEmit",
+    "test": "node --import tsx --test tests/*.test.mjs",
     "pack:dry-run": "npm pack --dry-run"
   },
   "peerDependencies": {
@@ -52,7 +53,8 @@
   },
   "devDependencies": {
     "@earendil-works/pi-coding-agent": "^0.80.0",
-    "@types/node": "^25.9.3",
+    "@types/node": "^26.1.0",
+    "tsx": "^4.23.0",
     "typescript": "^6.0.3"
   },
   "publishConfig": {

package/src/sleep-inhibitor.ts

@@ -13,7 +13,7 @@ export class MacSleepInhibitor {
   private readonly pid: number;
   private readonly caffeinatePath: string;
   private child: ChildProcess | undefined;
-  private activeRequests = 0;
+  private active = false;
 
   constructor(options: MacSleepInhibitorOptions = {}) {
     this.platform = options.platform ?? process.platform;
@@ -26,7 +26,7 @@ export class MacSleepInhibitor {
   }
 
   get isActive(): boolean {
-    return this.activeRequests > 0;
+    return this.active;
   }
 
   get isInhibiting(): boolean {
@@ -36,7 +36,7 @@ export class MacSleepInhibitor {
   acquire(): boolean {
     if (!this.isSupported) return false;
 
-    this.activeRequests += 1;
+    this.active = true;
     this.start();
     return this.isInhibiting;
   }
@@ -44,12 +44,12 @@ export class MacSleepInhibitor {
   release(): void {
     if (!this.isSupported) return;
 
-    if (this.activeRequests > 0) this.activeRequests -= 1;
-    if (this.activeRequests === 0) this.stop();
+    this.active = false;
+    this.stop();
   }
 
   forceStop(): void {
-    this.activeRequests = 0;
+    this.active = false;
     this.stop();
   }