-
Notifications
You must be signed in to change notification settings - Fork 64
Expand file tree
/
Copy pathrequest.js
More file actions
352 lines (301 loc) · 8.52 KB
/
request.js
File metadata and controls
352 lines (301 loc) · 8.52 KB
Edit and raw actions
OlderNewer
1
var capability = require('./capability')
2
var inherits = require('inherits')
3
var response = require('./response')
4
var stream = require('readable-stream')
5
6
var IncomingMessage = response.IncomingMessage
7
var rStates = response.readyStates
8
9
function decideMode (preferBinary, useFetch) {
10
if (capability.fetch && useFetch) {
11
return 'fetch'
12
} else if (capability.mozchunkedarraybuffer) {
13
return 'moz-chunked-arraybuffer'
14
} else if (capability.msstream) {
15
return 'ms-stream'
16
} else if (capability.arraybuffer && preferBinary) {
17
return 'arraybuffer'
18
} else {
19
return 'text'
20
}
21
}
22
23
var ClientRequest = module.exports = function (opts) {
24
var self = this
25
stream.Writable.call(self)
26
27
self._opts = opts
28
self._body = []
29
self._headers = {}
30
if (opts.auth)
31
self.setHeader('Authorization', 'Basic ' + Buffer.from(opts.auth).toString('base64'))
32
Object.keys(opts.headers).forEach(function (name) {
33
self.setHeader(name, opts.headers[name])
34
})
35
36
var preferBinary
37
var useFetch = true
38
if (opts.mode === 'disable-fetch' || ('requestTimeout' in opts && !capability.abortController)) {
39
// If the use of XHR should be preferred. Not typically needed.
40
useFetch = false
41
preferBinary = true
42
} else if (opts.mode === 'prefer-streaming') {
43
// If streaming is a high priority but binary compatibility and
44
// the accuracy of the 'content-type' header aren't
45
preferBinary = false
46
} else if (opts.mode === 'allow-wrong-content-type') {
47
// If streaming is more important than preserving the 'content-type' header
48
preferBinary = !capability.overrideMimeType
49
} else if (!opts.mode || opts.mode === 'default' || opts.mode === 'prefer-fast') {
50
// Use binary if text streaming may corrupt data or the content-type header, or for speed
51
preferBinary = true
52
} else {
53
throw new Error('Invalid value for opts.mode')
54
}
55
self._mode = decideMode(preferBinary, useFetch)
56
self._fetchTimer = null
57
self._socketTimeout = null
58
self._socketTimer = null
59
60
self.on('finish', function () {
61
self._onFinish()
62
})
63
}
64
65
inherits(ClientRequest, stream.Writable)
66
67
ClientRequest.prototype.setHeader = function (name, value) {
68
var self = this
69
var lowerName = name.toLowerCase()
70
// This check is not necessary, but it prevents warnings from browsers about setting unsafe
71
// headers. To be honest I'm not entirely sure hiding these warnings is a good thing, but
72
// http-browserify did it, so I will too.
73
if (unsafeHeaders.indexOf(lowerName) !== -1)
74
return
75
76
self._headers[lowerName] = {
77
name: name,
78
value: value
79
}
80
}
81
82
ClientRequest.prototype.getHeader = function (name) {
83
var header = this._headers[name.toLowerCase()]
84
if (header)
85
return header.value
86
return null
87
}
88
89
ClientRequest.prototype.removeHeader = function (name) {
90
var self = this
91
delete self._headers[name.toLowerCase()]
92
}
93
94
ClientRequest.prototype._onFinish = function () {
95
var self = this
96
97
if (self._destroyed)
98
return
99
var opts = self._opts
100
101
if ('timeout' in opts && opts.timeout !== 0) {
102
self.setTimeout(opts.timeout)
103
}
104
105
var headersObj = self._headers
106
var body = null
107
if (opts.method !== 'GET' && opts.method !== 'HEAD') {
108
body = new Blob(self._body, {
109
type: (headersObj['content-type'] || {}).value || ''
110
});
111
}
112
113
// create flattened list of headers
114
var headersList = []
115
Object.keys(headersObj).forEach(function (keyName) {
116
var name = headersObj[keyName].name
117
var value = headersObj[keyName].value
118
if (Array.isArray(value)) {
119
value.forEach(function (v) {
120
headersList.push([name, v])
121
})
122
} else {
123
headersList.push([name, value])
124
}
125
})
126
127
if (self._mode === 'fetch') {
128
var signal = null
129
if (capability.abortController) {
130
var controller = new AbortController()
131
signal = controller.signal
132
self._fetchAbortController = controller
133
134
if ('requestTimeout' in opts && opts.requestTimeout !== 0) {
135
self._fetchTimer = global.setTimeout(function () {
136
self.emit('requestTimeout')
137
if (self._fetchAbortController)
138
self._fetchAbortController.abort()
139
}, opts.requestTimeout)
140
}
141
}
142
143
global.fetch(self._opts.url, {
144
method: self._opts.method,
145
headers: headersList,
146
body: body || undefined,
147
mode: 'cors',
148
credentials: opts.withCredentials ? 'include' : 'same-origin',
149
signal: signal
150
}).then(function (response) {
151
self._fetchResponse = response
152
self._resetTimers(false)
153
self._connect()
154
}, function (reason) {
155
self._resetTimers(true)
156
if (!self._destroyed)
157
self.emit('error', reason)
158
})
159
} else {
160
var xhr = self._xhr = new global.XMLHttpRequest()
161
try {
162
xhr.open(self._opts.method, self._opts.url, true)
163
} catch (err) {
164
process.nextTick(function () {
165
self.emit('error', err)
166
})
167
return
168
}
169
170
// Can't set responseType on really old browsers
171
if ('responseType' in xhr)
172
xhr.responseType = self._mode
173
174
if ('withCredentials' in xhr)
175
xhr.withCredentials = !!opts.withCredentials
176
177
if (self._mode === 'text' && 'overrideMimeType' in xhr)
178
xhr.overrideMimeType('text/plain; charset=x-user-defined')
179
180
if ('requestTimeout' in opts) {
181
xhr.timeout = opts.requestTimeout
182
xhr.ontimeout = function () {
183
self.emit('requestTimeout')
184
}
185
}
186
187
headersList.forEach(function (header) {
188
xhr.setRequestHeader(header[0], header[1])
189
})
190
191
self._response = null
192
xhr.onreadystatechange = function () {
193
switch (xhr.readyState) {
194
case rStates.LOADING:
195
case rStates.DONE:
196
self._onXHRProgress()
197
break
198
}
199
}
200
// Necessary for streaming in Firefox, since xhr.response is ONLY defined
201
// in onprogress, not in onreadystatechange with xhr.readyState = 3
202
if (self._mode === 'moz-chunked-arraybuffer') {
203
xhr.onprogress = function () {
204
self._onXHRProgress()
205
}
206
}
207
208
xhr.onerror = function () {
209
if (self._destroyed)
210
return
211
self._resetTimers(true)
212
self.emit('error', new Error('XHR error'))
213
}
214
215
try {
216
xhr.send(body)
217
} catch (err) {
218
process.nextTick(function () {
219
self.emit('error', err)
220
})
221
return
222
}
223
}
224
}
225
226
/**
227
* Checks if xhr.status is readable and non-zero, indicating no error.
228
* Even though the spec says it should be available in readyState 3,
229
* accessing it throws an exception in IE8
230
*/
231
function statusValid (xhr) {
232
try {
233
var status = xhr.status
234
return (status !== null && status !== 0)
235
} catch (e) {
236
return false
237
}
238
}
239
240
ClientRequest.prototype._onXHRProgress = function () {
241
var self = this
242
243
self._resetTimers(false)
244
245
if (!statusValid(self._xhr) || self._destroyed)
246
return
247
248
if (!self._response)
249
self._connect()
250
251
self._response._onXHRProgress(self._resetTimers.bind(self))
252
}
253
254
ClientRequest.prototype._connect = function () {
255
var self = this
256
257
if (self._destroyed)
258
return
259
260
self._response = new IncomingMessage(self._xhr, self._fetchResponse, self._mode, self._resetTimers.bind(self))
261
self._response.on('error', function(err) {
262
self.emit('error', err)
263
})
264
265
self.emit('response', self._response)
266
}
267
268
ClientRequest.prototype._write = function (chunk, encoding, cb) {
269
var self = this
270
271
self._body.push(chunk)
272
cb()
273
}
274
275
ClientRequest.prototype._resetTimers = function (done) {
276
var self = this
277
278
global.clearTimeout(self._socketTimer)
279
self._socketTimer = null
280
281
if (done) {
282
global.clearTimeout(self._fetchTimer)
283
self._fetchTimer = null
284
} else if (self._socketTimeout) {
285
self._socketTimer = global.setTimeout(function () {
286
self.emit('timeout')
287
}, self._socketTimeout)
288
}
289
}
290
291
ClientRequest.prototype.abort = ClientRequest.prototype.destroy = function (err) {
292
var self = this
293
self._destroyed = true
294
self._resetTimers(true)
295
if (self._response)
296
self._response._destroyed = true
297
if (self._xhr)
298
self._xhr.abort()
299
else if (self._fetchAbortController)
300
self._fetchAbortController.abort()
301
302
if (err)
303
self.emit('error', err)
304
}
305
306
ClientRequest.prototype.end = function (data, encoding, cb) {
307
var self = this
308
if (typeof data === 'function') {
309
cb = data
310
data = undefined
311
}
312
313
stream.Writable.prototype.end.call(self, data, encoding, cb)
314
}
315
316
ClientRequest.prototype.setTimeout = function (timeout, cb) {
317
var self = this
318
319
if (cb)
320
self.once('timeout', cb)
321
322
self._socketTimeout = timeout
323
self._resetTimers(false)
324
}
325
326
ClientRequest.prototype.flushHeaders = function () {}
327
ClientRequest.prototype.setNoDelay = function () {}
328
ClientRequest.prototype.setSocketKeepAlive = function () {}
329
330
// Taken from http://www.w3.org/TR/XMLHttpRequest/#the-setrequestheader%28%29-method
331
var unsafeHeaders = [
332
'accept-charset',
333
'accept-encoding',
334
'access-control-request-headers',
335
'access-control-request-method',
336
'connection',
337
'content-length',
338
'cookie',
339
'cookie2',
340
'date',
341
'dnt',
342
'expect',
343
'host',
344
'keep-alive',
345
'origin',
346
'referer',
347
'te',
348
'trailer',
349
'transfer-encoding',
350
'upgrade',
351
'via'
352
]