Class: SAPOCI::Connect::Middleware::FollowRedirects

Inherits:
Faraday::Middleware
  • Object
show all
Defined in:
lib/sapoci/connect/middleware/follow_redirects.rb

Constant Summary collapse

REDIRECTS = 
[301, 302, 303]
FOLLOW_LIMIT = 
5

Instance Method Summary collapse

Constructor Details

#initialize(app, options = {}) ⇒ FollowRedirects

Returns a new instance of FollowRedirects.



26
27
28
29
30
 
# File 'lib/sapoci/connect/middleware/follow_redirects.rb', line 26

def initialize(app, options = {})
  super(app)
  @options = options
  @limit   = options[:limit] || FOLLOW_LIMIT
end

Instance Method Details

#call(env) ⇒ Object 



32
33
34
 
# File 'lib/sapoci/connect/middleware/follow_redirects.rb', line 32

def call(env)
  process(@app.call(env), @limit)
end

#process(response, limit) ⇒ Object 



36
37
38
39
40
41
42
43
44
45
46
 
# File 'lib/sapoci/connect/middleware/follow_redirects.rb', line 36

def process(response, limit)
  response.on_complete do |env|
    if redirect?(response)
      raise RedirectLimitReached, response if limit.zero?
      raise RedirectWithoutLocation, response if response['location'].blank?
      env[:url] += response['location']
      env[:method] = :get
      response = process(@app.call(env), limit - 1)
    end
  end
end

#redirect?(response) ⇒ Boolean

Returns:

  • (Boolean) 


48
49
50
 
# File 'lib/sapoci/connect/middleware/follow_redirects.rb', line 48

def redirect?(response)
  REDIRECTS.include? response.status
end