Sha256: 120176cebc378002391411b2cd9e88d89241c859fd33c15c069a229773478e55

Contents?: true

Size: 1.66 KB

Versions: 33

Compression:

Stored size: 1.66 KB

Contents

# 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

Version data entries

33 entries across 33 versions & 2 rubygems

Version Path
excon-0.112.0 lib/excon/middlewares/decompress.rb
excon-0.111.0 lib/excon/middlewares/decompress.rb
excon-0.110.0 lib/excon/middlewares/decompress.rb
excon-0.109.0 lib/excon/middlewares/decompress.rb
excon-0.108.0 lib/excon/middlewares/decompress.rb
excon-0.107.0 lib/excon/middlewares/decompress.rb
excon-0.106.0 lib/excon/middlewares/decompress.rb
excon-0.105.0 lib/excon/middlewares/decompress.rb
excon-0.104.0 lib/excon/middlewares/decompress.rb
excon-0.103.0 lib/excon/middlewares/decompress.rb
excon-0.102.0 lib/excon/middlewares/decompress.rb
excon-0.101.0 lib/excon/middlewares/decompress.rb
excon-0.100.0 lib/excon/middlewares/decompress.rb
excon-0.99.0 lib/excon/middlewares/decompress.rb
excon-0.98.0 lib/excon/middlewares/decompress.rb
excon-0.97.2 lib/excon/middlewares/decompress.rb
excon-0.97.1 lib/excon/middlewares/decompress.rb
excon-0.97.0 lib/excon/middlewares/decompress.rb
excon-0.96.0 lib/excon/middlewares/decompress.rb
excon-0.95.0 lib/excon/middlewares/decompress.rb