fix bug with certain headings (#246)

tscanlin · Oct 21, 2021 · bd4a8a8 · bd4a8a8
1 parent 27abc7a
commit bd4a8a8
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/src/components/Template/Tocbot/index.js b/src/components/Template/Tocbot/index.js
@@ -3,7 +3,7 @@ import React from 'react'
 const TOCBOT_OPTIONS = {
   tocSelector: '.js-toc',
   contentSelector: '.js-toc-content',
-  headingSelector: 'h2, h3, h4',
+  headingSelector: 'h2, h3, h4, h5, h6',
   positionFixedSelector: '.js-toc',
   includeHtml: true,
   includeTitleTags: false,

diff --git a/src/js/parse-content.js b/src/js/parse-content.js
@@ -78,7 +78,10 @@ module.exports = function parseContent (options) {
 
     while (counter > 0) {
       lastItem = getLastItem(array)
-      if (lastItem && lastItem.children !== undefined) {
+      // Handle case where there are multiple h5+ in a row.
+      if (lastItem && level === lastItem.headingLevel) {
+        break
+      } else if (lastItem && lastItem.children !== undefined) {
         array = lastItem.children
       }
       counter--