treevisitor 0.1.4

data/lib/treevisitor/visitors/block_tree_node_visitor.rb

@@ -0,0 +1,21 @@
+# -*- coding: utf-8 -*-
+module TreeVisitor
+  #
+  # It calls a block when visit a tree_node or leaf_node
+  #
+  class BlockTreeNodeVisitor < TreeNodeVisitor
+
+    def initialize( &action )
+      @block = action
+    end
+
+    def enter_tree_node( tree_node )
+      @block.call( tree_node )
+    end
+
+    def visit_leaf_node( leaf_node )
+      @block.call( leaf_node )
+    end
+
+  end
+end

data/lib/treevisitor/visitors/build_dir_tree_visitor.rb

@@ -0,0 +1,53 @@
+# -*- coding: utf-8 -*-
+module TreeVisitor
+  #
+  # Builds a TreeNode from a filesystem directory
+  # It similar to CloneTreeNodeVisitor
+  #
+  class BuildDirTreeVisitor < TreeNodeVisitor
+  
+    attr_reader :root
+
+    #
+    # Number of visited directory (aka nr_nodes - nr_leaf)
+    #
+    attr_reader :nr_directories
+
+
+    #
+    # Number of visited directory (nr_leaves)
+    # @see AbsNode#nr_leaves
+    #
+    attr_reader :nr_files
+  
+    def initialize
+      super
+      @root = nil
+      @stack = []
+      @nr_directories = 0
+      @nr_files = 0
+    end
+  
+    def enter_tree_node( pathname )
+      if @stack.empty?
+        tree_node = TreeNode.new( File.basename( pathname ) )
+        @root = tree_node
+      else
+        tree_node = TreeNode.new( File.basename( pathname ), @stack.last )
+      end
+      @nr_directories += 1
+      @stack.push( tree_node )
+    end
+
+    def exit_tree_node( pathname )
+      @stack.pop
+    end
+
+    def visit_leaf_node( pathname )
+      @nr_files  += 1
+      # connect the leaf_node created to the last tree_node on the stack
+      LeafNode.new( File.basename(pathname), @stack.last )
+    end
+
+  end
+end

data/lib/treevisitor/visitors/callback_tree_node_visitor.rb

@@ -0,0 +1,43 @@
+# -*- coding: utf-8 -*-
+module TreeVisitor
+  #
+  # Executes a block when enter in a node
+  # The block are defined from on_enter_X methods
+  # The blocks take as argument only the node
+  #
+  class CallbackTreeNodeVisitor < TreeNodeVisitor
+
+    def initialize
+      super()
+      @root = nil
+      @stack = []
+      @action_enter_tree_node = nil
+      @action_visit_leaf_node = nil
+    end
+
+    def on_enter_tree_node( &action )
+      @action_enter_tree_node = action
+    end
+
+    def on_visit_leaf_node( &action )
+      @action_visit_leaf_node = action
+    end
+
+    def enter_tree_node( tree_node )
+      # parent_node = @stack.empty? ? nil : @stack.last
+      @root = tree_node if @stack.empty?
+      @stack.push( tree_node )
+      @action_enter_tree_node.call( tree_node ) if @action_enter_tree_node
+    end
+
+    def exit_tree_node( tree_node )
+      @stack.pop
+    end
+
+    def visit_leaf_node( leaf_node )
+      # parent_node = @stack.last
+      @action_visit_leaf_node.call( leaf_node ) if @action_visit_leaf_node
+    end
+
+  end
+end

data/lib/treevisitor/visitors/callback_tree_node_visitor2.rb

@@ -0,0 +1,61 @@
+# -*- coding: utf-8 -*-
+module TreeVisitor
+  #
+  # Executes a block when enter in a node
+  # The block are defined from on_enter_X methods
+  # The blocks take as argument the node and the parent_node
+  #
+  class CallbackTreeNodeVisitor2 < TreeNodeVisitor
+
+    attr_reader :root
+
+    def initialize( delegate = nil)
+      super()
+      @stack = []
+      @root = nil
+      @delegate = delegate
+    end
+
+    def on_enter_tree_node( &action )
+      @action_enter_tree_node = action
+    end
+
+    def on_visit_leaf_node( &action )
+      @action_visit_leaf_node = action
+    end
+
+    def _on_enter_tree_node( src_tree_node, dst_parent_node )
+      if @action_enter_tree_node
+        dst_tree_node = @action_enter_tree_node.call( src_tree_node, dst_parent_node )
+      elsif @delegate
+        dst_tree_node = @delegate.on_enter_tree_node( src_tree_node, dst_parent_node )
+      end
+      dst_tree_node
+    end
+
+    def _on_visit_leaf_node( src_leaf_node, parent_node )
+      if @action_visit_leaf_node
+        @action_visit_leaf_node.call( src_leaf_node, parent_node )
+      elsif @delegate
+        @delegate.on_visit_leaf_node( src_leaf_node, parent_node )
+      end
+    end
+
+    def enter_tree_node( src_tree_node )
+      dst_parent_node = @stack.empty? ? nil : @stack.last
+      dst_tree_node = _on_enter_tree_node( src_tree_node, dst_parent_node )
+      @root = dst_tree_node if @stack.empty?
+      @stack.push( dst_tree_node )
+    end
+
+    def exit_tree_node( src_tree_node )
+      @stack.pop
+    end
+
+    def visit_leaf_node( src_leaf_node )
+      parent_node = @stack.last
+      _on_visit_leaf_node( src_leaf_node, parent_node )
+    end
+  end
+
+end

data/lib/treevisitor/visitors/clone_tree_node_visitor.rb

@@ -0,0 +1,39 @@
+# -*- coding: utf-8 -*-
+module TreeVisitor
+  #
+  # Clone a tree_node, nodes are duplicated.
+  # Node content are not duplicated!
+  #
+  class CloneTreeNodeVisitor < TreeNodeVisitor
+
+    #
+    # Contains the cloned tree node after the visit
+    #
+    attr_reader :cloned_root
+
+    def initialize
+      super
+      @cloned_root = nil
+      @stack = []
+    end
+
+    def enter_tree_node( tree_node )
+      if @stack.empty?
+        cloned_tree_node = TreeNode.new( tree_node.content )
+        @cloned_root = cloned_tree_node
+      else
+        cloned_tree_node = TreeNode.new( tree_node.content, @stack.last )
+      end
+      @stack.push( cloned_tree_node )
+    end
+
+    def exit_tree_node( tree_node )
+      @stack.pop
+    end
+
+    def visit_leaf_node( leaf_node )
+      LeafNode.new( leaf_node.content, @stack.last )
+    end
+
+  end
+end

data/lib/treevisitor/visitors/depth_tree_node_visitor.rb

@@ -0,0 +1,27 @@
+# -*- coding: utf-8 -*-
+module TreeVisitor
+  #
+  # Simple visitor: show how calculate the depth of a tree
+  #
+  class DepthTreeNodeVisitor < TreeNodeVisitor
+
+    attr_reader :depth
+
+    def initialize
+      super
+      @depth = 0
+    end
+
+    def enter_tree_node( tree_node )
+      @depth += 1
+    end
+
+    def exit_tree_node( tree_node )
+      @depth -= 1
+    end
+
+    def visit_leaf_node( leaf_node )
+    end
+
+  end
+end

data/lib/treevisitor/visitors/flat_print_tree_node_visitors.rb

@@ -0,0 +1,20 @@
+# -*- coding: utf-8 -*-
+module TreeVisitor
+  #
+  # Print for every node the name
+  #
+  class FlatPrintTreeNodeVisitor < TreeNodeVisitor
+
+    def enter_tree_node( tree_node )
+      puts tree_node.name
+    end
+
+    def exit_tree_node( tree_node )
+    end
+
+    def visit_leaf_node( leaf_node )
+      puts leaf_node.name
+    end
+
+  end
+end

data/lib/treevisitor/visitors/print_dir_tree_visitor.rb

@@ -0,0 +1,21 @@
+# -*- coding: utf-8 -*-
+module TreeVisitor
+  #
+  # Visitor for DirTreeWalker
+  # Prints the node at enter
+  # TODO: join this con PrintTreeNodeVisitor
+  class PrintDirTreeVisitor < TreeNodeVisitor
+
+    def enter_tree_node( pathname )
+      puts pathname
+    end
+
+    def exit_tree_node( pathname )
+    end
+
+    def visit_leaf_node( pathname )
+      puts pathname
+    end
+
+  end
+end

data/lib/treevisitor/visitors/print_tree_node_visitor.rb

@@ -0,0 +1,40 @@
+# -*- coding: utf-8 -*-
+module TreeVisitor
+  #
+  # Prints TreeNode names indenting according to depth
+  #
+  class PrintTreeNodeVisitor < TreeNodeVisitor
+
+    def initialize
+      @depth = 0
+    end
+
+    def enter_tree_node( tree_node )
+      str = ""
+      (0...@depth).step {
+        str << " |-"
+      }
+
+      if @depth == 0
+        puts str + tree_node.name.to_s
+      else
+        puts str + tree_node.name.to_s
+      end
+      @depth += 1
+    end
+
+    def exit_tree_node( tree_node )
+      @depth -= 1
+    end
+
+    def visit_leaf_node( leaf_node )
+      str = ""
+      (0...@depth-1).step {
+        str << " |-"
+      }
+      str << " |  "
+      puts str + leaf_node.name.to_s
+    end
+
+  end
+end

data/lib/treevisitor_cli.rb

@@ -0,0 +1,7 @@
+#
+# cli
+#
+
+require 'optparse'
+require 'treevisitor'
+require "treevisitor/cli/cli_tree"

data/spec/spec_helper.rb

@@ -0,0 +1,29 @@
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+
+require 'treevisitor'
+require 'treevisitor_cli'
+
+include TreeVisitor
+
+TEST_DIR = File.expand_path( File.join( File.dirname(__FILE__), "fixtures", "test_dir" ) )
+
+
+# Spec::Runner.configure do |config|
+# end
+
+
+require 'test/unit'
+require "stringio"
+
+def with_stdout_captured
+  old_stdout = $stdout
+  out = StringIO.new
+  $stdout = out
+  begin
+    yield
+  ensure
+    $stdout = old_stdout
+  end
+  out.string
+end

data/spec/treevisitor/cli/cli_tree_spec.rb

@@ -0,0 +1,58 @@
+# -*- coding: utf-8 -*-
+require File.join(File.dirname(__FILE__), "..", "..", "spec_helper")
+
+describe CliTree do
+
+  it "help message" do
+    out = with_stdout_captured do
+      args = %w{-h}
+      CliTree.new.parse_args(args)
+    end
+    out.should match /Usage:/
+  end
+
+  it "version" do
+    out = with_stdout_captured do
+      args = %w{--version}
+      CliTree.new.parse_args(args)
+    end
+    version = "0.1.4"
+    out.should match version
+  end
+
+  it "should accepts switch -d (directories only)" do
+    out = with_stdout_captured do
+      args = %w{-d}
+      args << TEST_DIR
+      CliTree.new.parse_args(args)
+    end
+    # pp out
+    out.split("\n").length.should == 6
+
+    out = with_stdout_captured do
+      args = %w{-da}
+      args << TEST_DIR
+      CliTree.new.parse_args(args)
+    end
+    #puts out
+    out.split("\n").length.should == 7
+  end
+
+  it "should accepts switch -a (all files)" do
+    out = with_stdout_captured do
+      args = %w{-a}
+      args << TEST_DIR
+      CliTree.new.parse_args(args)
+    end
+    # pp out
+    out.split("\n").length.should == 11
+
+    out = with_stdout_captured do
+      args = []
+      args << TEST_DIR
+      CliTree.new.parse_args(args)
+    end
+    # puts out
+    out.split("\n").length.should == 9
+  end
+end