Contents

class Eco::API::UseCases::DefaultCases::CsvToTree
  module Treeify
    def treeify(nodes, &block)
      get_children(nil, parents_hash(nodes), &block)
    end

    private

    def parents_hash(nodes)
      nodes.each_with_object({}) do |node, parents|
        (parents[node.parentId] ||= []).push(node)
      end
    end

    def get_children(node_id, parents, &block)
      (parents[node_id] ||= []).each_with_object([]) do |child, results|
        node_hash = {
          "id"    => child.id,
          "name"  => child.name
        }

        if block_given?
          yield_hash = yield(child)
          node_hash.merge(yield_hash) if yield_hash.is_a?(Hash)
        end
        
        results << node_hash.merge({
          "nodes" => get_children(child.id, parents, &block).compact
        })
      end
    end
  end
end

Version data entries

23 entries across 23 versions & 1 rubygems

Version	Path
eco-helpers-2.2.3	lib/eco/api/usecases/default_cases/csv_to_tree_case/treeify.rb
eco-helpers-2.2.2	lib/eco/api/usecases/default_cases/csv_to_tree_case/treeify.rb
eco-helpers-2.2.1	lib/eco/api/usecases/default_cases/csv_to_tree_case/treeify.rb