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.
		
		
		
		
		
			
		
			
				
					
					
						
							174 lines
						
					
					
						
							4.4 KiB
						
					
					
				
			
		
		
	
	
							174 lines
						
					
					
						
							4.4 KiB
						
					
					
				// 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() | 
						|
  } | 
						|
}
 | 
						|
 |