Class DesktopBridgeServer

src/node/services/desktop/DesktopBridgeServer.ts:94–402

}

export class DesktopBridgeServer {
  private readonly desktopSessionManager: Pick<DesktopSessionManager, "getLiveSessionConnection">;
  private readonly desktopTokenManager: Pick<DesktopTokenManager, "validate">;
  private readonly wss: WebSocketServer;
  private readonly activePairs = new Set<BridgePair>();
  // Keep upgrade rejection aligned with stop() so httpServer.close() cannot hang on sockets
  // that reconnect after shutdown snapshots the current bridge clients.
  private isStopping = false;
  private stopPromise: Promise<void> | null = null;

  constructor(options: DesktopBridgeServerOptions) {
    assert(options.desktopSessionManager, "DesktopBridgeServer requires a DesktopSessionManager");
    assert(options.desktopTokenManager, "DesktopBridgeServer requires a DesktopTokenManager");

    this.desktopSessionManager = options.desktopSessionManager;
    this.desktopTokenManager = options.desktopTokenManager;
    this.wss = new WebSocketServer({ noServer: true });
  }

  public ensureReady(): void {
    assert(this.wss, "DesktopBridgeServer WebSocketServer must be initialized");
  }

  public handleUpgrade(request: IncomingMessage, socket: Duplex, head: Buffer): void {
    this.ensureReady();
    if (this.isStopping) {
      log.debug("DesktopBridgeServer: rejecting upgrade while stopping", { url: request.url });
      rejectUpgrade(socket);
      return;
    }

    this.wss.handleUpgrade(request, socket, head, (ws) => {
      void this.handleUpgradedConnection(ws, request);
    });
  }

  async stop(): Promise<void> {
    if (this.stopPromise) {
      await this.stopPromise;
      return;
    }

    this.isStopping = true;
    const stopPromise = (async () => {
      const activePairs = Array.from(this.activePairs);
      const trackedWebSockets = new Set(activePairs.map((pair) => pair.ws));
      const activePairClosePromises = activePairs.map((pair) => waitForWebSocketClose(pair.ws));

      for (const pair of activePairs) {
        this.cleanupPair(pair, {
          closeCode: SERVER_STOPPING_CLOSE_CODE,
          closeReason: "server stopping",
        });
      }
      await Promise.allSettled(activePairClosePromises);

      const orphanClientClosePromises: Array<Promise<void>> = [];
      for (const ws of this.wss.clients) {