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
|
# frozen_string_literal: true
module Seahorse
module Client
module Plugins
class ContentLength < Plugin
# @api private
class Handler < Client::Handler
# https://github.com/ruby/net-http/blob/master/lib/net/http/requests.rb
# Methods without body are forwards compatible, because content-length
# may be set for requests without body but is technically incorrect.
METHODS_WITHOUT_BODY = Set.new(
%w[GET HEAD DELETE OPTIONS TRACE COPY MOVE]
)
def call(context)
body = context.http_request.body
method = context.http_request.http_method
# We use Net::HTTP with body_stream which doesn't do this by default
if body.respond_to?(:size) && !METHODS_WITHOUT_BODY.include?(method)
context.http_request.headers['Content-Length'] = body.size
end
@handler.call(context)
end
end
handler(Handler, step: :sign, priority: 0)
end
end
end
end
|
