123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111 |
- 'use strict';
- var through2 = require('through2');
- var EE = require('events').EventEmitter;
- var gutil = require('gulp-util');
- 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) {
- gutil.log(
- gutil.colors.cyan('Plumber') + gutil.colors.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 gutil.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;
- };
|