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

package/dist/main.js

@@ -9206,6 +9206,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 +11366,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.21",
   "description": "Workstation server for tiflis-code - manages agent sessions and terminal access",
   "author": "Roman Barinov <rbarinov@gmail.com>",
   "license": "FSL-1.1-NC",