123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111 |
- module.exports = Walker
- var path = require('path')
- , fs = require('fs')
- , util = require('util')
- , EventEmitter = require('events').EventEmitter
- , makeError = require('makeerror')
- /**
- * To walk a directory. It's complicated (but it's async, so it must be fast).
- *
- * @param root {String} the directory to start with
- */
- function Walker(root) {
- if (!(this instanceof Walker)) return new Walker(root)
- EventEmitter.call(this)
- this._pending = 0
- this._filterDir = function() { return true }
- this.go(root)
- }
- util.inherits(Walker, EventEmitter)
- /**
- * Errors of this type are thrown when the type of a file could not be
- * determined.
- */
- var UnknownFileTypeError = Walker.UnknownFileTypeError = makeError(
- 'UnknownFileTypeError',
- 'The type of this file could not be determined.'
- )
- /**
- * Setup a function to filter out directory entries.
- *
- * @param fn {Function} a function that will be given a directory name, which
- * if returns true will include the directory and it's children
- */
- Walker.prototype.filterDir = function(fn) {
- this._filterDir = fn
- return this
- }
- /**
- * Process a file or directory.
- */
- Walker.prototype.go = function(entry) {
- var that = this
- this._pending++
- fs.lstat(entry, function(er, stat) {
- if (er) {
- that.emit('error', er, entry, stat)
- that.doneOne()
- return
- }
- if (stat.isDirectory()) {
- if (!that._filterDir(entry, stat)) {
- that.doneOne()
- } else {
- fs.readdir(entry, function(er, files) {
- if (er) {
- that.emit('error', er, entry, stat)
- that.doneOne()
- return
- }
- that.emit('entry', entry, stat)
- that.emit('dir', entry, stat)
- files.forEach(function(part) {
- that.go(path.join(entry, part))
- })
- that.doneOne()
- })
- }
- } else if (stat.isSymbolicLink()) {
- that.emit('entry', entry, stat)
- that.emit('symlink', entry, stat)
- that.doneOne()
- } else if (stat.isBlockDevice()) {
- that.emit('entry', entry, stat)
- that.emit('blockDevice', entry, stat)
- that.doneOne()
- } else if (stat.isCharacterDevice()) {
- that.emit('entry', entry, stat)
- that.emit('characterDevice', entry, stat)
- that.doneOne()
- } else if (stat.isFIFO()) {
- that.emit('entry', entry, stat)
- that.emit('fifo', entry, stat)
- that.doneOne()
- } else if (stat.isSocket()) {
- that.emit('entry', entry, stat)
- that.emit('socket', entry, stat)
- that.doneOne()
- } else if (stat.isFile()) {
- that.emit('entry', entry, stat)
- that.emit('file', entry, stat)
- that.doneOne()
- } else {
- that.emit('error', UnknownFileTypeError(), entry, stat)
- that.doneOne()
- }
- })
- return this
- }
- Walker.prototype.doneOne = function() {
- if (--this._pending === 0) this.emit('end')
- return this
- }
|