'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');