Serializers hot plugged

app/controllers/devise_token_auth/application_controller.rb

@@ -4,38 +4,19 @@ class ApplicationController < DeviseController
     respond_to :json
 
     def success_message(message = nil)
-      json_response = { status: 'success' }
-      json_response[:message] = message if message
-      json_response
+      serializer = DeviseTokenAuth.success_message_serializer || SuccessMessageSerializer
+      serializer.new(message)
     end
 
     def error_messages(*args)
-      {
-        status: 'error',
-        errors: args
-      }
-    end
-
-    def resource_serializer(resource)
-      {
-        status: "success",
-        data: resource.as_json(except: [:tokens, :created_at, :updated_at])
-      }
+      serializer = DeviseTokenAuth.error_messages_serializer || ErrorMessagesSerializer
+      serializer.new(*args)
     end
 
     def error_serializer(*args)
-      resource = args[0]
-      response = {
-        status: "error",
-        data: resource.as_json(except: [:tokens, :created_at, :updated_at])
-      }
-      if args.length > 1
-        args.shift
-        response[:errors] = args
-      else
-        response[:errors] = resource.errors.to_hash.merge(full_messages: resource.errors.full_messages)
-      end
-      response
+      serializer = DeviseTokenAuth.error_serializer || ResourceErrorsSerializer
+      serializer.new(*args)
     end
+
   end
 end

app/controllers/devise_token_auth/passwords_controller.rb

@@ -99,5 +99,9 @@ def password_resource_params
       devise_parameter_sanitizer.sanitize(:account_update)
     end
 
+    def resource_serializer(user)
+      serializer = DeviseTokenAuth.password_serializer || ResourceSerializer
+      serializer.new(user)
+    end
   end
 end

app/controllers/devise_token_auth/registrations_controller.rb

@@ -90,5 +90,10 @@ def sign_up_params
     def account_update_params
       devise_parameter_sanitizer.sanitize(:account_update)
     end
+
+    def resource_serializer(user)
+      serializer = DeviseTokenAuth.registration_serializer || ResourceSerializer
+      serializer.new(user)
+    end
   end
 end

app/controllers/devise_token_auth/sessions_controller.rb

@@ -42,5 +42,9 @@ def destroy
       end
     end
 
+    def resource_serializer(user)
+      serializer = DeviseTokenAuth.session_serializer || ResourceSerializer
+      serializer.new(user)
+    end
   end
 end

app/controllers/devise_token_auth/token_validations_controller.rb

@@ -13,5 +13,10 @@ def validate_token
     end
 
 
+    def resource_serializer(user)
+      serializer = DeviseTokenAuth.token_validation_serializer || ResourceSerializer
+      serializer.new(user)
+    end
+
   end
 end

lib/devise_token_auth/engine.rb

@@ -12,7 +12,14 @@ class Engine < ::Rails::Engine
   mattr_accessor :change_headers_on_each_request,
                  :token_lifespan,
                  :batch_request_buffer_throttle,
-                 :omniauth_prefix
+                 :omniauth_prefix,
+                 :session_serializer,
+                 :registration_serializer,
+                 :token_validation_serializer,
+                 :password_serializer,
+                 :error_serializer,
+                 :error_messages_serializer,
+                 :success_message_serializer
 
   self.change_headers_on_each_request = true
   self.token_lifespan                 = 2.weeks