Move getFiles to fs module

1 files changed, 34 insertions, 0 deletions

diff --git a/lib/generate/fs.js b/lib/generate/fs.js
index f7ae772..c43225e 100644
--- a/lib/generate/fs.js
+++ b/lib/generate/fs.js
@@ -1,8 +1,42 @@
 var Q = require("q");
 var fs = require("fs");
 var fsExtra = require("fs-extra");
+var Ignore = require("fstream-ignore");
+
+var getFiles = function(path) {
+    var d = Q.defer();
+
+    // Our list of files
+    var files = [];
+
+    var ig = Ignore({
+        path: path,
+        ignoreFiles: ['.ignore', '.gitignore']
+    });
+
+    // Add extra rules to ignore common folders
+    ig.addIgnoreRules([
+        '.git/'
+    ], '__custom_stuff');
+
+    // Push each file to our list
+    ig.on('child', function (c) {
+        files.push(
+            c.path.substr(c.root.path.length + 1) + (c.props.Directory === true ? '/' : '')
+        );
+    });
+
+    ig.on('end', function() {
+        d.resolve(files);
+    });
+
+    ig.on('error', d.reject);
+
+    return d.promise;
+}
 
 module.exports = {
+    list: getFiles,
     readFile: Q.denodeify(fs.readFile),
     writeFile: Q.denodeify(fs.writeFile),
     mkdirp: Q.denodeify(fsExtra.mkdirp),