fix(gatsby-plugin-mdx): fix filter by heading depth (#16344)

gatsbyjs · Aug 6, 2019 · 234863c · 234863c
1 parent f9d51bb
commit 234863c
Showing 1 changed file with 4 additions and 9 deletions.
diff --git a/packages/gatsby-plugin-mdx/gatsby/source-nodes.js b/packages/gatsby-plugin-mdx/gatsby/source-nodes.js
@@ -59,7 +59,7 @@ module.exports = (
   const { createTypes } = actions;
 
   const options = defaultOptions(pluginOptions);
-
+  const headingsMdx = ['h1', 'h2', 'h3', 'h4', 'h5', 'h6'];
   createTypes(`
     type MdxFrontmatter {
       title: String!
@@ -71,12 +71,7 @@ module.exports = (
     }
 
     enum HeadingsMdx {
-      h1,
-      h2,
-      h3,
-      h4,
-      h5,
-      h6
+      ${headingsMdx}
     }
 
     type MdxWordCount {
@@ -172,8 +167,8 @@ module.exports = (
               depth: heading.depth
             });
           });
-          if (typeof depth === `number`) {
-            headings = headings.filter(heading => heading.depth === depth);
+          if (headingsMdx.includes(depth)) {
+            headings = headings.filter(heading => `h${heading.depth}` === depth);
           }
           return headings;
         }