Blame view

node_modules/request/lib/redirect.js 4.54 KB
f7563de62   Palak Handa   first commit
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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
  'use strict'
  
  var url = require('url')
  var isUrl = /^https?:/
  
  function Redirect (request) {
    this.request = request
    this.followRedirect = true
    this.followRedirects = true
    this.followAllRedirects = false
    this.followOriginalHttpMethod = false
    this.allowRedirect = function () {return true}
    this.maxRedirects = 10
    this.redirects = []
    this.redirectsFollowed = 0
    this.removeRefererHeader = false
  }
  
  Redirect.prototype.onRequest = function (options) {
    var self = this
  
    if (options.maxRedirects !== undefined) {
      self.maxRedirects = options.maxRedirects
    }
    if (typeof options.followRedirect === 'function') {
      self.allowRedirect = options.followRedirect
    }
    if (options.followRedirect !== undefined) {
      self.followRedirects = !!options.followRedirect
    }
    if (options.followAllRedirects !== undefined) {
      self.followAllRedirects = options.followAllRedirects
    }
    if (self.followRedirects || self.followAllRedirects) {
      self.redirects = self.redirects || []
    }
    if (options.removeRefererHeader !== undefined) {
      self.removeRefererHeader = options.removeRefererHeader
    }
    if (options.followOriginalHttpMethod !== undefined) {
      self.followOriginalHttpMethod = options.followOriginalHttpMethod
    }
  }
  
  Redirect.prototype.redirectTo = function (response) {
    var self = this
      , request = self.request
  
    var redirectTo = null
    if (response.statusCode >= 300 && response.statusCode < 400 && response.caseless.has('location')) {
      var location = response.caseless.get('location')
      request.debug('redirect', location)
  
      if (self.followAllRedirects) {
        redirectTo = location
      } else if (self.followRedirects) {
        switch (request.method) {
          case 'PATCH':
          case 'PUT':
          case 'POST':
          case 'DELETE':
            // Do not follow redirects
            break
          default:
            redirectTo = location
            break
        }
      }
    } else if (response.statusCode === 401) {
      var authHeader = request._auth.onResponse(response)
      if (authHeader) {
        request.setHeader('authorization', authHeader)
        redirectTo = request.uri
      }
    }
    return redirectTo
  }
  
  Redirect.prototype.onResponse = function (response) {
    var self = this
      , request = self.request
  
    var redirectTo = self.redirectTo(response)
    if (!redirectTo || !self.allowRedirect.call(request, response)) {
      return false
    }
  
    request.debug('redirect to', redirectTo)
  
    // ignore any potential response body.  it cannot possibly be useful
    // to us at this point.
    // response.resume should be defined, but check anyway before calling. Workaround for browserify.
    if (response.resume) {
      response.resume()
    }
  
    if (self.redirectsFollowed >= self.maxRedirects) {
      request.emit('error', new Error('Exceeded maxRedirects. Probably stuck in a redirect loop ' + request.uri.href))
      return false
    }
    self.redirectsFollowed += 1
  
    if (!isUrl.test(redirectTo)) {
      redirectTo = url.resolve(request.uri.href, redirectTo)
    }
  
    var uriPrev = request.uri
    request.uri = url.parse(redirectTo)
  
    // handle the case where we change protocol from https to http or vice versa
    if (request.uri.protocol !== uriPrev.protocol) {
      delete request.agent
    }
  
    self.redirects.push(
      { statusCode : response.statusCode
      , redirectUri: redirectTo
      }
    )
    if (self.followAllRedirects && request.method !== 'HEAD'
      && response.statusCode !== 401 && response.statusCode !== 307) {
      request.method = self.followOriginalHttpMethod ? request.method : 'GET'
    }
    // request.method = 'GET' // Force all redirects to use GET || commented out fixes #215
    delete request.src
    delete request.req
    delete request._started
    if (response.statusCode !== 401 && response.statusCode !== 307) {
      // Remove parameters from the previous response, unless this is the second request
      // for a server that requires digest authentication.
      delete request.body
      delete request._form
      if (request.headers) {
        request.removeHeader('host')
        request.removeHeader('content-type')
        request.removeHeader('content-length')
        if (request.uri.hostname !== request.originalHost.split(':')[0]) {
          // Remove authorization if changing hostnames (but not if just
          // changing ports or protocols).  This matches the behavior of curl:
          // https://github.com/bagder/curl/blob/6beb0eee/lib/http.c#L710
          request.removeHeader('authorization')
        }
      }
    }
  
    if (!self.removeRefererHeader) {
      request.setHeader('referer', uriPrev.href)
    }
  
    request.emit('redirect')
  
    request.init()
  
    return true
  }
  
  exports.Redirect = Redirect