class Sinatra::Response

def body=(value)
  value = value.body while Rack::Response === value
  @body = String === value ? [value.to_str] : value
end

def calculate_content_length?
  headers["Content-Type"] and not headers["Content-Length"] and Array === body
end

def drop_body?
  DROP_BODY_RESPONSES.include?(status.to_i)
end

def drop_content_info?
  status.to_i / 100 == 1 or drop_body?
end

def each
  block_given? ? super : enum_for(:each)
end

def finish
  result = body
  if drop_content_info?
    headers.delete "Content-Length"
    headers.delete "Content-Type"
  end
  if drop_body?
    close
    result = []
  end
  if calculate_content_length?
    # if some other code has already set Content-Length, don't muck with it
    # currently, this would be the static file-handler
    headers["Content-Length"] = body.inject(0) { |l, p| l + Rack::Utils.bytesize(p) }.to_s
  end
  [status.to_i, headers, result]
end

def initialize(*)
  super
  headers['Content-Type'] ||= 'text/html'
end