~ubuntu-branches/ubuntu/utopic/ruby-excon/utopic

module Excon

  module Middleware

    class ResponseParser < Excon::Middleware::Base

      def response_call(datum)

        unless datum.has_key?(:response)

          datum = Excon::Response.parse(datum[:connection].send(:socket), datum)

 

          # only requests without a :response_block add 'deflate, gzip' to the TE header.

          unless datum[:response_block]

            if key = datum[:response][:headers].keys.detect {|k| k.casecmp('Transfer-Encoding') == 0 }

              encodings = Utils.split_header_value(datum[:response][:headers][key])

              if encoding = encodings.last

                if encoding.casecmp('deflate') == 0

                  # assume inflate omits header

                  datum[:response][:body] = Zlib::Inflate.new(-Zlib::MAX_WBITS).inflate(datum[:response][:body])

                  encodings.pop

                elsif encoding.casecmp('gzip') == 0 || encoding.casecmp('x-gzip') == 0

                  datum[:response][:body] = Zlib::GzipReader.new(StringIO.new(datum[:response][:body])).read

                  encodings.pop

                end

                datum[:response][:headers][key] = encodings.join(', ')

              end

            end

          end

        end

        @stack.response_call(datum)

      end

    end

  end

end