demarches-normaliennes/app/models/concerns/treeable_concern.rb

module TreeableConcern
  extend ActiveSupport::Concern

  MAX_DEPTH = 6 # deepest level for header_sections is 3.
  # but a repetition can be nested an header_section, so 3+3=6=MAX_DEPTH

  included do
    # as we progress in the list of ordered champs
    #   we keep a reference to each level of nesting (walk)
    # when we encounter an header_section, it depends of its own depth of nesting minus 1, ie:
    #   h1 belongs to prior (rooted_tree)
    #   h2 belongs to prior h1
    #   h3 belongs to prior h2
    #   h1 belongs to prior (rooted_tree)
    # then, each and every champs which are not an header_section
    #   are added to the current_tree
    # given a root_depth at 0, we build a full tree
    # given a root_depth > 0, we build a partial tree (aka, a repetition)
    def to_tree(champs:)
      rooted_tree = []
      walk = Array.new(MAX_DEPTH)
      walk[0] = rooted_tree
      current_tree = rooted_tree

      champs.each do |champ|
        if champ.header_section?
          new_tree = [champ]
          walk[champ.header_section_level_value - 1].push(new_tree)
          current_tree = walk[champ.header_section_level_value] = new_tree
        else
          current_tree.push(champ)
        end
      end
      rooted_tree
    end
  end
end