class SplitIoClient::FaradayMiddleware::Gzip

def call(env)
  env[:request_headers][ACCEPT_ENCODING] ||= SUPPORTED_ENCODINGS
  @app.call(env).on_complete do |response_env|
    case response_env[:response_headers][CONTENT_ENCODING]
    when 'gzip'
      reset_body(response_env, &method(:uncompress_gzip))
    when 'deflate'
      reset_body(response_env, &method(:inflate))
    end
  end
end

def inflate(body)
  # Inflate as a DEFLATE (RFC 1950+RFC 1951) stream
  Zlib::Inflate.inflate(body)
rescue Zlib::DataError
  # Fall back to inflating as a "raw" deflate stream which
  # Microsoft servers return
  inflate = Zlib::Inflate.new(-Zlib::MAX_WBITS)
  begin
    inflate.inflate(body)
  ensure
    inflate.close
  end
end

def reset_body(env)
  env[:body] = yield(env[:body])
  env[:response_headers].delete(CONTENT_ENCODING)
  env[:response_headers][CONTENT_LENGTH] = env[:body].length
end

def uncompress_gzip(body)
  io = StringIO.new(body)
  gzip_reader = if RUBY_ENCODING
                  Zlib::GzipReader.new(io, encoding: 'ASCII-8BIT')
                else
                  Zlib::GzipReader.new(io)
                end
  gzip_reader.read
end