pidnap 0.0.0-dev.3 → 0.0.0-dev.5

package/src/tree-kill.ts

@@ -1,131 +0,0 @@
-import { spawn } from "node:child_process";
-
-type ProcessTree = Record<number, number[]>;
-
-/**
- * Kill a process and all its descendants (children, grandchildren, etc.)
- * @param pid - The process ID to kill
- * @param signal - The signal to send (default: SIGTERM)
- * @returns Promise that resolves when all processes have been signaled
- */
-export async function treeKill(pid: number, signal: NodeJS.Signals = "SIGTERM"): Promise<void> {
-  if (Number.isNaN(pid)) {
-    throw new Error("pid must be a number");
-  }
-
-  const tree: ProcessTree = { [pid]: [] };
-  const pidsToProcess = new Set<number>([pid]);
-
-  // Build the process tree
-  await buildProcessTree(pid, tree, pidsToProcess);
-
-  // Kill all processes in the tree (children first, then parents)
-  killAll(tree, signal);
-}
-
-/**
- * Build a tree of all child processes recursively
- */
-async function buildProcessTree(
-  parentPid: number,
-  tree: ProcessTree,
-  pidsToProcess: Set<number>,
-): Promise<void> {
-  return new Promise((resolve) => {
-    const ps = spawn("ps", ["-o", "pid", "--no-headers", "--ppid", String(parentPid)]);
-
-    let allData = "";
-
-    ps.stdout.on("data", (data: Buffer) => {
-      allData += data.toString("ascii");
-    });
-
-    ps.on("close", async (code) => {
-      pidsToProcess.delete(parentPid);
-
-      if (code !== 0) {
-        // No child processes found for this parent
-        if (pidsToProcess.size === 0) {
-          resolve();
-        }
-        return;
-      }
-
-      const childPids = allData.match(/\d+/g);
-      if (!childPids) {
-        if (pidsToProcess.size === 0) {
-          resolve();
-        }
-        return;
-      }
-
-      // Process all child PIDs concurrently
-      const promises: Promise<void>[] = [];
-
-      for (const pidStr of childPids) {
-        const childPid = parseInt(pidStr, 10);
-        tree[parentPid].push(childPid);
-        tree[childPid] = [];
-        pidsToProcess.add(childPid);
-        promises.push(buildProcessTree(childPid, tree, pidsToProcess));
-      }
-
-      await Promise.all(promises);
-
-      if (pidsToProcess.size === 0) {
-        resolve();
-      }
-    });
-
-    ps.on("error", () => {
-      pidsToProcess.delete(parentPid);
-      if (pidsToProcess.size === 0) {
-        resolve();
-      }
-    });
-  });
-}
-
-/**
- * Kill all processes in the tree
- * Kills children before parents to ensure clean shutdown
- */
-function killAll(tree: ProcessTree, signal: NodeJS.Signals): void {
-  const killed = new Set<number>();
-
-  // Get all PIDs and sort by depth (deepest first)
-  const allPids = Object.keys(tree).map(Number);
-
-  // Kill children first, then parents
-  for (const pid of allPids) {
-    // Kill all children of this pid
-    for (const childPid of tree[pid]) {
-      if (!killed.has(childPid)) {
-        killPid(childPid, signal);
-        killed.add(childPid);
-      }
-    }
-  }
-
-  // Kill all parent pids
-  for (const pid of allPids) {
-    if (!killed.has(pid)) {
-      killPid(pid, signal);
-      killed.add(pid);
-    }
-  }
-}
-
-/**
- * Kill a single process, ignoring ESRCH errors (process already dead)
- */
-function killPid(pid: number, signal: NodeJS.Signals): void {
-  try {
-    process.kill(pid, signal);
-  } catch (err) {
-    // ESRCH = No such process (already dead) - ignore this
-    if ((err as NodeJS.ErrnoException).code !== "ESRCH") {
-      throw err;
-    }
-  }
-}