1
|
var pump = require('pump')
|
2
|
var inherits = require('inherits')
|
3
|
var Duplexify = require('duplexify')
|
4
|
|
5
|
var toArray = function(args) {
|
6
|
if (!args.length) return []
|
7
|
return Array.isArray(args[0]) ? args[0] : Array.prototype.slice.call(args)
|
8
|
}
|
9
|
|
10
|
var define = function(opts) {
|
11
|
var Pumpify = function() {
|
12
|
var streams = toArray(arguments)
|
13
|
if (!(this instanceof Pumpify)) return new Pumpify(streams)
|
14
|
Duplexify.call(this, null, null, opts)
|
15
|
if (streams.length) this.setPipeline(streams)
|
16
|
}
|
17
|
|
18
|
inherits(Pumpify, Duplexify)
|
19
|
|
20
|
Pumpify.prototype.setPipeline = function() {
|
21
|
var streams = toArray(arguments)
|
22
|
var self = this
|
23
|
var ended = false
|
24
|
var w = streams[0]
|
25
|
var r = streams[streams.length-1]
|
26
|
|
27
|
r = r.readable ? r : null
|
28
|
w = w.writable ? w : null
|
29
|
|
30
|
var onclose = function() {
|
31
|
streams[0].emit('error', new Error('stream was destroyed'))
|
32
|
}
|
33
|
|
34
|
this.on('close', onclose)
|
35
|
this.on('prefinish', function() {
|
36
|
if (!ended) self.cork()
|
37
|
})
|
38
|
|
39
|
pump(streams, function(err) {
|
40
|
self.removeListener('close', onclose)
|
41
|
if (err) return self.destroy(err.message === 'premature close' ? null : err)
|
42
|
ended = true
|
43
|
// pump ends after the last stream is not writable *but*
|
44
|
// pumpify still forwards the readable part so we need to catch errors
|
45
|
// still, so reenable autoDestroy in this case
|
46
|
if (self._autoDestroy === false) self._autoDestroy = true
|
47
|
self.uncork()
|
48
|
})
|
49
|
|
50
|
if (this.destroyed) return onclose()
|
51
|
this.setWritable(w)
|
52
|
this.setReadable(r)
|
53
|
}
|
54
|
|
55
|
return Pumpify
|
56
|
}
|
57
|
|
58
|
module.exports = define({autoDestroy:false, destroy:false})
|
59
|
module.exports.obj = define({autoDestroy: false, destroy:false, objectMode:true, highWaterMark:16})
|
60
|
module.exports.ctor = define
|