pi-awaken 0.1.0 → 0.1.2

package/index.ts

@@ -22,7 +22,7 @@ const sleep = (ms: number) => new Promise((done) => setTimeout(done, ms));
 
 function installCli() {
   const target = join(agentBin(), "pi-wake");
-  const shim = `#!/bin/sh\nexec node "${cliPath}" "$@"\n`;
+  const shim = `#!/bin/sh\nexec "${process.execPath}" "${cliPath}" "$@"\n`;
   try {
     if (existsSync(target) && readFileSync(target, "utf8") === shim) return;
     mkdirSync(agentBin(), { recursive: true });
@@ -93,29 +93,40 @@ export default function (pi: ExtensionAPI) {
     if (queued.has(wake.key)) return;
     queued.add(wake.key);
     queue.push(wake);
-    flushTimer ??= setTimeout(flush, FLUSH_MS);
+    flushTimer ??= setTimeout(() => guarded(flush), FLUSH_MS);
   };
 
+  function guarded(fn: () => void) {
+    try {
+      fn();
+    } catch (error) {
+      if (error instanceof Error && error.message.includes("ctx is stale")) shutdown();
+      else throw error;
+    }
+  }
+
   function flush() {
     flushTimer = undefined;
     const s = store;
     const items = queue.splice(0);
     if (!s || !items.length) return;
-    const body = items.map(renderWake).join("\n\n");
-    const title = items.length > 1 ? `[pi-wake] ${items.length} background events` : "[pi-wake]";
+    const fresh = items.filter((item) => item.kind !== "exit" || !s.get(item.job.id)?.delivered);
+    for (const item of items) queued.delete(item.key);
+    if (!fresh.length) return;
+    const body = fresh.map(renderWake).join("\n\n");
+    const title = fresh.length > 1 ? `[pi-wake] ${fresh.length} background events` : "[pi-wake]";
     pi.sendMessage(
       {
         customType: "pi-wake",
         content: `${title}\n${body}`,
         display: true,
-        details: { events: items.map((item) => item.key) },
+        details: { events: fresh.map((item) => item.key) },
       },
       { triggerTurn: true, deliverAs: "steer" },
     );
-    for (const item of items) {
+    for (const item of fresh) {
       if (item.kind === "exit") s.markDelivered(item.job.id);
       if (item.kind === "message") s.consume(item.message.file);
-      queued.delete(item.key);
     }
   }
 
@@ -186,7 +197,7 @@ export default function (pi: ExtensionAPI) {
     store.ensure();
     JobStore.prune();
     installCli();
-    pollTimer = setInterval(tick, POLL_MS);
+    pollTimer = setInterval(() => guarded(tick), POLL_MS);
     pollTimer.unref?.();
     tick();
   });
@@ -245,8 +256,13 @@ export default function (pi: ExtensionAPI) {
           PATH: [agentBin(), process.env.PATH ?? ""].join(delimiter),
         },
       });
-      const early = await waitForExit(s, job.id, INLINE_WINDOW_MS, signal);
-      if (early) return finishedResult(s, early);
+      busy.add(job.id);
+      try {
+        const early = await waitForExit(s, job.id, INLINE_WINDOW_MS, signal);
+        if (early) return finishedResult(s, early);
+      } finally {
+        busy.delete(job.id);
+      }
       tick();
       const text = [
         `Started ${header(job)} in the background (pid ${job.pid}).`,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-awaken",
-  "version": "0.1.0",
+  "version": "0.1.2",
   "description": "Pi extension: run long shell scripts in the background and get woken when they finish, match output, or anything calls pi-wake.",
   "type": "module",
   "keywords": [