walker.js
2.83 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
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
}