lib/seahorse/client/plugins/content_length.rb


# 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