Sha256: f557685de2759b69db2b5e7cdc3eecc366f7cd08bf7eea77f51c8192fbdfdb0a

Contents?: true

Size: 1.21 KB

Versions: 11

Compression:

Stored size: 1.21 KB

Contents

module GovukTechDocs
  module TableOfContents
    class HeadingTreeRenderer
      DEFAULT_MAX_LEVEL = Float::INFINITY
      DEFAULT_INDENTATION = ''.freeze
      INDENTATION_INCREMENT = '  '.freeze

      def initialize(heading_tree, max_level: nil)
        @heading_tree = heading_tree
        @max_level = max_level || DEFAULT_MAX_LEVEL
      end

      def html
        render_tree(@heading_tree, level: 0)
      end

    private

      def render_tree(tree, indentation: DEFAULT_INDENTATION, level: nil)
        output = ''

        if tree.heading
          output += indentation + %{<a href="#{tree.heading.href}"><span>#{tree.heading.title}</span></a>\n}
        end

        if tree.children.any? && level < @max_level
          output += indentation + "<ul>\n"

          tree.children.each do |child|
            output += indentation + INDENTATION_INCREMENT + "<li>\n"
            output += render_tree(
              child,
              indentation: indentation + INDENTATION_INCREMENT * 2,
              level: level + 1
            )
            output += indentation + INDENTATION_INCREMENT + "</li>\n"
          end

          output += indentation + "</ul>\n"
        end

        output
      end
    end
  end
end

Version data entries

11 entries across 11 versions & 1 rubygems

Version Path
govuk_tech_docs-2.0.7 lib/govuk_tech_docs/table_of_contents/heading_tree_renderer.rb
govuk_tech_docs-2.0.6 lib/govuk_tech_docs/table_of_contents/heading_tree_renderer.rb
govuk_tech_docs-2.0.5 lib/govuk_tech_docs/table_of_contents/heading_tree_renderer.rb
govuk_tech_docs-2.0.4 lib/govuk_tech_docs/table_of_contents/heading_tree_renderer.rb
govuk_tech_docs-2.0.3 lib/govuk_tech_docs/table_of_contents/heading_tree_renderer.rb
govuk_tech_docs-2.0.2 lib/govuk_tech_docs/table_of_contents/heading_tree_renderer.rb
govuk_tech_docs-2.0.1 lib/govuk_tech_docs/table_of_contents/heading_tree_renderer.rb
govuk_tech_docs-2.0.0.pre.test.pre.release.pre.525 lib/govuk_tech_docs/table_of_contents/heading_tree_renderer.rb
govuk_tech_docs-2.0.0.pre.test.pre.release.pre.524 lib/govuk_tech_docs/table_of_contents/heading_tree_renderer.rb
govuk_tech_docs-2.0.0.test.pre.release.521 lib/govuk_tech_docs/table_of_contents/heading_tree_renderer.rb
govuk_tech_docs-2.0.0 lib/govuk_tech_docs/table_of_contents/heading_tree_renderer.rb