Sha256: d1af51b200eaab6c4ab8456a6bb0b8d833f8c87cedacc431618876e7fe6a671b
Contents?: true
Size: 1.88 KB
Versions: 7
Compression:
Stored size: 1.88 KB
Contents
require 'http/headers'
module Reel
class Response
CONTENT_LENGTH = 'Content-Length'.freeze
TRANSFER_ENCODING = 'Transfer-Encoding'.freeze
CHUNKED = 'chunked'.freeze
# Use status code tables from the HTTP gem
STATUS_CODES = HTTP::Response::Status::REASONS
SYMBOL_TO_STATUS_CODE = Hash[STATUS_CODES.map { |k, v| [v.downcase.gsub(/\s|-/, '_').to_sym, k] }].freeze
attr_reader :status # Status has a special setter to coerce symbol names
attr_accessor :reason # Reason can be set explicitly if desired
attr_reader :headers, :body, :version
def initialize(status, body_or_headers = nil, body = nil)
self.status = status
if body_or_headers.is_a?(Hash)
headers = body_or_headers.dup
@body = body
else
headers = {}
@body = body_or_headers
end
case @body
when String
headers[CONTENT_LENGTH] ||= @body.bytesize
when IO
headers[CONTENT_LENGTH] ||= @body.stat.size
when Enumerable
headers[TRANSFER_ENCODING] ||= CHUNKED
when NilClass
else raise TypeError, "can't render #{@body.class} as a response body"
end
@headers = HTTP::Headers.coerce(headers)
@version = http_version
end
def chunked?
headers[TRANSFER_ENCODING].to_s == CHUNKED
end
# Set the status
def status=(status, reason=nil)
case status
when Integer
@status = status
@reason ||= STATUS_CODES[status]
when Symbol
if code = SYMBOL_TO_STATUS_CODE[status]
self.status = code
else
raise ArgumentError, "unrecognized status symbol: #{status}"
end
else
raise TypeError, "invalid status type: #{status.inspect}"
end
end
def http_version
# FIXME: real HTTP versioning
"HTTP/1.1".freeze
end
private :http_version
end
end
Version data entries
7 entries across 7 versions & 1 rubygems