class CompressedRequests

# File lib/logstash/util/http_compressed_requests.rb, line 2
def initialize(app)
  @app = app
end

# File lib/logstash/util/http_compressed_requests.rb, line 14
def call(env)
  if method_handled?(env) && encoding_handled?(env)
    begin
      extracted = decode(env['rack.input'], env['HTTP_CONTENT_ENCODING'])
    rescue Zlib::Error
      return [400, {'Content-Type' => 'text/plain'}, ["Failed to decompress body"]]
    end

    env.delete('HTTP_CONTENT_ENCODING')
    env['CONTENT_LENGTH'] = extracted.bytesize
    env['rack.input'] = StringIO.new(extracted)
  end

  @app.call(env)
end

# File lib/logstash/util/http_compressed_requests.rb, line 30
def decode(input, content_encoding)
  case content_encoding
    when 'gzip' then
      Zlib::GzipReader.new(input).read
    when 'deflate' then
      Zlib::Inflate.inflate(input.read)
  end
end

# File lib/logstash/util/http_compressed_requests.rb, line 10
def encoding_handled?(env)
  ['gzip', 'deflate'].include? env['HTTP_CONTENT_ENCODING']
end

# File lib/logstash/util/http_compressed_requests.rb, line 6
def method_handled?(env)
  !!(env['REQUEST_METHOD'] =~ /(POST|PUT)/)
end