| var Transform = require('readable-stream').Transform |
| , inherits = require('util').inherits |
| , xtend = require('xtend') |
| |
| function DestroyableTransform(opts) { |
| Transform.call(this, opts) |
| this._destroyed = false |
| } |
| |
| inherits(DestroyableTransform, Transform) |
| |
| DestroyableTransform.prototype.destroy = function(err) { |
| if (this._destroyed) return |
| this._destroyed = true |
| |
| var self = this |
| process.nextTick(function() { |
| if (err) |
| self.emit('error', err) |
| self.emit('close') |
| }) |
| } |
| |
| // a noop _transform function |
| function noop (chunk, enc, callback) { |
| callback(null, chunk) |
| } |
| |
| |
| // create a new export function, used by both the main export and |
| // the .ctor export, contains common logic for dealing with arguments |
| function through2 (construct) { |
| return function (options, transform, flush) { |
| if (typeof options == 'function') { |
| flush = transform |
| transform = options |
| options = {} |
| } |
| |
| if (typeof transform != 'function') |
| transform = noop |
| |
| if (typeof flush != 'function') |
| flush = null |
| |
| return construct(options, transform, flush) |
| } |
| } |
| |
| |
| // main export, just make me a transform stream! |
| module.exports = through2(function (options, transform, flush) { |
| var t2 = new DestroyableTransform(options) |
| |
| t2._transform = transform |
| |
| if (flush) |
| t2._flush = flush |
| |
| return t2 |
| }) |
| |
| |
| // make me a reusable prototype that I can `new`, or implicitly `new` |
| // with a constructor call |
| module.exports.ctor = through2(function (options, transform, flush) { |
| function Through2 (override) { |
| if (!(this instanceof Through2)) |
| return new Through2(override) |
| |
| this.options = xtend(options, override) |
| |
| DestroyableTransform.call(this, this.options) |
| } |
| |
| inherits(Through2, DestroyableTransform) |
| |
| Through2.prototype._transform = transform |
| |
| if (flush) |
| Through2.prototype._flush = flush |
| |
| return Through2 |
| }) |
| |
| |
| module.exports.obj = through2(function (options, transform, flush) { |
| var t2 = new DestroyableTransform(xtend({ objectMode: true, highWaterMark: 16 }, options)) |
| |
| t2._transform = transform |
| |
| if (flush) |
| t2._flush = flush |
| |
| return t2 |
| }) |