Sha256: 9bee157848b2698c30ee38072ee44cf81943603073dac4bb1a8c2102797d0815
Contents?: true
Size: 1.7 KB
Versions: 11
Compression:
Stored size: 1.7 KB
Contents
module HTTP
class Redirector
# Notifies that we reached max allowed redirect hops
class TooManyRedirectsError < ResponseError; end
# Notifies that following redirects got into an endless loop
class EndlessRedirectError < TooManyRedirectsError; end
# HTTP status codes which indicate redirects
REDIRECT_CODES = [300, 301, 302, 303, 307, 308].freeze
# :nodoc:
def initialize(options = nil)
options = {:max_hops => 5} unless options.respond_to?(:fetch)
@max_hops = options.fetch(:max_hops, 5)
@max_hops = false if @max_hops && 1 > @max_hops.to_i
end
# Follows redirects until non-redirect response found
def perform(request, response, &block)
reset(request, response)
follow(&block)
end
private
# Reset redirector state
def reset(request, response)
@request, @response = request, response
@visited = []
end
# Follow redirects
def follow
while REDIRECT_CODES.include?(@response.code)
@visited << @request.uri.to_s
fail TooManyRedirectsError if too_many_hops?
fail EndlessRedirectError if endless_loop?
uri = @response.headers['Location']
fail StateError, 'no Location header in redirect' unless uri
if 303 == @response.code
@request = @request.redirect uri, :get
else
@request = @request.redirect uri
end
@response = yield @request
end
@response
end
# Check if we reached max amount of redirect hops
def too_many_hops?
@max_hops < @visited.count if @max_hops
end
# Check if we got into an endless loop
def endless_loop?
2 < @visited.count(@visited.last)
end
end
end
Version data entries
11 entries across 11 versions & 1 rubygems