work.suroh.tk/node_modules/require-directory/index.js

87 lines
2.8 KiB
JavaScript

'use strict';
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;