114 lines
2.2 KiB
JavaScript
114 lines
2.2 KiB
JavaScript
|
'use strict';
|
||
|
|
|
||
|
|
var through2 = require('through2');
|
||
|
|
var EE = require('events').EventEmitter;
|
||
|
|
var fancyLog = require('fancy-log');
|
||
|
|
var chalk = require('chalk');
|
||
|
|
var PluginError = require('plugin-error');
|
||
|
|
|
||
|
|
function removeDefaultHandler(stream, event) {
|
||
|
|
var found = false;
|
||
|
|
stream.listeners(event).forEach(function (item) {
|
||
|
|
if (item.name === 'on' + event) {
|
||
|
|
found = item;
|
||
|
|
this.removeListener(event, item);
|
||
|
|
}
|
||
|
|
}, stream);
|
||
|
|
return found;
|
||
|
|
}
|
||
|
|
|
||
|
|
function wrapPanicOnErrorHandler(stream) {
|
||
|
|
var oldHandler = removeDefaultHandler(stream, 'error');
|
||
|
|
if (oldHandler) {
|
||
|
|
stream.on('error', function onerror2(er) {
|
||
|
|
if (EE.listenerCount(stream, 'error') === 1) {
|
||
|
|
this.removeListener('error', onerror2);
|
||
|
|
oldHandler.call(stream, er);
|
||
|
|
}
|
||
|
|
});
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
function defaultErrorHandler(error) {
|
||
|
|
// onerror2 and this handler
|
||
|
|
if (EE.listenerCount(this, 'error') < 3) {
|
||
|
|
fancyLog(
|
||
|
|
chalk.cyan('Plumber') + chalk.red(' found unhandled error:\n'),
|
||
|
|
error.toString()
|
||
|
|
);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
function plumber(opts) {
|
||
|
|
opts = opts || {};
|
||
|
|
|
||
|
|
if (typeof opts === 'function') {
|
||
|
|
opts = {errorHandler: opts};
|
||
|
|
}
|
||
|
|
|
||
|
|
var through = through2.obj();
|
||
|
|
through._plumber = true;
|
||
|
|
|
||
|
|
if (opts.errorHandler !== false) {
|
||
|
|
through.errorHandler = (typeof opts.errorHandler === 'function') ?
|
||
|
|
opts.errorHandler :
|
||
|
|
defaultErrorHandler;
|
||
|
|
}
|
||
|
|
|
||
|
|
function patchPipe(stream) {
|
||
|
|
if (stream.pipe2) {
|
||
|
|
wrapPanicOnErrorHandler(stream);
|
||
|
|
stream._pipe = stream._pipe || stream.pipe;
|
||
|
|
stream.pipe = stream.pipe2;
|
||
|
|
stream._plumbed = true;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
through.pipe2 = function pipe2(dest) {
|
||
|
|
if (!dest) {
|
||
|
|
throw new PluginError('plumber', 'Can\'t pipe to undefined');
|
||
|
|
}
|
||
|
|
|
||
|
|
this._pipe.apply(this, arguments);
|
||
|
|
|
||
|
|
if (dest._unplumbed) {
|
||
|
|
return dest;
|
||
|
|
}
|
||
|
|
|
||
|
|
removeDefaultHandler(this, 'error');
|
||
|
|
|
||
|
|
if (dest._plumber) {
|
||
|
|
return dest;
|
||
|
|
}
|
||
|
|
|
||
|
|
dest.pipe2 = pipe2;
|
||
|
|
|
||
|
|
// Patching pipe method
|
||
|
|
if (opts.inherit !== false) {
|
||
|
|
patchPipe(dest);
|
||
|
|
}
|
||
|
|
|
||
|
|
// Placing custom on error handler
|
||
|
|
if (this.errorHandler) {
|
||
|
|
dest.errorHandler = this.errorHandler;
|
||
|
|
dest.on('error', this.errorHandler.bind(dest));
|
||
|
|
}
|
||
|
|
|
||
|
|
dest._plumbed = true;
|
||
|
|
|
||
|
|
return dest;
|
||
|
|
};
|
||
|
|
|
||
|
|
patchPipe(through);
|
||
|
|
|
||
|
|
return through;
|
||
|
|
}
|
||
|
|
|
||
|
|
module.exports = plumber;
|
||
|
|
|
||
|
|
module.exports.stop = function () {
|
||
|
|
var through = through2.obj();
|
||
|
|
through._unplumbed = true;
|
||
|
|
return through;
|
||
|
|
};
|