Fix json root

Author: Michaël Villeneuve

Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    active_mappers (0.1.1)
+    active_mappers (0.2.0)
       activesupport (>= 4.2)
 
 GEM
@@ -12,7 +12,7 @@ GEM
       i18n (>= 0.7, < 2)
       minitest (~> 5.1)
       tzinfo (~> 1.1)
-    concurrent-ruby (1.0.5)
+    concurrent-ruby (1.1.3)
     i18n (1.1.1)
       concurrent-ruby (~> 1.0)
     minitest (5.11.3)

active_mappers-0.2.0.gem

@@ -1,6 +1,6 @@
 Gem::Specification.new do |s|
   s.name        = 'active_mappers'
-  s.version     = '0.1.1'
+  s.version     = '0.2.0'
   s.date        = '2018-10-14'
   s.summary     = 'Slick, fast view layer for you Rails API.'
   s.description = 'Fast, simple, declarative way to design your API\'s view layer'

active_mappers.gemspec

@@ -1,6 +1,7 @@
 require 'active_support'
 require 'active_support/core_ext/object/try'
 require 'active_support/core_ext/string/inflections'
+require 'active_support/core_ext/string'
 require_relative 'core_ext/hash'
 
 module ActiveMappers
@@ -34,21 +35,21 @@ def self.relation(key, mapper = nil, optional_path = nil)
 
       each do |resource|
         mapper ||= "::#{resource.send(key).class.name}Mapper".constantize
-        { key => mapper.with(path.to_s.split('.').inject(resource, :try)) }
+        { key => mapper.with(path.to_s.split('.').inject(resource, :try), rootless: true) }
       end
     end
 
     def self.polymorphic(key)
       each do |resource|
         resource_mapper = "::#{resource.send("#{key}_type")}Mapper".constantize
-        { key => resource_mapper.with(resource.send(key)) }
+        { key => resource_mapper.with(resource.send(key), rootless: true) }
       end
     end
 
     def self.acts_as_polymorph
       each do |resource|
         mapper = "::#{resource.class}Mapper".constantize
-        mapper.with(resource)
+        mapper.with(resource, rootless: true)
       rescue NameError
         raise NotImplementedError, 'No mapper found for this type of resource'
       end
@@ -58,8 +59,21 @@ def self.each(&block)
       @@renderers[name] = (@@renderers[name] || []) << block
     end
 
-    def self.with(args)
-      args.respond_to?(:each) ? all(args) : one(args)
+    def self.with(args, options = {})
+      if options[:rootless]
+        args.respond_to?(:each) ? all(args) : one(args)
+      else
+        render_with_root(args, options)
+      end
+    end
+
+    def self.render_with_root(args, options = {})
+      resource = options[:root] || self.name.gsub('Mapper', '').downcase
+      if args.respond_to?(:each)
+        { resource.tableize.gsub('/', '_').to_sym => all(args) }
+      else
+        { resource.to_sym => one(args) }
+      end
     end
 
     def self.all(collection)

lib/active_mappers.rb

@@ -40,23 +40,23 @@ def test_can_render_a_list_of_resources
 
     mapped_users = UserMapper.with(users)
 
-    assert_equal 5, mapped_users.size
-    assert_equal '123', mapped_users[0][:id]
+    assert_equal 5, mapped_users[:users].size
+    assert_equal '123', mapped_users[:users][0][:id]
   end
 
   def test_can_render_a_single_resource
     user = User.new('123', 'Michael', nil)
-    assert_equal user.id, UserMapper.with(user)[:id]
+    assert_equal user.id, UserMapper.with(user)[:user][:id]
   end
 
   def test_each_can_be_used_to_declare_custom_attrs
     user = User.new('123', 'Michael', nil)
-    assert_equal 'lol', UserMapper.with(user)[:lol]
+    assert_equal 'lol', UserMapper.with(user)[:user][:lol]
   end
 
   def test_each_can_be_chained
     user = User.new('123', 'Michael', nil)
-    assert_equal 'lola', UserMapper.with(user)[:lola]
+    assert_equal 'lola', UserMapper.with(user)[:user][:lola]
   end
 
   class FriendShipMapper < ActiveMappers::Base
@@ -66,8 +66,7 @@ class FriendShipMapper < ActiveMappers::Base
   def test_relation_can_query_other_mapepr
     friend = Friend.new('124', 'Nicolas', nil)
     user = User.new('123', 'Michael', friend)
-
-    assert_equal 'Nicolas', FriendShipMapper.with(user)[:friend][:name]
+    assert_equal 'Nicolas', FriendShipMapper.with(user, root: :user)[:user][:friend][:name]
   end
 
   class ProfileMapper < ActiveMappers::Base
@@ -76,7 +75,7 @@ class ProfileMapper < ActiveMappers::Base
   def test_delegate_can_remap_attributes
     friend = Friend.new('124', 'Nicolas', nil)
     user = User.new('123', 'Michael', friend)
-    assert_equal 'Nicolas', ProfileMapper.with(user)[:name]
+    assert_equal 'Nicolas', ProfileMapper.with(user, root: :user)[:user][:name]
   end
 
   def test_core_extensions_work_as_expected