now accepts promises as return of supplied functions

ts/index.ts

@@ -1,42 +1,60 @@
 /// <reference path="typings/main.d.ts" />
-var through = require("through2");
-var path = require("path");
-var beautylog = require("beautylog");
+
+var plugins = {
+    beautylog: require("beautylog"),
+    Q: require("q"),
+    through: require("through2")
+}
+
+
 
 
 module.exports = function (functionsToExecuteArg:any|any[],executionModeArg:string = 'forEach') {
     //important vars
-    var gulpFunction = {
-        executionMode: executionModeArg, //can be forEach or atEnd
-        functionsToExecute: functionsToExecuteArg
+    var executionMode = executionModeArg; //can be forEach or atEnd
+    var functionsToExecute = functionsToExecuteArg;
+    var promiseArray = [];
+    var runFunction = function(functionArg){
+        var returnValue = functionArg();
+        if (typeof returnValue !== "undefined" && typeof returnValue.then !== "undefined") {
+            promiseArray.push(returnValue);
+        }
     };
 
-    var runFunctionNames = function () {
-        if (typeof gulpFunction.functionsToExecute === "function" ) {
-            gulpFunction.functionsToExecute();
-        } else if (Array.isArray(gulpFunction.functionsToExecute)) {
-            for (var anyFunction in gulpFunction.functionsToExecute) {
-                gulpFunction.functionsToExecute[anyFunction]();
+    var checkAndRunFunction = function () {
+        if (typeof functionsToExecute === "function" ) {
+            runFunction(functionsToExecute);
+        } else if (Array.isArray(functionsToExecute)) {
+            for (var anyFunction in functionsToExecute) {
+                runFunction(functionsToExecute[anyFunction]);
             }
         } else {
-            beautylog.error('gulp-callfunction: something is strange with the given arguments');
+            plugins.beautylog.error('gulp-callfunction: something is strange with the given arguments');
         }
+        return plugins.Q.all(promiseArray);
     };
 
 
     var forEach = function (file, enc, cb) {
-        if (gulpFunction.executionMode === 'forEach') {
-            runFunctionNames();
+        if (executionMode === 'forEach') {
+            checkAndRunFunction().then(function(){
+                cb(null, file);
+            });
+        } else {
+            cb(null, file);
         }
         //tell gulp that we are complete
-        return cb(null, file);
+
     };
 
     var atEnd = function(cb) {
-        if (gulpFunction.executionMode === "atEnd") {
-            runFunctionNames();
+        if (executionMode === "atEnd") {
+            checkAndRunFunction().then(function(){
+                cb();
+            });
+        } else {
+            cb();
         }
-        cb();
     };
-    return through.obj(forEach,atEnd);
+    return plugins.through.obj(forEach,atEnd);
 };

ts/test.ts

@@ -2,32 +2,49 @@
 var gulp = require("gulp");
 var gulpFunction = require("../index.js");
 var beautylog = require("beautylog");
+var Q = require("q");
 
 var myFunction = function () {
+    var done = Q.defer()
     beautylog.log("Function executed");
+    done.resolve();
+    return done.promise;
 };
 var myFunction2 = function () {
-    beautylog.log("Function2 executed");
+    var done = Q.defer();
+    beautylog.ok("Function2 executed");
+    done.resolve();
+    return done.promise;
+};
+var myFunction3 = function () {
+    var done = Q.defer();
+    beautylog.success("Function3 executed");
+    done.resolve();
+    return done.promise;
 };
 
 describe("gulpFunction",function(){
-    it("should run through smoothly with " + "'forEach'".blue,function(){
+    it("should run through smoothly with " + "'forEach'".blue,function(done){
         gulp.src('./test/*.md')
             .pipe(gulpFunction(myFunction,'forEach'))
             .pipe(gulp.dest("./test/result/"));
+
         gulp.src('./test/*.md')
-            .pipe(gulpFunction([myFunction,myFunction2],'forEach'))
-            .pipe(gulp.dest("./test/result/"));
+            .pipe(gulpFunction([myFunction2,myFunction3],'forEach'))
+            .pipe(gulp.dest("./test/result/"))
+            .pipe(gulpFunction(done,"atEnd"));
 
     });
 
-    it("should run through smoothly with " + "'atEnd'".blue,function(){
+    it("should run through smoothly with " + "'atEnd'".blue,function(done){
         gulp.src('./test/*.md')
             .pipe(gulpFunction(myFunction,'atEnd'))
             .pipe(gulp.dest("./test/result/"));
+
         gulp.src('./test/*.md')
-            .pipe(gulpFunction([myFunction,myFunction2],'atEnd'))
-            .pipe(gulp.dest("./test/result/"));
+            .pipe(gulpFunction([myFunction2,myFunction3],'atEnd'))
+            .pipe(gulp.dest("./test/result/"))
+            .pipe(gulpFunction(done,"atEnd"));
     });
 });