@graphql-mesh/transport-ws/dist/index.js

import { process } from '@graphql-mesh/cross-helpers';
import { getInterpolatedHeadersFactory } from '@graphql-mesh/string-interpolation';
import { makeAsyncDisposable, isDisposable, dispose } from '@graphql-mesh/utils';
import { buildGraphQLWSExecutor } from '@graphql-tools/executor-graphql-ws';

function switchProtocols(url) {
  if (url.startsWith("https://")) {
    return url.replace("https://", "wss://");
  }
  if (url.startsWith("http://")) {
    return url.replace("http://", "ws://");
  }
  return url;
}
var index = {
  getSubgraphExecutor({ transportEntry, logger }, onClient) {
    const wsExecutorMap = /* @__PURE__ */ new Map();
    if (!transportEntry.location) {
      throw new Error(
        "WS Transport: location is required in the transport entry"
      );
    }
    const wsUrl = switchProtocols(transportEntry.location);
    const connectionParamsFactory = transportEntry.options?.connectionParams ? getInterpolatedHeadersFactory(transportEntry.options.connectionParams) : void 0;
    const headersFactory = transportEntry.headers ? getInterpolatedHeadersFactory(
      Object.fromEntries(transportEntry.headers)
    ) : void 0;
    const mergedExecutor = function mergedExecutor2(execReq) {
      const connectionParams = connectionParamsFactory?.({
        env: process.env,
        root: execReq.rootValue,
        context: execReq.context,
        info: execReq.info
      });
      const headers = headersFactory?.({
        env: process.env,
        root: execReq.rootValue,
        context: execReq.context,
        info: execReq.info
      });
      const hash = JSON.stringify({ wsUrl, connectionParams, headers });
      let wsExecutor = wsExecutorMap.get(hash);
      if (!wsExecutor) {
        const executorLogger = logger?.child({
          executor: "GraphQL WS",
          wsUrl,
          connectionParams,
          headers
        });
        wsExecutor = buildGraphQLWSExecutor({
          headers,
          url: wsUrl,
          lazy: true,
          lazyCloseTimeout: 3e3,
          ...transportEntry.options,
          connectionParams,
          on: {
            connecting(isRetry) {
              executorLogger?.debug("connecting", { isRetry });
            },
            opened(socket) {
              executorLogger?.debug("opened", { socket });
            },
            connected(socket, payload) {
              executorLogger?.debug("connected", { socket, payload });
            },
            ping(received, payload) {
              executorLogger?.debug("ping", { received, payload });
            },
            pong(received, payload) {
              executorLogger?.debug("pong", { received, payload });
            },
            message(message) {
              executorLogger?.debug("message", { message });
            },
            closed(event) {
              executorLogger?.debug("closed", { event });
              wsExecutorMap.delete(hash);
            },
            error(error) {
              executorLogger?.debug("error", { error });
            }
          },
          onClient
        });
        wsExecutorMap.set(hash, wsExecutor);
      }
      return wsExecutor(execReq);
    };
    return makeAsyncDisposable(
      mergedExecutor,
      () => Promise.all(
        Array.from(wsExecutorMap.values()).map(
          (executor) => isDisposable(executor) && dispose(executor)
        )
      ).then(() => {
      })
    );
  }
};

export { index as default };