1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
|
require 'net/https'
class RedirectFollower
REDIRECT_DEFAULT_LIMIT = 5
class TooManyRedirects < StandardError; end
attr_accessor :url, :body, :redirect_limit, :response, :headers
def initialize(url, limit = REDIRECT_DEFAULT_LIMIT, options = {})
if limit.is_a? Hash
options = limit
limit = REDIRECT_DEFAULT_LIMIT
end
@url, @redirect_limit = url, limit
@headers = options[:headers] || {}
end
def resolve
raise TooManyRedirects if redirect_limit < 0
uri = URI.parse(URI.escape(url))
http = Net::HTTP.new(uri.host, uri.port)
if uri.scheme == 'https'
http.use_ssl = true
http.verify_mode = OpenSSL::SSL::VERIFY_PEER
end
self.response = http.request_get(uri.request_uri, @headers)
if response.kind_of?(Net::HTTPRedirection)
self.url = redirect_url
self.redirect_limit -= 1
resolve
end
self.body = response.body
self
end
def redirect_url
if response['location'].nil?
response.body.match(/<a href=\"([^>]+)\">/i)[1]
else
response['location']
end
end
end
|
