Contents

module GovukTechDocs
  module TableOfContents
    class HeadingTreeBuilder
      def initialize(headings)
        @headings = headings
        @tree = HeadingTree.new
        @pointer = @tree
      end

      def tree
        @headings.each do |heading|
          move_to_depth(heading.size)

          @pointer.children << HeadingTree.new(parent: @pointer, heading:)
        end

        @tree
      end

    private

      def move_to_depth(depth)
        if depth > @pointer.depth
          @pointer = @pointer.children.last

          if depth > @pointer.depth
            @pointer.children << HeadingTree.new(parent: @pointer)

            move_to_depth(depth)
          end
        end

        if depth < @pointer.depth
          @pointer = @pointer.parent

          move_to_depth(depth)
        end
      end
    end
  end
end

Version data entries

5 entries across 5 versions & 1 rubygems

Version	Path
govuk_tech_docs-4.2.0	lib/govuk_tech_docs/table_of_contents/heading_tree_builder.rb
govuk_tech_docs-4.1.2	lib/govuk_tech_docs/table_of_contents/heading_tree_builder.rb
govuk_tech_docs-4.1.1	lib/govuk_tech_docs/table_of_contents/heading_tree_builder.rb
govuk_tech_docs-4.1.0	lib/govuk_tech_docs/table_of_contents/heading_tree_builder.rb
govuk_tech_docs-4.0.0	lib/govuk_tech_docs/table_of_contents/heading_tree_builder.rb