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.
		
		
		
		
		
			
		
			
				
					239 lines
				
				5.8 KiB
			
		
		
			
		
	
	
					239 lines
				
				5.8 KiB
			| 
								 
											2 years ago
										 
									 | 
							
								var stream = require('readable-stream')
							 | 
						||
| 
								 | 
							
								var eos = require('end-of-stream')
							 | 
						||
| 
								 | 
							
								var inherits = require('inherits')
							 | 
						||
| 
								 | 
							
								var shift = require('stream-shift')
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								var SIGNAL_FLUSH = (Buffer.from && Buffer.from !== Uint8Array.from)
							 | 
						||
| 
								 | 
							
								  ? Buffer.from([0])
							 | 
						||
| 
								 | 
							
								  : new Buffer([0])
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								var onuncork = function(self, fn) {
							 | 
						||
| 
								 | 
							
								  if (self._corked) self.once('uncork', fn)
							 | 
						||
| 
								 | 
							
								  else fn()
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								var autoDestroy = function (self, err) {
							 | 
						||
| 
								 | 
							
								  if (self._autoDestroy) self.destroy(err)
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								var destroyer = function(self, end) {
							 | 
						||
| 
								 | 
							
								  return function(err) {
							 | 
						||
| 
								 | 
							
								    if (err) autoDestroy(self, err.message === 'premature close' ? null : err)
							 | 
						||
| 
								 | 
							
								    else if (end && !self._ended) self.end()
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								var end = function(ws, fn) {
							 | 
						||
| 
								 | 
							
								  if (!ws) return fn()
							 | 
						||
| 
								 | 
							
								  if (ws._writableState && ws._writableState.finished) return fn()
							 | 
						||
| 
								 | 
							
								  if (ws._writableState) return ws.end(fn)
							 | 
						||
| 
								 | 
							
								  ws.end()
							 | 
						||
| 
								 | 
							
								  fn()
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								var noop = function() {}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								var toStreams2 = function(rs) {
							 | 
						||
| 
								 | 
							
								  return new (stream.Readable)({objectMode:true, highWaterMark:16}).wrap(rs)
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								var Duplexify = function(writable, readable, opts) {
							 | 
						||
| 
								 | 
							
								  if (!(this instanceof Duplexify)) return new Duplexify(writable, readable, opts)
							 | 
						||
| 
								 | 
							
								  stream.Duplex.call(this, opts)
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  this._writable = null
							 | 
						||
| 
								 | 
							
								  this._readable = null
							 | 
						||
| 
								 | 
							
								  this._readable2 = null
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  this._autoDestroy = !opts || opts.autoDestroy !== false
							 | 
						||
| 
								 | 
							
								  this._forwardDestroy = !opts || opts.destroy !== false
							 | 
						||
| 
								 | 
							
								  this._forwardEnd = !opts || opts.end !== false
							 | 
						||
| 
								 | 
							
								  this._corked = 1 // start corked
							 | 
						||
| 
								 | 
							
								  this._ondrain = null
							 | 
						||
| 
								 | 
							
								  this._drained = false
							 | 
						||
| 
								 | 
							
								  this._forwarding = false
							 | 
						||
| 
								 | 
							
								  this._unwrite = null
							 | 
						||
| 
								 | 
							
								  this._unread = null
							 | 
						||
| 
								 | 
							
								  this._ended = false
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  this.destroyed = false
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  if (writable) this.setWritable(writable)
							 | 
						||
| 
								 | 
							
								  if (readable) this.setReadable(readable)
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								inherits(Duplexify, stream.Duplex)
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								Duplexify.obj = function(writable, readable, opts) {
							 | 
						||
| 
								 | 
							
								  if (!opts) opts = {}
							 | 
						||
| 
								 | 
							
								  opts.objectMode = true
							 | 
						||
| 
								 | 
							
								  opts.highWaterMark = 16
							 | 
						||
| 
								 | 
							
								  return new Duplexify(writable, readable, opts)
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								Duplexify.prototype.cork = function() {
							 | 
						||
| 
								 | 
							
								  if (++this._corked === 1) this.emit('cork')
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								Duplexify.prototype.uncork = function() {
							 | 
						||
| 
								 | 
							
								  if (this._corked && --this._corked === 0) this.emit('uncork')
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								Duplexify.prototype.setWritable = function(writable) {
							 | 
						||
| 
								 | 
							
								  if (this._unwrite) this._unwrite()
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  if (this.destroyed) {
							 | 
						||
| 
								 | 
							
								    if (writable && writable.destroy) writable.destroy()
							 | 
						||
| 
								 | 
							
								    return
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  if (writable === null || writable === false) {
							 | 
						||
| 
								 | 
							
								    this.end()
							 | 
						||
| 
								 | 
							
								    return
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  var self = this
							 | 
						||
| 
								 | 
							
								  var unend = eos(writable, {writable:true, readable:false}, destroyer(this, this._forwardEnd))
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  var ondrain = function() {
							 | 
						||
| 
								 | 
							
								    var ondrain = self._ondrain
							 | 
						||
| 
								 | 
							
								    self._ondrain = null
							 | 
						||
| 
								 | 
							
								    if (ondrain) ondrain()
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  var clear = function() {
							 | 
						||
| 
								 | 
							
								    self._writable.removeListener('drain', ondrain)
							 | 
						||
| 
								 | 
							
								    unend()
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  if (this._unwrite) process.nextTick(ondrain) // force a drain on stream reset to avoid livelocks
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  this._writable = writable
							 | 
						||
| 
								 | 
							
								  this._writable.on('drain', ondrain)
							 | 
						||
| 
								 | 
							
								  this._unwrite = clear
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  this.uncork() // always uncork setWritable
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								Duplexify.prototype.setReadable = function(readable) {
							 | 
						||
| 
								 | 
							
								  if (this._unread) this._unread()
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  if (this.destroyed) {
							 | 
						||
| 
								 | 
							
								    if (readable && readable.destroy) readable.destroy()
							 | 
						||
| 
								 | 
							
								    return
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  if (readable === null || readable === false) {
							 | 
						||
| 
								 | 
							
								    this.push(null)
							 | 
						||
| 
								 | 
							
								    this.resume()
							 | 
						||
| 
								 | 
							
								    return
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  var self = this
							 | 
						||
| 
								 | 
							
								  var unend = eos(readable, {writable:false, readable:true}, destroyer(this))
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  var onreadable = function() {
							 | 
						||
| 
								 | 
							
								    self._forward()
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  var onend = function() {
							 | 
						||
| 
								 | 
							
								    self.push(null)
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  var clear = function() {
							 | 
						||
| 
								 | 
							
								    self._readable2.removeListener('readable', onreadable)
							 | 
						||
| 
								 | 
							
								    self._readable2.removeListener('end', onend)
							 | 
						||
| 
								 | 
							
								    unend()
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  this._drained = true
							 | 
						||
| 
								 | 
							
								  this._readable = readable
							 | 
						||
| 
								 | 
							
								  this._readable2 = readable._readableState ? readable : toStreams2(readable)
							 | 
						||
| 
								 | 
							
								  this._readable2.on('readable', onreadable)
							 | 
						||
| 
								 | 
							
								  this._readable2.on('end', onend)
							 | 
						||
| 
								 | 
							
								  this._unread = clear
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  this._forward()
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								Duplexify.prototype._read = function() {
							 | 
						||
| 
								 | 
							
								  this._drained = true
							 | 
						||
| 
								 | 
							
								  this._forward()
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								Duplexify.prototype._forward = function() {
							 | 
						||
| 
								 | 
							
								  if (this._forwarding || !this._readable2 || !this._drained) return
							 | 
						||
| 
								 | 
							
								  this._forwarding = true
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  var data
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  while (this._drained && (data = shift(this._readable2)) !== null) {
							 | 
						||
| 
								 | 
							
								    if (this.destroyed) continue
							 | 
						||
| 
								 | 
							
								    this._drained = this.push(data)
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  this._forwarding = false
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								Duplexify.prototype.destroy = function(err, cb) {
							 | 
						||
| 
								 | 
							
								  if (!cb) cb = noop
							 | 
						||
| 
								 | 
							
								  if (this.destroyed) return cb(null)
							 | 
						||
| 
								 | 
							
								  this.destroyed = true
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  var self = this
							 | 
						||
| 
								 | 
							
								  process.nextTick(function() {
							 | 
						||
| 
								 | 
							
								    self._destroy(err)
							 | 
						||
| 
								 | 
							
								    cb(null)
							 | 
						||
| 
								 | 
							
								  })
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								Duplexify.prototype._destroy = function(err) {
							 | 
						||
| 
								 | 
							
								  if (err) {
							 | 
						||
| 
								 | 
							
								    var ondrain = this._ondrain
							 | 
						||
| 
								 | 
							
								    this._ondrain = null
							 | 
						||
| 
								 | 
							
								    if (ondrain) ondrain(err)
							 | 
						||
| 
								 | 
							
								    else this.emit('error', err)
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  if (this._forwardDestroy) {
							 | 
						||
| 
								 | 
							
								    if (this._readable && this._readable.destroy) this._readable.destroy()
							 | 
						||
| 
								 | 
							
								    if (this._writable && this._writable.destroy) this._writable.destroy()
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  this.emit('close')
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								Duplexify.prototype._write = function(data, enc, cb) {
							 | 
						||
| 
								 | 
							
								  if (this.destroyed) return
							 | 
						||
| 
								 | 
							
								  if (this._corked) return onuncork(this, this._write.bind(this, data, enc, cb))
							 | 
						||
| 
								 | 
							
								  if (data === SIGNAL_FLUSH) return this._finish(cb)
							 | 
						||
| 
								 | 
							
								  if (!this._writable) return cb()
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  if (this._writable.write(data) === false) this._ondrain = cb
							 | 
						||
| 
								 | 
							
								  else if (!this.destroyed) cb()
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								Duplexify.prototype._finish = function(cb) {
							 | 
						||
| 
								 | 
							
								  var self = this
							 | 
						||
| 
								 | 
							
								  this.emit('preend')
							 | 
						||
| 
								 | 
							
								  onuncork(this, function() {
							 | 
						||
| 
								 | 
							
								    end(self._forwardEnd && self._writable, function() {
							 | 
						||
| 
								 | 
							
								      // haxx to not emit prefinish twice
							 | 
						||
| 
								 | 
							
								      if (self._writableState.prefinished === false) self._writableState.prefinished = true
							 | 
						||
| 
								 | 
							
								      self.emit('prefinish')
							 | 
						||
| 
								 | 
							
								      onuncork(self, cb)
							 | 
						||
| 
								 | 
							
								    })
							 | 
						||
| 
								 | 
							
								  })
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								Duplexify.prototype.end = function(data, enc, cb) {
							 | 
						||
| 
								 | 
							
								  if (typeof data === 'function') return this.end(null, null, data)
							 | 
						||
| 
								 | 
							
								  if (typeof enc === 'function') return this.end(data, null, enc)
							 | 
						||
| 
								 | 
							
								  this._ended = true
							 | 
						||
| 
								 | 
							
								  if (data) this.write(data)
							 | 
						||
| 
								 | 
							
								  if (!this._writableState.ending && !this._writableState.destroyed) this.write(SIGNAL_FLUSH)
							 | 
						||
| 
								 | 
							
								  return stream.Writable.prototype.end.call(this, cb)
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								module.exports = Duplexify
							 |