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

import type WebSocket from 'ws';
import type { WebSocketLike } from '@hocuspocus/server';

/**
 * Wrapper around ws WebSocket that Hocuspocus only writes to.
 * Incoming socket events are forwarded separately by the gateway,
 * which prevents double-handling with RedisSyncExtension.
 */
export class WsSocketWrapper implements WebSocketLike {
  private ws: WebSocket;
  readyState = 1;

  constructor(ws: WebSocket) {
    this.ws = ws;
  }

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

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