claude-task-worker 0.12.0 → 0.14.0

package/dist/index.js

@@ -193,6 +193,7 @@ __export(dispatcher_exports, {
   POLL_INTERVAL_MS: () => POLL_INTERVAL_MS,
   SHUTDOWN_RETRY_TIMEOUT_MS: () => SHUTDOWN_RETRY_TIMEOUT_MS,
   SHUTDOWN_TIMEOUT_MS: () => SHUTDOWN_TIMEOUT_MS,
+  createDispatcherShutdownHandler: () => createDispatcherShutdownHandler,
   formatUptime: () => formatUptime,
   monitorSessions: () => monitorSessions,
   pollOnce: () => pollOnce,
@@ -338,11 +339,9 @@ function monitorSessions(sessions, herdr, options) {
       }
     });
   }, pollIntervalMs);
-  pollInterval.unref();
   const renderInterval2 = setInterval(() => {
     renderSessionTable(sessions);
   }, renderIntervalMs);
-  renderInterval2.unref();
   return {
     stop: finish,
     done
@@ -352,7 +351,7 @@ async function sendCtrlCToAllSessions(sessions, herdr) {
   await Promise.all(
     [...sessions.values()].map(async (session) => {
       try {
-        await herdr.paneSendKeys(session.paneId, "ctrl-c");
+        await herdr.paneSendKeys(session.paneId, CTRL_C_KEY);
       } catch (error) {
         console.error(`[dispatcher] failed to send ctrl-c to session "${session.name}": ${error}`);
       }
@@ -436,7 +435,34 @@ async function shutdownDispatcher(sessions, monitorHandle, options) {
     shutdownPromise = void 0;
   }
 }
-var getDisplayWidth2, truncateToWidth2, padToWidth2, POLL_INTERVAL_MS, SHUTDOWN_TIMEOUT_MS, SHUTDOWN_RETRY_TIMEOUT_MS, shutdownPromise;
+function createDispatcherShutdownHandler(shutdown2) {
+  let shuttingDown2 = false;
+  let forceKilling = false;
+  const handle = async () => {
+    if (shuttingDown2) {
+      if (forceKilling) return;
+      forceKilling = true;
+      console.log("\n[dispatcher] Force killing sessions immediately...");
+      try {
+        await shutdown2({ forceKill: true });
+      } catch (error) {
+        console.error(`[dispatcher] force-kill shutdown failed: ${error}`);
+        process.exit(1);
+      }
+      return;
+    }
+    shuttingDown2 = true;
+    console.log("\n[dispatcher] Stopping sessions. Waiting for them to finish... (Press Ctrl-C again to force kill)");
+    try {
+      await shutdown2();
+    } catch (error) {
+      console.error(`[dispatcher] shutdown failed: ${error}`);
+      process.exit(1);
+    }
+  };
+  return { handle, isShuttingDown: () => shuttingDown2 };
+}
+var getDisplayWidth2, truncateToWidth2, padToWidth2, POLL_INTERVAL_MS, SHUTDOWN_TIMEOUT_MS, SHUTDOWN_RETRY_TIMEOUT_MS, CTRL_C_KEY, shutdownPromise;
 var init_dispatcher = __esm({
   async "src/dispatcher.ts"() {
     "use strict";
@@ -444,6 +470,7 @@ var init_dispatcher = __esm({
     POLL_INTERVAL_MS = 7 * 1e3;
     SHUTDOWN_TIMEOUT_MS = 10 * 60 * 1e3;
     SHUTDOWN_RETRY_TIMEOUT_MS = 90 * 1e3;
+    CTRL_C_KEY = "ctrl+c";
   }
 });
 
@@ -2572,20 +2599,28 @@ if (hasProjectFilter()) {
   (async () => {
     let sessions = /* @__PURE__ */ new Map();
     let monitorHandle;
-    const handleShutdown = async () => {
-      const { shutdownDispatcher: shutdownDispatcher2 } = await init_dispatcher().then(() => dispatcher_exports);
-      await shutdownDispatcher2(sessions, monitorHandle);
-    };
-    process.on("SIGTERM", handleShutdown);
-    process.on("SIGINT", handleShutdown);
     const herdr = await Promise.resolve().then(() => (init_herdr(), herdr_exports));
+    const dispatcher = await init_dispatcher().then(() => dispatcher_exports);
+    const shutdownController = dispatcher.createDispatcherShutdownHandler(
+      (options) => dispatcher.shutdownDispatcher(sessions, monitorHandle, options)
+    );
+    process.on("SIGTERM", shutdownController.handle);
+    process.on("SIGINT", shutdownController.handle);
     try {
       const config = loadProjectsConfig();
       const projects = resolveTargetProjects(parseProjectFilters(), config);
       const forwardedCommand = buildForwardedCommand(process.argv.slice(2));
-      const { runDispatcher: runDispatcher2, monitorSessions: monitorSessions2 } = await init_dispatcher().then(() => dispatcher_exports);
-      sessions = await runDispatcher2(projects, forwardedCommand);
-      monitorHandle = monitorSessions2(sessions, herdr);
+      sessions = await dispatcher.runDispatcher(projects, forwardedCommand);
+      if (sessions.size === 0) {
+        console.log("[dispatcher] no sessions were dispatched, exiting");
+        process.exit(0);
+      }
+      monitorHandle = dispatcher.monitorSessions(sessions, herdr);
+      await monitorHandle.done;
+      if (!shutdownController.isShuttingDown()) {
+        console.log("[dispatcher] all sessions finished, exiting");
+        process.exit(0);
+      }
     } catch (err) {
       if (err instanceof ProjectsConfigError || err instanceof herdr.HerdrUnavailableError) {
         console.error(`[dispatcher] ${err.message}`);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "claude-task-worker",
-  "version": "0.12.0",
+  "version": "0.14.0",
   "description": "CLI tool that polls GitHub Issues/PRs and delegates work to Claude CLI",
   "type": "module",
   "main": "dist/index.js",