forked from projectmoon/tenebrous-dicebot
75 lines
1.8 KiB
JavaScript
75 lines
1.8 KiB
JavaScript
|
var http = require('http');
|
||
|
var https = require('https');
|
||
|
var url = require('url');
|
||
|
|
||
|
var isHttpResource = require('../utils/is-http-resource');
|
||
|
var isHttpsResource = require('../utils/is-https-resource');
|
||
|
var override = require('../utils/override');
|
||
|
|
||
|
var HTTP_PROTOCOL = 'http:';
|
||
|
|
||
|
function loadRemoteResource(uri, inlineRequest, inlineTimeout, callback) {
|
||
|
var proxyProtocol = inlineRequest.protocol || inlineRequest.hostname;
|
||
|
var errorHandled = false;
|
||
|
var requestOptions;
|
||
|
var fetch;
|
||
|
|
||
|
requestOptions = override(
|
||
|
url.parse(uri),
|
||
|
inlineRequest || {}
|
||
|
);
|
||
|
|
||
|
if (inlineRequest.hostname !== undefined) {
|
||
|
// overwrite as we always expect a http proxy currently
|
||
|
requestOptions.protocol = inlineRequest.protocol || HTTP_PROTOCOL;
|
||
|
requestOptions.path = requestOptions.href;
|
||
|
}
|
||
|
|
||
|
fetch = (proxyProtocol && !isHttpsResource(proxyProtocol)) || isHttpResource(uri) ?
|
||
|
http.get :
|
||
|
https.get;
|
||
|
|
||
|
fetch(requestOptions, function (res) {
|
||
|
var chunks = [];
|
||
|
var movedUri;
|
||
|
|
||
|
if (errorHandled) {
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
if (res.statusCode < 200 || res.statusCode > 399) {
|
||
|
return callback(res.statusCode, null);
|
||
|
} else if (res.statusCode > 299) {
|
||
|
movedUri = url.resolve(uri, res.headers.location);
|
||
|
return loadRemoteResource(movedUri, inlineRequest, inlineTimeout, callback);
|
||
|
}
|
||
|
|
||
|
res.on('data', function (chunk) {
|
||
|
chunks.push(chunk.toString());
|
||
|
});
|
||
|
res.on('end', function () {
|
||
|
var body = chunks.join('');
|
||
|
callback(null, body);
|
||
|
});
|
||
|
})
|
||
|
.on('error', function (res) {
|
||
|
if (errorHandled) {
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
errorHandled = true;
|
||
|
callback(res.message, null);
|
||
|
})
|
||
|
.on('timeout', function () {
|
||
|
if (errorHandled) {
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
errorHandled = true;
|
||
|
callback('timeout', null);
|
||
|
})
|
||
|
.setTimeout(inlineTimeout);
|
||
|
}
|
||
|
|
||
|
module.exports = loadRemoteResource;
|