Seven Languages in Seven Weeks - Ruby - Day 2 - Tree

tree_hash.rb

class Tree
    attr_accessor :children, :node_name, :depth

    def initialize(structure, depth)
        @node_name = structure.keys.first
        @depth = depth
        @children = structure.values.first.map { |name,child| Tree.new({name,child}, depth+1) }
    end

    def visit_all(&block)
        visit &block
        children.each {|c| c.visit_all &block}
    end

    def visit(&block)
        block.call self
    end
end

treeHash = 
    {
        'grandpa' => 
        {
            'dad' => 
            {
                'child1' => {},
                'child2' => {}
            },
            'uncle' => 
            {
               'child3' => {},
               'child4' => {}
            }
        }
    }

tree = Tree.new(treeHash, 1) 

puts "Visiting entire tree"
puts 
tree.visit_all do |node| 
    (node.depth-1).times { print "--" }
    puts node.node_name
end