treevisitor 0.1.4

data/spec/treevisitor/dir_processor_spec.rb

@@ -0,0 +1,15 @@
+# -*- coding: utf-8 -*-
+require File.join(File.dirname(__FILE__), "..", "spec_helper")
+
+describe DirProcessor do
+
+  it do
+    files = []
+    dp = DirProcessor.new { |f| files << f }
+    dp.process(TEST_DIR)
+    files.length.should == 3
+  end
+
+end
+
+

data/spec/treevisitor/dir_tree_walker_spec.rb

@@ -0,0 +1,33 @@
+# -*- coding: utf-8 -*-
+require File.join(File.dirname(__FILE__), "..", "spec_helper")
+
+describe DirTreeWalker do
+
+  it "should accumulate file names" do
+    dir_tree_walker = DirTreeWalker.new( TEST_DIR )
+    dir_tree_walker.ignore_dir( ".svn" )
+
+    accumulator = []
+    visitor = BlockTreeNodeVisitor.new { |pathname| accumulator << File.basename( pathname ) }
+    dir_tree_walker.run( visitor )
+    accumulator.length.should == 9
+    accumulator.sort.should ==  %w{ test_dir dir.1 dir.1.2 file.1.2.1 file.1.1 dir.2 file.2.1 .dir_with_dot dummy.txt}.sort
+  end
+
+  it "should ignore files" do
+    dtp = DirTreeWalker.new( "." )
+
+    dtp.ignore(/^\./)
+    dtp.ignore_file?( ".thumbnails" ).should be_true
+    dtp.ignore_dir?( ".thumbnails" ).should be_true
+
+    dtp.ignore_dir("thumbnails")
+    dtp.ignore_dir?( ".thumbnails" ).should be_true
+    dtp.ignore_dir?( "thumbnails" ).should be_true
+    dtp.ignore_dir?( "pippo" ).should be_false
+
+    dtp.ignore_file("xvpics")
+    dtp.ignore_file?( "xvpics" ).should be_true
+  end
+
+end

data/spec/treevisitor/tree_node_dsl_spec.rb

@@ -0,0 +1,153 @@
+# -*- coding: utf-8 -*-
+require File.join(File.dirname(__FILE__), "..", "spec_helper")
+
+describe "TreeNodeDsl" do
+
+  class DTreeNode < TreeNode
+    def to_s
+      "dt: #{content}"
+    end
+  end
+
+  class DLeafNode < LeafNode
+    def to_s
+      "dl: #{content}"
+    end
+  end
+
+  class ArgsTreeNode < TreeNode
+
+    attr_reader :description
+
+    def initialize(name, description, parent)
+      super(name, parent)
+      @description =  description
+    end
+
+    def to_s
+      "a: #{description}"
+    end
+  end
+
+  class ArgsLeafNode < LeafNode
+
+    attr_reader :description
+
+    def initialize(name, description, parent)
+      super(name, parent)
+      @description =  description
+    end
+
+    def to_s
+      "a: #{description}"
+    end
+  end
+
+
+
+  it "test_dsl" do
+    tree = TreeNode.create do
+      node "root" do
+        leaf "l1"
+        leaf "l2"
+        node "sub" do
+          leaf "l3"
+        end
+        node "woleaves"
+      end
+    end
+
+    # puts tree.to_str
+    out =<<EOS
+root
+|-- l1
+|-- l2
+|-- sub
+|   `-- l3
+`-- woleaves
+EOS
+    tree.to_str.should == out
+  end
+
+  it "test_dsl_block_with_arg" do
+    tree = TreeNode.create do
+      node "root" do |node|
+        node.prefix_path=("pre/")        
+        leaf "l1"
+        leaf "l2"
+        node "sub" do
+          leaf "l3" do |leaf|            
+          end
+        end
+        node "woleaves"
+      end
+    end
+
+    # puts tree.to_str
+    out =<<EOS
+root
+|-- l1
+|-- l2
+|-- sub
+|   `-- l3
+`-- woleaves
+EOS
+    tree.to_str.should ==  out
+    tree.find("l3").path_with_prefix.should == "pre/root/sub/l3"
+  end
+
+  it "test_derivated" do
+    tree = TreeNode.create(DTreeNode, DLeafNode) do
+      node "root" do
+        leaf "l1"
+        leaf "l2"
+        node "sub" do
+          leaf "l3"
+        end
+      end
+    end
+
+    # puts tree.to_str
+    out =<<EOS
+dt: root
+|-- dl: l1
+|-- dl: l2
+`-- dt: sub
+    `-- dl: l3
+EOS
+    tree.to_str.should == out
+
+    tree = DTreeNode.create(DLeafNode) do
+      node "root" do
+        leaf "l1"
+        leaf "l2"
+        node "sub" do
+          leaf "l3"
+        end
+      end
+    end
+    tree.to_str.should == out
+  end
+
+  it "test_derivated_args" do
+    tree = TreeNode.create(ArgsTreeNode, ArgsLeafNode) do
+      node "root", "droot" do
+        leaf "l1", "dl1"
+        leaf "l2", "dl2"
+        node "sub", "dsub" do
+          leaf "l3", "dl3"
+        end
+      end
+    end
+
+    # puts tree.to_str
+    out =<<EOS
+a: droot
+|-- a: dl1
+|-- a: dl2
+`-- a: dsub
+    `-- a: dl3
+EOS
+    tree.to_str.should == out
+  end
+end

data/spec/treevisitor/tree_node_spec.rb

@@ -0,0 +1,153 @@
+# -*- coding: utf-8 -*-
+require File.join(File.dirname(__FILE__), "..", "spec_helper")
+
+describe TreeNode do
+
+  it "should initialize correctly" do
+    ta = TreeNode.new("a")
+    ta.should be_root
+
+    ln1 = LeafNode.new("1", ta)
+    ln1.parent.should == ta
+    ln1.should_not be_root
+
+    ln2 = LeafNode.new("2", ta)
+    tb  = TreeNode.new("b", ta)
+    ln3 = LeafNode.new("3", tb)
+    ln3.parent.should == tb
+
+    # test depth
+    tb.depth.should == 2
+    ln3.depth.should == 3
+
+    # test nr_nodes
+    tb.nr_nodes.should == 1
+    ta.nr_nodes.should == 4
+
+    # puts ta.to_str
+  end
+
+  it "should add child and leaf" do
+    ta = TreeNode.new("a")
+    ta.should be_root
+
+    ln1 = LeafNode.new("1")
+    ta.add_leaf(ln1)
+    ln1.parent.should == ta
+
+    ln2 = LeafNode.new("2", ta)
+    ta.add_leaf(ln2)
+
+    tb  = TreeNode.new("b", ta)
+    ln3 = LeafNode.new("3", tb)
+    tb.add_leaf(ln3)
+    ln3.parent.should == tb
+
+    ta.add_child(tb)
+    # puts ta.to_str
+  end
+
+  it "next and prev" do
+    ta = TreeNode.new("a")
+    ta.prev.should be_nil
+    ta.next.should be_nil
+
+    ln1 = LeafNode.new("1", ta)
+    ln1.prev.should be_nil
+    ln1.next.should be_nil
+
+    ln2 = LeafNode.new("2", ta)
+    ln2.prev.should == ln1
+    ln2.next.should be_nil
+
+    ln3 = LeafNode.new("3", ta)
+    ln2.next.should == ln3
+
+    tb = TreeNode.new("b", ta)
+    tb.next.should be_nil
+    tb.prev.should be_nil
+
+    tc = TreeNode.new("c", ta)
+    tc.prev.should == tb
+    tc.next.should be_nil
+  end
+
+  context "navigate tree" do
+
+    before do
+      @tree     = TreeNode.new("a")
+      ln1       = LeafNode.new("1", @tree)
+      ln2       = LeafNode.new("2", @tree)
+      @sub_tree = TreeNode.new("b", @tree)
+      @ln3       = LeafNode.new("3", @sub_tree)
+    end
+
+    it "nr_nodes and nr_leaves and nr_children" do
+      @tree.nr_nodes.should == 4
+      @tree.nr_leaves.should == 3
+      @tree.nr_children.should == 1
+
+      @sub_tree.nr_nodes.should == 1
+      @sub_tree.nr_leaves.should == 1
+      @sub_tree.nr_children.should == 0
+    end
+
+    it "prefix_path" do
+      @tree.path.should == "a"
+      @sub_tree.path.should == "a/b"
+      @tree.path_with_prefix.should == "a"
+      @sub_tree.path_with_prefix.should == "a/b"
+
+      @tree.prefix_path= "<root>/"
+
+      @tree.prefix_path.should == "<root>/"
+      @tree.path.should == "a"
+      @sub_tree.path.should == "a/b"
+      @tree.path_with_prefix.should == "<root>/a"
+      @sub_tree.path_with_prefix.should == "<root>/a/b"
+    end
+
+    it "invalidate" do
+      @tree.prefix_path="root/"
+      @sub_tree.path.should == "a/b"
+      @sub_tree.path_with_prefix.should == "root/a/b"
+      @sub_tree.depth.should == 2
+
+      r = TreeNode.new("r")
+      r.add_child(@tree)
+      @sub_tree.path.should == "r/a/b"
+      @sub_tree.path_with_prefix.should == "r/a/b"
+
+      r.prefix_path="new_root/"
+      @sub_tree.path.should == "r/a/b"
+      @sub_tree.path_with_prefix.should == "new_root/r/a/b"
+      @sub_tree.depth.should == 3
+    end
+
+    it "find without block" do
+      @tree.find("a").should === @tree
+      @tree.find("b").should === @sub_tree
+      @tree.find("3").should === @ln3
+      @tree.find("not existent").should be_nil
+    end
+
+    it "find with block" do
+      @tree.find {|e| e == "a"}.should === @tree
+      @tree.find {|e| e == "b"}.should === @sub_tree
+      @tree.find {|e| e == "3"}.should === @ln3
+      @tree.find {|e| e == "not existent"}.should be_nil
+    end
+
+    it "to_str" do
+      out = <<EOS
+a
+|-- 1
+|-- 2
+`-- b
+    `-- 3
+EOS
+      @tree.to_str.should == out
+    end
+
+  end
+end

data/spec/treevisitor/tree_node_visitor_spec.rb

@@ -0,0 +1,70 @@
+# -*- coding: utf-8 -*-
+require File.join(File.dirname(__FILE__), "..", "spec_helper")
+
+require 'treevisitor/visitors/block_tree_node_visitor'
+require 'treevisitor/visitors/callback_tree_node_visitor'
+require 'treevisitor/visitors/callback_tree_node_visitor2'
+require 'treevisitor/visitors/clone_tree_node_visitor'
+require 'treevisitor/visitors/depth_tree_node_visitor'
+
+describe "TreeNodeVisitors" do
+
+  before do
+    ta = TreeNode.new( "a", nil )
+    LeafNode.new("1", ta )
+    LeafNode.new("2", ta )
+    
+    tb = TreeNode.new( "b", ta )
+    LeafNode.new( "3", tb ) 
+    
+    @tree = ta
+  end
+
+  it BlockTreeNodeVisitor do
+    accumulator = []
+    visitor = BlockTreeNodeVisitor.new { |node| accumulator << node.content}
+    @tree.accept( visitor )
+    accumulator.length.should == 5
+    accumulator.should == %w{ a 1 2 b 3 }
+  end
+  
+  it DepthTreeNodeVisitor do
+    visitor = DepthTreeNodeVisitor.new
+    @tree.accept( visitor )
+    visitor.depth.should == 0
+
+    visitor = CloneTreeNodeVisitor.new
+    @tree.accept( visitor )
+    visitor.cloned_root.nr_nodes.should ==  @tree.nr_nodes
+  end
+  
+  it CallbackTreeNodeVisitor do
+    accumulator = []
+    visitor = CallbackTreeNodeVisitor.new
+    visitor.on_enter_tree_node{ |tree_node| accumulator << tree_node.content }
+    visitor.on_visit_leaf_node{ |leaf_node| accumulator << leaf_node.content }
+    @tree.accept( visitor )
+    accumulator.length.should == 5
+    accumulator.should == %w{ a 1 2 b 3 }
+  end
+  
+  it CallbackTreeNodeVisitor2 do
+    visitor = CallbackTreeNodeVisitor2.new
+    visitor.on_enter_tree_node{ |tree_node, new_parent_node|
+      TreeNode.new("n" + tree_node.content, new_parent_node)
+    }
+    visitor.on_visit_leaf_node{ |leaf_node, new_parent_node|
+      LeafNode.new( "n" + leaf_node.content, new_parent_node )
+    }
+    @tree.accept( visitor )
+    new_root = visitor.root
+    new_root.content.should ==  "n" + @tree.content
+    
+    accumulator = []
+    visitor = BlockTreeNodeVisitor.new { |node| accumulator << node.content}
+    new_root.accept( visitor )
+    accumulator.length.should == 5
+    accumulator.should == %w{ na n1 n2 nb n3 }
+  end
+
+end

data/tasks/jeweler.rake

@@ -0,0 +1,55 @@
+# -*- coding: utf-8 -*-
+#
+# jeweler
+#
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gem|
+    # gem is a Gem::Specification... see http://www.rubygems.org/read/chapter/20 for additional settings
+    gem.name = "treevisitor"
+    gem.summary = "Implementation of visitor design pattern"
+    gem.description = <<-EOF
+      Implementation of visitor design pattern. It contains a 'tree.rb'
+      command line clone of the tree unix tool.
+    EOF
+
+    gem.authors = ["Tokiro"]
+    gem.email = "tokiro.oyama@gmail.com"
+    gem.homepage = "http://github.com/tokiro/treevisitor"
+
+    #
+    # dependecies
+    #
+    gem.add_development_dependency "rspec"
+
+    #
+    # bin
+    #
+    gem.executables = %w{ tree.rb }
+
+    #
+    # files
+    #
+    gem.files  = %w{LICENSE README.rdoc Rakefile VERSION.yml dircat.gemspec}
+    gem.files.concat Dir['lib/**/*.rb']
+    gem.files.concat Dir['tasks/**/*.rake']
+    gem.files.concat Dir['examples/*.rb']
+
+
+    #
+    # test files
+    #
+    gem.test_files = Dir['spec/**/*.rb']
+    gem.test_files.concat Dir['spec/fixtures/**/*']
+    gem.test_files.concat Dir['spec/fixtures/**/.dir_with_dot/*']
+
+    #
+    # rubyforge
+    #
+    # gem.rubyforge_project = 'treevisitor'
+  end
+  Jeweler::GemcutterTasks.new
+rescue LoadError
+  puts "Jeweler (or a dependency) not available. Install it with: sudo gem install jeweler"
+end