1 files changed, 45 insertions, 0 deletions
diff --git a/lib/autoInject.js b/lib/autoInject.js
new file mode 100644
index 0000000..1b6fea4
--- /dev/null
+++ b/lib/autoInject.js
@@ -0,0 +1,45 @@
+import auto from './auto';
+import forOwn from 'lodash/forOwn';
+import arrayMap from 'lodash/_arrayMap';
+import isArray from 'lodash/isArray';
+
+var argsRegex =  /^function\s*[^\(]*\(\s*([^\)]*)\)/m;
+
+function parseParams(func) {
+    return  func.toString().match(argsRegex)[1].split(/\s*\,\s*/);
+}
+
+export default function autoInject(tasks, callback) {
+    var newTasks = {};
+
+    forOwn(tasks, function (taskFn, key) {
+        var params;
+
+        if (isArray(taskFn)) {
+            params = [...taskFn];
+            taskFn = params.pop();
+
+            newTasks[key] = [...params].concat(newTask);
+        } else if (taskFn.length === 0) {
+            throw new Error("autoInject task functions require explicit parameters.");
+        } else if (taskFn.length === 1) {
+            // no dependencies
+            newTasks[key] = taskFn;
+        } else {
+            params = parseParams(taskFn);
+            params.pop();
+
+            newTasks[key] = [...params].concat(newTask);
+
+        }
+
+        function newTask(results, taskCb) {
+            var newArgs = arrayMap(params, function (name) {
+                return results[name];
+            });
+            taskFn(...newArgs.concat(taskCb));
+        }
+    });
+
+    auto(newTasks, callback);
+}