@tiflis-io/tiflis-code-workstation 0.3.20 → 0.3.22

package/dist/main.js

@@ -2965,6 +2965,9 @@ var TerminalOutputBatcher = class {
     if (this.timeout === null) {
       const adaptiveInterval = this.outputRate > 1e3 ? this.batchIntervalMs : Math.min(8, this.batchIntervalMs);
       this.timeout = setTimeout(() => this.flush(), adaptiveInterval);
+      if (this.timeout.unref) {
+        this.timeout.unref();
+      }
     }
   }
   /**
@@ -9186,6 +9189,19 @@ function handleAuthMessageViaTunnel(data, tunnelClient, authenticateClient, logg
     tunnelClient.send(JSON.stringify(errorResponse));
   }
 }
+var emergencySignalCount = 0;
+process.on("SIGINT", () => {
+  emergencySignalCount++;
+  console.log("\n[SIGINT received]");
+  if (emergencySignalCount >= 2) {
+    console.log("[Force exit on repeated Ctrl+C]");
+    process.exit(1);
+  }
+});
+process.on("SIGTERM", () => {
+  console.log("\n[SIGTERM received]");
+  process.exit(1);
+});
 async function bootstrap() {
   const workstationVersion = getWorkstationVersion();
   const protocolVersion = getProtocolVersion();
@@ -9206,6 +9222,34 @@ async function bootstrap() {
     },
     "Starting workstation server"
   );
+  const registerSignalHandlers = (shutdown2) => {
+    let signalCount = 0;
+    const handleSignal = (signal) => {
+      signalCount++;
+      logger.info({ signal, count: signalCount }, "Signal received, initiating shutdown");
+      if (signalCount === 1) {
+        console.log("\nShutting down gracefully...");
+        shutdown2(signal).then(() => {
+          process.exit(0);
+        }).catch((error) => {
+          logger.error({ error }, "Error during shutdown");
+          process.exit(1);
+        });
+      } else {
+        console.log("\nForce exiting...");
+        process.exit(1);
+      }
+    };
+    process.on("SIGTERM", () => handleSignal("SIGTERM"));
+    process.on("SIGINT", () => handleSignal("SIGINT"));
+    process.on("unhandledRejection", (reason, promise) => {
+      logger.error({ reason, promise }, "Unhandled rejection");
+    });
+    process.on("uncaughtException", (error) => {
+      logger.fatal({ error }, "Uncaught exception");
+      process.exit(1);
+    });
+  };
   const dataDir = env.DATA_DIR;
   initDatabase(dataDir);
   logger.info({ dataDir }, "Database initialized");
@@ -11338,25 +11382,7 @@ async function bootstrap() {
       process.exit(1);
     }
   };
-  let signalCount = 0;
-  const handleSignal = (signal) => {
-    signalCount++;
-    if (signalCount === 1) {
-      void shutdown(signal);
-    } else {
-      logger.warn({ signal, count: signalCount }, "Force exit on repeated signal");
-      process.exit(1);
-    }
-  };
-  process.on("SIGTERM", () => handleSignal("SIGTERM"));
-  process.on("SIGINT", () => handleSignal("SIGINT"));
-  process.on("unhandledRejection", (reason, promise) => {
-    logger.error({ reason, promise }, "Unhandled rejection");
-  });
-  process.on("uncaughtException", (error) => {
-    logger.fatal({ error }, "Uncaught exception");
-    process.exit(1);
-  });
+  registerSignalHandlers(shutdown);
 }
 bootstrap().catch((error) => {
   console.error("Failed to bootstrap:", error);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tiflis-io/tiflis-code-workstation",
-  "version": "0.3.20",
+  "version": "0.3.22",
   "description": "Workstation server for tiflis-code - manages agent sessions and terminal access",
   "author": "Roman Barinov <rbarinov@gmail.com>",
   "license": "FSL-1.1-NC",