Class RelayClient

desktop/src/shared/api/relayClientSession.ts:53–1016

const STALL_IDLE_TIMEOUT_MS = 60_000;

export class RelayClient {
  private wsId: number | null = null;
  private relayUrl: string | null = null;
  private connectPromise: Promise<void> | null = null;
  private reconnectTimeout: number | null = null;
  private reconnectDelayMs = RECONNECT_BASE_DELAY_MS;
  private keepAliveRequested = false;
  private authRequest: {
    pendingEventId: string;
    resolve: () => void;
    reject: (error: Error) => void;
    timeout: number;
  } | null = null;
  private subscriptions = new Map<string, RelaySubscription>();
  private pendingEvents = new Map<string, PendingEvent>();
  private eventBuffer: Array<{ subId: string; event: RelayEvent }> = [];
  private flushTimeout: number | null = null;
  private reconnectListeners = new Set<() => void>();
  private hasConnectedOnce = false;
  private notifyReconnectListeners = false;
  private onMessageChannel: Channel<unknown> | null = null;
  private connectionGeneration = 0;

  /**
   * Sticky terminal flag. Set when `resetConnection` is called with
   * `reconnect: false` (today: auth rejection). Acts as a hard guard against
   * the reconnect-timer / retry-wrapper paths racing back to "reconnecting"
   * after we've already declared the session dead.
   *
   * Cleared only on explicit user re-engagement: `disconnect()` (workspace
   * switch — the singleton is being reused for a different workspace) and
   * `preconnect()` (caller is asking us to come back up).
   */
  private terminal = false;

  private connectionStateEmitter = new RelayConnectionStateEmitter("idle");
  private stallWatchdog = new RelayStallWatchdog({
    intervalMs: STALL_CHECK_INTERVAL_MS,
    idleTimeoutMs: STALL_IDLE_TIMEOUT_MS,
    onStall: (error) => {
      this.connectionStateEmitter.set("stalled");
      this.resetConnection(error);
    },
  });

  /**
   * Cleanly tear down the connection without scheduling a reconnect.
   * Used during workspace switches to reset the singleton before the
   * new workspace applies.
   */
  disconnect() {
    const error = new Error("Relay disconnected for workspace switch.");

    if (this.reconnectTimeout) {
      window.clearTimeout(this.reconnectTimeout);
      this.reconnectTimeout = null;
    }
    this.stallWatchdog.stop();