You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
113 lines
2.2 KiB
113 lines
2.2 KiB
'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; |
|
};
|
|
|