@nmakarov/cli-toolkit 0.72.0 → 0.75.0

package/dist/init.js

@@ -2783,6 +2783,7 @@ function setup(opts = {}) {
   const partialContext = {
     emitter: new EventEmitter(),
     isStop: () => false,
+    isKill: () => false,
     cleanupFunctions: [],
     registerCleanup: (fn) => {
       partialContext.cleanupFunctions.push(fn);
@@ -2804,6 +2805,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
@@ -2853,6 +2855,7 @@ function printAllParameters(context) {
 }
 async function init(flow, opts = {}) {
   let stop = false;
+  let kill = false;
   let context = null;
   let cleanupRan = false;
   const runRegisteredCleanups = async (ctx) => {
@@ -2882,6 +2885,7 @@ async function init(flow, 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"));
@@ -2920,6 +2924,14 @@ async function init(flow, 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 flow(context);
   } catch (error) {
     const errorLocation = error instanceof Error && error.stack ? error.stack.split("\n")[1]?.trim() || "Unknown location" : "Unknown location";
@@ -2946,7 +2958,7 @@ async function init(flow, opts = {}) {
     }
     if (process.exitCode && process.exitCode !== 0) {
       process.exit(process.exitCode);
-    } else if (stop) {
+    } else if (stop || kill) {
       process.exit(0);
     }
   }