feat(BA-2806): Implement WebSocket connectionParams to HTTP headers conversion

changes/6379.feature.md

@@ -0,0 +1 @@
+Implement WebSocket connectionParams to HTTP headers conversion

configs/graphql/gateway.config.ts

@@ -1,13 +1,54 @@
-import { defineConfig } from '@graphql-hive/gateway';
+import { defineConfig, WSTransportOptions, type GatewayPlugin } from '@graphql-hive/gateway';
+
+// Custom plugin to convert WebSocket connectionParams to HTTP headers
+// This allows authentication parameters sent via WebSocket connection_init
+// to be forwarded as HTTP headers when the gateway makes federation requests to subgraphs
+const useConnectionParamsToHeadersPlugin = (): GatewayPlugin => {
+  return {
+    onSubgraphExecute({ executionRequest, executor, setExecutor }) {
+      // Access WebSocket connectionParams from context
+      const connectionParams = executionRequest.context?.connectionParams;
+
+      if (connectionParams && typeof connectionParams === 'object') {
+        // Wrap executor to inject connectionParams as HTTP headers
+        const originalExecutor = executor;
+        const wrappedExecutor = async (execRequest: any) => {
+          // Extract Authorization header from context if it exists
+          let reqHeaders = execRequest.context.headers || {};
+          delete reqHeaders['content-length'];
+          // Merge connectionParams with Authorization (if present)
+          const mergedHeaders = {
+            ...reqHeaders,
+            ...connectionParams,
+          };
+          const modifiedRequest = {
+            ...execRequest,
+            extensions: {
+              ...execRequest.extensions,
+              headers: mergedHeaders,
+            },
+          };
+          return originalExecutor(modifiedRequest);
+        };
+
+        setExecutor(wrappedExecutor);
+      }
+    },
+  };
+};
 
 export const gatewayConfig = defineConfig({
+  plugins: (ctx) => [
+    useConnectionParamsToHeadersPlugin(),
+  ],
   propagateHeaders: {
-    fromClientToSubgraphs({ request }) {
-      let headers = Object.fromEntries(request.headers.entries());
+    fromClientToSubgraphs({ context }) {
+      let headers = context.headers || {};
       delete headers['content-length'];
       return headers;
-    }
+    },
   },
+  logging: false,
   supergraph: '/gateway/supergraph.graphql',
   graphqlEndpoint: '/admin/gql',
   skipValidation: true,