-
Notifications
You must be signed in to change notification settings - Fork 64
Expand file tree
/
Copy pathrequest.js
More file actions
211 lines (179 loc) · 5.21 KB
/
request.js
File metadata and controls
211 lines (179 loc) · 5.21 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 copy (to, from) {
13
// if (!Array.isArray(from))
14
// from = [from]
15
// from.forEach(function (obj) {
16
// Object.keys(function (key), {
17
// to[key] = from[key]
18
// })
19
// })
20
// return to
21
// }
22
23
function decideMode (preferBinary) {
24
if (capability.fetch) {
25
return 'fetch'
26
} else if (capability.mozchunkedarraybuffer) {
27
return 'moz-chunked-arraybuffer'
28
} else if (capability.msstream) {
29
return 'ms-stream'
30
} else if (capability.arraybuffer && preferBinary) {
31
return 'arraybuffer'
32
} else {
33
return 'text'
34
}
35
}
36
37
var ClientRequest = module.exports = function (opts) {
38
var self = this
39
stream.Writable.call(self)
40
41
self._opts = opts
42
self._body = []
43
self._fullHeaders = {}
44
foreach(keys(opts.headers), function (name) {
45
self.setHeader(name, opts.headers[name])
46
})
47
48
var preferBinary
49
if (opts.mode === 'prefer-stream') {
50
// If streaming is a high priority but binary compatibility isn't
51
preferBinary = false
52
} else if (opts.mode === 'prefer-binary') {
53
// If binary compatibility is the highest priority
54
preferBinary = true
55
} else if (!opts.mode || opts.mode === 'default') {
56
// By default, use binary if text streaming may corrupt data
57
preferBinary = !capability.overrideMimeType
58
} else {
59
throw new Error('Invalid value for opts.mode')
60
}
61
self._mode = decideMode(preferBinary)
62
63
self.on('finish', function () {
64
self._onFinish()
65
})
66
}
67
68
util.inherits(ClientRequest, stream.Writable)
69
70
ClientRequest.prototype.setHeader = function (name, value) {
71
var self = this
72
self._fullHeaders[name.toLowerCase()] = value
73
}
74
75
ClientRequest.prototype.getHeader = function (name) {
76
var self = this
77
return self._fullHeaders[name.toLowerCase()]
78
}
79
80
ClientRequest.prototype.removeHeader = function (name) {
81
var self = this
82
delete self._fullHeaders[name.toLowerCase()]
83
}
84
85
ClientRequest.prototype._onFinish = function () {
86
var self = this
87
88
var opts = self._opts
89
var url = opts.protocol + '//' + opts.hostname +
90
(opts.port ? ':' + opts.port : '') + opts.path
91
92
var user, pass
93
if (opts.auth) {
94
var authMatch = opts.auth.match(/^([^:]*):(.*)$/)
95
user = authMatch[0]
96
pass = authMatch[1]
97
}
98
99
// process and send data
100
var fullHeaders = self._fullHeaders
101
var body
102
if (opts.method in ['PUT', 'POST']) {
103
if (typeof window.Blob === 'function') {
104
body = new window.Blob(self._body.map(function (buffer) {
105
return buffer.toArrayBuffer()
106
}), {
107
type: fullHeaders['content-type'] || ''
108
})
109
} else {
110
// get utf8 string
111
body = Buffer.concat(self._body).toString()
112
}
113
}
114
115
if (self._mode === 'fetch') {
116
var headers = keys(fullHeaders).map(function (name) {
117
return [name, fullHeaders[name]]
118
})
119
120
window.fetch(url, {
121
method: self._opts.method,
122
headers: headers,
123
body: body,
124
mode: 'cors',
125
credentials: opts.credentials ? 'include' : 'omit'
126
}).then(function (response) {
127
self._fetchResponse = response
128
self._connect()
129
})
130
} else {
131
var xhr = self._xhr = new window.XMLHttpRequest() // TODO: old IE
132
xhr.open(self._opts.method, url, true, user, pass)
133
134
// Can't set responseType on really old browsers
135
if ('responseType' in xhr)
136
xhr.responseType = self._mode
137
138
if ('withCredentials' in xhr)
139
xhr.withCredentials = !!opts.credentials
140
141
if (self._mode === 'text' && 'overrideMimeType' in xhr)
142
xhr.overrideMimeType('text/plain; charset=x-user-defined')
143
144
keys(fullHeaders, function (name) {
145
xhr.setRequestHeader(name, headers[name])
146
})
147
148
xhr.onreadystatechange = function () {
149
switch (xhr.readyState) {
150
case rStates.HEADERS_RECEIVED:
151
self._connect()
152
break
153
case rStates.LOADING:
154
case rStates.DONE:
155
self._response._onXHRReadyStateChange()
156
break
157
}
158
}
159
// Necessary for streaming in Firefox, since xhr.response is ONLY defined
160
// in onprogress, not in onreadystatechange with xhr.readyState = 3
161
if (self._mode === 'moz-chunked-arraybuffer') {
162
xhr.onprogress = function () {
163
self._response._onXHRReadyStateChange()
164
}
165
}
166
167
xhr.send(body)
168
// This is the best approximation to where 'socket' should be fired
169
process.nextTick(function () {
170
self.emit('socket')
171
})
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 () {}