class WEBrick::HTTPResponse

def [](field)
  @header[field.downcase]
end

def []=(field, value)
  @chunked = value.to_s.downcase == 'chunked' if field.downcase == 'transfer-encoding'
  @header[field.downcase] = value.to_s
end

def _write_data(socket, data)
  socket << data
end

def check_header(header_value)
  header_value = header_value.to_s
  if /[\r\n]/ =~ header_value
    raise InvalidHeader
  else
    header_value
  end
end

def chunked=(val)
  @chunked = val ? true : false
end

def chunked?
  @chunked
end

def content_length
  if len = self['content-length']
    return Integer(len)
  end
end

def content_length=(len)
  self['content-length'] = len.to_s
end

def content_type
  self['content-type']
end

def content_type=(type)
  self['content-type'] = type
end

def each
  @header.each{|field, value|  yield(field, value) }
end

def error_body(backtrace, ex, host, port)
  @body = +""
  @body << <<-_end_of_html_
CTYPE HTML PUBLIC "-//W3C//DTD HTML 4.0//EN">
L>
EAD><TITLE>#{HTMLUtils::escape(@reason_phrase)}</TITLE></HEAD>
ODY>
<H1>#{HTMLUtils::escape(@reason_phrase)}</H1>
#{HTMLUtils::escape(ex.message)}
<HR>
  _end_of_html_
  if backtrace && $DEBUG
    @body << "backtrace of '#{HTMLUtils::escape(ex.class.to_s)}' "
    @body << "#{HTMLUtils::escape(ex.message)}"
    @body << "<PRE>"
    ex.backtrace.each{|line| @body << "\t#{line}\n"}
    @body << "</PRE><HR>"
  end
  @body << <<-_end_of_html_
<ADDRESS>
 #{HTMLUtils::escape(@config[:ServerSoftware])} at
 #{host}:#{port}
</ADDRESS>
BODY>
ML>
  _end_of_html_
end

def initialize(config)
  @config = config
  @buffer_size = config[:OutputBufferSize]
  @logger = config[:Logger]
  @header = Hash.new
  @status = HTTPStatus::RC_OK
  @reason_phrase = nil
  @http_version = HTTPVersion::convert(@config[:HTTPVersion])
  @body = +""
  @keep_alive = true
  @cookies = []
  @request_method = nil
  @request_uri = nil
  @request_http_version = @http_version  # temporary
  @chunked = false
  @filename = nil
  @sent_size = 0
  @bodytempfile = nil
end

def keep_alive?
  @keep_alive
end

def make_body_tempfile # :nodoc:
  return if @bodytempfile
  bodytempfile = Tempfile.create("webrick")
  if @body.nil?
    # nothing
  elsif @body.respond_to? :readpartial
    IO.copy_stream(@body, bodytempfile)
    @body.close
  elsif @body.respond_to? :call
    @body.call(bodytempfile)
  else
    bodytempfile.write @body
  end
  bodytempfile.rewind
  @body = @bodytempfile = bodytempfile
  @header['content-length'] = bodytempfile.stat.size.to_s
end

def remove_body_tempfile # :nodoc:
  if @bodytempfile
    @bodytempfile.close
    File.unlink @bodytempfile.path
    @bodytempfile = nil
  end
end

def send_body(socket) # :nodoc:
  if @body.respond_to? :readpartial then
    send_body_io(socket)
  elsif @body.respond_to?(:call) then
    send_body_proc(socket)
  else
    send_body_string(socket)
  end
end

def send_body_io(socket)
  begin
    if @request_method == "HEAD"
      # do nothing
    elsif chunked?
      buf = +''
      begin
        @body.readpartial(@buffer_size, buf)
        size = buf.bytesize
        data = +"#{size.to_s(16)}#{CRLF}#{buf}#{CRLF}"
        socket.write(data)
        data.clear
        @sent_size += size
      rescue EOFError
        break
      end while true
      buf.clear
      socket.write("0#{CRLF}#{CRLF}")
    else
      if %r{\Abytes (\d+)-(\d+)/\d+\z} =~ @header['content-range']
        offset = $1.to_i
        size = $2.to_i - offset + 1
      else
        offset = nil
        size = @header['content-length']
        size = size.to_i if size
      end
      begin
        @sent_size = IO.copy_stream(@body, socket, size, offset)
      rescue NotImplementedError
        @body.seek(offset, IO::SEEK_SET)
        @sent_size = IO.copy_stream(@body, socket, size)
      end
    end
  ensure
    @body.close
  end
  remove_body_tempfile
end

def send_body_proc(socket)
  if @request_method == "HEAD"
    # do nothing
  elsif chunked?
    @body.call(ChunkedWrapper.new(socket, self))
    socket.write("0#{CRLF}#{CRLF}")
  else
    if @bodytempfile
      @bodytempfile.rewind
      IO.copy_stream(@bodytempfile, socket)
    else
      @body.call(socket)
    end
    if content_length = @header['content-length']
      @sent_size = content_length.to_i
    end
  end
end

def send_body_string(socket)
  if @request_method == "HEAD"
    # do nothing
  elsif chunked?
    body ? @body.bytesize : 0
    while buf = @body[@sent_size, @buffer_size]
      break if buf.empty?
      size = buf.bytesize
      data = "#{size.to_s(16)}#{CRLF}#{buf}#{CRLF}"
      buf.clear
      socket.write(data)
      @sent_size += size
    end
    socket.write("0#{CRLF}#{CRLF}")
  else
    if @body && @body.bytesize > 0
      socket.write(@body)
      @sent_size = @body.bytesize
    end
  end
end

def send_header(socket) # :nodoc:
  if @http_version.major > 0
    data = status_line().dup
    @header.each{|key, value|
      tmp = key.gsub(/\bwww|^te$|\b\w/){ $&.upcase }
      data << "#{tmp}: #{check_header(value)}" << CRLF
    }
    @cookies.each{|cookie|
      data << "Set-Cookie: " << check_header(cookie.to_s) << CRLF
    }
    data << CRLF
    socket.write(data)
  end
rescue InvalidHeader => e
  @header.clear
  @cookies.clear
  set_error e
  retry
end

def send_response(socket) # :nodoc:
  begin
    setup_header()
    send_header(socket)
    send_body(socket)
  rescue Errno::EPIPE, Errno::ECONNRESET, Errno::ENOTCONN => ex
    @logger.debug(ex)
    @keep_alive = false
  rescue Exception => ex
    @logger.error(ex)
    @keep_alive = false
  end
end

def set_error(ex, backtrace=false)
  case ex
  when HTTPStatus::Status
    @keep_alive = false if HTTPStatus::error?(ex.code)
    self.status = ex.code
  else
    @keep_alive = false
    self.status = HTTPStatus::RC_INTERNAL_SERVER_ERROR
  end
  @header['content-type'] = "text/html; charset=ISO-8859-1"
  if respond_to?(:create_error_page)
    create_error_page()
    return
  end
  if @request_uri
    host, port = @request_uri.host, @request_uri.port
  else
    host, port = @config[:ServerName], @config[:Port]
  end
  error_body(backtrace, ex, host, port)
end

def set_redirect(status, url)
  url = URI(url).to_s
  @body = "<HTML><A HREF=\"#{url}\">#{url}</A>.</HTML>\n"
  @header['location'] = url
  raise status
end

def setup_header() # :nodoc:
  @reason_phrase    ||= HTTPStatus::reason_phrase(@status)
  @header['server'] ||= @config[:ServerSoftware]
  @header['date']   ||= Time.now.httpdate
  if @upgrade
    @header['connection'] = 'upgrade'
    @header['upgrade'] = @upgrade
    @keep_alive = false
    return
  end
  # HTTP/0.9 features
  if @request_http_version < "1.0"
    @http_version = HTTPVersion.new("0.9")
    @keep_alive = false
  end
  # HTTP/1.0 features
  if @request_http_version < "1.1"
    if chunked?
      @chunked = false
      ver = @request_http_version.to_s
      msg = "chunked is set for an HTTP/#{ver} request. (ignored)"
      @logger.warn(msg)
    end
  end
  # Determine the message length (RFC2616 -- 4.4 Message Length)
  if @status == 304 || @status == 204 || HTTPStatus::info?(@status)
    @header.delete('content-length')
    @body = ""
  elsif chunked?
    @header["transfer-encoding"] = "chunked"
    @header.delete('content-length')
  elsif %r{^multipart/byteranges} =~ @header['content-type']
    @header.delete('content-length')
  elsif @header['content-length'].nil?
    if @body.respond_to?(:bytesize)
      @header['content-length'] = @body.bytesize.to_s
    else
      @header['connection'] = 'close'
    end
  end
  # Keep-Alive connection.
  if @header['connection'] == "close"
     @keep_alive = false
  elsif keep_alive?
    if chunked? || @header['content-length'] || @status == 304 || @status == 204 || HTTPStatus.info?(@status)
      @header['connection'] = "Keep-Alive"
    else
      msg = "Could not determine content-length of response body. Set content-length of the response or set Response#chunked = true"
      @logger.warn(msg)
      @header['connection'] = "close"
      @keep_alive = false
    end
  else
    @header['connection'] = "close"
  end
  # Location is a single absoluteURI.
  if location = @header['location']
    if @request_uri
      @header['location'] = @request_uri.merge(location).to_s
    end
  end
end

def status=(status)
  @status = status
  @reason_phrase = HTTPStatus::reason_phrase(status)
end

def status_line
  "HTTP/#@http_version #@status #@reason_phrase".rstrip << CRLF
end

def upgrade!(protocol)
  @upgrade = protocol
  @keep_alive = false
  @chunked = false
end