138 lines
3.7 KiB
Ruby
138 lines
3.7 KiB
Ruby
require 'delegate'
|
|
require 'http/headers'
|
|
require 'http/content_type'
|
|
require 'http/mime_type'
|
|
|
|
module HTTP
|
|
class Response
|
|
include HTTP::Headers::Mixin
|
|
|
|
STATUS_CODES = {
|
|
100 => 'Continue',
|
|
101 => 'Switching Protocols',
|
|
102 => 'Processing',
|
|
200 => 'OK',
|
|
201 => 'Created',
|
|
202 => 'Accepted',
|
|
203 => 'Non-Authoritative Information',
|
|
204 => 'No Content',
|
|
205 => 'Reset Content',
|
|
206 => 'Partial Content',
|
|
207 => 'Multi-Status',
|
|
226 => 'IM Used',
|
|
300 => 'Multiple Choices',
|
|
301 => 'Moved Permanently',
|
|
302 => 'Found',
|
|
303 => 'See Other',
|
|
304 => 'Not Modified',
|
|
305 => 'Use Proxy',
|
|
306 => 'Reserved',
|
|
307 => 'Temporary Redirect',
|
|
400 => 'Bad Request',
|
|
401 => 'Unauthorized',
|
|
402 => 'Payment Required',
|
|
403 => 'Forbidden',
|
|
404 => 'Not Found',
|
|
405 => 'Method Not Allowed',
|
|
406 => 'Not Acceptable',
|
|
407 => 'Proxy Authentication Required',
|
|
408 => 'Request Timeout',
|
|
409 => 'Conflict',
|
|
410 => 'Gone',
|
|
411 => 'Length Required',
|
|
412 => 'Precondition Failed',
|
|
413 => 'Request Entity Too Large',
|
|
414 => 'Request-URI Too Long',
|
|
415 => 'Unsupported Media Type',
|
|
416 => 'Requested Range Not Satisfiable',
|
|
417 => 'Expectation Failed',
|
|
418 => "I'm a Teapot",
|
|
422 => 'Unprocessable Entity',
|
|
423 => 'Locked',
|
|
424 => 'Failed Dependency',
|
|
426 => 'Upgrade Required',
|
|
500 => 'Internal Server Error',
|
|
501 => 'Not Implemented',
|
|
502 => 'Bad Gateway',
|
|
503 => 'Service Unavailable',
|
|
504 => 'Gateway Timeout',
|
|
505 => 'HTTP Version Not Supported',
|
|
506 => 'Variant Also Negotiates',
|
|
507 => 'Insufficient Storage',
|
|
510 => 'Not Extended'
|
|
}
|
|
STATUS_CODES.freeze
|
|
|
|
SYMBOL_TO_STATUS_CODE = Hash[STATUS_CODES.map { |code, msg| [msg.downcase.gsub(/\s|-/, '_').to_sym, code] }]
|
|
SYMBOL_TO_STATUS_CODE.freeze
|
|
|
|
attr_reader :status
|
|
attr_reader :body
|
|
attr_reader :uri
|
|
|
|
# Status aliases! TIMTOWTDI!!! (Want to be idiomatic? Just use status :)
|
|
alias_method :code, :status
|
|
alias_method :status_code, :status
|
|
|
|
def initialize(status, version, headers, body, uri = nil) # rubocop:disable ParameterLists
|
|
@status, @version, @body, @uri = status, version, body, uri
|
|
@headers = HTTP::Headers.coerce(headers || {})
|
|
end
|
|
|
|
# Obtain the 'Reason-Phrase' for the response
|
|
def reason
|
|
STATUS_CODES[@status]
|
|
end
|
|
|
|
# Returns an Array ala Rack: `[status, headers, body]`
|
|
def to_a
|
|
[status, headers.to_h, body.to_s]
|
|
end
|
|
|
|
# Return the response body as a string
|
|
def to_s
|
|
body.to_s
|
|
end
|
|
alias_method :to_str, :to_s
|
|
|
|
# Flushes body and returns self-reference
|
|
def flush
|
|
body.to_s
|
|
self
|
|
end
|
|
|
|
# Parsed Content-Type header
|
|
# @return [HTTP::ContentType]
|
|
def content_type
|
|
@content_type ||= ContentType.parse headers['Content-Type']
|
|
end
|
|
|
|
# MIME type of response (if any)
|
|
# @return [String, nil]
|
|
def mime_type
|
|
@mime_type ||= content_type.mime_type
|
|
end
|
|
|
|
# Charset of response (if any)
|
|
# @return [String, nil]
|
|
def charset
|
|
@charset ||= content_type.charset
|
|
end
|
|
|
|
# Parse response body with corresponding MIME type adapter.
|
|
#
|
|
# @param [#to_s] as Parse as given MIME type
|
|
# instead of the one determined from headers
|
|
# @raise [Error] if adapter not found
|
|
# @return [Object]
|
|
def parse(as = nil)
|
|
MimeType[as || mime_type].decode to_s
|
|
end
|
|
|
|
# Inspect a response
|
|
def inspect
|
|
"#<#{self.class}/#{@version} #{status} #{reason} headers=#{headers.inspect}>"
|
|
end
|
|
end
|
|
end
|