Refactor table of contents building

1 files changed, 11 insertions, 19 deletions

diff --git a/lib/banzai/filter/table_of_contents_filter.rb b/lib/banzai/filter/table_of_contents_filter.rb
index 5522986ba21..47151626208 100644
--- a/lib/banzai/filter/table_of_contents_filter.rb
+++ b/lib/banzai/filter/table_of_contents_filter.rb
@@ -43,15 +43,7 @@ module Banzai
           end
         end
 
-        if header_root.children.length > 0
-          result[:toc] = %q{<ul class="section-nav">}
-
-          header_root.children.each do |child|
-            push_toc(child)
-          end
-
-          result[:toc] << '</ul>'
-        end
+        push_toc(header_root.children, root: true)
 
         doc
       end
@@ -62,19 +54,19 @@ module Banzai
         %Q{<a id="user-content-#{href}" class="anchor" href="##{href}" aria-hidden="true"></a>}
       end
 
-      def push_toc(header_node)
-        result[:toc] << %Q{<li><a href="##{header_node.href}">#{header_node.text}</a>}
-
-        if header_node.children.length > 0
-          result[:toc] << '<ul>'
+      def push_toc(children, root: false)
+        return if children.empty?
 
-          header_node.children.each do |child|
-            push_toc(child)
-          end
+        klass = ' class="section-nav"' if root
 
-          result[:toc] << '</ul>'
-        end
+        result[:toc] << "<ul#{klass}>"
+        children.each { |child| push_anchor(child) }
+        result[:toc] << '</ul>'
+      end
 
+      def push_anchor(header_node)
+        result[:toc] << %Q{<li><a href="##{header_node.href}">#{header_node.text}</a>}
+        push_toc(header_node.children)
         result[:toc] << '</li>'
       end