fix(graphiql): fix operation detection on multi-op documents (#1191)

graphile · Nov 20, 2019 · 49b2176 · 49b2176
1 parent 8f1ac84
commit 49b2176
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/postgraphiql/src/components/PostGraphiQL.js b/postgraphiql/src/components/PostGraphiQL.js
@@ -1,6 +1,6 @@
 import React from 'react';
 import GraphiQL from 'graphiql';
-import { parse } from 'graphql';
+import { getOperationAST, parse } from 'graphql';
 import GraphiQLExplorer from 'graphiql-explorer';
 import StorageAPI from 'graphiql/dist/utility/StorageAPI';
 import './postgraphiql.css';
@@ -41,11 +41,13 @@ const defaultQuery = `\
 #
 `;
 
-const isSubscription = ({ query }) =>
-  parse(query).definitions.some(
-    definition =>
-      definition.kind === 'OperationDefinition' && definition.operation === 'subscription',
-  );
+const isSubscription = ({ query, operationName }) => {
+  const node = parse(query);
+
+  const operation = getOperationAST(node, operationName);
+
+  return operation && operation.operation === 'subscription';
+};
 
 const {
   POSTGRAPHILE_CONFIG = {