happy path async funtions in asyncify

1 files changed, 20 insertions, 11 deletions

diff --git a/lib/asyncify.js b/lib/asyncify.js
index e6d63c0..168f88a 100644
--- a/lib/asyncify.js
+++ b/lib/asyncify.js
@@ -1,5 +1,6 @@
 import initialParams from './internal/initialParams';
 import setImmediate from './internal/setImmediate';
+import { isAsync } from './internal/wrapAsync'
 
 /**
  * Take a sync function and make it async, passing its return value to a
@@ -58,6 +59,14 @@ import setImmediate from './internal/setImmediate';
  * q.push(files);
  */
 export default function asyncify(func) {
+    if (isAsync(func)) {
+        return function (...args/*, callback*/) {
+            const callback = args.pop()
+            const promise = func.apply(this, args)
+            return handlePromise(promise, callback)
+        }
+    }
+
     return initialParams(function (args, callback) {
         var result;
         try {
@@ -67,25 +76,25 @@ export default function asyncify(func) {
         }
         // if result is Promise object
         if (result && typeof result.then === 'function') {
-            result.then(function(value) {
-                invokeCallback(callback, null, value);
-            }, function(err) {
-                invokeCallback(callback, err.message ? err : new Error(err));
-            });
+            return handlePromise(result, callback)
         } else {
             callback(null, result);
         }
     });
 }
 
+function handlePromise(promise, callback) {
+    return promise.then(value => {
+        invokeCallback(callback, null, value);
+    }, err => {
+        invokeCallback(callback, err.message ? err : new Error(err));
+    });
+}
+
 function invokeCallback(callback, error, value) {
     try {
         callback(error, value);
-    } catch (e) {
-        setImmediate(rethrow, e);
+    } catch (err) {
+        setImmediate(e => { throw e }, err);
     }
 }
-
-function rethrow(error) {
-    throw error;
-}