@nmakarov/cli-toolkit 0.73.0 → 0.75.0

package/dist/cli-runner.js

@@ -2794,6 +2794,7 @@ function setup(opts = {}) {
   const partialContext = {
     emitter: new EventEmitter(),
     isStop: () => false,
+    isKill: () => false,
     cleanupFunctions: [],
     registerCleanup: (fn) => {
       partialContext.cleanupFunctions.push(fn);
@@ -2815,6 +2816,7 @@ function setup(opts = {}) {
     logger,
     emitter: partialContext.emitter,
     isStop: partialContext.isStop,
+    isKill: partialContext.isKill,
     cleanupFunctions: partialContext.cleanupFunctions,
     registerCleanup: partialContext.registerCleanup,
     // For long-running scripts (servers): with --showUsedParams=top, print
@@ -2864,6 +2866,7 @@ function printAllParameters(context) {
 }
 async function init(flow2, opts = {}) {
   let stop = false;
+  let kill = false;
   let context = null;
   let cleanupRan = false;
   const runRegisteredCleanups = async (ctx) => {
@@ -2893,6 +2896,7 @@ async function init(flow2, opts = {}) {
     }
     context = setup(opts);
     context.isStop = () => stop;
+    context.isKill = () => kill;
     context = await setupModules(context, opts);
     const stopAfter = context.args.get("stopAfter");
     const stopAllowanceSec = Number(context.params.get("stopAllowance", "number default 60"));
@@ -2931,6 +2935,14 @@ async function init(flow2, opts = {}) {
       );
       context.emitter.emit("stop", stopAllowanceMs);
     });
+    process.on("SIGUSR2", () => {
+      if (!context || kill) return;
+      kill = true;
+      stop = true;
+      context.logger.warn(">> SIGUSR2: emitting kill (urgent stop)");
+      context.emitter.emit("kill");
+      context.emitter.emit("stop", stopAllowanceMs);
+    });
     await flow2(context);
   } catch (error) {
     const errorLocation = error instanceof Error && error.stack ? error.stack.split("\n")[1]?.trim() || "Unknown location" : "Unknown location";
@@ -2957,7 +2969,7 @@ async function init(flow2, opts = {}) {
     }
     if (process.exitCode && process.exitCode !== 0) {
       process.exit(process.exitCode);
-    } else if (stop) {
+    } else if (stop || kill) {
       process.exit(0);
     }
   }