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;