msw 2.14.7 → 3.0.0

package/lib/core/handlers/RequestHandler.mjs

@@ -1,283 +0,0 @@
-import { Headers as HeadersPolyfill } from "headers-polyfill";
-import { getCallFrame } from '../utils/internal/getCallFrame.mjs';
-import {
-  isIterable
-} from '../utils/internal/isIterable.mjs';
-import {
-} from '../HttpResponse.mjs';
-import { devUtils } from '../utils/internal/devUtils.mjs';
-import { getRawSetCookie } from '../utils/HttpResponse/decorators.mjs';
-class RequestHandler {
-  static cache = /* @__PURE__ */ new WeakMap();
-  kind = "request";
-  resolver;
-  resolverIterator;
-  resolverIteratorResult;
-  resolverIteratorCleanups;
-  options;
-  scheduledCleanups;
-  info;
-  /**
-   * Indicates whether this request handler has been used
-   * (its resolver has successfully executed).
-   */
-  isUsed;
-  constructor(args) {
-    this.resolver = args.resolver;
-    this.options = args.options;
-    this.scheduledCleanups = /* @__PURE__ */ new Map();
-    const callFrame = getCallFrame(new Error());
-    this.info = {
-      ...args.info,
-      callFrame
-    };
-    this.isUsed = false;
-  }
-  /**
-   * Reset the runtime state accumulated during response resolution,
-   * such as generator iterator progress. Called when this handler is
-   * removed from the active handlers list so re-adding it later starts
-   * from a clean state.
-   */
-  reset() {
-    this.scheduledCleanups.clear();
-    const iterator = this.resolverIterator;
-    this.resolverIterator = void 0;
-    this.resolverIteratorResult = void 0;
-    this.resolverIteratorCleanups = void 0;
-    if (typeof iterator?.return === "function") {
-      void Promise.resolve(iterator.return());
-    }
-  }
-  /**
-   * Restore this handler so it can match requests again after being
-   * exhausted (e.g. via `{ once: true }`). Also clears any accumulated
-   * resolution state.
-   */
-  restore() {
-    if (this.options?.once) {
-      this.reset();
-      this.isUsed = false;
-    }
-  }
-  /**
-   * Parse the intercepted request to extract additional information from it.
-   * Parsed result is then exposed to other methods of this request handler.
-   */
-  async parse(_args) {
-    return {};
-  }
-  /**
-   * Test if this handler matches the given request.
-   *
-   * This method is not used internally but is exposed
-   * as a convenience method for consumers writing custom
-   * handlers.
-   */
-  async test(args) {
-    const parsedResult = await this.parse({
-      request: args.request,
-      resolutionContext: args.resolutionContext
-    });
-    return this.predicate({
-      request: args.request,
-      parsedResult,
-      resolutionContext: args.resolutionContext
-    });
-  }
-  extendResolverArgs(_args) {
-    return {};
-  }
-  // Clone the request instance before it's passed to the handler phases
-  // and the response resolver so we can always read it for logging.
-  // We only clone it once per request to avoid unnecessary overhead.
-  cloneRequestOrGetFromCache(request) {
-    const existingClone = RequestHandler.cache.get(request);
-    if (typeof existingClone !== "undefined") {
-      return existingClone;
-    }
-    const clonedRequest = request.clone();
-    RequestHandler.cache.set(request, clonedRequest);
-    return clonedRequest;
-  }
-  /**
-   * Execute this request handler and produce a mocked response
-   * using the given resolver function.
-   */
-  async run(args) {
-    if (this.isUsed && this.options?.once) {
-      return null;
-    }
-    const requestClone = this.cloneRequestOrGetFromCache(args.request);
-    const parsedResult = await this.parse({
-      request: args.request,
-      resolutionContext: args.resolutionContext
-    });
-    const shouldInterceptRequest = await this.predicate({
-      request: args.request,
-      parsedResult,
-      resolutionContext: args.resolutionContext
-    });
-    if (!shouldInterceptRequest) {
-      return null;
-    }
-    if (this.isUsed && this.options?.once) {
-      return null;
-    }
-    this.isUsed = true;
-    const executeResolver = this.wrapResolver(this.resolver);
-    const resolverExtras = this.extendResolverArgs({
-      request: args.request,
-      parsedResult
-    });
-    const listenerController = new AbortController();
-    args.request.signal.addEventListener(
-      "abort",
-      () => this.runScheduledCleanups(args.requestId),
-      {
-        once: true,
-        signal: listenerController.signal
-      }
-    );
-    const mockedResponsePromise = executeResolver({
-      ...resolverExtras,
-      finalize: (callback) => {
-        this.scheduleCleanup(args.requestId, callback);
-      },
-      requestId: args.requestId,
-      request: args.request
-    }).catch((errorOrResponse) => {
-      if (errorOrResponse instanceof Response) {
-        return errorOrResponse;
-      }
-      throw errorOrResponse;
-    }).finally(() => {
-      listenerController.abort();
-    });
-    const mockedResponse = await mockedResponsePromise;
-    if (mockedResponse) {
-      forwardResponseCookies(mockedResponse);
-    }
-    const executionResult = this.createExecutionResult({
-      // Pass the cloned request to the result so that logging
-      // and other consumers could read its body once more.
-      request: requestClone,
-      requestId: args.requestId,
-      response: mockedResponse,
-      parsedResult
-    });
-    return executionResult;
-  }
-  wrapResolver(resolver) {
-    return async (info) => {
-      if (!this.resolverIterator) {
-        let result;
-        try {
-          result = await resolver(info);
-        } catch (error) {
-          await this.runScheduledCleanups(info.requestId);
-          throw error;
-        }
-        if (!isIterable(result)) {
-          await this.runScheduledCleanups(info.requestId);
-          return result;
-        }
-        const existingCleanups = this.scheduledCleanups.get(info.requestId);
-        if (existingCleanups != null && existingCleanups.length > 0) {
-          this.resolverIteratorCleanups = existingCleanups;
-          this.scheduledCleanups.delete(info.requestId);
-        }
-        this.resolverIterator = Symbol.iterator in result ? result[Symbol.iterator]() : result[Symbol.asyncIterator]();
-      }
-      this.isUsed = false;
-      const { done, value } = await this.resolverIterator.next();
-      const nextResponse = await value;
-      if (nextResponse) {
-        this.resolverIteratorResult = nextResponse.clone();
-      }
-      if (done) {
-        this.isUsed = true;
-        await this.runScheduledCleanups(info.requestId);
-        return this.resolverIteratorResult?.clone();
-      }
-      return nextResponse;
-    };
-  }
-  createExecutionResult(args) {
-    return {
-      handler: this,
-      request: args.request,
-      requestId: args.requestId,
-      response: args.response,
-      parsedResult: args.parsedResult
-    };
-  }
-  scheduleCleanup(requestId, callback) {
-    if (this.resolverIterator) {
-      ;
-      (this.resolverIteratorCleanups ||= []).unshift(callback);
-      return;
-    }
-    const cleanups = this.scheduledCleanups.get(requestId) || [];
-    cleanups.unshift(callback);
-    this.scheduledCleanups.set(requestId, cleanups);
-  }
-  async exhaustCleanups(cleanups) {
-    const errors = [];
-    for (const cleanup of cleanups) {
-      try {
-        await cleanup();
-      } catch (error) {
-        if (error instanceof Error) {
-          errors.push(error);
-        }
-      }
-    }
-    if (errors.length > 0) {
-      devUtils.error(
-        'Failed to execute cleanup for request handler "%s"',
-        this.info.header,
-        new AggregateError(
-          errors,
-          `Failed to execute cleanup for request handler "${this.info.header}"`
-        )
-      );
-    }
-  }
-  async runScheduledCleanups(requestId) {
-    if (this.resolverIterator && this.resolverIteratorCleanups != null && this.resolverIteratorCleanups.length > 0) {
-      try {
-        await this.exhaustCleanups(this.resolverIteratorCleanups);
-      } finally {
-        this.resolverIteratorCleanups = void 0;
-      }
-      return;
-    }
-    const cleanups = this.scheduledCleanups.get(requestId);
-    if (!cleanups || cleanups.length == 0) {
-      return;
-    }
-    await this.exhaustCleanups(cleanups);
-    this.scheduledCleanups.delete(requestId);
-  }
-}
-function forwardResponseCookies(response) {
-  if (typeof document === "undefined") {
-    return;
-  }
-  const responseCookies = getRawSetCookie(response);
-  if (!responseCookies) {
-    return;
-  }
-  const allResponseCookies = HeadersPolyfill.prototype.getSetCookie.call(
-    new Headers([["set-cookie", responseCookies]])
-  );
-  for (const cookieString of allResponseCookies) {
-    document.cookie = cookieString;
-  }
-}
-export {
-  RequestHandler,
-  forwardResponseCookies
-};
-//# sourceMappingURL=RequestHandler.mjs.map

package/lib/core/handlers/RequestHandler.mjs.map

@@ -1 +0,0 @@
-{"version":3,"sources":["../../../src/core/handlers/RequestHandler.ts"],"sourcesContent":["import { Headers as HeadersPolyfill } from 'headers-polyfill'\nimport { getCallFrame } from '../utils/internal/getCallFrame'\nimport {\n  isIterable,\n  type AsyncIterable,\n  type Iterable,\n} from '../utils/internal/isIterable'\nimport type { ResponseResolutionContext } from '../utils/executeHandlers'\nimport type { MaybePromise } from '../typeUtils'\nimport type { HttpResponse } from '../HttpResponse'\nimport {\n  type StrictRequest,\n  type DefaultUnsafeFetchResponse,\n} from '../HttpResponse'\nimport type { GraphQLRequestBody } from './GraphQLHandler'\nimport { devUtils } from '../utils/internal/devUtils'\nimport { getRawSetCookie } from '../utils/HttpResponse/decorators'\n\nexport type DefaultRequestMultipartBody = Record<\n  string,\n  string | File | Array<string | File>\n>\n\nexport type DefaultBodyType =\n  | Record<string, any>\n  | DefaultRequestMultipartBody\n  | string\n  | number\n  | boolean\n  | null\n  | undefined\n\nexport type JsonBodyType =\n  | Record<string, any>\n  | string\n  | number\n  | boolean\n  | null\n  | undefined\n\nexport interface RequestHandlerDefaultInfo {\n  header: string\n}\n\nexport interface RequestHandlerInternalInfo {\n  callFrame?: string\n}\n\nexport type ResponseResolverReturnType<\n  ResponseBodyType extends DefaultBodyType = undefined,\n> =\n  // If ResponseBodyType is a union and one of the types is `undefined`,\n  // allow plain Response as the type.\n  | ([ResponseBodyType] extends [undefined]\n      ? Response\n      : /**\n         * Treat GraphQL response body type as a special case.\n         * For esome reason, making the default HttpResponse<T> | DefaultUnsafeFetchResponse\n         * union breaks the body type inference for HTTP requests.\n         * @see https://github.com/mswjs/msw/issues/2130\n         */\n        ResponseBodyType extends GraphQLRequestBody<any>\n        ? HttpResponse<ResponseBodyType> | DefaultUnsafeFetchResponse\n        : HttpResponse<ResponseBodyType>)\n  | undefined\n  | void\n\nexport type MaybeAsyncResponseResolverReturnType<\n  ResponseBodyType extends DefaultBodyType,\n> = MaybePromise<ResponseResolverReturnType<ResponseBodyType>>\n\nexport type AsyncResponseResolverReturnType<\n  ResponseBodyType extends DefaultBodyType,\n> = MaybePromise<\n  | ResponseResolverReturnType<ResponseBodyType>\n  | Iterable<\n      MaybeAsyncResponseResolverReturnType<ResponseBodyType>,\n      MaybeAsyncResponseResolverReturnType<ResponseBodyType>,\n      MaybeAsyncResponseResolverReturnType<ResponseBodyType>\n    >\n  | AsyncIterable<\n      MaybeAsyncResponseResolverReturnType<ResponseBodyType>,\n      MaybeAsyncResponseResolverReturnType<ResponseBodyType>,\n      MaybeAsyncResponseResolverReturnType<ResponseBodyType>\n    >\n>\n\nexport type ResponseResolverInfo<\n  ResolverExtraInfo extends Record<string, unknown>,\n  RequestBodyType extends DefaultBodyType = DefaultBodyType,\n> = {\n  request: StrictRequest<RequestBodyType>\n  requestId: string\n  /**\n   * Schedule a callback to run after this response resolver completes.\n   * Handy for cleaning up the side effects introduced in the resolver.\n   * @example\n   * sse('/', ({ client, finalize }) => {\n   *   const interval = setInterval(() => client.send({ data: 'ping' }))\n   *   finalize(() => clearInterval(interval))\n   * })\n   */\n  finalize: ResponseResolverFinalizeFunction\n} & ResolverExtraInfo\n\ntype ResponseResolverFinalizeFunction = (\n  callback: () => MaybePromise<void>,\n) => void\n\nexport type ResponseResolver<\n  ResolverExtraInfo extends Record<string, unknown> = Record<string, unknown>,\n  RequestBodyType extends DefaultBodyType = DefaultBodyType,\n  ResponseBodyType extends DefaultBodyType = undefined,\n> = (\n  info: ResponseResolverInfo<ResolverExtraInfo, RequestBodyType>,\n) => AsyncResponseResolverReturnType<ResponseBodyType>\n\nexport interface RequestHandlerArgs<\n  HandlerInfo,\n  HandlerOptions extends RequestHandlerOptions,\n> {\n  info: HandlerInfo\n  resolver: ResponseResolver<any>\n  options?: HandlerOptions\n}\n\nexport interface RequestHandlerOptions {\n  once?: boolean\n}\n\nexport interface RequestHandlerExecutionResult<\n  ParsedResult extends object | undefined,\n> {\n  handler: RequestHandler\n  parsedResult?: ParsedResult\n  request: Request\n  requestId: string\n  response?: Response\n}\n\nexport abstract class RequestHandler<\n  HandlerInfo extends RequestHandlerDefaultInfo = RequestHandlerDefaultInfo,\n  ParsedResult extends Record<string, any> | undefined = any,\n  ResolverExtras extends Record<string, unknown> = any,\n  HandlerOptions extends RequestHandlerOptions = RequestHandlerOptions,\n> {\n  static cache = new WeakMap<\n    StrictRequest<DefaultBodyType>,\n    StrictRequest<DefaultBodyType>\n  >()\n\n  public readonly kind = 'request' as const\n\n  protected resolver: ResponseResolver<ResolverExtras, any, any>\n  private resolverIterator?:\n    | Iterator<\n        MaybeAsyncResponseResolverReturnType<any>,\n        MaybeAsyncResponseResolverReturnType<any>,\n        MaybeAsyncResponseResolverReturnType<any>\n      >\n    | AsyncIterator<\n        MaybeAsyncResponseResolverReturnType<any>,\n        MaybeAsyncResponseResolverReturnType<any>,\n        MaybeAsyncResponseResolverReturnType<any>\n      >\n  private resolverIteratorResult?: Response | HttpResponse<any>\n  private resolverIteratorCleanups?: Array<() => MaybePromise<void>>\n  private options?: HandlerOptions\n  private scheduledCleanups: Map<string, Array<() => MaybePromise<void>>>\n\n  public info: HandlerInfo & RequestHandlerInternalInfo\n\n  /**\n   * Indicates whether this request handler has been used\n   * (its resolver has successfully executed).\n   */\n  public isUsed: boolean\n\n  constructor(args: RequestHandlerArgs<HandlerInfo, HandlerOptions>) {\n    this.resolver = args.resolver\n    this.options = args.options\n    this.scheduledCleanups = new Map()\n\n    const callFrame = getCallFrame(new Error())\n\n    this.info = {\n      ...args.info,\n      callFrame,\n    }\n\n    this.isUsed = false\n  }\n\n  /**\n   * Reset the runtime state accumulated during response resolution,\n   * such as generator iterator progress. Called when this handler is\n   * removed from the active handlers list so re-adding it later starts\n   * from a clean state.\n   */\n  protected reset(): void {\n    this.scheduledCleanups.clear()\n\n    const iterator = this.resolverIterator\n    this.resolverIterator = undefined\n    this.resolverIteratorResult = undefined\n    this.resolverIteratorCleanups = undefined\n\n    if (typeof iterator?.return === 'function') {\n      void Promise.resolve(iterator.return())\n    }\n  }\n\n  /**\n   * Restore this handler so it can match requests again after being\n   * exhausted (e.g. via `{ once: true }`). Also clears any accumulated\n   * resolution state.\n   */\n  protected restore(): void {\n    if (this.options?.once) {\n      this.reset()\n      this.isUsed = false\n    }\n  }\n\n  /**\n   * Determine if the intercepted request should be mocked.\n   */\n  abstract predicate(args: {\n    request: Request\n    parsedResult: ParsedResult\n    resolutionContext?: ResponseResolutionContext\n  }): boolean | Promise<boolean>\n\n  /**\n   * Print out the successfully handled request.\n   */\n  abstract log(args: {\n    request: Request\n    response: Response\n    parsedResult: ParsedResult\n  }): void\n\n  /**\n   * Parse the intercepted request to extract additional information from it.\n   * Parsed result is then exposed to other methods of this request handler.\n   */\n  async parse(_args: {\n    request: Request\n    resolutionContext?: ResponseResolutionContext\n  }): Promise<ParsedResult> {\n    return {} as ParsedResult\n  }\n\n  /**\n   * Test if this handler matches the given request.\n   *\n   * This method is not used internally but is exposed\n   * as a convenience method for consumers writing custom\n   * handlers.\n   */\n  public async test(args: {\n    request: Request\n    resolutionContext?: ResponseResolutionContext\n  }): Promise<boolean> {\n    const parsedResult = await this.parse({\n      request: args.request,\n      resolutionContext: args.resolutionContext,\n    })\n\n    return this.predicate({\n      request: args.request,\n      parsedResult,\n      resolutionContext: args.resolutionContext,\n    })\n  }\n\n  protected extendResolverArgs(_args: {\n    request: Request\n    parsedResult: ParsedResult\n  }): ResolverExtras {\n    return {} as ResolverExtras\n  }\n\n  // Clone the request instance before it's passed to the handler phases\n  // and the response resolver so we can always read it for logging.\n  // We only clone it once per request to avoid unnecessary overhead.\n  private cloneRequestOrGetFromCache(\n    request: StrictRequest<DefaultBodyType>,\n  ): StrictRequest<DefaultBodyType> {\n    const existingClone = RequestHandler.cache.get(request)\n\n    if (typeof existingClone !== 'undefined') {\n      return existingClone\n    }\n\n    const clonedRequest = request.clone()\n    RequestHandler.cache.set(request, clonedRequest)\n\n    return clonedRequest\n  }\n\n  /**\n   * Execute this request handler and produce a mocked response\n   * using the given resolver function.\n   */\n  public async run(args: {\n    request: StrictRequest<any>\n    requestId: string\n    resolutionContext?: ResponseResolutionContext\n  }): Promise<RequestHandlerExecutionResult<ParsedResult> | null> {\n    if (this.isUsed && this.options?.once) {\n      return null\n    }\n\n    // Clone the request.\n    // If this is the first time MSW handles this request, a fresh clone\n    // will be created and cached. Upon further handling of the same request,\n    // the request clone from the cache will be reused to prevent abundant\n    // \"abort\" listeners and save up resources on cloning.\n    const requestClone = this.cloneRequestOrGetFromCache(args.request)\n\n    const parsedResult = await this.parse({\n      request: args.request,\n      resolutionContext: args.resolutionContext,\n    })\n    const shouldInterceptRequest = await this.predicate({\n      request: args.request,\n      parsedResult,\n      resolutionContext: args.resolutionContext,\n    })\n\n    if (!shouldInterceptRequest) {\n      return null\n    }\n\n    // Re-check isUsed, in case another request hit this handler while we were\n    // asynchronously parsing the request.\n    if (this.isUsed && this.options?.once) {\n      return null\n    }\n\n    // Preemptively mark the handler as used.\n    // Generators will undo this because only when the resolver reaches the\n    // \"done\" state of the generator that it considers the handler used.\n    this.isUsed = true\n\n    // Create a response extraction wrapper around the resolver\n    // since it can be both an async function and a generator.\n    const executeResolver = this.wrapResolver(this.resolver)\n\n    const resolverExtras = this.extendResolverArgs({\n      request: args.request,\n      parsedResult,\n    })\n\n    const listenerController = new AbortController()\n\n    args.request.signal.addEventListener(\n      'abort',\n      () => this.runScheduledCleanups(args.requestId),\n      {\n        once: true,\n        signal: listenerController.signal,\n      },\n    )\n\n    const mockedResponsePromise = (\n      executeResolver({\n        ...resolverExtras,\n        finalize: (callback) => {\n          this.scheduleCleanup(args.requestId, callback)\n        },\n        requestId: args.requestId,\n        request: args.request,\n      }) as Promise<Response>\n    )\n      .catch((errorOrResponse) => {\n        // Allow throwing a Response instance in a response resolver.\n        if (errorOrResponse instanceof Response) {\n          return errorOrResponse\n        }\n\n        // Otherwise, throw the error as-is.\n        throw errorOrResponse\n      })\n      .finally(() => {\n        listenerController.abort()\n      })\n\n    const mockedResponse = await mockedResponsePromise\n\n    if (mockedResponse) {\n      forwardResponseCookies(mockedResponse)\n    }\n\n    const executionResult = this.createExecutionResult({\n      // Pass the cloned request to the result so that logging\n      // and other consumers could read its body once more.\n      request: requestClone,\n      requestId: args.requestId,\n      response: mockedResponse,\n      parsedResult,\n    })\n\n    return executionResult\n  }\n\n  private wrapResolver(\n    resolver: ResponseResolver<ResolverExtras>,\n  ): ResponseResolver<ResolverExtras> {\n    return async (info): Promise<ResponseResolverReturnType<any>> => {\n      if (!this.resolverIterator) {\n        let result\n\n        try {\n          result = await resolver(info)\n        } catch (error) {\n          // Ensure cleanup if the resolver throws, regardless of the returned type.\n          await this.runScheduledCleanups(info.requestId)\n\n          // Re-throw the error for the \"run()\" method to handle (e.g. thrown responses).\n          throw error\n        }\n\n        if (!isIterable(result)) {\n          // Otherwise, run the cleanup immediately if it's a plain response resolver.\n          await this.runScheduledCleanups(info.requestId)\n          return result\n        }\n\n        /**\n         * @note Carry over any previously registered cleanups onto the iterator cleanups.\n         * This is only relevant if \"finalize()\" is called in a regular resolver that\n         * returns an iterator.\n         * @example\n         * http.get('/', async ({ finalize }) => {\n         *   finalize(cleanup)\n         *   return (async function*() {})()\n         * })\n         */\n        const existingCleanups = this.scheduledCleanups.get(info.requestId)\n        if (existingCleanups != null && existingCleanups.length > 0) {\n          this.resolverIteratorCleanups = existingCleanups\n          this.scheduledCleanups.delete(info.requestId)\n        }\n\n        this.resolverIterator =\n          Symbol.iterator in result\n            ? result[Symbol.iterator]()\n            : result[Symbol.asyncIterator]()\n      }\n\n      // Opt-out from marking this handler as used.\n      this.isUsed = false\n\n      const { done, value } = await this.resolverIterator.next()\n      const nextResponse = await value\n\n      if (nextResponse) {\n        this.resolverIteratorResult = nextResponse.clone()\n      }\n\n      if (done) {\n        // A one-time generator resolver stops affecting the network\n        // only after it's been completely exhausted.\n        this.isUsed = true\n\n        await this.runScheduledCleanups(info.requestId)\n\n        // Clone the previously stored response so it can be read\n        // when receiving it repeatedly from the \"done\" generator.\n        return this.resolverIteratorResult?.clone()\n      }\n\n      return nextResponse\n    }\n  }\n\n  private createExecutionResult(args: {\n    request: Request\n    requestId: string\n    parsedResult: ParsedResult\n    response?: Response\n  }): RequestHandlerExecutionResult<ParsedResult> {\n    return {\n      handler: this,\n      request: args.request,\n      requestId: args.requestId,\n      response: args.response,\n      parsedResult: args.parsedResult,\n    }\n  }\n\n  private scheduleCleanup(\n    requestId: string,\n    callback: () => MaybePromise<void>,\n  ): void {\n    if (this.resolverIterator) {\n      ;(this.resolverIteratorCleanups ||= []).unshift(callback)\n      return\n    }\n\n    const cleanups = this.scheduledCleanups.get(requestId) || []\n    cleanups.unshift(callback)\n    this.scheduledCleanups.set(requestId, cleanups)\n  }\n\n  private async exhaustCleanups(\n    cleanups: Array<() => MaybePromise<void>>,\n  ): Promise<void> {\n    const errors: Array<Error> = []\n\n    for (const cleanup of cleanups) {\n      try {\n        await cleanup()\n      } catch (error) {\n        if (error instanceof Error) {\n          errors.push(error)\n        }\n      }\n    }\n\n    if (errors.length > 0) {\n      devUtils.error(\n        'Failed to execute cleanup for request handler \"%s\"',\n        this.info.header,\n        new AggregateError(\n          errors,\n          `Failed to execute cleanup for request handler \"${this.info.header}\"`,\n        ),\n      )\n    }\n  }\n\n  private async runScheduledCleanups(requestId: string): Promise<void> {\n    if (\n      this.resolverIterator &&\n      this.resolverIteratorCleanups != null &&\n      this.resolverIteratorCleanups.length > 0\n    ) {\n      try {\n        await this.exhaustCleanups(this.resolverIteratorCleanups)\n      } finally {\n        this.resolverIteratorCleanups = undefined\n      }\n      return\n    }\n\n    const cleanups = this.scheduledCleanups.get(requestId)\n\n    if (!cleanups || cleanups.length == 0) {\n      return\n    }\n\n    await this.exhaustCleanups(cleanups)\n    this.scheduledCleanups.delete(requestId)\n  }\n}\n\n/**\n * Forwards the cookies from the given response to `document.cookie`.\n */\nexport function forwardResponseCookies(response: Response): void {\n  // Cookie forwarding is only relevant in the browser.\n  if (typeof document === 'undefined') {\n    return\n  }\n\n  const responseCookies = getRawSetCookie(response)\n\n  if (!responseCookies) {\n    return\n  }\n\n  // Write the mocked response cookies to the document.\n  // Use `headers-polyfill` to get the Set-Cookie header value correctly.\n  // This is an alternative until TypeScript 5.2\n  // and Node.js v20 become the minimum supported versions\n  // and \"Headers.prototype.getSetCookie\" can be used directly.\n  const allResponseCookies = HeadersPolyfill.prototype.getSetCookie.call(\n    new Headers([['set-cookie', responseCookies]]),\n  )\n\n  for (const cookieString of allResponseCookies) {\n    document.cookie = cookieString\n  }\n}\n"],"mappings":"AAAA,SAAS,WAAW,uBAAuB;AAC3C,SAAS,oBAAoB;AAC7B;AAAA,EACE;AAAA,OAGK;AAIP;AAAA,OAGO;AAEP,SAAS,gBAAgB;AACzB,SAAS,uBAAuB;AA4HzB,MAAe,eAKpB;AAAA,EACA,OAAO,QAAQ,oBAAI,QAGjB;AAAA,EAEc,OAAO;AAAA,EAEb;AAAA,EACF;AAAA,EAWA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAED;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA;AAAA,EAEP,YAAY,MAAuD;AACjE,SAAK,WAAW,KAAK;AACrB,SAAK,UAAU,KAAK;AACpB,SAAK,oBAAoB,oBAAI,IAAI;AAEjC,UAAM,YAAY,aAAa,IAAI,MAAM,CAAC;AAE1C,SAAK,OAAO;AAAA,MACV,GAAG,KAAK;AAAA,MACR;AAAA,IACF;AAEA,SAAK,SAAS;AAAA,EAChB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQU,QAAc;AACtB,SAAK,kBAAkB,MAAM;AAE7B,UAAM,WAAW,KAAK;AACtB,SAAK,mBAAmB;AACxB,SAAK,yBAAyB;AAC9B,SAAK,2BAA2B;AAEhC,QAAI,OAAO,UAAU,WAAW,YAAY;AAC1C,WAAK,QAAQ,QAAQ,SAAS,OAAO,CAAC;AAAA,IACxC;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOU,UAAgB;AACxB,QAAI,KAAK,SAAS,MAAM;AACtB,WAAK,MAAM;AACX,WAAK,SAAS;AAAA,IAChB;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA,EAwBA,MAAM,MAAM,OAGc;AACxB,WAAO,CAAC;AAAA,EACV;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,MAAa,KAAK,MAGG;AACnB,UAAM,eAAe,MAAM,KAAK,MAAM;AAAA,MACpC,SAAS,KAAK;AAAA,MACd,mBAAmB,KAAK;AAAA,IAC1B,CAAC;AAED,WAAO,KAAK,UAAU;AAAA,MACpB,SAAS,KAAK;AAAA,MACd;AAAA,MACA,mBAAmB,KAAK;AAAA,IAC1B,CAAC;AAAA,EACH;AAAA,EAEU,mBAAmB,OAGV;AACjB,WAAO,CAAC;AAAA,EACV;AAAA;AAAA;AAAA;AAAA,EAKQ,2BACN,SACgC;AAChC,UAAM,gBAAgB,eAAe,MAAM,IAAI,OAAO;AAEtD,QAAI,OAAO,kBAAkB,aAAa;AACxC,aAAO;AAAA,IACT;AAEA,UAAM,gBAAgB,QAAQ,MAAM;AACpC,mBAAe,MAAM,IAAI,SAAS,aAAa;AAE/C,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAa,IAAI,MAI+C;AAC9D,QAAI,KAAK,UAAU,KAAK,SAAS,MAAM;AACrC,aAAO;AAAA,IACT;AAOA,UAAM,eAAe,KAAK,2BAA2B,KAAK,OAAO;AAEjE,UAAM,eAAe,MAAM,KAAK,MAAM;AAAA,MACpC,SAAS,KAAK;AAAA,MACd,mBAAmB,KAAK;AAAA,IAC1B,CAAC;AACD,UAAM,yBAAyB,MAAM,KAAK,UAAU;AAAA,MAClD,SAAS,KAAK;AAAA,MACd;AAAA,MACA,mBAAmB,KAAK;AAAA,IAC1B,CAAC;AAED,QAAI,CAAC,wBAAwB;AAC3B,aAAO;AAAA,IACT;AAIA,QAAI,KAAK,UAAU,KAAK,SAAS,MAAM;AACrC,aAAO;AAAA,IACT;AAKA,SAAK,SAAS;AAId,UAAM,kBAAkB,KAAK,aAAa,KAAK,QAAQ;AAEvD,UAAM,iBAAiB,KAAK,mBAAmB;AAAA,MAC7C,SAAS,KAAK;AAAA,MACd;AAAA,IACF,CAAC;AAED,UAAM,qBAAqB,IAAI,gBAAgB;AAE/C,SAAK,QAAQ,OAAO;AAAA,MAClB;AAAA,MACA,MAAM,KAAK,qBAAqB,KAAK,SAAS;AAAA,MAC9C;AAAA,QACE,MAAM;AAAA,QACN,QAAQ,mBAAmB;AAAA,MAC7B;AAAA,IACF;AAEA,UAAM,wBACJ,gBAAgB;AAAA,MACd,GAAG;AAAA,MACH,UAAU,CAAC,aAAa;AACtB,aAAK,gBAAgB,KAAK,WAAW,QAAQ;AAAA,MAC/C;AAAA,MACA,WAAW,KAAK;AAAA,MAChB,SAAS,KAAK;AAAA,IAChB,CAAC,EAEA,MAAM,CAAC,oBAAoB;AAE1B,UAAI,2BAA2B,UAAU;AACvC,eAAO;AAAA,MACT;AAGA,YAAM;AAAA,IACR,CAAC,EACA,QAAQ,MAAM;AACb,yBAAmB,MAAM;AAAA,IAC3B,CAAC;AAEH,UAAM,iBAAiB,MAAM;AAE7B,QAAI,gBAAgB;AAClB,6BAAuB,cAAc;AAAA,IACvC;AAEA,UAAM,kBAAkB,KAAK,sBAAsB;AAAA;AAAA;AAAA,MAGjD,SAAS;AAAA,MACT,WAAW,KAAK;AAAA,MAChB,UAAU;AAAA,MACV;AAAA,IACF,CAAC;AAED,WAAO;AAAA,EACT;AAAA,EAEQ,aACN,UACkC;AAClC,WAAO,OAAO,SAAmD;AAC/D,UAAI,CAAC,KAAK,kBAAkB;AAC1B,YAAI;AAEJ,YAAI;AACF,mBAAS,MAAM,SAAS,IAAI;AAAA,QAC9B,SAAS,OAAO;AAEd,gBAAM,KAAK,qBAAqB,KAAK,SAAS;AAG9C,gBAAM;AAAA,QACR;AAEA,YAAI,CAAC,WAAW,MAAM,GAAG;AAEvB,gBAAM,KAAK,qBAAqB,KAAK,SAAS;AAC9C,iBAAO;AAAA,QACT;AAYA,cAAM,mBAAmB,KAAK,kBAAkB,IAAI,KAAK,SAAS;AAClE,YAAI,oBAAoB,QAAQ,iBAAiB,SAAS,GAAG;AAC3D,eAAK,2BAA2B;AAChC,eAAK,kBAAkB,OAAO,KAAK,SAAS;AAAA,QAC9C;AAEA,aAAK,mBACH,OAAO,YAAY,SACf,OAAO,OAAO,QAAQ,EAAE,IACxB,OAAO,OAAO,aAAa,EAAE;AAAA,MACrC;AAGA,WAAK,SAAS;AAEd,YAAM,EAAE,MAAM,MAAM,IAAI,MAAM,KAAK,iBAAiB,KAAK;AACzD,YAAM,eAAe,MAAM;AAE3B,UAAI,cAAc;AAChB,aAAK,yBAAyB,aAAa,MAAM;AAAA,MACnD;AAEA,UAAI,MAAM;AAGR,aAAK,SAAS;AAEd,cAAM,KAAK,qBAAqB,KAAK,SAAS;AAI9C,eAAO,KAAK,wBAAwB,MAAM;AAAA,MAC5C;AAEA,aAAO;AAAA,IACT;AAAA,EACF;AAAA,EAEQ,sBAAsB,MAKkB;AAC9C,WAAO;AAAA,MACL,SAAS;AAAA,MACT,SAAS,KAAK;AAAA,MACd,WAAW,KAAK;AAAA,MAChB,UAAU,KAAK;AAAA,MACf,cAAc,KAAK;AAAA,IACrB;AAAA,EACF;AAAA,EAEQ,gBACN,WACA,UACM;AACN,QAAI,KAAK,kBAAkB;AACzB;AAAC,OAAC,KAAK,6BAA6B,CAAC,GAAG,QAAQ,QAAQ;AACxD;AAAA,IACF;AAEA,UAAM,WAAW,KAAK,kBAAkB,IAAI,SAAS,KAAK,CAAC;AAC3D,aAAS,QAAQ,QAAQ;AACzB,SAAK,kBAAkB,IAAI,WAAW,QAAQ;AAAA,EAChD;AAAA,EAEA,MAAc,gBACZ,UACe;AACf,UAAM,SAAuB,CAAC;AAE9B,eAAW,WAAW,UAAU;AAC9B,UAAI;AACF,cAAM,QAAQ;AAAA,MAChB,SAAS,OAAO;AACd,YAAI,iBAAiB,OAAO;AAC1B,iBAAO,KAAK,KAAK;AAAA,QACnB;AAAA,MACF;AAAA,IACF;AAEA,QAAI,OAAO,SAAS,GAAG;AACrB,eAAS;AAAA,QACP;AAAA,QACA,KAAK,KAAK;AAAA,QACV,IAAI;AAAA,UACF;AAAA,UACA,kDAAkD,KAAK,KAAK,MAAM;AAAA,QACpE;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAc,qBAAqB,WAAkC;AACnE,QACE,KAAK,oBACL,KAAK,4BAA4B,QACjC,KAAK,yBAAyB,SAAS,GACvC;AACA,UAAI;AACF,cAAM,KAAK,gBAAgB,KAAK,wBAAwB;AAAA,MAC1D,UAAE;AACA,aAAK,2BAA2B;AAAA,MAClC;AACA;AAAA,IACF;AAEA,UAAM,WAAW,KAAK,kBAAkB,IAAI,SAAS;AAErD,QAAI,CAAC,YAAY,SAAS,UAAU,GAAG;AACrC;AAAA,IACF;AAEA,UAAM,KAAK,gBAAgB,QAAQ;AACnC,SAAK,kBAAkB,OAAO,SAAS;AAAA,EACzC;AACF;AAKO,SAAS,uBAAuB,UAA0B;AAE/D,MAAI,OAAO,aAAa,aAAa;AACnC;AAAA,EACF;AAEA,QAAM,kBAAkB,gBAAgB,QAAQ;AAEhD,MAAI,CAAC,iBAAiB;AACpB;AAAA,EACF;AAOA,QAAM,qBAAqB,gBAAgB,UAAU,aAAa;AAAA,IAChE,IAAI,QAAQ,CAAC,CAAC,cAAc,eAAe,CAAC,CAAC;AAAA,EAC/C;AAEA,aAAW,gBAAgB,oBAAoB;AAC7C,aAAS,SAAS;AAAA,EACpB;AACF;","names":[]}

package/lib/core/handlers/WebSocketHandler.d.mts

@@ -1,49 +0,0 @@
-import { Emitter } from 'strict-event-emitter';
-import { WebSocketClientConnectionProtocol, WebSocketServerConnectionProtocol, WebSocketConnectionData } from '@mswjs/interceptors/WebSocket';
-import { Path, PathParams, Match } from '../utils/matching/matchRequestUrl.mjs';
-
-type WebSocketHandlerParsedResult = {
-    match: Match;
-};
-type WebSocketHandlerEventMap = {
-    connection: [args: WebSocketHandlerConnection];
-};
-interface WebSocketHandlerConnection {
-    client: WebSocketClientConnectionProtocol;
-    server: WebSocketServerConnectionProtocol;
-    info: WebSocketConnectionData['info'];
-    params: PathParams;
-}
-interface WebSocketResolutionContext {
-    baseUrl?: string;
-    [kAutoConnect]?: boolean;
-}
-declare const kEmitter: unique symbol;
-declare const kSender: unique symbol;
-declare const kConnect: unique symbol;
-declare const kAutoConnect: unique symbol;
-declare class WebSocketHandler {
-    #private;
-    protected readonly url: Path;
-    id: string;
-    callFrame?: string;
-    kind: "websocket";
-    protected [kEmitter]: Emitter<WebSocketHandlerEventMap>;
-    constructor(url: Path);
-    parse(args: {
-        url: string | URL;
-        resolutionContext?: WebSocketResolutionContext;
-    }): WebSocketHandlerParsedResult;
-    predicate(args: {
-        url: string | URL;
-        parsedResult: WebSocketHandlerParsedResult;
-    }): boolean;
-    test(url: string | URL, resolutionContext?: WebSocketResolutionContext & {
-        strict?: boolean;
-    }): boolean;
-    run(connection: WebSocketConnectionData, resolutionContext?: WebSocketResolutionContext): Promise<WebSocketHandlerConnection | null>;
-    protected [kConnect](connection: WebSocketHandlerConnection): boolean;
-    log(connection: WebSocketConnectionData): () => void;
-}
-
-export { WebSocketHandler, type WebSocketHandlerConnection, type WebSocketHandlerEventMap, type WebSocketResolutionContext, kAutoConnect, kConnect, kEmitter, kSender };

package/lib/core/handlers/WebSocketHandler.d.ts

@@ -1,49 +0,0 @@
-import { Emitter } from 'strict-event-emitter';
-import { WebSocketClientConnectionProtocol, WebSocketServerConnectionProtocol, WebSocketConnectionData } from '@mswjs/interceptors/WebSocket';
-import { Path, PathParams, Match } from '../utils/matching/matchRequestUrl.js';
-
-type WebSocketHandlerParsedResult = {
-    match: Match;
-};
-type WebSocketHandlerEventMap = {
-    connection: [args: WebSocketHandlerConnection];
-};
-interface WebSocketHandlerConnection {
-    client: WebSocketClientConnectionProtocol;
-    server: WebSocketServerConnectionProtocol;
-    info: WebSocketConnectionData['info'];
-    params: PathParams;
-}
-interface WebSocketResolutionContext {
-    baseUrl?: string;
-    [kAutoConnect]?: boolean;
-}
-declare const kEmitter: unique symbol;
-declare const kSender: unique symbol;
-declare const kConnect: unique symbol;
-declare const kAutoConnect: unique symbol;
-declare class WebSocketHandler {
-    #private;
-    protected readonly url: Path;
-    id: string;
-    callFrame?: string;
-    kind: "websocket";
-    protected [kEmitter]: Emitter<WebSocketHandlerEventMap>;
-    constructor(url: Path);
-    parse(args: {
-        url: string | URL;
-        resolutionContext?: WebSocketResolutionContext;
-    }): WebSocketHandlerParsedResult;
-    predicate(args: {
-        url: string | URL;
-        parsedResult: WebSocketHandlerParsedResult;
-    }): boolean;
-    test(url: string | URL, resolutionContext?: WebSocketResolutionContext & {
-        strict?: boolean;
-    }): boolean;
-    run(connection: WebSocketConnectionData, resolutionContext?: WebSocketResolutionContext): Promise<WebSocketHandlerConnection | null>;
-    protected [kConnect](connection: WebSocketHandlerConnection): boolean;
-    log(connection: WebSocketConnectionData): () => void;
-}
-
-export { WebSocketHandler, type WebSocketHandlerConnection, type WebSocketHandlerEventMap, type WebSocketResolutionContext, kAutoConnect, kConnect, kEmitter, kSender };

package/lib/core/handlers/WebSocketHandler.js

@@ -1,178 +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 WebSocketHandler_exports = {};
-__export(WebSocketHandler_exports, {
-  WebSocketHandler: () => WebSocketHandler,
-  kAutoConnect: () => kAutoConnect,
-  kConnect: () => kConnect,
-  kEmitter: () => kEmitter,
-  kSender: () => kSender
-});
-module.exports = __toCommonJS(WebSocketHandler_exports);
-var import_strict_event_emitter = require("strict-event-emitter");
-var import_interceptors = require("@mswjs/interceptors");
-var import_matchRequestUrl = require("../utils/matching/matchRequestUrl");
-var import_getCallFrame = require("../utils/internal/getCallFrame");
-var import_attachWebSocketLogger = require("../ws/utils/attachWebSocketLogger");
-const kEmitter = /* @__PURE__ */ Symbol("kEmitter");
-const kSender = /* @__PURE__ */ Symbol("kSender");
-const kConnect = /* @__PURE__ */ Symbol("kConnect");
-const kAutoConnect = /* @__PURE__ */ Symbol("kAutoConnect");
-const kStopPropagationPatched = /* @__PURE__ */ Symbol("kStopPropagationPatched");
-const KOnStopPropagation = /* @__PURE__ */ Symbol("KOnStopPropagation");
-class WebSocketHandler {
-  constructor(url) {
-    this.url = url;
-    this.id = (0, import_interceptors.createRequestId)();
-    this[kEmitter] = new import_strict_event_emitter.Emitter();
-    this.callFrame = (0, import_getCallFrame.getCallFrame)(new Error());
-  }
-  url;
-  id;
-  callFrame;
-  kind = "websocket";
-  [kEmitter];
-  parse(args) {
-    const clientUrl = new URL(args.url);
-    const resolvedHandlerUrl = this.url instanceof RegExp || this.url.startsWith("*") ? this.url : this.#resolveWebSocketUrl(this.url, args.resolutionContext?.baseUrl);
-    clientUrl.pathname = clientUrl.pathname.replace(/^\/socket.io\//, "/");
-    const match = (0, import_matchRequestUrl.matchRequestUrl)(
-      clientUrl,
-      resolvedHandlerUrl,
-      args.resolutionContext?.baseUrl
-    );
-    return {
-      match
-    };
-  }
-  predicate(args) {
-    return args.parsedResult.match.matches;
-  }
-  test(url, resolutionContext) {
-    return this.#match(url, resolutionContext) != null;
-  }
-  async run(connection, resolutionContext) {
-    const parsedResult = this.#match(connection.client.url, resolutionContext);
-    if (parsedResult == null) {
-      return null;
-    }
-    const resolvedConnection = {
-      ...connection,
-      params: parsedResult.match.params || {}
-    };
-    if (resolutionContext?.[kAutoConnect] ?? true) {
-      if (this[kConnect](resolvedConnection)) {
-        return resolvedConnection;
-      }
-      return null;
-    }
-    return resolvedConnection;
-  }
-  #match(url, resolutionContext) {
-    const resolvedUrl = this.#resolveWebSocketUrl(
-      url.toString(),
-      resolutionContext?.baseUrl
-    );
-    const parsedResult = this.parse({
-      url: resolvedUrl,
-      resolutionContext
-    });
-    if (this.predicate({
-      url,
-      parsedResult
-    })) {
-      return parsedResult;
-    }
-    return null;
-  }
-  [kConnect](connection) {
-    connection.client.addEventListener(
-      "message",
-      createStopPropagationListener(this)
-    );
-    connection.client.addEventListener(
-      "close",
-      createStopPropagationListener(this)
-    );
-    connection.server.addEventListener(
-      "open",
-      createStopPropagationListener(this)
-    );
-    connection.server.addEventListener(
-      "message",
-      createStopPropagationListener(this)
-    );
-    connection.server.addEventListener(
-      "error",
-      createStopPropagationListener(this)
-    );
-    connection.server.addEventListener(
-      "close",
-      createStopPropagationListener(this)
-    );
-    return this[kEmitter].emit("connection", connection);
-  }
-  log(connection) {
-    return (0, import_attachWebSocketLogger.attachWebSocketLogger)(connection);
-  }
-  #resolveWebSocketUrl(url, baseUrl) {
-    const resolvedUrl = (0, import_interceptors.resolveWebSocketUrl)(
-      baseUrl ? (
-        /**
-         * @note Resolve against the base URL preemtively because `resolveWebSocketUrl` only
-         * resolves against `location.href`, which is missing in Node.js. Base URL allows
-         * the handler to accept a relative URL in Node.js.
-         */
-        new URL(url, baseUrl)
-      ) : url
-    );
-    return resolvedUrl.replace(/\/$/, "");
-  }
-}
-function createStopPropagationListener(handler) {
-  return function stopPropagationListener(event) {
-    const propagationStoppedAt = Reflect.get(event, "kPropagationStoppedAt");
-    if (propagationStoppedAt && handler.id !== propagationStoppedAt) {
-      event.stopImmediatePropagation();
-      return;
-    }
-    Object.defineProperty(event, KOnStopPropagation, {
-      value() {
-        Object.defineProperty(event, "kPropagationStoppedAt", {
-          value: handler.id
-        });
-      },
-      configurable: true
-    });
-    if (!Reflect.get(event, kStopPropagationPatched)) {
-      event.stopPropagation = new Proxy(event.stopPropagation, {
-        apply: (target, thisArg, args) => {
-          Reflect.get(event, KOnStopPropagation)?.call(handler);
-          return Reflect.apply(target, thisArg, args);
-        }
-      });
-      Object.defineProperty(event, kStopPropagationPatched, {
-        value: true,
-        // If something else attempts to redefine this, throw.
-        configurable: false
-      });
-    }
-  };
-}
-//# sourceMappingURL=WebSocketHandler.js.map

package/lib/core/handlers/WebSocketHandler.js.map

@@ -1 +0,0 @@
-{"version":3,"sources":["../../../src/core/handlers/WebSocketHandler.ts"],"sourcesContent":["import { Emitter } from 'strict-event-emitter'\nimport { createRequestId, resolveWebSocketUrl } from '@mswjs/interceptors'\nimport type {\n  WebSocketClientConnectionProtocol,\n  WebSocketConnectionData,\n  WebSocketServerConnectionProtocol,\n} from '@mswjs/interceptors/WebSocket'\nimport {\n  type Match,\n  type Path,\n  type PathParams,\n  matchRequestUrl,\n} from '../utils/matching/matchRequestUrl'\nimport { getCallFrame } from '../utils/internal/getCallFrame'\nimport { attachWebSocketLogger } from '../ws/utils/attachWebSocketLogger'\n\ntype WebSocketHandlerParsedResult = {\n  match: Match\n}\n\nexport type WebSocketHandlerEventMap = {\n  connection: [args: WebSocketHandlerConnection]\n}\n\nexport interface WebSocketHandlerConnection {\n  client: WebSocketClientConnectionProtocol\n  server: WebSocketServerConnectionProtocol\n  info: WebSocketConnectionData['info']\n  params: PathParams\n}\n\nexport interface WebSocketResolutionContext {\n  baseUrl?: string\n  [kAutoConnect]?: boolean\n}\n\nexport const kEmitter = Symbol('kEmitter')\nexport const kSender = Symbol('kSender')\nexport const kConnect = Symbol('kConnect')\nexport const kAutoConnect = Symbol('kAutoConnect')\n\nconst kStopPropagationPatched = Symbol('kStopPropagationPatched')\nconst KOnStopPropagation = Symbol('KOnStopPropagation')\n\nexport class WebSocketHandler {\n  public id: string\n  public callFrame?: string\n  public kind = 'websocket' as const\n\n  protected [kEmitter]: Emitter<WebSocketHandlerEventMap>\n\n  constructor(protected readonly url: Path) {\n    this.id = createRequestId()\n\n    this[kEmitter] = new Emitter()\n    this.callFrame = getCallFrame(new Error())\n  }\n\n  public parse(args: {\n    url: string | URL\n    resolutionContext?: WebSocketResolutionContext\n  }): WebSocketHandlerParsedResult {\n    const clientUrl = new URL(args.url)\n\n    // Resolve the WebSocket handler path:\n    // - Plain string URLs resolved as per the specification (via Interceptors).\n    // - String URLs starting with a wildcard are preserved (prepending a scheme there will break them).\n    // - RegExp paths are preserved.\n    const resolvedHandlerUrl =\n      this.url instanceof RegExp || this.url.startsWith('*')\n        ? this.url\n        : this.#resolveWebSocketUrl(this.url, args.resolutionContext?.baseUrl)\n\n    /**\n     * @note Remove the Socket.IO path prefix from the WebSocket\n     * client URL. This is an exception to keep the users from\n     * including the implementation details in their handlers.\n     */\n    clientUrl.pathname = clientUrl.pathname.replace(/^\\/socket.io\\//, '/')\n\n    const match = matchRequestUrl(\n      clientUrl,\n      resolvedHandlerUrl,\n      args.resolutionContext?.baseUrl,\n    )\n\n    return {\n      match,\n    }\n  }\n\n  public predicate(args: {\n    url: string | URL\n    parsedResult: WebSocketHandlerParsedResult\n  }): boolean {\n    return args.parsedResult.match.matches\n  }\n\n  public test(\n    url: string | URL,\n    resolutionContext?: WebSocketResolutionContext & { strict?: boolean },\n  ): boolean {\n    return this.#match(url, resolutionContext) != null\n  }\n\n  public async run(\n    connection: WebSocketConnectionData,\n    resolutionContext?: WebSocketResolutionContext,\n  ): Promise<WebSocketHandlerConnection | null> {\n    const parsedResult = this.#match(connection.client.url, resolutionContext)\n\n    if (parsedResult == null) {\n      return null\n    }\n\n    const resolvedConnection: WebSocketHandlerConnection = {\n      ...connection,\n      params: parsedResult.match.params || {},\n    }\n\n    if (resolutionContext?.[kAutoConnect] ?? true) {\n      if (this[kConnect](resolvedConnection)) {\n        return resolvedConnection\n      }\n\n      return null\n    }\n\n    return resolvedConnection\n  }\n\n  #match(\n    url: string | URL,\n    resolutionContext?: WebSocketResolutionContext & { strict?: boolean },\n  ): WebSocketHandlerParsedResult | null {\n    const resolvedUrl = this.#resolveWebSocketUrl(\n      url.toString(),\n      resolutionContext?.baseUrl,\n    )\n    const parsedResult = this.parse({\n      url: resolvedUrl,\n      resolutionContext,\n    })\n\n    if (\n      this.predicate({\n        url,\n        parsedResult,\n      })\n    ) {\n      return parsedResult\n    }\n\n    return null\n  }\n\n  protected [kConnect](connection: WebSocketHandlerConnection): boolean {\n    // Support `event.stopPropagation()` for various client/server events.\n    connection.client.addEventListener(\n      'message',\n      createStopPropagationListener(this),\n    )\n    connection.client.addEventListener(\n      'close',\n      createStopPropagationListener(this),\n    )\n\n    connection.server.addEventListener(\n      'open',\n      createStopPropagationListener(this),\n    )\n    connection.server.addEventListener(\n      'message',\n      createStopPropagationListener(this),\n    )\n    connection.server.addEventListener(\n      'error',\n      createStopPropagationListener(this),\n    )\n    connection.server.addEventListener(\n      'close',\n      createStopPropagationListener(this),\n    )\n\n    /**\n     * @fixme Use \"rettime\" and await these events to have\n     * exceptions propagate properly.\n     */\n    return this[kEmitter].emit('connection', connection)\n  }\n\n  public log(connection: WebSocketConnectionData): () => void {\n    return attachWebSocketLogger(connection)\n  }\n\n  #resolveWebSocketUrl(url: string, baseUrl?: string): string {\n    const resolvedUrl = resolveWebSocketUrl(\n      baseUrl\n        ? /**\n           * @note Resolve against the base URL preemtively because `resolveWebSocketUrl` only\n           * resolves against `location.href`, which is missing in Node.js. Base URL allows\n           * the handler to accept a relative URL in Node.js.\n           */\n          new URL(url, baseUrl)\n        : url,\n    )\n\n    /**\n     * @note Omit the trailing slash.\n     * While the browser always produces a trailing slash at the end of a WebSocket URL,\n     * having it in as the handler's predicate would mean it is *required* in the actual URL.\n     */\n    return resolvedUrl.replace(/\\/$/, '')\n  }\n}\n\nfunction createStopPropagationListener(handler: WebSocketHandler) {\n  return function stopPropagationListener(event: Event) {\n    const propagationStoppedAt = Reflect.get(event, 'kPropagationStoppedAt') as\n      | string\n      | undefined\n\n    if (propagationStoppedAt && handler.id !== propagationStoppedAt) {\n      event.stopImmediatePropagation()\n      return\n    }\n\n    Object.defineProperty(event, KOnStopPropagation, {\n      value(this: WebSocketHandler) {\n        Object.defineProperty(event, 'kPropagationStoppedAt', {\n          value: handler.id,\n        })\n      },\n      configurable: true,\n    })\n\n    // Since the same event instance is shared between all client/server objects,\n    // make sure to patch its `stopPropagation` method only once.\n    if (!Reflect.get(event, kStopPropagationPatched)) {\n      event.stopPropagation = new Proxy(event.stopPropagation, {\n        apply: (target, thisArg, args) => {\n          Reflect.get(event, KOnStopPropagation)?.call(handler)\n          return Reflect.apply(target, thisArg, args)\n        },\n      })\n\n      Object.defineProperty(event, kStopPropagationPatched, {\n        value: true,\n        // If something else attempts to redefine this, throw.\n        configurable: false,\n      })\n    }\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,kCAAwB;AACxB,0BAAqD;AAMrD,6BAKO;AACP,0BAA6B;AAC7B,mCAAsC;AAsB/B,MAAM,WAAW,uBAAO,UAAU;AAClC,MAAM,UAAU,uBAAO,SAAS;AAChC,MAAM,WAAW,uBAAO,UAAU;AAClC,MAAM,eAAe,uBAAO,cAAc;AAEjD,MAAM,0BAA0B,uBAAO,yBAAyB;AAChE,MAAM,qBAAqB,uBAAO,oBAAoB;AAE/C,MAAM,iBAAiB;AAAA,EAO5B,YAA+B,KAAW;AAAX;AAC7B,SAAK,SAAK,qCAAgB;AAE1B,SAAK,QAAQ,IAAI,IAAI,oCAAQ;AAC7B,SAAK,gBAAY,kCAAa,IAAI,MAAM,CAAC;AAAA,EAC3C;AAAA,EAL+B;AAAA,EANxB;AAAA,EACA;AAAA,EACA,OAAO;AAAA,EAEd,CAAW,QAAQ;AAAA,EASZ,MAAM,MAGoB;AAC/B,UAAM,YAAY,IAAI,IAAI,KAAK,GAAG;AAMlC,UAAM,qBACJ,KAAK,eAAe,UAAU,KAAK,IAAI,WAAW,GAAG,IACjD,KAAK,MACL,KAAK,qBAAqB,KAAK,KAAK,KAAK,mBAAmB,OAAO;AAOzE,cAAU,WAAW,UAAU,SAAS,QAAQ,kBAAkB,GAAG;AAErE,UAAM,YAAQ;AAAA,MACZ;AAAA,MACA;AAAA,MACA,KAAK,mBAAmB;AAAA,IAC1B;AAEA,WAAO;AAAA,MACL;AAAA,IACF;AAAA,EACF;AAAA,EAEO,UAAU,MAGL;AACV,WAAO,KAAK,aAAa,MAAM;AAAA,EACjC;AAAA,EAEO,KACL,KACA,mBACS;AACT,WAAO,KAAK,OAAO,KAAK,iBAAiB,KAAK;AAAA,EAChD;AAAA,EAEA,MAAa,IACX,YACA,mBAC4C;AAC5C,UAAM,eAAe,KAAK,OAAO,WAAW,OAAO,KAAK,iBAAiB;AAEzE,QAAI,gBAAgB,MAAM;AACxB,aAAO;AAAA,IACT;AAEA,UAAM,qBAAiD;AAAA,MACrD,GAAG;AAAA,MACH,QAAQ,aAAa,MAAM,UAAU,CAAC;AAAA,IACxC;AAEA,QAAI,oBAAoB,YAAY,KAAK,MAAM;AAC7C,UAAI,KAAK,QAAQ,EAAE,kBAAkB,GAAG;AACtC,eAAO;AAAA,MACT;AAEA,aAAO;AAAA,IACT;AAEA,WAAO;AAAA,EACT;AAAA,EAEA,OACE,KACA,mBACqC;AACrC,UAAM,cAAc,KAAK;AAAA,MACvB,IAAI,SAAS;AAAA,MACb,mBAAmB;AAAA,IACrB;AACA,UAAM,eAAe,KAAK,MAAM;AAAA,MAC9B,KAAK;AAAA,MACL;AAAA,IACF,CAAC;AAED,QACE,KAAK,UAAU;AAAA,MACb;AAAA,MACA;AAAA,IACF,CAAC,GACD;AACA,aAAO;AAAA,IACT;AAEA,WAAO;AAAA,EACT;AAAA,EAEA,CAAW,QAAQ,EAAE,YAAiD;AAEpE,eAAW,OAAO;AAAA,MAChB;AAAA,MACA,8BAA8B,IAAI;AAAA,IACpC;AACA,eAAW,OAAO;AAAA,MAChB;AAAA,MACA,8BAA8B,IAAI;AAAA,IACpC;AAEA,eAAW,OAAO;AAAA,MAChB;AAAA,MACA,8BAA8B,IAAI;AAAA,IACpC;AACA,eAAW,OAAO;AAAA,MAChB;AAAA,MACA,8BAA8B,IAAI;AAAA,IACpC;AACA,eAAW,OAAO;AAAA,MAChB;AAAA,MACA,8BAA8B,IAAI;AAAA,IACpC;AACA,eAAW,OAAO;AAAA,MAChB;AAAA,MACA,8BAA8B,IAAI;AAAA,IACpC;AAMA,WAAO,KAAK,QAAQ,EAAE,KAAK,cAAc,UAAU;AAAA,EACrD;AAAA,EAEO,IAAI,YAAiD;AAC1D,eAAO,oDAAsB,UAAU;AAAA,EACzC;AAAA,EAEA,qBAAqB,KAAa,SAA0B;AAC1D,UAAM,kBAAc;AAAA,MAClB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,QAMI,IAAI,IAAI,KAAK,OAAO;AAAA,UACpB;AAAA,IACN;AAOA,WAAO,YAAY,QAAQ,OAAO,EAAE;AAAA,EACtC;AACF;AAEA,SAAS,8BAA8B,SAA2B;AAChE,SAAO,SAAS,wBAAwB,OAAc;AACpD,UAAM,uBAAuB,QAAQ,IAAI,OAAO,uBAAuB;AAIvE,QAAI,wBAAwB,QAAQ,OAAO,sBAAsB;AAC/D,YAAM,yBAAyB;AAC/B;AAAA,IACF;AAEA,WAAO,eAAe,OAAO,oBAAoB;AAAA,MAC/C,QAA8B;AAC5B,eAAO,eAAe,OAAO,yBAAyB;AAAA,UACpD,OAAO,QAAQ;AAAA,QACjB,CAAC;AAAA,MACH;AAAA,MACA,cAAc;AAAA,IAChB,CAAC;AAID,QAAI,CAAC,QAAQ,IAAI,OAAO,uBAAuB,GAAG;AAChD,YAAM,kBAAkB,IAAI,MAAM,MAAM,iBAAiB;AAAA,QACvD,OAAO,CAAC,QAAQ,SAAS,SAAS;AAChC,kBAAQ,IAAI,OAAO,kBAAkB,GAAG,KAAK,OAAO;AACpD,iBAAO,QAAQ,MAAM,QAAQ,SAAS,IAAI;AAAA,QAC5C;AAAA,MACF,CAAC;AAED,aAAO,eAAe,OAAO,yBAAyB;AAAA,QACpD,OAAO;AAAA;AAAA,QAEP,cAAc;AAAA,MAChB,CAAC;AAAA,IACH;AAAA,EACF;AACF;","names":[]}