151 lines
3.6 KiB
Ruby
151 lines
3.6 KiB
Ruby
require 'rack/request'
|
|
require 'rack/utils'
|
|
require 'time'
|
|
|
|
module Rack
|
|
# Rack::Response provides a convenient interface to create a Rack
|
|
# response.
|
|
#
|
|
# It allows setting of headers and cookies, and provides useful
|
|
# defaults (a OK response containing HTML).
|
|
#
|
|
# You can use Response#write to iteratively generate your response,
|
|
# but note that this is buffered by Rack::Response until you call
|
|
# +finish+. +finish+ however can take a block inside which calls to
|
|
# +write+ are syncronous with the Rack response.
|
|
#
|
|
# Your application's +call+ should end returning Response#finish.
|
|
|
|
class Response
|
|
attr_accessor :length
|
|
|
|
def initialize(body=[], status=200, header={}, &block)
|
|
@status = status.to_i
|
|
@header = Utils::HeaderHash.new({"Content-Type" => "text/html"}.
|
|
merge(header))
|
|
|
|
@writer = lambda { |x| @body << x }
|
|
@block = nil
|
|
@length = 0
|
|
|
|
@body = []
|
|
|
|
if body.respond_to? :to_str
|
|
write body.to_str
|
|
elsif body.respond_to?(:each)
|
|
body.each { |part|
|
|
write part.to_s
|
|
}
|
|
else
|
|
raise TypeError, "stringable or iterable required"
|
|
end
|
|
|
|
yield self if block_given?
|
|
end
|
|
|
|
attr_reader :header
|
|
attr_accessor :status, :body
|
|
|
|
def [](key)
|
|
header[key]
|
|
end
|
|
|
|
def []=(key, value)
|
|
header[key] = value
|
|
end
|
|
|
|
def set_cookie(key, value)
|
|
Utils.set_cookie_header!(header, key, value)
|
|
end
|
|
|
|
def delete_cookie(key, value={})
|
|
Utils.delete_cookie_header!(header, key, value)
|
|
end
|
|
|
|
def redirect(target, status=302)
|
|
self.status = status
|
|
self["Location"] = target
|
|
end
|
|
|
|
def finish(&block)
|
|
@block = block
|
|
|
|
if [204, 304].include?(status.to_i)
|
|
header.delete "Content-Type"
|
|
[status.to_i, header, []]
|
|
else
|
|
[status.to_i, header, self]
|
|
end
|
|
end
|
|
alias to_a finish # For *response
|
|
|
|
def each(&callback)
|
|
@body.each(&callback)
|
|
@writer = callback
|
|
@block.call(self) if @block
|
|
end
|
|
|
|
# Append to body and update Content-Length.
|
|
#
|
|
# NOTE: Do not mix #write and direct #body access!
|
|
#
|
|
def write(str)
|
|
s = str.to_s
|
|
@length += Rack::Utils.bytesize(s)
|
|
@writer.call s
|
|
|
|
header["Content-Length"] = @length.to_s
|
|
str
|
|
end
|
|
|
|
def close
|
|
body.close if body.respond_to?(:close)
|
|
end
|
|
|
|
def empty?
|
|
@block == nil && @body.empty?
|
|
end
|
|
|
|
alias headers header
|
|
|
|
module Helpers
|
|
def invalid?; @status < 100 || @status >= 600; end
|
|
|
|
def informational?; @status >= 100 && @status < 200; end
|
|
def successful?; @status >= 200 && @status < 300; end
|
|
def redirection?; @status >= 300 && @status < 400; end
|
|
def client_error?; @status >= 400 && @status < 500; end
|
|
def server_error?; @status >= 500 && @status < 600; end
|
|
|
|
def ok?; @status == 200; end
|
|
def forbidden?; @status == 403; end
|
|
def not_found?; @status == 404; end
|
|
|
|
def redirect?; [301, 302, 303, 307].include? @status; end
|
|
def empty?; [201, 204, 304].include? @status; end
|
|
|
|
# Headers
|
|
attr_reader :headers, :original_headers
|
|
|
|
def include?(header)
|
|
!!headers[header]
|
|
end
|
|
|
|
def content_type
|
|
headers["Content-Type"]
|
|
end
|
|
|
|
def content_length
|
|
cl = headers["Content-Length"]
|
|
cl ? cl.to_i : cl
|
|
end
|
|
|
|
def location
|
|
headers["Location"]
|
|
end
|
|
end
|
|
|
|
include Helpers
|
|
end
|
|
end
|