Contents

module RestClient
  module Mixin
    module Response
      attr_reader :net_http_res

      # HTTP status code
      def code
        @code ||= @net_http_res.code.to_i
      end

      # A hash of the headers, beautified with symbols and underscores.
      # e.g. "Content-type" will become :content_type.
      def headers
        @headers ||= self.class.beautify_headers(@net_http_res.to_hash)
      end

      # The raw headers.
      def raw_headers
        @raw_headers ||= @net_http_res.to_hash
      end

      # Hash of cookies extracted from response headers
      def cookies
        @cookies ||= (self.headers[:set_cookie] || []).inject({}) do |out, cookie_content|
          # correctly parse comma-separated cookies containing HTTP dates (which also contain a comma)
          cookie_content.split(/,\s*/).inject([""]) { |array, blob| 
            blob =~ /expires=.+?$/ ? array.push(blob) : array.last.concat(blob)
            array
          }.each do |cookie|
            next if cookie.empty?
            key, *val = cookie.split(";").first.split("=")
            out[key] = val.join("=")
          end
          out
        end
      end

      # Return the default behavior corresponding to the response code:
      # the response itself for code in 200..206 and an exception in other cases
      def return!
        if (200..206).include? code
          self
        elsif Exceptions::EXCEPTIONS_MAP[code]
          raise Exceptions::EXCEPTIONS_MAP[code], self
        else
          raise RequestFailed, self
        end
      end

      def self.included(receiver)
        receiver.extend(RestClient::Mixin::Response::ClassMethods)
      end

      module ClassMethods
        def beautify_headers(headers)
          headers.inject({}) do |out, (key, value)|
            out[key.gsub(/-/, '_').downcase.to_sym] = %w{set-cookie}.include?(key.downcase) ? value : value.first
            out
          end
        end
      end
    end
  end
end

Version data entries

3 entries across 3 versions & 2 rubygems

Version	Path
rest-client-1.3.1	lib/restclient/mixin/response.rb
rest-client-1.3.0	lib/restclient/mixin/response.rb
rest-client-next-1.3.0	lib/restclient/mixin/response.rb