Class: RestCore::FollowRedirect

Inherits:
Object
  • Object
show all
Includes:
Middleware
Defined in:
lib/rest-core/middleware/follow_redirect.rb

Constant Summary

Constant Summary

Constants included from Middleware

Middleware::UNRESERVED

Constants included from RestCore

ASYNC, CLIENT, DRY, FAIL, HIJACK, LOG, PROMISE, REQUEST_HEADERS, REQUEST_METHOD, REQUEST_PATH, REQUEST_PAYLOAD, REQUEST_QUERY, REQUEST_URI, RESPONSE_BODY, RESPONSE_HEADERS, RESPONSE_KEY, RESPONSE_SOCKET, RESPONSE_STATUS, TIMER, Universal, VERSION

Class Method Summary collapse

Instance Method Summary collapse

Methods included from Middleware

contain_binary?, #error_callback, escape, #fail, #give_promise, #id, included, #log, merge_hash, percent_encode, request_uri, #run, string_keys

Methods included from RestCore

eagerload, id

Class Method Details

.membersObject



5
# File 'lib/rest-core/middleware/follow_redirect.rb', line 5

def self.members; [:max_redirects]; end

Instance Method Details

#call(env, &k) ⇒ Object



8
9
10
11
12
13
14
# File 'lib/rest-core/middleware/follow_redirect.rb', line 8

def call env, &k
  if env[DRY]
    app.call(env, &k)
  else
    app.call(env){ |res| process(res, k) }
  end
end

#process(res, k) ⇒ Object



16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
# File 'lib/rest-core/middleware/follow_redirect.rb', line 16

def process res, k
  return k.call(res) if max_redirects(res) <= 0
  return k.call(res) if ![301,302,303,307].include?(res[RESPONSE_STATUS])
  return k.call(res) if  [301,302    ,307].include?(res[RESPONSE_STATUS]) &&
                        ![:get, :head    ].include?(res[REQUEST_METHOD])

  location = [res[RESPONSE_HEADERS]['LOCATION']].flatten.first
  meth     = if res[RESPONSE_STATUS] == 303
               :get
             else
               res[REQUEST_METHOD]
             end

  give_promise(call(res.merge(
    REQUEST_METHOD => meth    ,
    REQUEST_PATH   => location,
    REQUEST_QUERY  => {}      ,
    'max_redirects' => max_redirects(res) - 1), &k))
end