msw 2.3.0-ws.rc-5 → 2.3.0

package/lib/core/ws.js

@@ -1,63 +0,0 @@
-"use strict";
-var __defProp = Object.defineProperty;
-var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
-var __getOwnPropNames = Object.getOwnPropertyNames;
-var __hasOwnProp = Object.prototype.hasOwnProperty;
-var __export = (target, all) => {
-  for (var name in all)
-    __defProp(target, name, { get: all[name], enumerable: true });
-};
-var __copyProps = (to, from, except, desc) => {
-  if (from && typeof from === "object" || typeof from === "function") {
-    for (let key of __getOwnPropNames(from))
-      if (!__hasOwnProp.call(to, key) && key !== except)
-        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
-  }
-  return to;
-};
-var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
-var ws_exports = {};
-__export(ws_exports, {
-  ws: () => ws
-});
-module.exports = __toCommonJS(ws_exports);
-var import_outvariant = require("outvariant");
-var import_WebSocketHandler = require("./handlers/WebSocketHandler.js");
-var import_matchRequestUrl = require("./utils/matching/matchRequestUrl.js");
-var import_WebSocketClientManager = require("./ws/WebSocketClientManager.js");
-const wsBroadcastChannel = new BroadcastChannel("msw:ws-client-manager");
-function createWebSocketLinkHandler(url) {
-  (0, import_outvariant.invariant)(url, "Expected a WebSocket server URL but got undefined");
-  (0, import_outvariant.invariant)(
-    (0, import_matchRequestUrl.isPath)(url),
-    "Expected a WebSocket server URL to be a valid path but got %s",
-    typeof url
-  );
-  const clientManager = new import_WebSocketClientManager.WebSocketClientManager(wsBroadcastChannel);
-  return {
-    clients: clientManager.clients,
-    on(event, listener) {
-      const handler = new import_WebSocketHandler.WebSocketHandler(url);
-      handler[import_WebSocketHandler.kEmitter].on("connection", ({ client }) => {
-        clientManager.addConnection(client);
-      });
-      handler[import_WebSocketHandler.kEmitter].on(event, listener);
-      return handler;
-    },
-    broadcast(data) {
-      this.broadcastExcept([], data);
-    },
-    broadcastExcept(clients, data) {
-      const ignoreClients = Array.prototype.concat(clients).map((client) => client.id);
-      clientManager.clients.forEach((otherClient) => {
-        if (!ignoreClients.includes(otherClient.id)) {
-          otherClient.send(data);
-        }
-      });
-    }
-  };
-}
-const ws = {
-  link: createWebSocketLinkHandler
-};
-//# sourceMappingURL=ws.js.map

package/lib/core/ws.js.map

@@ -1 +0,0 @@
-{"version":3,"sources":["../../src/core/ws.ts"],"sourcesContent":["import { invariant } from 'outvariant'\nimport type {\n  WebSocketClientConnectionProtocol,\n  WebSocketData,\n} from '@mswjs/interceptors/WebSocket'\nimport {\n  WebSocketHandler,\n  kEmitter,\n  type WebSocketHandlerEventMap,\n} from './handlers/WebSocketHandler'\nimport { Path, isPath } from './utils/matching/matchRequestUrl'\nimport { WebSocketClientManager } from './ws/WebSocketClientManager'\n\nconst wsBroadcastChannel = new BroadcastChannel('msw:ws-client-manager')\n\nexport type WebSocketLink = {\n  /**\n   * A set of all WebSocket clients connected\n   * to this link.\n   */\n  clients: Set<WebSocketClientConnectionProtocol>\n\n  on<EventType extends keyof WebSocketHandlerEventMap>(\n    event: EventType,\n    listener: (...args: WebSocketHandlerEventMap[EventType]) => void,\n  ): WebSocketHandler\n\n  /**\n   * Broadcasts the given data to all WebSocket clients.\n   *\n   * @example\n   * const service = ws.link('wss://example.com')\n   * service.on('connection', () => {\n   *   service.broadcast('hello, everyone!')\n   * })\n   */\n  broadcast(data: WebSocketData): void\n\n  /**\n   * Broadcasts the given data to all WebSocket clients\n   * except the ones provided in the `clients` argument.\n   *\n   * @example\n   * const service = ws.link('wss://example.com')\n   * service.on('connection', ({ client }) => {\n   *   service.broadcastExcept(client, 'hi, the rest of you!')\n   * })\n   */\n  broadcastExcept(\n    clients:\n      | WebSocketClientConnectionProtocol\n      | Array<WebSocketClientConnectionProtocol>,\n    data: WebSocketData,\n  ): void\n}\n\n/**\n * Intercepts outgoing WebSocket connections to the given URL.\n *\n * @example\n * const chat = ws.link('wss://chat.example.com')\n * chat.on('connection', ({ client }) => {\n *   client.send('hello from server!')\n * })\n */\nfunction createWebSocketLinkHandler(url: Path): WebSocketLink {\n  invariant(url, 'Expected a WebSocket server URL but got undefined')\n\n  invariant(\n    isPath(url),\n    'Expected a WebSocket server URL to be a valid path but got %s',\n    typeof url,\n  )\n\n  const clientManager = new WebSocketClientManager(wsBroadcastChannel)\n\n  return {\n    clients: clientManager.clients,\n    on(event, listener) {\n      const handler = new WebSocketHandler(url)\n\n      // Add the connection event listener for when the\n      // handler matches and emits a connection event.\n      // When that happens, store that connection in the\n      // set of all connections for reference.\n      handler[kEmitter].on('connection', ({ client }) => {\n        clientManager.addConnection(client)\n      })\n\n      // The \"handleWebSocketEvent\" function will invoke\n      // the \"run()\" method on the WebSocketHandler.\n      // If the handler matches, it will emit the \"connection\"\n      // event. Attach the user-defined listener to that event.\n      handler[kEmitter].on(event, listener)\n\n      return handler\n    },\n\n    broadcast(data) {\n      // This will invoke \"send()\" on the immediate clients\n      // in this runtime and post a message to the broadcast channel\n      // to trigger send for the clients in other runtimes.\n      this.broadcastExcept([], data)\n    },\n\n    broadcastExcept(clients, data) {\n      const ignoreClients = Array.prototype\n        .concat(clients)\n        .map((client) => client.id)\n\n      clientManager.clients.forEach((otherClient) => {\n        if (!ignoreClients.includes(otherClient.id)) {\n          otherClient.send(data)\n        }\n      })\n    },\n  }\n}\n\nexport const ws = {\n  link: createWebSocketLinkHandler,\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,wBAA0B;AAK1B,8BAIO;AACP,6BAA6B;AAC7B,oCAAuC;AAEvC,MAAM,qBAAqB,IAAI,iBAAiB,uBAAuB;AAoDvE,SAAS,2BAA2B,KAA0B;AAC5D,mCAAU,KAAK,mDAAmD;AAElE;AAAA,QACE,+BAAO,GAAG;AAAA,IACV;AAAA,IACA,OAAO;AAAA,EACT;AAEA,QAAM,gBAAgB,IAAI,qDAAuB,kBAAkB;AAEnE,SAAO;AAAA,IACL,SAAS,cAAc;AAAA,IACvB,GAAG,OAAO,UAAU;AAClB,YAAM,UAAU,IAAI,yCAAiB,GAAG;AAMxC,cAAQ,gCAAQ,EAAE,GAAG,cAAc,CAAC,EAAE,OAAO,MAAM;AACjD,sBAAc,cAAc,MAAM;AAAA,MACpC,CAAC;AAMD,cAAQ,gCAAQ,EAAE,GAAG,OAAO,QAAQ;AAEpC,aAAO;AAAA,IACT;AAAA,IAEA,UAAU,MAAM;AAId,WAAK,gBAAgB,CAAC,GAAG,IAAI;AAAA,IAC/B;AAAA,IAEA,gBAAgB,SAAS,MAAM;AAC7B,YAAM,gBAAgB,MAAM,UACzB,OAAO,OAAO,EACd,IAAI,CAAC,WAAW,OAAO,EAAE;AAE5B,oBAAc,QAAQ,QAAQ,CAAC,gBAAgB;AAC7C,YAAI,CAAC,cAAc,SAAS,YAAY,EAAE,GAAG;AAC3C,sBAAY,KAAK,IAAI;AAAA,QACvB;AAAA,MACF,CAAC;AAAA,IACH;AAAA,EACF;AACF;AAEO,MAAM,KAAK;AAAA,EAChB,MAAM;AACR;","names":[]}

package/lib/core/ws.mjs

@@ -1,46 +0,0 @@
-import { invariant } from "outvariant";
-import {
-  WebSocketHandler,
-  kEmitter
-} from './handlers/WebSocketHandler.mjs';
-import { isPath } from './utils/matching/matchRequestUrl.mjs';
-import { WebSocketClientManager } from './ws/WebSocketClientManager.mjs';
-const wsBroadcastChannel = new BroadcastChannel("msw:ws-client-manager");
-function createWebSocketLinkHandler(url) {
-  invariant(url, "Expected a WebSocket server URL but got undefined");
-  invariant(
-    isPath(url),
-    "Expected a WebSocket server URL to be a valid path but got %s",
-    typeof url
-  );
-  const clientManager = new WebSocketClientManager(wsBroadcastChannel);
-  return {
-    clients: clientManager.clients,
-    on(event, listener) {
-      const handler = new WebSocketHandler(url);
-      handler[kEmitter].on("connection", ({ client }) => {
-        clientManager.addConnection(client);
-      });
-      handler[kEmitter].on(event, listener);
-      return handler;
-    },
-    broadcast(data) {
-      this.broadcastExcept([], data);
-    },
-    broadcastExcept(clients, data) {
-      const ignoreClients = Array.prototype.concat(clients).map((client) => client.id);
-      clientManager.clients.forEach((otherClient) => {
-        if (!ignoreClients.includes(otherClient.id)) {
-          otherClient.send(data);
-        }
-      });
-    }
-  };
-}
-const ws = {
-  link: createWebSocketLinkHandler
-};
-export {
-  ws
-};
-//# sourceMappingURL=ws.mjs.map

package/lib/core/ws.mjs.map

@@ -1 +0,0 @@
-{"version":3,"sources":["../../src/core/ws.ts"],"sourcesContent":["import { invariant } from 'outvariant'\nimport type {\n  WebSocketClientConnectionProtocol,\n  WebSocketData,\n} from '@mswjs/interceptors/WebSocket'\nimport {\n  WebSocketHandler,\n  kEmitter,\n  type WebSocketHandlerEventMap,\n} from './handlers/WebSocketHandler'\nimport { Path, isPath } from './utils/matching/matchRequestUrl'\nimport { WebSocketClientManager } from './ws/WebSocketClientManager'\n\nconst wsBroadcastChannel = new BroadcastChannel('msw:ws-client-manager')\n\nexport type WebSocketLink = {\n  /**\n   * A set of all WebSocket clients connected\n   * to this link.\n   */\n  clients: Set<WebSocketClientConnectionProtocol>\n\n  on<EventType extends keyof WebSocketHandlerEventMap>(\n    event: EventType,\n    listener: (...args: WebSocketHandlerEventMap[EventType]) => void,\n  ): WebSocketHandler\n\n  /**\n   * Broadcasts the given data to all WebSocket clients.\n   *\n   * @example\n   * const service = ws.link('wss://example.com')\n   * service.on('connection', () => {\n   *   service.broadcast('hello, everyone!')\n   * })\n   */\n  broadcast(data: WebSocketData): void\n\n  /**\n   * Broadcasts the given data to all WebSocket clients\n   * except the ones provided in the `clients` argument.\n   *\n   * @example\n   * const service = ws.link('wss://example.com')\n   * service.on('connection', ({ client }) => {\n   *   service.broadcastExcept(client, 'hi, the rest of you!')\n   * })\n   */\n  broadcastExcept(\n    clients:\n      | WebSocketClientConnectionProtocol\n      | Array<WebSocketClientConnectionProtocol>,\n    data: WebSocketData,\n  ): void\n}\n\n/**\n * Intercepts outgoing WebSocket connections to the given URL.\n *\n * @example\n * const chat = ws.link('wss://chat.example.com')\n * chat.on('connection', ({ client }) => {\n *   client.send('hello from server!')\n * })\n */\nfunction createWebSocketLinkHandler(url: Path): WebSocketLink {\n  invariant(url, 'Expected a WebSocket server URL but got undefined')\n\n  invariant(\n    isPath(url),\n    'Expected a WebSocket server URL to be a valid path but got %s',\n    typeof url,\n  )\n\n  const clientManager = new WebSocketClientManager(wsBroadcastChannel)\n\n  return {\n    clients: clientManager.clients,\n    on(event, listener) {\n      const handler = new WebSocketHandler(url)\n\n      // Add the connection event listener for when the\n      // handler matches and emits a connection event.\n      // When that happens, store that connection in the\n      // set of all connections for reference.\n      handler[kEmitter].on('connection', ({ client }) => {\n        clientManager.addConnection(client)\n      })\n\n      // The \"handleWebSocketEvent\" function will invoke\n      // the \"run()\" method on the WebSocketHandler.\n      // If the handler matches, it will emit the \"connection\"\n      // event. Attach the user-defined listener to that event.\n      handler[kEmitter].on(event, listener)\n\n      return handler\n    },\n\n    broadcast(data) {\n      // This will invoke \"send()\" on the immediate clients\n      // in this runtime and post a message to the broadcast channel\n      // to trigger send for the clients in other runtimes.\n      this.broadcastExcept([], data)\n    },\n\n    broadcastExcept(clients, data) {\n      const ignoreClients = Array.prototype\n        .concat(clients)\n        .map((client) => client.id)\n\n      clientManager.clients.forEach((otherClient) => {\n        if (!ignoreClients.includes(otherClient.id)) {\n          otherClient.send(data)\n        }\n      })\n    },\n  }\n}\n\nexport const ws = {\n  link: createWebSocketLinkHandler,\n}\n"],"mappings":"AAAA,SAAS,iBAAiB;AAK1B;AAAA,EACE;AAAA,EACA;AAAA,OAEK;AACP,SAAe,cAAc;AAC7B,SAAS,8BAA8B;AAEvC,MAAM,qBAAqB,IAAI,iBAAiB,uBAAuB;AAoDvE,SAAS,2BAA2B,KAA0B;AAC5D,YAAU,KAAK,mDAAmD;AAElE;AAAA,IACE,OAAO,GAAG;AAAA,IACV;AAAA,IACA,OAAO;AAAA,EACT;AAEA,QAAM,gBAAgB,IAAI,uBAAuB,kBAAkB;AAEnE,SAAO;AAAA,IACL,SAAS,cAAc;AAAA,IACvB,GAAG,OAAO,UAAU;AAClB,YAAM,UAAU,IAAI,iBAAiB,GAAG;AAMxC,cAAQ,QAAQ,EAAE,GAAG,cAAc,CAAC,EAAE,OAAO,MAAM;AACjD,sBAAc,cAAc,MAAM;AAAA,MACpC,CAAC;AAMD,cAAQ,QAAQ,EAAE,GAAG,OAAO,QAAQ;AAEpC,aAAO;AAAA,IACT;AAAA,IAEA,UAAU,MAAM;AAId,WAAK,gBAAgB,CAAC,GAAG,IAAI;AAAA,IAC/B;AAAA,IAEA,gBAAgB,SAAS,MAAM;AAC7B,YAAM,gBAAgB,MAAM,UACzB,OAAO,OAAO,EACd,IAAI,CAAC,WAAW,OAAO,EAAE;AAE5B,oBAAc,QAAQ,QAAQ,CAAC,gBAAgB;AAC7C,YAAI,CAAC,cAAc,SAAS,YAAY,EAAE,GAAG;AAC3C,sBAAY,KAAK,IAAI;AAAA,QACvB;AAAA,MACF,CAAC;AAAA,IACH;AAAA,EACF;AACF;AAEO,MAAM,KAAK;AAAA,EAChB,MAAM;AACR;","names":[]}

package/src/core/handlers/WebSocketHandler.ts

@@ -1,71 +0,0 @@
-import { Emitter } from 'strict-event-emitter'
-import type { WebSocketConnectionData } from '@mswjs/interceptors/WebSocket'
-import {
-  type Match,
-  type Path,
-  type PathParams,
-  matchRequestUrl,
-} from '../utils/matching/matchRequestUrl'
-import { getCallFrame } from '../utils/internal/getCallFrame'
-
-type WebSocketHandlerParsedResult = {
-  match: Match
-}
-
-export type WebSocketHandlerEventMap = {
-  connection: [args: WebSocketHandlerConnection]
-}
-
-interface WebSocketHandlerConnection extends WebSocketConnectionData {
-  params: PathParams
-}
-
-export const kEmitter = Symbol('kEmitter')
-export const kDispatchEvent = Symbol('kDispatchEvent')
-export const kSender = Symbol('kSender')
-
-export class WebSocketHandler {
-  public callFrame?: string
-
-  protected [kEmitter]: Emitter<WebSocketHandlerEventMap>
-
-  constructor(private readonly url: Path) {
-    this[kEmitter] = new Emitter()
-    this.callFrame = getCallFrame(new Error())
-  }
-
-  public parse(args: {
-    event: MessageEvent<WebSocketConnectionData>
-  }): WebSocketHandlerParsedResult {
-    const connection = args.event.data
-    const match = matchRequestUrl(connection.client.url, this.url)
-
-    return {
-      match,
-    }
-  }
-
-  public predicate(args: {
-    event: MessageEvent<WebSocketConnectionData>
-    parsedResult: WebSocketHandlerParsedResult
-  }): boolean {
-    return args.parsedResult.match.matches
-  }
-
-  async [kDispatchEvent](
-    event: MessageEvent<WebSocketConnectionData>,
-  ): Promise<void> {
-    const parsedResult = this.parse({ event })
-    const connection = event.data
-
-    const resolvedConnection: WebSocketHandlerConnection = {
-      client: connection.client,
-      server: connection.server,
-      params: parsedResult.match.params || {},
-    }
-
-    // Emit the connection event on the handler.
-    // This is what the developer adds listeners for.
-    this[kEmitter].emit('connection', resolvedConnection)
-  }
-}

package/src/core/ws/WebSocketClientManager.test.ts

@@ -1,159 +0,0 @@
-/**
- * @vitest-environment node-websocket
- */
-import { randomUUID } from 'node:crypto'
-import {
-  WebSocketClientConnection,
-  WebSocketTransport,
-} from '@mswjs/interceptors/WebSocket'
-import {
-  WebSocketClientManager,
-  WebSocketBroadcastChannelMessage,
-  WebSocketRemoteClientConnection,
-} from './WebSocketClientManager'
-
-const channel = new BroadcastChannel('test:channel')
-vi.spyOn(channel, 'postMessage')
-
-const socket = new WebSocket('ws://localhost')
-const transport = {
-  onOutgoing: vi.fn(),
-  onIncoming: vi.fn(),
-  onClose: vi.fn(),
-  send: vi.fn(),
-  close: vi.fn(),
-} satisfies WebSocketTransport
-
-afterEach(() => {
-  vi.resetAllMocks()
-})
-
-it('adds a client from this runtime to the list of clients', () => {
-  const manager = new WebSocketClientManager(channel)
-  const connection = new WebSocketClientConnection(socket, transport)
-
-  manager.addConnection(connection)
-
-  // Must add the client to the list of clients.
-  expect(Array.from(manager.clients.values())).toEqual([connection])
-
-  // Must emit the connection open event to notify other runtimes.
-  expect(channel.postMessage).toHaveBeenCalledWith({
-    type: 'connection:open',
-    payload: {
-      clientId: connection.id,
-      url: socket.url,
-    },
-  } satisfies WebSocketBroadcastChannelMessage)
-})
-
-it('adds a client from another runtime to the list of clients', async () => {
-  const clientId = randomUUID()
-  const url = new URL('ws://localhost')
-  const manager = new WebSocketClientManager(channel)
-
-  channel.dispatchEvent(
-    new MessageEvent<WebSocketBroadcastChannelMessage>('message', {
-      data: {
-        type: 'connection:open',
-        payload: {
-          clientId,
-          url: url.href,
-        },
-      },
-    }),
-  )
-
-  await vi.waitFor(() => {
-    expect(Array.from(manager.clients.values())).toEqual([
-      new WebSocketRemoteClientConnection(clientId, url, channel),
-    ])
-  })
-})
-
-it('replays a "send" event coming from another runtime', async () => {
-  const manager = new WebSocketClientManager(channel)
-  const connection = new WebSocketClientConnection(socket, transport)
-  manager.addConnection(connection)
-  vi.spyOn(connection, 'send')
-
-  // Emulate another runtime signaling this connection to receive data.
-  channel.dispatchEvent(
-    new MessageEvent<WebSocketBroadcastChannelMessage>('message', {
-      data: {
-        type: 'extraneous:send',
-        payload: {
-          clientId: connection.id,
-          data: 'hello',
-        },
-      },
-    }),
-  )
-
-  await vi.waitFor(() => {
-    // Must execute the requested operation on the connection.
-    expect(connection.send).toHaveBeenCalledWith('hello')
-    expect(connection.send).toHaveBeenCalledTimes(1)
-  })
-})
-
-it('replays a "close" event coming from another runtime', async () => {
-  const manager = new WebSocketClientManager(channel)
-  const connection = new WebSocketClientConnection(socket, transport)
-  manager.addConnection(connection)
-  vi.spyOn(connection, 'close')
-
-  // Emulate another runtime signaling this connection to close.
-  channel.dispatchEvent(
-    new MessageEvent<WebSocketBroadcastChannelMessage>('message', {
-      data: {
-        type: 'extraneous:close',
-        payload: {
-          clientId: connection.id,
-          code: 1000,
-          reason: 'Normal closure',
-        },
-      },
-    }),
-  )
-
-  await vi.waitFor(() => {
-    // Must execute the requested operation on the connection.
-    expect(connection.close).toHaveBeenCalledWith(1000, 'Normal closure')
-    expect(connection.close).toHaveBeenCalledTimes(1)
-  })
-})
-
-it('removes the extraneous message listener when the connection closes', async () => {
-  const manager = new WebSocketClientManager(channel)
-  const connection = new WebSocketClientConnection(socket, transport)
-  vi.spyOn(connection, 'close').mockImplementationOnce(() => {
-    /**
-     * @note This is a nasty hack so we don't have to uncouple
-     * the connection from transport. Creating a mock transport
-     * is difficult because it relies on the `WebSocketOverride` class.
-     * All we care here is that closing the connection triggers
-     * the transport closure, which it always does.
-     */
-    connection['transport'].onClose()
-  })
-  vi.spyOn(connection, 'send')
-
-  manager.addConnection(connection)
-  connection.close()
-
-  // Signals from other runtimes have no effect on the closed connection.
-  channel.dispatchEvent(
-    new MessageEvent<WebSocketBroadcastChannelMessage>('message', {
-      data: {
-        type: 'extraneous:send',
-        payload: {
-          clientId: connection.id,
-          data: 'hello',
-        },
-      },
-    }),
-  )
-
-  expect(connection.send).not.toHaveBeenCalled()
-})

package/src/core/ws/WebSocketClientManager.ts

@@ -1,170 +0,0 @@
-import type {
-  WebSocketData,
-  WebSocketClientConnection,
-  WebSocketClientConnectionProtocol,
-} from '@mswjs/interceptors/WebSocket'
-
-export type WebSocketBroadcastChannelMessage =
-  | {
-      type: 'connection:open'
-      payload: {
-        clientId: string
-        url: string
-      }
-    }
-  | {
-      type: 'extraneous:send'
-      payload: {
-        clientId: string
-        data: WebSocketData
-      }
-    }
-  | {
-      type: 'extraneous:close'
-      payload: {
-        clientId: string
-        code?: number
-        reason?: string
-      }
-    }
-
-export const kAddByClientId = Symbol('kAddByClientId')
-
-/**
- * A manager responsible for accumulating WebSocket client
- * connections across different browser runtimes.
- */
-export class WebSocketClientManager {
-  /**
-   * All active WebSocket client connections.
-   */
-  public clients: Set<WebSocketClientConnectionProtocol>
-
-  constructor(private channel: BroadcastChannel) {
-    this.clients = new Set()
-
-    this.channel.addEventListener('message', (message) => {
-      const { type, payload } = message.data as WebSocketBroadcastChannelMessage
-
-      switch (type) {
-        case 'connection:open': {
-          // When another runtime notifies about a new connection,
-          // create a connection wrapper class and add it to the set.
-          this.onRemoteConnection(payload.clientId, new URL(payload.url))
-          break
-        }
-      }
-    })
-  }
-
-  /**
-   * Adds the given `WebSocket` client connection to the set
-   * of all connections. The given connection is always the complete
-   * connection object because `addConnection()` is called only
-   * for the opened connections in the same runtime.
-   */
-  public addConnection(client: WebSocketClientConnection): void {
-    this.clients.add(client)
-
-    // Signal to other runtimes about this connection.
-    this.channel.postMessage({
-      type: 'connection:open',
-      payload: {
-        clientId: client.id,
-        url: client.url.toString(),
-      },
-    } as WebSocketBroadcastChannelMessage)
-
-    // Instruct the current client how to handle events
-    // coming from other runtimes (e.g. when calling `.broadcast()`).
-    const handleExtraneousMessage = (
-      message: MessageEvent<WebSocketBroadcastChannelMessage>,
-    ) => {
-      const { type, payload } = message.data
-
-      // Ignore broadcasted messages for other clients.
-      if (
-        typeof payload === 'object' &&
-        'clientId' in payload &&
-        payload.clientId !== client.id
-      ) {
-        return
-      }
-
-      switch (type) {
-        case 'extraneous:send': {
-          client.send(payload.data)
-          break
-        }
-
-        case 'extraneous:close': {
-          client.close(payload.code, payload.reason)
-          break
-        }
-      }
-    }
-
-    const abortController = new AbortController()
-
-    this.channel.addEventListener('message', handleExtraneousMessage, {
-      signal: abortController.signal,
-    })
-
-    // Once closed, this connection cannot be operated on.
-    // This must include the extraneous runtimes as well.
-    client.addEventListener('close', () => abortController.abort(), {
-      once: true,
-    })
-  }
-
-  /**
-   * Adds a client connection wrapper to operate with
-   * WebSocket client connections in other runtimes.
-   */
-  private onRemoteConnection(id: string, url: URL): void {
-    this.clients.add(
-      // Create a connection-compatible instance that can
-      // operate with this client from a different runtime
-      // using the BroadcastChannel messages.
-      new WebSocketRemoteClientConnection(id, url, this.channel),
-    )
-  }
-}
-
-/**
- * A wrapper class to operate with WebSocket client connections
- * from other runtimes. This class maintains 1-1 public API
- * compatibility to the `WebSocketClientConnection` but relies
- * on the given `BroadcastChannel` to communicate instructions
- * with the client connections from other runtimes.
- */
-export class WebSocketRemoteClientConnection
-  implements WebSocketClientConnectionProtocol
-{
-  constructor(
-    public readonly id: string,
-    public readonly url: URL,
-    private channel: BroadcastChannel,
-  ) {}
-
-  send(data: WebSocketData): void {
-    this.channel.postMessage({
-      type: 'extraneous:send',
-      payload: {
-        clientId: this.id,
-        data,
-      },
-    } as WebSocketBroadcastChannelMessage)
-  }
-
-  close(code?: number | undefined, reason?: string | undefined): void {
-    this.channel.postMessage({
-      type: 'extraneous:close',
-      payload: {
-        clientId: this.id,
-        code,
-        reason,
-      },
-    } as WebSocketBroadcastChannelMessage)
-  }
-}

package/src/core/ws/handleWebSocketEvent.ts

@@ -1,56 +0,0 @@
-import type { WebSocketConnectionData } from '@mswjs/interceptors/lib/browser/interceptors/WebSocket'
-import { RequestHandler } from '../handlers/RequestHandler'
-import { WebSocketHandler, kDispatchEvent } from '../handlers/WebSocketHandler'
-import { webSocketInterceptor } from './webSocketInterceptor'
-
-interface HandleWebSocketEventOptions {
-  getHandlers: () => Array<RequestHandler | WebSocketHandler>
-  onMockedConnection: (connection: WebSocketConnectionData) => void
-  onPassthroughConnection: (onnection: WebSocketConnectionData) => void
-}
-
-export function handleWebSocketEvent(options: HandleWebSocketEventOptions) {
-  webSocketInterceptor.on('connection', (connection) => {
-    const handlers = options.getHandlers()
-
-    const connectionEvent = new MessageEvent('connection', {
-      data: connection,
-    })
-
-    // First, filter only those WebSocket handlers that
-    // match the "ws.link()" endpoint predicate. Don't dispatch
-    // anything yet so the logger can be attached to the connection
-    // before it potentially sends events.
-    const matchingHandlers = handlers.filter<WebSocketHandler>(
-      (handler): handler is WebSocketHandler => {
-        if (handler instanceof WebSocketHandler) {
-          return handler.predicate({
-            event: connectionEvent,
-            parsedResult: handler.parse({
-              event: connectionEvent,
-            }),
-          })
-        }
-
-        return false
-      },
-    )
-
-    if (matchingHandlers.length > 0) {
-      options?.onMockedConnection(connection)
-
-      // Iterate over the handlers and forward the connection
-      // event to WebSocket event handlers. This is equivalent
-      // to dispatching that event onto multiple listeners.
-      for (const handler of matchingHandlers) {
-        handler[kDispatchEvent](connectionEvent)
-      }
-    } else {
-      options?.onPassthroughConnection(connection)
-
-      // If none of the "ws" handlers matched,
-      // establish the WebSocket connection as-is.
-      connection.server.connect()
-    }
-  })
-}