added tree.depth and tree.full_tree

schacon · schacon · commit 476d943baabc · 2008-03-08T18:33:34.000-08:00
diff --git a/lib/git/lib.rb b/lib/git/lib.rb
@@ -181,6 +181,14 @@ def ls_tree(sha)
       
       data
     end
+    
+    def full_tree(sha)
+      command_lines('ls-tree', ['-r', sha.to_s])
+    end
+            
+    def tree_depth(sha)
+      full_tree(sha).size
+    end
 
     def branches_all
       head = File.read(File.join(@git_dir, 'HEAD'))
diff --git a/lib/git/object.rb b/lib/git/object.rb
@@ -124,6 +124,14 @@ def trees
       alias_method :subtrees, :trees
       alias_method :subdirectories, :trees
        
+      def full_tree
+        @base.lib.full_tree(@objectish)
+      end
+                  
+      def depth
+        @base.lib.tree_depth(@objectish)
+      end
+       
       private
       
         def setup
diff --git a/tests/units/test_object.rb b/tests/units/test_object.rb
@@ -72,6 +72,11 @@ def test_tree
     assert_equal(1, o.blobs.size)
     assert_equal(1, o.subtrees.size)
     assert_equal(1, o.trees['ex_dir'].blobs.size)
+
+    assert_equal(2, o.full_tree.size)
+    assert_equal("100644 blob e69de29bb2d1d6434b8b29ae775ad8c2e48c5391\tex_dir/ex.txt", o.full_tree.first)
+    
+    assert_equal(2, o.depth)
     
     o = @git.gtree('94c827875e2cadb8bc8d4cdd900f19aa9e8634c7')
     assert(o.is_a?(Git::Object::Tree))
