module Ethon::Easy::ResponseCallbacks

def body(chunk)
  if defined?(@on_body) and not @on_body.nil?
    @on_body.each{ |callback| callback.call(chunk, self) }
  else
    :unyielded
  end
end

def complete
  if defined?(@on_complete) and not @on_complete.nil?
    @on_complete.each{ |callback| callback.call(self) }
  end
end

def headers
  if defined?(@on_headers) and not @on_headers.nil?
    @on_headers.each{ |callback| callback.call(self) }
  end
end

def on_body(&block)
  @on_body ||= []
  @on_body << block if block_given?
  @on_body
end

def on_complete(&block)
  @on_complete ||= []
  @on_complete << block if block_given?
  @on_complete
end

def on_headers(&block)
  @on_headers ||= []
  @on_headers << block if block_given?
  @on_headers
end