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
|
# frozen_string_literal: true
module Excon
module Middleware
class Decompress < Excon::Middleware::Base
INFLATE_ZLIB_OR_GZIP = 47 # Zlib::MAX_WBITS + 32
INFLATE_RAW = -15 # Zlib::MAX_WBITS * -1
def request_call(datum)
unless datum.has_key?(:response_block)
key = datum[:headers].keys.detect {|k| k.to_s.casecmp('Accept-Encoding') == 0 } || 'Accept-Encoding'
if datum[:headers][key].to_s.empty?
datum[:headers][key] = 'deflate, gzip'
end
end
@stack.request_call(datum)
end
def response_call(datum)
body = datum[:response][:body]
unless datum.has_key?(:response_block) || body.nil? || body.empty?
if (key = datum[:response][:headers].keys.detect {|k| k.casecmp('Content-Encoding') == 0 })
encodings = Utils.split_header_value(datum[:response][:headers][key])
if (encoding = encodings.last)
if encoding.casecmp('deflate') == 0
datum[:response][:body] = begin
Zlib::Inflate.new(INFLATE_ZLIB_OR_GZIP).inflate(body)
rescue Zlib::DataError # fallback to raw on error
Zlib::Inflate.new(INFLATE_RAW).inflate(body)
end
encodings.pop
elsif encoding.casecmp('gzip') == 0 || encoding.casecmp('x-gzip') == 0
datum[:response][:body] = Zlib::GzipReader.new(StringIO.new(body)).read
encodings.pop
end
datum[:response][:headers][key] = encodings.join(', ')
end
end
end
@stack.response_call(datum)
end
end
end
end
|
