Spaces:
Running
Running
; | |
var fs = require('fs'), | |
join = require('path').join, | |
resolve = require('path').resolve, | |
dirname = require('path').dirname, | |
defaultOptions = { | |
extensions: ['js', 'json', 'coffee'], | |
recurse: true, | |
rename: function (name) { | |
return name; | |
}, | |
visit: function (obj) { | |
return obj; | |
} | |
}; | |
function checkFileInclusion(path, filename, options) { | |
return ( | |
// verify file has valid extension | |
(new RegExp('\\.(' + options.extensions.join('|') + ')$', 'i').test(filename)) && | |
// if options.include is a RegExp, evaluate it and make sure the path passes | |
!(options.include && options.include instanceof RegExp && !options.include.test(path)) && | |
// if options.include is a function, evaluate it and make sure the path passes | |
!(options.include && typeof options.include === 'function' && !options.include(path, filename)) && | |
// if options.exclude is a RegExp, evaluate it and make sure the path doesn't pass | |
!(options.exclude && options.exclude instanceof RegExp && options.exclude.test(path)) && | |
// if options.exclude is a function, evaluate it and make sure the path doesn't pass | |
!(options.exclude && typeof options.exclude === 'function' && options.exclude(path, filename)) | |
); | |
} | |
function requireDirectory(m, path, options) { | |
var retval = {}; | |
// path is optional | |
if (path && !options && typeof path !== 'string') { | |
options = path; | |
path = null; | |
} | |
// default options | |
options = options || {}; | |
for (var prop in defaultOptions) { | |
if (typeof options[prop] === 'undefined') { | |
options[prop] = defaultOptions[prop]; | |
} | |
} | |
// if no path was passed in, assume the equivelant of __dirname from caller | |
// otherwise, resolve path relative to the equivalent of __dirname | |
path = !path ? dirname(m.filename) : resolve(dirname(m.filename), path); | |
// get the path of each file in specified directory, append to current tree node, recurse | |
fs.readdirSync(path).forEach(function (filename) { | |
var joined = join(path, filename), | |
files, | |
key, | |
obj; | |
if (fs.statSync(joined).isDirectory() && options.recurse) { | |
// this node is a directory; recurse | |
files = requireDirectory(m, joined, options); | |
// exclude empty directories | |
if (Object.keys(files).length) { | |
retval[options.rename(filename, joined, filename)] = files; | |
} | |
} else { | |
if (joined !== m.filename && checkFileInclusion(joined, filename, options)) { | |
// hash node key shouldn't include file extension | |
key = filename.substring(0, filename.lastIndexOf('.')); | |
obj = m.require(joined); | |
retval[options.rename(key, joined, filename)] = options.visit(obj, joined, filename) || obj; | |
} | |
} | |
}); | |
return retval; | |
} | |
module.exports = requireDirectory; | |
module.exports.defaults = defaultOptions; | |