Class NodeWebSocket

scripts/e2e-webrtc-reliability.mjs:160–349

}

class NodeWebSocket {
  constructor(url) {
    this.url = new URL(url);
    this.listeners = new Map();
    this.buffer = Buffer.alloc(0);
    this.handshakeComplete = false;
    this.opened = this.connect();
  }

  addEventListener(type, listener, options = {}) {
    const listeners = this.listeners.get(type) ?? [];
    listeners.push({ listener, once: Boolean(options.once) });
    this.listeners.set(type, listeners);
  }

  removeEventListener(type, listener) {
    this.listeners.set(
      type,
      (this.listeners.get(type) ?? []).filter(
        (entry) => entry.listener !== listener,
      ),
    );
  }

  emit(type, event = {}) {
    const listeners = this.listeners.get(type) ?? [];
    this.listeners.set(
      type,
      listeners.filter(({ listener, once }) => {
        listener(event);
        return !once;
      }),
    );
  }

  async connect() {
    if (this.url.protocol !== "ws:") {
      throw new Error(
        `Unsupported CDP WebSocket protocol: ${this.url.protocol}`,
      );
    }
    const port = Number(this.url.port || 80);
    const key = randomBytes(16).toString("base64");
    this.socket = net.createConnection({
      host: this.url.hostname,
      port,
    });
    this.socket.on("data", (chunk) => this.handleData(chunk));
    this.socket.on("error", (error) => this.emit("error", error));
    this.socket.on("close", () => this.emit("close", {}));
    await new Promise((resolve, reject) => {
      this.socket.once("connect", resolve);
      this.socket.once("error", reject);
    });
    const path = `${this.url.pathname}${this.url.search}`;
    this.socket.write(
      [
        `GET ${path} HTTP/1.1`,