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

package/dist/main.js

@@ -5070,21 +5070,7 @@ var MessageBroadcasterImpl = class {
    * connected to the tunnel and will re-authenticate soon.
    */
   broadcastToAll(message) {
-    const clients = this.deps.clientRegistry.getAll();
-    const authenticatedCount = clients.filter((c) => c.isAuthenticated).length;
-    this.logger.info(
-      {
-        totalClients: clients.length,
-        authenticatedClients: authenticatedCount,
-        messagePreview: message.slice(0, 100)
-      },
-      "broadcastToAll called"
-    );
-    const sent = this.deps.tunnelClient.send(message);
-    this.logger.info(
-      { sent, authenticatedClients: authenticatedCount },
-      "Broadcast to all - message sent to tunnel"
-    );
+    this.deps.tunnelClient.send(message);
   }
   /**
    * Broadcasts a message to all clients subscribed to a session.
@@ -5096,15 +5082,7 @@ var MessageBroadcasterImpl = class {
       (c) => c.isAuthenticated
     ).length;
     if (authenticatedCount > 0) {
-      const sent = this.deps.tunnelClient.send(message);
-      this.logger.debug(
-        {
-          sessionId: sessionId.value,
-          sent,
-          authenticatedSubscribers: authenticatedCount
-        },
-        "Broadcast to session"
-      );
+      this.deps.tunnelClient.send(message);
     }
   }
   /**
@@ -5118,17 +5096,6 @@ var MessageBroadcasterImpl = class {
    *    but not yet re-authenticated with the workstation
    */
   sendToClient(deviceId, message) {
-    const device = new DeviceId(deviceId);
-    const client = this.deps.clientRegistry.getByDeviceId(device);
-    this.logger.info(
-      {
-        deviceId,
-        clientInRegistry: !!client,
-        clientAuthenticated: client?.isAuthenticated,
-        messagePreview: message.slice(0, 100)
-      },
-      "sendToClient - sending via tunnel (supports HTTP polling clients)"
-    );
     return this.deps.tunnelClient.sendToDevice(deviceId, message);
   }
   /**
@@ -5141,37 +5108,12 @@ var MessageBroadcasterImpl = class {
     const authenticatedSubscribers = subscribers.filter(
       (c) => c.isAuthenticated
     );
-    const allClients = this.deps.clientRegistry.getAll();
-    this.logger.info(
-      {
-        sessionId,
-        totalClients: allClients.length,
-        totalSubscribers: subscribers.length,
-        authenticatedSubscribers: authenticatedSubscribers.length,
-        clientsInfo: allClients.map((c) => ({
-          deviceId: c.deviceId.value,
-          isAuthenticated: c.isAuthenticated,
-          isConnected: c.isConnected,
-          status: c.status,
-          subscriptions: c.getSubscriptions()
-        }))
-      },
-      "broadcastToSubscribers - client state"
-    );
     if (authenticatedSubscribers.length === 0) {
-      this.logger.warn(
-        { sessionId, totalClients: allClients.length },
-        "broadcastToSubscribers - no authenticated subscribers found"
-      );
       return;
     }
     for (const client of authenticatedSubscribers) {
       this.deps.tunnelClient.sendToDevice(client.deviceId.value, message);
     }
-    this.logger.debug(
-      { sessionId, subscriberCount: authenticatedSubscribers.length },
-      "Broadcast to subscribers - messages sent"
-    );
   }
 };
 
@@ -9264,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");
@@ -11396,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.19",
+  "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",