81 lines
1.9 KiB
JavaScript
81 lines
1.9 KiB
JavaScript
'use strict';
|
|
|
|
|
|
var ProbeError = require('./lib/common').ProbeError;
|
|
var needle = require('needle');
|
|
var lodashMerge = require('lodash.merge');
|
|
var pkg = require('./package.json');
|
|
var probeStream = require('./stream');
|
|
var URL = require('url').URL;
|
|
|
|
var defaultAgent = pkg.name + '/' + pkg.version + '(+https://github.com/nodeca/probe-image-size)';
|
|
|
|
var defaults = {
|
|
open_timeout: 10000,
|
|
response_timeout: 60000,
|
|
read_timeout: 60000,
|
|
follow_max: 10,
|
|
parse_response: false,
|
|
// Use to ignore bad certificates.
|
|
//rejectUnauthorized: false,
|
|
headers: {
|
|
'User-Agent': defaultAgent
|
|
}
|
|
};
|
|
|
|
|
|
module.exports = function probeHttp(src, options) {
|
|
return new Promise(function (resolve, reject) {
|
|
var stream, len, finalUrl = src;
|
|
|
|
try {
|
|
var needleOptions = lodashMerge({}, defaults, options);
|
|
stream = needle.get(src, needleOptions);
|
|
} catch (err) {
|
|
reject(err);
|
|
return;
|
|
}
|
|
|
|
stream.on('redirect', function (location) {
|
|
try {
|
|
finalUrl = new URL(location, finalUrl).href;
|
|
} catch (err) {
|
|
reject(err);
|
|
stream.request.abort();
|
|
}
|
|
});
|
|
|
|
stream.on('header', function (statusCode, headers) {
|
|
if (statusCode !== 200) {
|
|
reject(new ProbeError('bad status code: ' + statusCode, null, statusCode));
|
|
stream.request.abort();
|
|
return;
|
|
}
|
|
|
|
len = headers['content-length'];
|
|
});
|
|
|
|
stream.on('err', function (err) {
|
|
reject(err);
|
|
stream.request.abort();
|
|
});
|
|
|
|
probeStream(stream, true)
|
|
.then(function (result) {
|
|
if (len && len.match(/^\d+$/)) result.length = +len;
|
|
|
|
result.url = finalUrl;
|
|
|
|
resolve(result);
|
|
stream.request.abort();
|
|
})
|
|
.catch(function (err) {
|
|
reject(err);
|
|
stream.request.abort();
|
|
});
|
|
});
|
|
};
|
|
|
|
|
|
module.exports.parsers = require('./lib/parsers_stream');
|