@teamvibe/poller 0.1.28 → 0.1.29

package/dist/claude-spawner.d.ts

@@ -11,3 +11,4 @@ export declare function spawnClaudeCode(msg: TeamVibeQueueMessage, sessionLog: S
 }>;
 export declare function getActiveProcessCount(): number;
 export declare function isAtCapacity(): boolean;
+export declare function killAllActiveProcesses(): Promise<void>;

package/dist/claude-spawner.js

@@ -298,3 +298,26 @@ export function getActiveProcessCount() {
 export function isAtCapacity() {
     return activeProcesses.size >= config.MAX_CONCURRENT_SESSIONS;
 }
+export function killAllActiveProcesses() {
+    if (activeProcesses.size === 0)
+        return Promise.resolve();
+    logger.info(`Killing ${activeProcesses.size} active Claude process(es)...`);
+    const killPromises = Array.from(activeProcesses.entries()).map(([id, proc]) => new Promise((resolve) => {
+        if (proc.exitCode !== null) {
+            resolve();
+            return;
+        }
+        const forceKillTimer = setTimeout(() => {
+            logger.warn(`Process ${id} did not exit after SIGTERM, sending SIGKILL`);
+            proc.kill('SIGKILL');
+        }, 5000);
+        proc.once('close', () => {
+            clearTimeout(forceKillTimer);
+            resolve();
+        });
+        proc.kill('SIGTERM');
+    }));
+    return Promise.all(killPromises).then(() => {
+        logger.info('All active processes terminated');
+    });
+}

package/dist/poller.js

@@ -1,7 +1,7 @@
 import { config } from './config.js';
 import { logger } from './logger.js';
 import { pollMessages, deleteMessage, extendVisibility, } from './sqs-poller.js';
-import { spawnClaudeCode, isAtCapacity, getActiveProcessCount } from './claude-spawner.js';
+import { spawnClaudeCode, isAtCapacity, getActiveProcessCount, killAllActiveProcesses } from './claude-spawner.js';
 import { sendSlackError, addReaction, getUserInfo, startTypingIndicator } from './slack-client.js';
 import { acquireSessionLock, releaseSessionLock, updateSessionId } from './session-store.js';
 import { getBrainPath, ensureDirectories, ensureBaseBrain, pushBrainChanges } from './brain-manager.js';
@@ -191,7 +191,7 @@ async function processMessage(received) {
 }
 async function pollLoop() {
     logger.info('Poll loop started');
-    while (true) {
+    while (!shuttingDown) {
         try {
             if (isAtCapacity()) {
                 logger.debug(`At capacity (${getActiveProcessCount()}/${config.MAX_CONCURRENT_SESSIONS}), waiting...`);
@@ -229,6 +229,16 @@ async function shutdown(signal) {
     shuttingDown = true;
     logger.info(`${signal} received, shutting down gracefully...`);
     stopRefresh();
+    // Unblock all messages waiting on thread completion so they can exit
+    for (const [threadId, signals] of threadCompletionSignals.entries()) {
+        logger.info(`Unblocking ${signals.length} waiting message(s) on thread ${threadId}`);
+        signals.forEach((resolve) => resolve());
+    }
+    threadCompletionSignals.clear();
+    waitingCountByThread.clear();
+    // Kill all active Claude child processes
+    await killAllActiveProcesses();
+    // Wait for processingMessages to drain (with timeout)
     const shutdownTimeout = 30000;
     const startTime = Date.now();
     while (processingMessages.size > 0) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@teamvibe/poller",
-  "version": "0.1.28",
+  "version": "0.1.29",
   "type": "module",
   "main": "./dist/index.js",
   "bin": {