1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
# frozen_string_literal: true
require 'faraday'
module FaradayMiddleware
# 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.
class EncodeJson < Faraday::Middleware
CONTENT_TYPE = 'Content-Type'
MIME_TYPE = 'application/json'
MIME_TYPE_REGEX = %r{^application/(vnd\..+\+)?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.generate data
end
def match_content_type(env)
return unless process_request?(env)
env[:request_headers][CONTENT_TYPE] ||= MIME_TYPE
yield env[:body] unless env[:body].respond_to?(:to_str)
end
def process_request?(env)
type = request_type(env)
has_body?(env) && (type.empty? || MIME_TYPE_REGEX =~ type)
end
def has_body?(env)
(body = env[:body]) && !(body.respond_to?(:to_str) && body.empty?)
end
def request_type(env)
type = env[:request_headers][CONTENT_TYPE].to_s
type = type.split(';', 2).first if type.index(';')
type
end
end
end
|
