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.
		
		
		
		
			
				
					175 lines
				
				4.4 KiB
			
		
		
			
		
	
	
					175 lines
				
				4.4 KiB
			| 
								 
											4 years ago
										 
									 | 
							
								// It is expected that, when .add() returns false, the consumer
							 | 
						||
| 
								 | 
							
								// of the DirWriter will pause until a "drain" event occurs. Note
							 | 
						||
| 
								 | 
							
								// that this is *almost always going to be the case*, unless the
							 | 
						||
| 
								 | 
							
								// thing being written is some sort of unsupported type, and thus
							 | 
						||
| 
								 | 
							
								// skipped over.
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								module.exports = DirWriter
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								var Writer = require('./writer.js')
							 | 
						||
| 
								 | 
							
								var inherits = require('inherits')
							 | 
						||
| 
								 | 
							
								var mkdir = require('mkdirp')
							 | 
						||
| 
								 | 
							
								var path = require('path')
							 | 
						||
| 
								 | 
							
								var collect = require('./collect.js')
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								inherits(DirWriter, Writer)
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								function DirWriter (props) {
							 | 
						||
| 
								 | 
							
								  var self = this
							 | 
						||
| 
								 | 
							
								  if (!(self instanceof DirWriter)) {
							 | 
						||
| 
								 | 
							
								    self.error('DirWriter must be called as constructor.', null, true)
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  // should already be established as a Directory type
							 | 
						||
| 
								 | 
							
								  if (props.type !== 'Directory' || !props.Directory) {
							 | 
						||
| 
								 | 
							
								    self.error('Non-directory type ' + props.type + ' ' +
							 | 
						||
| 
								 | 
							
								      JSON.stringify(props), null, true)
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  Writer.call(this, props)
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								DirWriter.prototype._create = function () {
							 | 
						||
| 
								 | 
							
								  var self = this
							 | 
						||
| 
								 | 
							
								  mkdir(self._path, Writer.dirmode, function (er) {
							 | 
						||
| 
								 | 
							
								    if (er) return self.error(er)
							 | 
						||
| 
								 | 
							
								    // ready to start getting entries!
							 | 
						||
| 
								 | 
							
								    self.ready = true
							 | 
						||
| 
								 | 
							
								    self.emit('ready')
							 | 
						||
| 
								 | 
							
								    self._process()
							 | 
						||
| 
								 | 
							
								  })
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								// a DirWriter has an add(entry) method, but its .write() doesn't
							 | 
						||
| 
								 | 
							
								// do anything.  Why a no-op rather than a throw?  Because this
							 | 
						||
| 
								 | 
							
								// leaves open the door for writing directory metadata for
							 | 
						||
| 
								 | 
							
								// gnu/solaris style dumpdirs.
							 | 
						||
| 
								 | 
							
								DirWriter.prototype.write = function () {
							 | 
						||
| 
								 | 
							
								  return true
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								DirWriter.prototype.end = function () {
							 | 
						||
| 
								 | 
							
								  this._ended = true
							 | 
						||
| 
								 | 
							
								  this._process()
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								DirWriter.prototype.add = function (entry) {
							 | 
						||
| 
								 | 
							
								  var self = this
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  // console.error('\tadd', entry._path, '->', self._path)
							 | 
						||
| 
								 | 
							
								  collect(entry)
							 | 
						||
| 
								 | 
							
								  if (!self.ready || self._currentEntry) {
							 | 
						||
| 
								 | 
							
								    self._buffer.push(entry)
							 | 
						||
| 
								 | 
							
								    return false
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  // create a new writer, and pipe the incoming entry into it.
							 | 
						||
| 
								 | 
							
								  if (self._ended) {
							 | 
						||
| 
								 | 
							
								    return self.error('add after end')
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  self._buffer.push(entry)
							 | 
						||
| 
								 | 
							
								  self._process()
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  return this._buffer.length === 0
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								DirWriter.prototype._process = function () {
							 | 
						||
| 
								 | 
							
								  var self = this
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  // console.error('DW Process p=%j', self._processing, self.basename)
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  if (self._processing) return
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  var entry = self._buffer.shift()
							 | 
						||
| 
								 | 
							
								  if (!entry) {
							 | 
						||
| 
								 | 
							
								    // console.error("DW Drain")
							 | 
						||
| 
								 | 
							
								    self.emit('drain')
							 | 
						||
| 
								 | 
							
								    if (self._ended) self._finish()
							 | 
						||
| 
								 | 
							
								    return
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  self._processing = true
							 | 
						||
| 
								 | 
							
								  // console.error("DW Entry", entry._path)
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  self.emit('entry', entry)
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  // ok, add this entry
							 | 
						||
| 
								 | 
							
								  //
							 | 
						||
| 
								 | 
							
								  // don't allow recursive copying
							 | 
						||
| 
								 | 
							
								  var p = entry
							 | 
						||
| 
								 | 
							
								  var pp
							 | 
						||
| 
								 | 
							
								  do {
							 | 
						||
| 
								 | 
							
								    pp = p._path || p.path
							 | 
						||
| 
								 | 
							
								    if (pp === self.root._path || pp === self._path ||
							 | 
						||
| 
								 | 
							
								      (pp && pp.indexOf(self._path) === 0)) {
							 | 
						||
| 
								 | 
							
								      // console.error('DW Exit (recursive)', entry.basename, self._path)
							 | 
						||
| 
								 | 
							
								      self._processing = false
							 | 
						||
| 
								 | 
							
								      if (entry._collected) entry.pipe()
							 | 
						||
| 
								 | 
							
								      return self._process()
							 | 
						||
| 
								 | 
							
								    }
							 | 
						||
| 
								 | 
							
								    p = p.parent
							 | 
						||
| 
								 | 
							
								  } while (p)
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  // console.error("DW not recursive")
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  // chop off the entry's root dir, replace with ours
							 | 
						||
| 
								 | 
							
								  var props = {
							 | 
						||
| 
								 | 
							
								    parent: self,
							 | 
						||
| 
								 | 
							
								    root: self.root || self,
							 | 
						||
| 
								 | 
							
								    type: entry.type,
							 | 
						||
| 
								 | 
							
								    depth: self.depth + 1
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  pp = entry._path || entry.path || entry.props.path
							 | 
						||
| 
								 | 
							
								  if (entry.parent) {
							 | 
						||
| 
								 | 
							
								    pp = pp.substr(entry.parent._path.length + 1)
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								  // get rid of any ../../ shenanigans
							 | 
						||
| 
								 | 
							
								  props.path = path.join(self.path, path.join('/', pp))
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  // if i have a filter, the child should inherit it.
							 | 
						||
| 
								 | 
							
								  props.filter = self.filter
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  // all the rest of the stuff, copy over from the source.
							 | 
						||
| 
								 | 
							
								  Object.keys(entry.props).forEach(function (k) {
							 | 
						||
| 
								 | 
							
								    if (!props.hasOwnProperty(k)) {
							 | 
						||
| 
								 | 
							
								      props[k] = entry.props[k]
							 | 
						||
| 
								 | 
							
								    }
							 | 
						||
| 
								 | 
							
								  })
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  // not sure at this point what kind of writer this is.
							 | 
						||
| 
								 | 
							
								  var child = self._currentChild = new Writer(props)
							 | 
						||
| 
								 | 
							
								  child.on('ready', function () {
							 | 
						||
| 
								 | 
							
								    // console.error("DW Child Ready", child.type, child._path)
							 | 
						||
| 
								 | 
							
								    // console.error("  resuming", entry._path)
							 | 
						||
| 
								 | 
							
								    entry.pipe(child)
							 | 
						||
| 
								 | 
							
								    entry.resume()
							 | 
						||
| 
								 | 
							
								  })
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  // XXX Make this work in node.
							 | 
						||
| 
								 | 
							
								  // Long filenames should not break stuff.
							 | 
						||
| 
								 | 
							
								  child.on('error', function (er) {
							 | 
						||
| 
								 | 
							
								    if (child._swallowErrors) {
							 | 
						||
| 
								 | 
							
								      self.warn(er)
							 | 
						||
| 
								 | 
							
								      child.emit('end')
							 | 
						||
| 
								 | 
							
								      child.emit('close')
							 | 
						||
| 
								 | 
							
								    } else {
							 | 
						||
| 
								 | 
							
								      self.emit('error', er)
							 | 
						||
| 
								 | 
							
								    }
							 | 
						||
| 
								 | 
							
								  })
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  // we fire _end internally *after* end, so that we don't move on
							 | 
						||
| 
								 | 
							
								  // until any "end" listeners have had their chance to do stuff.
							 | 
						||
| 
								 | 
							
								  child.on('close', onend)
							 | 
						||
| 
								 | 
							
								  var ended = false
							 | 
						||
| 
								 | 
							
								  function onend () {
							 | 
						||
| 
								 | 
							
								    if (ended) return
							 | 
						||
| 
								 | 
							
								    ended = true
							 | 
						||
| 
								 | 
							
								    // console.error("* DW Child end", child.basename)
							 | 
						||
| 
								 | 
							
								    self._currentChild = null
							 | 
						||
| 
								 | 
							
								    self._processing = false
							 | 
						||
| 
								 | 
							
								    self._process()
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								}
							 |