1
|
'use strict';
|
2
|
|
3
|
var fs = require('fs'),
|
4
|
join = require('path').join,
|
5
|
resolve = require('path').resolve,
|
6
|
dirname = require('path').dirname,
|
7
|
defaultOptions = {
|
8
|
extensions: ['js', 'json', 'coffee'],
|
9
|
recurse: true,
|
10
|
rename: function (name) {
|
11
|
return name;
|
12
|
},
|
13
|
visit: function (obj) {
|
14
|
return obj;
|
15
|
}
|
16
|
};
|
17
|
|
18
|
function checkFileInclusion(path, filename, options) {
|
19
|
return (
|
20
|
// verify file has valid extension
|
21
|
(new RegExp('\\.(' + options.extensions.join('|') + ')$', 'i').test(filename)) &&
|
22
|
|
23
|
// if options.include is a RegExp, evaluate it and make sure the path passes
|
24
|
!(options.include && options.include instanceof RegExp && !options.include.test(path)) &&
|
25
|
|
26
|
// if options.include is a function, evaluate it and make sure the path passes
|
27
|
!(options.include && typeof options.include === 'function' && !options.include(path, filename)) &&
|
28
|
|
29
|
// if options.exclude is a RegExp, evaluate it and make sure the path doesn't pass
|
30
|
!(options.exclude && options.exclude instanceof RegExp && options.exclude.test(path)) &&
|
31
|
|
32
|
// if options.exclude is a function, evaluate it and make sure the path doesn't pass
|
33
|
!(options.exclude && typeof options.exclude === 'function' && options.exclude(path, filename))
|
34
|
);
|
35
|
}
|
36
|
|
37
|
function requireDirectory(m, path, options) {
|
38
|
var retval = {};
|
39
|
|
40
|
// path is optional
|
41
|
if (path && !options && typeof path !== 'string') {
|
42
|
options = path;
|
43
|
path = null;
|
44
|
}
|
45
|
|
46
|
// default options
|
47
|
options = options || {};
|
48
|
for (var prop in defaultOptions) {
|
49
|
if (typeof options[prop] === 'undefined') {
|
50
|
options[prop] = defaultOptions[prop];
|
51
|
}
|
52
|
}
|
53
|
|
54
|
// if no path was passed in, assume the equivelant of __dirname from caller
|
55
|
// otherwise, resolve path relative to the equivalent of __dirname
|
56
|
path = !path ? dirname(m.filename) : resolve(dirname(m.filename), path);
|
57
|
|
58
|
// get the path of each file in specified directory, append to current tree node, recurse
|
59
|
fs.readdirSync(path).forEach(function (filename) {
|
60
|
var joined = join(path, filename),
|
61
|
files,
|
62
|
key,
|
63
|
obj;
|
64
|
|
65
|
if (fs.statSync(joined).isDirectory() && options.recurse) {
|
66
|
// this node is a directory; recurse
|
67
|
files = requireDirectory(m, joined, options);
|
68
|
// exclude empty directories
|
69
|
if (Object.keys(files).length) {
|
70
|
retval[options.rename(filename, joined, filename)] = files;
|
71
|
}
|
72
|
} else {
|
73
|
if (joined !== m.filename && checkFileInclusion(joined, filename, options)) {
|
74
|
// hash node key shouldn't include file extension
|
75
|
key = filename.substring(0, filename.lastIndexOf('.'));
|
76
|
obj = m.require(joined);
|
77
|
retval[options.rename(key, joined, filename)] = options.visit(obj, joined, filename) || obj;
|
78
|
}
|
79
|
}
|
80
|
});
|
81
|
|
82
|
return retval;
|
83
|
}
|
84
|
|
85
|
module.exports = requireDirectory;
|
86
|
module.exports.defaults = defaultOptions;
|