lib/munson/middleware/encode_json_api.rb in munson-0.2.0 vs lib/munson/middleware/encode_json_api.rb in munson-0.3.0

@@ -2,23 +2,29 @@
   module Middleware
     class EncodeJsonApi < Faraday::Middleware
       CONTENT_TYPE = 'Content-Type'.freeze
       ACCEPT       = 'Accept'.freeze
       MIME_TYPE    = 'application/vnd.api+json'.freeze
-      USER_AGENT   = 'User-Agent'
+      USER_AGENT   = 'User-Agent'.freeze
 
+      def initialize(app, key_formatter = nil)
+        super(app)
+        @key_formatter = key_formatter
+      end
+
       def call(env)
         env[:request_headers][USER_AGENT] = "Munson v#{Munson::VERSION}"
         env[:request_headers][ACCEPT] ||= MIME_TYPE
         match_content_type(env) do |data|
           env[:body] = encode data
         end
         @app.call env
       end
 
       def encode(data)
-        ::JSON.dump data
+        json = @key_formatter ? @key_formatter.externalize(data) : data
+        ::JSON.dump(json)
       end
 
       def match_content_type(env)
         if process_request?(env)
           env[:request_headers][CONTENT_TYPE] ||= MIME_TYPE
@@ -41,5 +47,7 @@
         type
       end
     end
   end
 end
+
+Faraday::Request.register_middleware :"Munson::Middleware::EncodeJsonApi" => Munson::Middleware::EncodeJsonApi