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
						
					
					
				| // 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()
 | |
|   }
 | |
| }
 |