-
Notifications
You must be signed in to change notification settings - Fork 233
/
sendProxyRequest.js
78 lines (66 loc) · 2.25 KB
/
sendProxyRequest.js
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
'use strict';
var chunkLength = require('../../lib/chunkLength');
function sendProxyRequest(Container) {
var req = Container.user.req;
var bodyContent = Container.proxy.bodyContent;
var reqOpt = Container.proxy.reqBuilder;
var options = Container.options;
return new Promise(function(resolve, reject) {
var protocol = Container.proxy.requestModule;
var proxyReq = Container.proxy.req = protocol.request(reqOpt, function(rsp) {
if (options.stream) {
Container.proxy.res = rsp;
return resolve(Container);
}
var chunks = [];
rsp.on('data', function(chunk) { chunks.push(chunk); });
rsp.on('end', function() {
Container.proxy.res = rsp;
Container.proxy.resData = Buffer.concat(chunks, chunkLength(chunks));
resolve(Container);
});
rsp.on('error', reject);
});
proxyReq.on('socket', function(socket) {
if (options.timeout) {
socket.setTimeout(options.timeout, function() {
proxyReq.abort();
});
}
});
proxyReq.on('error', reject);
// this guy should go elsewhere, down the chain
if (options.parseReqBody) {
// We are parsing the body ourselves so we need to write the body content
// and then manually end the request.
//if (bodyContent instanceof Object) {
//throw new Error
//debugger;
//bodyContent = JSON.stringify(bodyContent);
//}
if (bodyContent.length) {
var body = bodyContent;
var contentType = proxyReq.getHeader('Content-Type');
if (contentType === 'x-www-form-urlencoded' || contentType === 'application/x-www-form-urlencoded') {
try {
var params = JSON.parse(body);
body = Object.keys(params).map(function(k) { return k + '=' + params[k]; }).join('&');
} catch (e) {
// bodyContent is not json-format
}
}
proxyReq.setHeader('Content-Length', Buffer.byteLength(body));
proxyReq.write(body);
}
proxyReq.end();
} else {
// Pipe will call end when it has completely read from the request.
req.pipe(proxyReq);
}
req.on('aborted', function() {
// reject?
proxyReq.abort();
});
});
}
module.exports = sendProxyRequest;