#3 - Updating the promise handler so that it can be used in coffeescript

atsid · Mar 18, 2015 · 4157062 · 4157062
1 parent 9f7fec0
commit 4157062
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/src/proxies/promise-handler.js b/src/proxies/promise-handler.js
@@ -11,13 +11,9 @@ module.exports = {
     name: 'Promise Handler',
     init: (delegate) => {
         return (req, res, next) => {
-            if (delegate.length === 2) {
-                JPromise.resolve(true)
-                    .then(() => delegate(req, res))
-                    .then(() => next())
-                    .catch(next);
-            } else {
-                delegate(req, res, next);
+            let possiblePromise = delegate(req, res, next);
+            if (possiblePromise && possiblePromise.then) {
+                possiblePromise.then(() => next()).catch(next);
             }
         };
     }