Class WebSocketClient

packages/web/src/clients/websocket/client.ts:13–149

const DEFAULT_RETRY_IN_MS = 5000;

export default class WebSocketClient {
  private url: string;

  private socket: WebSocket | null = null;

  private queue: { topic: string; event: string; payload: Record<string, any> }[] = [];

  private callbacks: Record<string, Array<(event: string, payload: Record<string, any>) => void>> =
    {};

  private currentRetry = 0;

  constructor(url: string) {
    this.url = url;
    this.connect();
  }

  get open() {
    return this.socket !== null && this.socket.readyState === WebSocket.OPEN;
  }

  on(topic: string, callback: (event: string, payload: Record<string, any>) => void) {
    this.callbacks[topic] = this.callbacks[topic] || [];
    this.callbacks[topic]?.push(callback);
  }

  off(topic: string, callback: (event: string, payload: Record<string, any>) => void) {
    const callbacks = (this.callbacks[topic] || []).filter((cb) => cb !== callback);

    if (callbacks.length === 0) {
      delete this.callbacks[topic];
    } else {
      this.callbacks[topic] = callbacks;
    }
  }

  push(topic: string, event: string, payload: Record<string, any>) {
    if (this.open) {
      this.send(topic, event, payload);
    } else {
      this.queue.push({ topic, event, payload });
    }
  }

  private flush() {
    while (this.queue.length > 0) {
      const message = this.queue.shift()!;
      this.send(message.topic, message.event, message.payload);
    }
  }

  private send(topic: string, event: string, payload: Record<string, any>) {
    const message = JSON.stringify([topic, event, payload]);

    if (this.socket && this.open) {
      this.socket.send(message);
    } else {
      console.error(