@tiflis-io/tiflis-code-tunnel 0.3.0 → 0.3.2

package/dist/main.js

@@ -967,29 +967,51 @@ var WebSocketServerWrapper = class {
     return this.wss?.clients.size ?? 0;
   }
   /**
-   * Closes the WebSocket server gracefully.
+   * Closes the WebSocket server gracefully with timeout.
+   * @param timeoutMs - Maximum time to wait for graceful close (default: 5000ms)
    */
-  async close() {
+  async close(timeoutMs = 5e3) {
     if (this.heartbeatInterval) {
       clearInterval(this.heartbeatInterval);
       this.heartbeatInterval = null;
     }
     const wss = this.wss;
     if (!wss) return;
+    const clientCount = wss.clients.size;
+    this.logger.info({ clientCount }, "Closing WebSocket server");
     wss.clients.forEach((client) => {
-      client.close(1001, "Server shutting down");
+      try {
+        client.close(1001, "Server shutting down");
+      } catch {
+      }
     });
-    return new Promise((resolve, reject) => {
+    const closePromise = new Promise((resolve, reject) => {
       wss.close((err) => {
         if (err) {
           this.logger.error({ error: err }, "Error closing WebSocket server");
           reject(err);
         } else {
-          this.logger.info("WebSocket server closed");
+          this.logger.info("WebSocket server closed gracefully");
           resolve();
         }
       });
     });
+    const timeoutPromise = new Promise((resolve) => {
+      setTimeout(() => {
+        this.logger.warn(
+          { timeoutMs, remainingClients: wss.clients.size },
+          "WebSocket graceful close timed out, forcing termination"
+        );
+        wss.clients.forEach((client) => {
+          try {
+            client.terminate();
+          } catch {
+          }
+        });
+        resolve();
+      }, timeoutMs);
+    });
+    await Promise.race([closePromise, timeoutPromise]);
   }
 };
 
@@ -2111,14 +2133,21 @@ async function bootstrap() {
     logger.fatal({ error }, "Failed to start server");
     process.exit(1);
   }
+  const SHUTDOWN_TIMEOUT_MS = 1e4;
   const shutdown = async (signal) => {
     logger.info({ signal }, "Shutdown signal received");
+    const forceExitTimer = setTimeout(() => {
+      logger.error("Shutdown timed out, forcing exit");
+      process.exit(1);
+    }, SHUTDOWN_TIMEOUT_MS);
     try {
       await wsServer.close();
       await app.close();
+      clearTimeout(forceExitTimer);
       logger.info("Shutdown complete");
       process.exit(0);
     } catch (error) {
+      clearTimeout(forceExitTimer);
       logger.error({ error }, "Error during shutdown");
       process.exit(1);
     }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tiflis-io/tiflis-code-tunnel",
-  "version": "0.3.0",
+  "version": "0.3.2",
   "description": "Tunnel server for tiflis-code - reverse proxy for workstation connections",
   "author": "Roman Barinov <rbarinov@gmail.com>",
   "license": "FSL-1.1-NC",
@@ -53,7 +53,7 @@
     "vitest": "^2.1.8"
   },
   "engines": {
-    "node": ">=22.0.0"
+    "node": ">=24.0.0"
   },
   "scripts": {
     "dev": "tsx watch src/main.ts",