Repo#ls_tree

lib/rugged_adapter/git_layer_rugged.rb

@@ -651,12 +651,25 @@ def log(commit = Gollum::Git.default_ref_for_repo(@repo), path = nil, options =
         git.log(commit, path, **options)
       end
 
-      def lstree(sha, options = {})
-        results = []
-        @repo.lookup(sha).tree.walk(:postorder) do |root, entry|
-          results << ::Gollum::Git::Tree.tree_entry_from_rugged_hash(entry, root)
+      def lstree(sha, path = '/', options = {})
+
+        tree = @repo.lookup(sha).tree
+        if path
+          tree = Gollum::Git::Tree.rugged_tree_path(tree, path)
+          return nil unless tree  
+        end
+
+        if options.fetch(:recursive, true)
+          results = []
+          tree.walk(:postorder) do |root, entry|
+            results << ::Gollum::Git::Tree.tree_entry_from_rugged_hash(entry, root)
+          end
+          results
+        else
+          tree.map do |entry|
+            Gollum::Git::Tree.tree_entry_from_rugged_hash(entry, path)
+          end
         end
-        results
       end
 
       def path
@@ -685,6 +698,19 @@ def find_branch(search_list)
 
     class Tree
 
+      attr_reader :tree
+
+      def self.rugged_tree_path(tree, path)
+        return tree if path == '/'
+        begin
+          obj = tree.path(path)
+          rescue Rugged::TreeError
+            return nil
+          end
+          return nil if obj.nil?
+          tree.owner.lookup(obj[:oid])
+      end
+
       def self.tree_entry_from_rugged_hash(entry, root = '')
         {
           sha:  entry[:oid],
@@ -711,16 +737,10 @@ def id
         @tree.oid
       end
 
-      def /(file)
-        return self if file == '/'
-        begin
-        obj = @tree.path(file)
-        rescue Rugged::TreeError
-          return nil
-        end
+      def /(path)
+        obj = self.class.rugged_tree_path(@tree, path)
         return nil if obj.nil?
-        obj = @tree.owner.lookup(obj[:oid])
-        obj.is_a?(Rugged::Tree) ? Gollum::Git::Tree.new(obj) : Gollum::Git::Blob.new(obj)
+        (obj).is_a?(Rugged::Tree) ? Gollum::Git::Tree.new(obj) : Gollum::Git::Blob.new(obj)
       end
 
       def blobs