'use strict'; const spawn = require('child_process').spawn; const path = require('path'); const format = require('util').format; const lazyRequire = require('lazy-req')(require); const configstore = lazyRequire('configstore'); const chalk = lazyRequire('chalk'); const semverDiff = lazyRequire('semver-diff'); const latestVersion = lazyRequire('latest-version'); const isNpm = lazyRequire('is-npm'); const boxen = lazyRequire('boxen'); const xdgBasedir = lazyRequire('xdg-basedir'); const ONE_DAY = 1000 * 60 * 60 * 24; class UpdateNotifier { constructor(options) { this.options = options = options || {}; options.pkg = options.pkg || {}; // Reduce pkg to the essential keys. with fallback to deprecated options // TODO: Remove deprecated options at some point far into the future options.pkg = { name: options.pkg.name || options.packageName, version: options.pkg.version || options.packageVersion }; if (!options.pkg.name || !options.pkg.version) { throw new Error('pkg.name and pkg.version required'); } this.packageName = options.pkg.name; this.packageVersion = options.pkg.version; this.updateCheckInterval = typeof options.updateCheckInterval === 'number' ? options.updateCheckInterval : ONE_DAY; this.hasCallback = typeof options.callback === 'function'; this.callback = options.callback || (() => {}); if (!this.hasCallback) { try { const ConfigStore = configstore(); this.config = new ConfigStore(`update-notifier-${this.packageName}`, { optOut: false, // Init with the current time so the first check is only // after the set interval, so not to bother users right away lastUpdateCheck: Date.now() }); } catch (err) { // Expecting error code EACCES or EPERM const msg = chalk().yellow(format(' %s update check failed ', options.pkg.name)) + format('\n Try running with %s or get access ', chalk().cyan('sudo')) + '\n to the local update config store via \n' + chalk().cyan(format(' sudo chown -R $USER:$(id -gn $USER) %s ', xdgBasedir().config)); process.on('exit', () => { console.error('\n' + boxen()(msg, {align: 'center'})); }); } } } check() { if (this.hasCallback) { this.checkNpm() .then(update => this.callback(null, update)) .catch(err => this.callback(err)); return; } if ( !this.config || this.config.get('optOut') || 'NO_UPDATE_NOTIFIER' in process.env || process.argv.indexOf('--no-update-notifier') !== -1 ) { return; } this.update = this.config.get('update'); if (this.update) { this.config.delete('update'); } // Only check for updates on a set interval if (Date.now() - this.config.get('lastUpdateCheck') < this.updateCheckInterval) { return; } // Spawn a detached process, passing the options as an environment property spawn(process.execPath, [path.join(__dirname, 'check.js'), JSON.stringify(this.options)], { detached: true, stdio: 'ignore' }).unref(); } checkNpm() { return latestVersion()(this.packageName).then(latestVersion => { return { latest: latestVersion, current: this.packageVersion, type: semverDiff()(this.packageVersion, latestVersion) || 'latest', name: this.packageName }; }); } notify(opts) { if (!process.stdout.isTTY || isNpm() || !this.update) { return this; } opts = Object.assign({isGlobal: true}, opts); opts.message = opts.message || 'Update available ' + chalk().dim(this.update.current) + chalk().reset(' → ') + chalk().green(this.update.latest) + ' \nRun ' + chalk().cyan('npm i ' + (opts.isGlobal ? '-g ' : '') + this.packageName) + ' to update'; opts.boxenOpts = opts.boxenOpts || { padding: 1, margin: 1, align: 'center', borderColor: 'yellow', borderStyle: 'round' }; const message = '\n' + boxen()(opts.message, opts.boxenOpts); if (opts.defer === false) { console.error(message); } else { process.on('exit', () => { console.error(message); }); process.on('SIGINT', () => { console.error(''); process.exit(); }); } return this; } } module.exports = options => { const updateNotifier = new UpdateNotifier(options); updateNotifier.check(); return updateNotifier; }; module.exports.UpdateNotifier = UpdateNotifier;