-
Notifications
You must be signed in to change notification settings - Fork 64
Expand file tree
/
Copy pathrequest.js
More file actions
211 lines (180 loc) · 5.38 KB
/
request.js
File metadata and controls
211 lines (180 loc) · 5.38 KB
Edit and raw actions
OlderNewer
1
// var Base64 = require('Base64')
2
var capability = require('./capability')
3
var foreach = require('foreach')
4
var keys = require('object-keys')
5
var response = require('./response')
6
var stream = require('stream')
7
var util = require('util')
8
9
var IncomingMessage = response.IncomingMessage
10
var rStates = response.readyStates
11
12
function decideMode (preferBinary) {
13
if (capability.fetch) {
14
return 'fetch'
15
} else if (capability.mozchunkedarraybuffer) {
16
return 'moz-chunked-arraybuffer'
17
} else if (capability.msstream) {
18
return 'ms-stream'
19
} else if (capability.arraybuffer && preferBinary) {
20
return 'arraybuffer'
21
} else if (capability.vbArray && preferBinary) {
22
return 'text:vbarray'
23
} else {
24
return 'text'
25
}
26
}
27
28
var ClientRequest = module.exports = function (opts) {
29
var self = this
30
stream.Writable.call(self)
31
32
self._opts = opts
33
self._url = opts.protocol + '//' + opts.hostname + ':' + opts.port + opts.path
34
self._body = []
35
self._headers = {}
36
if (opts.auth)
37
self.setHeader('Authorization', 'Basic ' + new Buffer(opts.auth).toString('base64'))
38
foreach(keys(opts.headers), function (name) {
39
self.setHeader(name, opts.headers[name])
40
})
41
42
var preferBinary
43
if (opts.mode === 'prefer-streaming') {
44
// If streaming is a high priority but binary compatibility isn't
45
preferBinary = false
46
} else if (opts.mode === 'prefer-binary') {
47
// If binary compatibility is the highest priority
48
preferBinary = true
49
} else if (!opts.mode || opts.mode === 'default') {
50
// By default, use binary if text streaming may corrupt data
51
preferBinary = !capability.overrideMimeType
52
} else {
53
throw new Error('Invalid value for opts.mode')
54
}
55
self._mode = decideMode(preferBinary)
56
57
self.on('finish', function () {
58
self._onFinish()
59
})
60
}
61
62
util.inherits(ClientRequest, stream.Writable)
63
64
ClientRequest.prototype.setHeader = function (name, value) {
65
var self = this
66
self._headers[name.toLowerCase()] = {
67
name: name,
68
value: value
69
}
70
}
71
72
ClientRequest.prototype.getHeader = function (name) {
73
var self = this
74
return self._headers[name.toLowerCase()].value
75
}
76
77
ClientRequest.prototype.removeHeader = function (name) {
78
var self = this
79
delete self._headers[name.toLowerCase()]
80
}
81
82
ClientRequest.prototype._onFinish = function () {
83
var self = this
84
85
var opts = self._opts
86
87
var headersObj = self._headers
88
var body
89
if (opts.method in ['PUT', 'POST']) {
90
if (util.isFunction(window.Blob)) {
91
body = new window.Blob(self._body.map(function (buffer) {
92
return buffer.toArrayBuffer()
93
}), {
94
type: (headersObj['content-type'] || {}).value || ''
95
})
96
} else {
97
// get utf8 string
98
body = Buffer.concat(self._body).toString()
99
}
100
}
101
102
if (self._mode === 'fetch') {
103
var headers = keys(headersObj).map(function (name) {
104
return [headersObj[name].name, headersObj[name].value]
105
})
106
107
window.fetch(self._url, {
108
method: self._opts.method,
109
headers: headers,
110
body: body,
111
mode: 'cors',
112
credentials: opts.credentials ? 'include' : 'omit'
113
}).then(function (response) {
114
self._fetchResponse = response
115
self._connect()
116
})
117
} else {
118
var xhr = self._xhr = new window.XMLHttpRequest()
119
xhr.open(self._opts.method, self._url, true)
120
121
// Can't set responseType on really old browsers
122
if ('responseType' in xhr)
123
xhr.responseType = self._mode.split(':')[0]
124
125
if ('withCredentials' in xhr)
126
xhr.withCredentials = !!opts.withCredentials
127
128
if (self._mode === 'text' && 'overrideMimeType' in xhr)
129
xhr.overrideMimeType('text/plain; charset=x-user-defined')
130
131
foreach(keys(headersObj), function (name) {
132
xhr.setRequestHeader(headersObj[name].name, headersObj[name].value)
133
})
134
135
self._response = null
136
xhr.onreadystatechange = function () {
137
switch (xhr.readyState) {
138
case rStates.LOADING:
139
case rStates.DONE:
140
if (!self._response && statusValid(xhr))
141
self._connect()
142
if (self._response)
143
self._response._onXHRReadyStateChange()
144
break
145
}
146
}
147
// Necessary for streaming in Firefox, since xhr.response is ONLY defined
148
// in onprogress, not in onreadystatechange with xhr.readyState = 3
149
if (self._mode === 'moz-chunked-arraybuffer') {
150
xhr.onprogress = function () {
151
self._response._onXHRReadyStateChange()
152
}
153
}
154
155
xhr.send(body)
156
// This is the best approximation to where 'socket' should be fired
157
process.nextTick(function () {
158
self.emit('socket')
159
})
160
}
161
}
162
163
/**
164
* Checks if xhr.status is readable. Even though the spec says it should
165
* be available in readyState 3, accessing it throws an exception in IE8
166
*/
167
function statusValid (xhr) {
168
try {
169
return (xhr.status !== null)
170
} catch (e) {
171
return false
172
}
173
}
174
175
ClientRequest.prototype._connect = function () {
176
var self = this
177
178
self._response = new IncomingMessage(self._xhr, self._fetchResponse, self._mode)
179
self.emit('response', self._response)
180
}
181
182
ClientRequest.prototype._write = function (chunk, encoding, cb) {
183
var self = this
184
185
self._body.push(chunk)
186
cb()
187
}
188
189
ClientRequest.prototype.abort = function () {
190
var self = this
191
if (self._xhr)
192
self._xhr.abort()
193
}
194
195
ClientRequest.prototype.end = function (data, encoding, cb) {
196
var self = this
197
if (typeof data === 'function') {
198
cb = data
199
data = undefined
200
}
201
202
if (data)
203
stream.Writable.push.call(self, data, encoding)
204
205
stream.Writable.prototype.end.call(self, cb)
206
}
207
208
ClientRequest.prototype.flushHeaders = function () {}
209
ClientRequest.prototype.setTimeout = function () {}
210
ClientRequest.prototype.setNoDelay = function () {}
211
ClientRequest.prototype.setSocketKeepAlive = function () {}