Add new GraphQL resolver

Author: andrewbranch

gatsby-node.js

@@ -13,45 +13,15 @@ exports.createResolvers = ({
       grvscCodeBlocks: {
         type: ['GRVSCCodeBlock'],
         resolve(source, _, context) {
-          return getFromCache();
-
-          /** @param {boolean=} stop */
-          async function getFromCache(stop) {
-            const childNodes = await getChildNodes(cache, source.id, source.internal.contentDigest);
-            // Hack alert: ensure plugin has been run by querying htmlAst,
-            // which is set via `setFieldsOnGraphQLNodeType` by gatsby-transformer-remark,
-            // therefore might not have been run before this resolver runs.
-            if (!childNodes && !stop) {
-              await context.nodeModel.runQuery({
-                query: {
-                  filter: {
-                    id: { eq: source.id },
-                    htmlAst: { ne: null },
-                  },
-                },
-                type: 'MarkdownRemark',
-                firstOnly: true,
-              });
-              return getFromCache(true);
-            }
-            if (!childNodes) {
-              logger.error(
-                'gatsby-remark-vscode couldn’t retrieve up-to-date GRVSCCodeBlock GraphQL nodes. ' +
-                'The `GRVSCCodeBlocks` field may be missing, empty or stale. ' +
-                'The Gatsby cache is probably in a weird state. Try running `gatsby clean`, and file an ' +
-                'issue at https://github.com/andrewbranch/gatsby-remark-vscode/issues/new if the problem persists.'
-              );
-
-              return context.nodeModel.runQuery({
-                query: { parent: { id: { eq: source.id } } },
-                type: 'GRVSCCodeBlock',
-                firstOnly: false
-              });
-            }
-            return childNodes || [];
-          }
-        },
+          return getFromCache('GRVSCCodeBlock', cache, source, context);
+        }
       },
+      grvscCodeSpans: {
+        type: ['GRVSCCodeSpan'],
+        resolve(source, _, context) {
+          return getFromCache('GRVSCCodeSpan', cache, source, context);
+        }
+      }
     },
 
     Query: {
@@ -83,3 +53,45 @@ exports.createResolvers = ({
     }
   });
 };
+
+/**
+ * @param {string} type
+ * @param {any} cache
+ * @param {any} source
+ * @param {any} context
+ * @param {boolean=} stop
+ */
+async function getFromCache(type, cache, source, context, stop) {
+  const childNodes = await getChildNodes(cache, source.id, source.internal.contentDigest);
+  // Hack alert: ensure plugin has been run by querying htmlAst,
+  // which is set via `setFieldsOnGraphQLNodeType` by gatsby-transformer-remark,
+  // therefore might not have been run before this resolver runs.
+  if (!childNodes && !stop) {
+    await context.nodeModel.runQuery({
+      query: {
+        filter: {
+          id: { eq: source.id },
+          htmlAst: { ne: null },
+        },
+      },
+      type: 'MarkdownRemark',
+      firstOnly: true,
+    });
+    return getFromCache(cache, source, context, true);
+  }
+  if (!childNodes) {
+    logger.error(
+      'gatsby-remark-vscode couldn’t retrieve up-to-date GRVSCCodeBlock GraphQL nodes. ' +
+      'The `GRVSCCodeBlocks` field may be missing, empty or stale. ' +
+      'The Gatsby cache is probably in a weird state. Try running `gatsby clean`, and file an ' +
+      'issue at https://github.com/andrewbranch/gatsby-remark-vscode/issues/new if the problem persists.'
+    );
+
+    return context.nodeModel.runQuery({
+      query: { parent: { id: { eq: source.id } } },
+      type,
+      firstOnly: false
+    });
+  }
+  return childNodes || [];
+}