Contents

# -*- coding: utf-8 -*-

module TreeVisitor
  #
  # Build hash with directory structure
  #
  class DirectoryToHashVisitor # < TreeVisitor::BasicTreeNodeVisitor

    attr_reader :root

    def initialize(pathname)
      @stack = []
      @node = {}
      @root = @node
    end

    def enter_node(pathname)
      subnode                        = {}
      @node[File.basename(pathname)] = subnode
      @stack.push(@node)
      @node = subnode
    end

    def exit_node(pathname)
      @node = @stack.pop
    end

    def visit_leaf(pathname)
      @node[File.basename(pathname)] = File.stat(pathname).size
    end

  end
end

Version data entries

5 entries across 5 versions & 2 rubygems

Version	Path
tree.rb-0.3.0	lib/treevisitor/visitors/directory_to_hash_visitor.rb
treevisitor-0.2.3	lib/treevisitor/visitors/directory_to_hash_visitor.rb
treevisitor-0.2.2	lib/treevisitor/visitors/directory_to_hash_visitor.rb
treevisitor-0.2.1	lib/treevisitor/visitors/directory_to_hash_visitor.rb
treevisitor-0.2.0	lib/treevisitor/visitors/directory_to_hash_visitor.rb