tree_decorator 0.1.0 → 0.1.1

data/README.md

@@ -16,7 +16,7 @@ For example:
     end
 
     hanger = TreeDecorator::ObjectHanger.new(
-              Thing.roots.first,
+              Thing.roots,
               :children_method => :children,
               :content_method => :name
             )

data/lib/tree_decorator/object_hanger.rb

@@ -20,15 +20,14 @@ module TreeDecorator
     
     private
     def get_hash_from_objects
-      content = Hash.new
-      content[@root.send(content_method)] = populate_with_children(@root)
-      return content
+      objects = @root.kind_of?(Array) ? @root : [@root]
+      populate_with_children(objects)
     end
     
     def populate_with_children(object)
-      content = Hash.new
-      children = object.send(children_method)
-      
+      content = Hash.new          
+      children = object.kind_of?(Array) ? object : object.send(children_method)
+
       if children and !children.empty?
         children.each do |child|
           content[child.send(content_method)] = populate_with_children(child)

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: tree_decorator
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
   prerelease: 
 platform: ruby
 authors: