now accepts promises as return of supplied functions

index.d.ts

@@ -1,4 +1,6 @@
 /// <reference path="ts/typings/main.d.ts" />
-declare var through: any;
-declare var path: any;
-declare var beautylog: any;
+declare var plugins: {
+    beautylog: any;
+    Q: any;
+    through: any;
+};

index.js

@@ -1,41 +1,57 @@
 #!/usr/bin/env node
 
 /// <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, executionModeArg) {
     if (executionModeArg === void 0) { executionModeArg = 'forEach'; }
     //important vars
-    var gulpFunction = {
-        executionMode: executionModeArg,
-        functionsToExecute: functionsToExecuteArg
-    };
-    var runFunctionNames = function () {
-        if (typeof gulpFunction.functionsToExecute === "function") {
-            gulpFunction.functionsToExecute();
+    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);
         }
-        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);
 };

package.json

@@ -25,8 +25,9 @@
   },
   "homepage": "https://github.com/pushrocks/gulp-function",
   "dependencies": {
-    "beautylog": "2.1.1",
-    "through2": "2.0.1"
+    "beautylog": "^2.1.1",
+    "q": "^1.4.1",
+    "through2": "^2.0.1"
   },
   "devDependencies": {
     "gulp": "^3.9.1",

test/test.d.ts

@@ -2,5 +2,7 @@
 declare var gulp: any;
 declare var gulpFunction: any;
 declare var beautylog: any;
-declare var myFunction: () => void;
-declare var myFunction2: () => void;
+declare var Q: any;
+declare var myFunction: () => any;
+declare var myFunction2: () => any;
+declare var myFunction3: () => any;

test/test.js

@@ -4,27 +4,42 @@
 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"));
     });
 });

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"));
     });
 });