var URL = require('url') , request = require('request') , Stream = require('stream') , UError = require('./error').UserError , mystreams = require('./streams') , Logger = require('./logger') , utils = require('./utils') , parse_interval = require('./config').parse_interval , encode = encodeURIComponent // // Implements Storage interface // (same for storage.js, local-storage.js, up-storage.js) // function Storage(config, mainconfig) { if (!(this instanceof Storage)) return new Storage(config) this.config = config this.failed_requests = 0 this.userAgent = mainconfig.user_agent this.ca = config.ca this.logger = Logger.logger.child({sub: 'out'}) this.server_id = mainconfig.server_id this.url = URL.parse(this.config.url) if (this.url.hostname === 'registry.npmjs.org') { // npm registry is too slow working with ssl :( /*if (this.config._autogenerated) { // encrypt all the things! this.url.protocol = 'https' this.config.url = URL.format(this.url) }*/ } _setupProxy.call(this, this.url.hostname, config, mainconfig, this.url.protocol === 'https:') this.config.url = this.config.url.replace(/\/$/, '') this.maxage = parse_interval(this.config.maxage || '2m') if (Number(this.config.timeout) >= 1000) { this.logger.warn('Too big timeout value: ' + this.config.timeout + '\nWe changed time format to nginx-like one\n(see http://wiki.nginx.org/ConfigNotation)\nso please update your config accordingly') } this.timeout = parse_interval(this.config.timeout || '30s') this.max_fails = Number(this.config.max_fails) || 2 this.fail_timeout = parse_interval(this.config.fail_timeout || '5m') return this } function _setupProxy(hostname, config, mainconfig, isHTTPS) { var no_proxy var proxy_key = isHTTPS ? 'https_proxy' : 'http_proxy' // get http_proxy and no_proxy configs if (proxy_key in config) { this.proxy = config[proxy_key] } else if (proxy_key in mainconfig) { this.proxy = mainconfig[proxy_key] } if ('no_proxy' in config) { no_proxy = config.no_proxy } else if ('no_proxy' in mainconfig) { no_proxy = mainconfig.no_proxy } // use wget-like algorithm to determine if proxy shouldn't be used if (hostname[0] !== '.') hostname = '.' + hostname if (typeof(no_proxy) === 'string' && no_proxy.length) { no_proxy = no_proxy.split(',') } if (Array.isArray(no_proxy)) { for (var i=0; i= this.max_fails && Math.abs(Date.now() - this.last_request_time) < this.fail_timeout) { return false } else { return true } } else { if (alive) { if (this.failed_requests >= this.max_fails) { this.logger.warn({host: this.url.host}, 'host @{host} is back online') } this.failed_requests = 0 } else { this.failed_requests++ if (this.failed_requests === this.max_fails) { this.logger.warn({host: this.url.host}, 'host @{host} is now offline') } } this.last_request_time = Date.now() } } Storage.prototype.can_fetch_url = function(url) { url = URL.parse(url) return url.protocol === this.url.protocol && url.host === this.url.host && url.path.indexOf(this.url.path) === 0 } Storage.prototype.get_package = function(name, options, callback) { if (typeof(options) === 'function') callback = options, options = {} var headers = {} if (options.etag) { headers['If-None-Match'] = options.etag headers['Accept'] = 'application/octet-stream' } this._add_proxy_headers(options.req, headers) this.request({ uri: '/' + encode(name), json: true, headers: headers, }, function(err, res, body) { if (err) return callback(err) if (res.statusCode === 404) { return callback(new UError({ msg: 'package doesn\'t exist on uplink', status: 404, })) } if (!(res.statusCode >= 200 && res.statusCode < 300)) { return callback(new Error('bad status code: ' + res.statusCode)) } callback(null, body, res.headers.etag) }) } Storage.prototype.get_tarball = function(name, options, filename) { if (!options) options = {} return this.get_url(this.config.url + '/' + name + '/-/' + filename) } Storage.prototype.get_url = function(url) { var stream = new mystreams.ReadTarballStream() stream.abort = function() {} var current_length = 0, expected_length var rstream = this.request({ uri_full: url, encoding: null, headers: { Accept: 'application/octet-stream', }, }) rstream.on('response', function(res) { if (res.statusCode === 404) { return stream.emit('error', new UError({ msg: 'file doesn\'t exist on uplink', status: 404, })) } if (!(res.statusCode >= 200 && res.statusCode < 300)) { return stream.emit('error', new UError({ msg: 'bad uplink status code: ' + res.statusCode, status: 500, })) } if (res.headers['content-length']) { expected_length = res.headers['content-length'] stream.emit('content-length', res.headers['content-length']) } rstream.pipe(stream) }) rstream.on('error', function(err) { stream.emit('error', err) }) rstream.on('data', function(d) { current_length += d.length }) rstream.on('end', function(d) { if (d) current_length += d.length if (expected_length && current_length != expected_length) stream.emit('error', new Error('content length mismatch')) }) return stream } Storage.prototype._add_proxy_headers = function(req, headers) { if (req) { headers['X-Forwarded-For'] = ( (req && req.headers['x-forwarded-for']) ? req.headers['x-forwarded-for'] + ', ' : '' ) + req.connection.remoteAddress } // always attach Via header to avoid loops, even if we're not proxying headers['Via'] = (req && req.headers['via']) ? req.headers['via'] + ', ' : '' headers['Via'] += '1.1 ' + this.server_id + ' (Sinopia)' } module.exports = Storage