module RestClient::AbstractResponse

def self.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

def code
  @code ||= @net_http_res.code.to_i
end

def cookie_jar
  return @cookie_jar if @cookie_jar
  jar = HTTP::CookieJar.new
  headers.fetch(:set_cookie, []).each do |cookie|
    jar.parse(cookie, @request.url)
  end
  @cookie_jar = jar
end

def cookies
  hash = {}
  cookie_jar.cookies.each do |cookie|
    hash[cookie.name] = cookie.value
  end
  hash
end

def description
  "#{code} #{STATUSES[code]} | #{(headers[:content_type] || '').gsub(/;.*$/, '')} #{size} bytes\n"
end

def follow_redirection request = nil, result = nil, & block
  new_args = @args.dup
  url = headers[:location]
  if url !~ /^http/
    url = URI.parse(request.url).merge(url).to_s
  end
  new_args[:url] = url
  if request
    if request.max_redirects == 0
      raise MaxRedirectsReached
    end
    new_args[:password] = request.password
    new_args[:user] = request.user
    new_args[:headers] = request.headers
    new_args[:max_redirects] = request.max_redirects - 1
    # TODO: figure out what to do with original :cookie, :cookies values
    new_args[:headers]['Cookie'] = HTTP::Cookie.cookie_value(
      cookie_jar.cookies(new_args.fetch(:url)))
  end
  Request.execute(new_args, &block)
end

def headers
  @headers ||= AbstractResponse.beautify_headers(@net_http_res.to_hash)
end

def parse_cookie cookie_content
  out = {}
  CGI::Cookie::parse(cookie_content).each do |key, cookie|
    unless ['expires', 'path'].include? key
      out[CGI::escape(key)] = cookie.value[0] ? (CGI::escape(cookie.value[0]) || '') : ''
    end
  end
  out
end

def raw_headers
  @raw_headers ||= @net_http_res.to_hash
end

def response_set_vars(net_http_res, args, request)
  @net_http_res = net_http_res
  @args = args
  @request = request
end

def return! request = nil, result = nil, & block
  if (200..207).include? code
    self
  elsif [301, 302, 307].include? code
    unless [:get, :head].include? args[:method]
      raise Exceptions::EXCEPTIONS_MAP[code].new(self, code)
    else
      follow_redirection(request, result, & block)
    end
  elsif code == 303
    args[:method] = :get
    args.delete :payload
    follow_redirection(request, result, & block)
  elsif Exceptions::EXCEPTIONS_MAP[code]
    raise Exceptions::EXCEPTIONS_MAP[code].new(self, code)
  else
    raise RequestFailed.new(self, code)
  end
end

def to_i
  code
end