Contents

require 'faraday'
require 'faraday/middleware'

# Request middleware that encodes the body as JSON.
#
# Processes only requests with matching Content-type or those without a type.
# If a request doesn't have a type but has a body, it sets the Content-type
# to JSON MIME-type.
#
# Doesn't try to encode bodies that already are in string form.

module PRX
  module Middleware
    class Json < Faraday::Middleware

      CONTENT_TYPE = 'Content-Type'.freeze unless defined? CONTENT_TYPE

      class << self
        attr_accessor :mime_type
      end

      self.mime_type = 'application/json'.freeze

      dependency do
        require 'json' unless defined?(::JSON)
      end

      def call(env)
        match_content_type(env) do |data|
          env.body = encode(data)
        end
        @app.call env
      end

      def encode(data)
        ::JSON.dump(data)
      end

      def match_content_type(env)
        if process_request?(env)
          env.request_headers[CONTENT_TYPE] ||= self.class.mime_type
          yield(env.body) unless env.body.respond_to?(:to_str)
        end
      end

      def process_request?(env)
        type = request_type(env)
        has_body?(env) and (type.empty? or type == self.class.mime_type)
      end

      def request_type(env)
        type = env.request_headers[CONTENT_TYPE].to_s
        type = type.split(';', 2).first if type.index(';')
        type
      end

      def has_body?(env)
        body = env.body and !(body.respond_to?(:to_str) and body.empty?)
      end

    end
  end
end

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
prx_client-1.0.0	lib/prx/middleware/json.rb
prx_client-0.3.1	lib/prx/middleware/json.rb
prx_client-0.3.0	lib/prx/middleware/json.rb