class Excon::Response

def self.parse(socket, params={})
  response = new(:status => socket.readline[9, 11].to_i)
  block_given = block_given?
  until ((data = socket.readline).chop!).empty?
    key, value = data.split(/:\s*/, 2)
    response.headers[key] = ([*response.headers[key]] << value).compact.join(', ')
    if key.casecmp('Content-Length') == 0
      content_length = value.to_i
    elsif (key.casecmp('Transfer-Encoding') == 0) && (value.casecmp('chunked') == 0)
      transfer_encoding_chunked = true
    end
  end
  unless (params[:method].to_s.casecmp('HEAD') == 0) || NO_ENTITY.include?(response.status)
    # don't pass stuff into a block if there was an error
    if params[:expects] && ![*params[:expects]].include?(response.status)
      block_given = false
    end
    if block_given
      if transfer_encoding_chunked
        # 2 == "/r/n".length
        while (chunk_size = socket.readline.chop!.to_i(16)) > 0
          yield(socket.read(chunk_size + 2).chop!, nil, content_length)
        end
        socket.read(2)
      elsif remaining = content_length
        remaining = content_length
        while remaining > 0
          yield(socket.read([CHUNK_SIZE, remaining].min), [remaining - CHUNK_SIZE, 0].max, content_length)
          remaining -= CHUNK_SIZE
        end
      else
        while remaining = socket.read(CHUNK_SIZE)
          yield(remaining, remaining.length, content_length)
        end
      end
    else
      if transfer_encoding_chunked
        while (chunk_size = socket.readline.chop!.to_i(16)) > 0
          response.body << socket.read(chunk_size + 2).chop! # 2 == "/r/n".length
        end
        socket.read(2) # 2 == "/r/n".length
      elsif remaining = content_length
        while remaining > 0
          response.body << socket.read([CHUNK_SIZE, remaining].min)
          remaining -= CHUNK_SIZE
        end
      else
        response.body << socket.read
      end
    end
  end
  response
end

def attributes
  {
    :body     => body,
    :headers  => headers,
    :status   => status
  }
end

def get_header(name)
  headers.each do |key,value|
    if key.casecmp(name) == 0
      return value
    end
  end
  nil
end

def initialize(attrs={})
  @body    = attrs[:body]    || ''
  @headers = attrs[:headers] || {}
  @status  = attrs[:status]
end