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