Add option to select adapter by the request header

Author: akaKuruma

lib/action_controller/serialization.rb

@@ -25,6 +25,9 @@ def get_serializer(resource, options = {})
           "Please pass 'adapter: false' or see ActiveSupport::SerializableResource#serialize"
         options[:adapter] = false
       end
+
+      options[:adapter] ||= ActiveModel::Serializer::Adapter.by_request(request)
+
       ActiveModel::SerializableResource.serialize(resource, options) do |serializable_resource|
         if serializable_resource.serializer?
           serializable_resource.serialization_scope ||= serialization_scope

lib/active_model/serializer.rb

@@ -104,6 +104,22 @@ def self.adapter
       adapter_class
     end
 
+    def self.custom_media_type_adapters=(media_type_adapters)
+      config.custom_media_type_adapters ||= {}
+
+      unless media_type_adatets.is_a? Array
+        media_types_adapters = [media_types_adapters]
+      end
+
+      media_types_adapters.each do |media_type_adapter|
+        config.custom_media_type_adapters.merge!(media_type_adapter)
+      end
+    end
+
+    def self.custom_media_type_adapters
+      config.custom_media_type_adapters ||= {}
+    end
+
     def self.root_name
       name.demodulize.underscore.sub(/_serializer$/, '') if name
     end

lib/active_model/serializer/adapter.rb

@@ -11,6 +11,11 @@ class Adapter
 
       attr_reader :serializer
 
+      MEDIA_TYPE_ADAPTERS = {
+        "application/vnd.api+json" => "json_api",
+        "application/json" => "flatten_json"
+      }
+
       def initialize(serializer, options = {})
         @serializer = serializer
         @options = options
@@ -37,6 +42,13 @@ def self.adapter_class(adapter)
         "ActiveModel::Serializer::Adapter::#{adapter_name}".safe_constantize
       end
 
+      def self.by_request(request)
+        return unless request && request.accept
+        custom = ActiveModel::Serializer.custom_media_type_adapters
+        media_type_adapters = MEDIA_TYPE_ADAPTERS.merge(custom)
+        media_type_adapters[request.accept]
+      end
+
       def fragment_cache(*args)
         raise NotImplementedError, 'This is an abstract method. Should be implemented at the concrete adapter.'
       end

test/action_controller/adapter_selector_test.rb

@@ -18,6 +18,11 @@ def render_skipping_adapter
           @profile = Profile.new({ name: 'Name 1', description: 'Description 1', comments: 'Comments 1' })
           render json: @profile, adapter: false
         end
+
+        def render_selecting_adapter_by_accept_header_field
+          @profile = Profile.new({ name: 'Name 1', description: 'Description 1', comments: 'Comments 1' })
+          render json: @profile
+        end
       end
 
       tests AdapterSelectorTestController
@@ -48,6 +53,26 @@ def test_render_skipping_adapter
         get :render_skipping_adapter
         assert_equal '{"attributes":{"name":"Name 1","description":"Description 1","comments":"Comments 1"}}', response.body
       end
+
+      def test_render_using_adapter_selected_by_accept_header_field
+        request.headers['Accept'] = "application/vnd.api+json"
+
+        get :render_selecting_adapter_by_accept_header_field
+
+        expected = {
+          data: {
+            id: assigns(:profile).id.to_s,
+            type: "profiles",
+            attributes: {
+              name: "Name 1",
+              description: "Description 1",
+            }
+          }
+        }
+
+        assert_equal expected.to_json, response.body
+      end
+
     end
   end
 end