docmost/apps/server/src/collaboration/extensions/redis-sync/ws-socket-wrapper.ts

import { EventEmitter } from 'events';
import type WebSocket from 'ws';

/**
 * Wrapper around ws WebSocket that only receives events via emit().
 * This prevents double-handling when used with RedisSyncExtension.
 */
export class WsSocketWrapper extends EventEmitter {
  private ws: WebSocket;
  readyState = 1;

  constructor(ws: WebSocket) {
    super();
    this.ws = ws;
    this.once('close', () => {
      this.readyState = 3;
    });
  }

  close(code?: number, reason?: string) {
    if (this.readyState !== 1) return;
    this.readyState = 3;
    try {
      this.ws.close(code, reason);
    } catch (e) {
      // Socket already closed
    }
  }

  ping() {
    if (this.readyState !== 1) return;
    try {
      this.ws.ping();
    } catch (e) {
      // Socket already closed
    }
  }

  send(message: Uint8Array) {
    if (this.readyState !== 1) return;
    try {
      this.ws.send(message);
    } catch (e) {
      // Socket already closed
    }
  }
}