workflow-ai 1.0.50 → 1.0.52

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "workflow-ai",
-  "version": "1.0.50",
+  "version": "1.0.52",
   "description": "AI Agent Workflow Coordinator — kanban-based pipeline for AI coding agents",
   "type": "module",
   "bin": {

package/src/runner.mjs

@@ -2,46 +2,64 @@
 
 import fs from 'fs';
 import path from 'path';
-import { spawn, execSync, exec } from 'child_process';
+import { spawn, execSync } from 'child_process';
 import crypto from 'crypto';
 import yaml from './lib/js-yaml.mjs';
 import { findProjectRoot } from './lib/find-root.mjs';
 
 // ============================================================================
-// killProcessTree — убивает процесс и всех его потомков
-// На Windows child.kill() не убивает дочерние процессы (npx → node → python),
-// поэтому используем taskkill /T /F для уничтожения всего дерева.
+// ProcessTracker — snapshot-based process cleanup для Windows.
+//
+// Проблема: runner spawn → claude CLI → npx → node (MCP-сервер).
+// При завершении claude CLI, MCP-серверы становятся сиротами.
+// taskkill /T не находит их, wmic parent-tree разорвано.
+//
+// Решение: снимаем снапшот PID'ов ДО spawn, при cleanup берём текущие PID'ы,
+// разница = процессы порождённые нашим агентом (включая внуков и сирот).
 // ============================================================================
-function killProcessTree(child, logger = null) {
-  const pid = child.pid;
-  if (!pid) return;
+function getProcessSnapshot() {
+  if (process.platform !== 'win32') return new Set();
+  try {
+    const output = execSync(
+      'wmic process where "name=\'node.exe\' or name=\'python.exe\' or name=\'python3.exe\' or name=\'cmd.exe\'" get ProcessId /FORMAT:LIST',
+      { stdio: ['pipe', 'pipe', 'pipe'], encoding: 'utf-8', timeout: 5000 }
+    );
+    const pids = new Set();
+    for (const line of output.split('\n')) {
+      const match = line.match(/ProcessId=(\d+)/);
+      if (match) pids.add(parseInt(match[1], 10));
+    }
+    return pids;
+  } catch {
+    return new Set();
+  }
+}
 
+function killPids(pids, logger = null) {
+  if (pids.length === 0) return;
   if (process.platform === 'win32') {
-    try {
-      execSync(`taskkill /pid ${pid} /T /F`, { stdio: 'pipe' });
-    } catch {
-      // Процесс уже завершился — это нормально
+    for (const pid of pids) {
+      try {
+        execSync(`taskkill /pid ${pid} /F`, { stdio: 'pipe' });
+      } catch { /* already dead */ }
     }
   } else {
-    // Unix: отправляем SIGTERM, через 5 сек — SIGKILL если ещё жив
-    try {
-      child.kill('SIGTERM');
-    } catch {
-      return; // процесс уже завершился
+    for (const pid of pids) {
+      try { process.kill(pid, 'SIGKILL'); } catch { /* already dead */ }
     }
+  }
+  if (logger && pids.length > 0) {
+    logger.info(`Cleaned up ${pids.length} orphan process(es): ${pids.join(', ')}`, 'ProcessCleanup');
+  }
+}
 
-    setTimeout(() => {
-      try {
-        // Проверяем что процесс ещё жив (kill(0) не убивает, только проверяет)
-        process.kill(pid, 0);
-        child.kill('SIGKILL');
-        if (logger) {
-          logger.warn(`Process ${pid} did not exit after SIGTERM, sent SIGKILL`, 'ProcessCleanup');
-        }
-      } catch {
-        // Процесс уже завершился после SIGTERM — всё ок
-      }
-    }, 5000);
+function killChildProcess(child) {
+  if (!child.pid) return;
+  if (process.platform === 'win32') {
+    // taskkill /T /F — лучший effort для прямого дерева
+    try { execSync(`taskkill /pid ${child.pid} /T /F`, { stdio: 'pipe' }); } catch {}
+  } else {
+    try { child.kill('SIGTERM'); } catch {}
   }
 }
 
@@ -802,16 +820,34 @@ class StageExecutor {
 
     // Трекинг активных дочерних процессов для cleanup
     this.activeChildren = new Set();
+    // Снапшот PID'ов перед spawn — для вычисления порождённых процессов
+    this.preSpawnSnapshot = new Set();
   }
 
   /**
-   * Убивает все активные дочерние процессы
+   * Убивает все активные дочерние процессы и порождённых ими сирот.
+   * Snapshot-based: сравнивает текущие PID'ы с снапшотом до spawn,
+   * новые процессы = порождённые нашим агентом (включая внуков/MCP-серверов).
    */
   killAllChildren() {
+    // Убиваем прямых детей через taskkill /T (best effort)
     for (const child of this.activeChildren) {
-      killProcessTree(child, this.logger);
+      killChildProcess(child);
     }
     this.activeChildren.clear();
+
+    // Snapshot-based: убиваем все процессы, появившиеся после spawn
+    if (this.preSpawnSnapshot.size > 0) {
+      const currentPids = getProcessSnapshot();
+      const orphans = [];
+      for (const pid of currentPids) {
+        if (!this.preSpawnSnapshot.has(pid) && pid !== process.pid) {
+          orphans.push(pid);
+        }
+      }
+      killPids(orphans, this.logger);
+    }
+    this.preSpawnSnapshot = new Set();
   }
 
   /**
@@ -949,6 +985,11 @@ class StageExecutor {
         }
       }
 
+      // Снапшот PID'ов до spawn — для snapshot-based cleanup сирот
+      if (this.preSpawnSnapshot.size === 0) {
+        this.preSpawnSnapshot = getProcessSnapshot();
+      }
+
       const child = spawn(agent.command, args, {
         cwd: path.resolve(this.projectRoot, agent.workdir || '.'),
         stdio: ['pipe', 'pipe', 'pipe'],
@@ -970,11 +1011,10 @@ class StageExecutor {
       let stderr = '';
       let timedOut = false;
 
-      // Таймаут — убиваем всё дерево процессов (на Windows child.kill не убивает внуков)
+      // Таймаут — убиваем child + snapshot-based cleanup через killAllChildren
       const timeoutId = setTimeout(() => {
         timedOut = true;
-        killProcessTree(child, this.logger);
-        this.activeChildren.delete(child);
+        this.killAllChildren();  // убьёт child + сирот по snapshot
         if (this.logger) {
           this.logger.timeout(stageId, timeout);
         }
@@ -1215,6 +1255,8 @@ class PipelineRunner {
 
     // Текущий executor для доступа к activeChildren при shutdown
     this.currentExecutor = null;
+    // Снапшот PID'ов перед стартом пайплайна — для snapshot-based cleanup
+    this.preRunSnapshot = new Set();
 
     // Настройка graceful shutdown
     this.setupGracefulShutdown();
@@ -1281,6 +1323,9 @@ class PipelineRunner {
     const maxSteps = this.pipeline.execution?.max_steps || 100;
     const delayBetweenStages = this.pipeline.execution?.delay_between_stages || 5;
 
+    // Снапшот процессов до старта — для snapshot-based cleanup при выходе
+    this.preRunSnapshot = getProcessSnapshot();
+
     this.logger.info('=== Pipeline Runner Started ===', 'PipelineRunner');
     this.logger.info(`Entry stage: ${this.pipeline.entry}`, 'PipelineRunner');
     this.logger.info(`Max steps: ${maxSteps}`, 'PipelineRunner');
@@ -1367,6 +1412,9 @@ class PipelineRunner {
     // Записываем итоговый summary
     this.logger.writeSummary();
 
+    // Cleanup: убиваем процессы-сироты, порождённые агентами во время пайплайна
+    this._cleanupOrphans();
+
     return {
       steps: this.stepCount,
       tasksExecuted: this.tasksExecuted,
@@ -1493,24 +1541,46 @@ class PipelineRunner {
   /**
    * Настройка graceful shutdown
    */
+  /**
+   * Snapshot-based cleanup: убивает процессы, появившиеся после старта пайплайна.
+   * Сравнивает текущие PID'ы node/python/cmd с снапшотом до run() —
+   * разница = процессы порождённые агентами (включая MCP-серверы-сироты).
+   */
+  _cleanupOrphans() {
+    if (this.preRunSnapshot.size === 0) return;
+    const currentPids = getProcessSnapshot();
+    const orphans = [];
+    for (const pid of currentPids) {
+      if (!this.preRunSnapshot.has(pid) && pid !== process.pid) {
+        orphans.push(pid);
+      }
+    }
+    if (orphans.length > 0) {
+      killPids(orphans, this.logger);
+    }
+  }
+
   setupGracefulShutdown() {
     const shutdown = (signal) => {
       if (this.logger) {
         this.logger.info(`Received ${signal}. Killing child processes and shutting down...`, 'PipelineRunner');
       }
       this.running = false;
-      // Убиваем все активные дочерние процессы (включая внуков на Windows)
+      // Убиваем активные дочерние процессы текущего stage
       if (this.currentExecutor) {
         this.currentExecutor.killAllChildren();
       }
+      // Snapshot-based: убиваем всех сирот от всех stage'ей
+      this._cleanupOrphans();
     };
 
     process.on('SIGINT', () => shutdown('SIGINT'));
     process.on('SIGTERM', () => shutdown('SIGTERM'));
 
-    // Финальный cleanup при выходе процесса — синхронный, последний шанс убить детей
+    // Финальный синхронный cleanup при выходе — последний шанс убить сирот
     process.on('exit', () => {
-      if (this.currentExecutor && this.currentExecutor.activeChildren.size > 0) {
+      // Убиваем активные child objects (если exit во время работы stage)
+      if (this.currentExecutor) {
         for (const child of this.currentExecutor.activeChildren) {
           if (child.pid) {
             if (process.platform === 'win32') {
@@ -1521,6 +1591,8 @@ class PipelineRunner {
           }
         }
       }
+      // Snapshot-based cleanup
+      this._cleanupOrphans();
     });
   }
 }