Allow introspecting onSubscribe from app level. (#952)

* allow introspecting onSubscribe from app level * add changeset * fix type error * address feedback
PabloSzx · May 1, 2023 · 36e7c9e · 36e7c9e · vercel · May 1, 2023
1 parent 9b5eb5c
commit 36e7c9e
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 5 deletions.
diff --git a/.changeset/small-hairs-refuse.md b/.changeset/small-hairs-refuse.md
@@ -0,0 +1,5 @@
+---
+'@graphql-ez/plugin-websockets': patch
+---
+
+allow introspecting onSubscribe from app level
diff --git a/packages/plugin/websockets/src/core.ts b/packages/plugin/websockets/src/core.ts
@@ -14,10 +14,7 @@ export type FilteredSubscriptionsTransportOptions = Omit<
   'schema' | 'execute' | 'subscribe' | 'onConnect' | 'validate' | 'parse'
 >;
 
-export type FilteredGraphQLWSOptions = Omit<
-  GraphQLWSOptions,
-  'schema' | 'execute' | 'subscribe' | 'context' | 'validate' | 'onSubscribe'
->;
+export type FilteredGraphQLWSOptions = Omit<GraphQLWSOptions, 'schema' | 'execute' | 'subscribe' | 'context' | 'validate'>;
 
 export interface WebSocketsState {
   closing: boolean;
@@ -88,7 +85,19 @@ export function handleGraphQLWS(
       ...cleanObject(options),
       execute,
       subscribe,
-      async onSubscribe({ connectionParams, extra: { request, socket } }, { payload: { operationName, query, variables } }) {
+      async onSubscribe(context, message) {
+        const {
+          connectionParams,
+          extra: { request, socket },
+        } = context;
+        const {
+          payload: { operationName, query, variables },
+        } = message;
+
+        if (typeof options?.onSubscribe === 'function') {
+          options.onSubscribe(context, message);
+        }
+
         const contextArgsData = {
           req: request,
           ws: {