Added loose Promise detection

nuxy · Mar 28, 2023 · 11e8a90 · 11e8a90
1 parent 257247d
commit 11e8a90
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/src/router/Stack.js b/src/router/Stack.js
@@ -121,7 +121,7 @@ class RouterStack {
         lastItem = index++ === funcs.length;
         nextItem = false;
 
-        if (isAsyncFunc(func) || isPromise(func)) {
+        if (isAsyncFunc(func) || isPromise(func) || hasReturn(func)) {
 
           // Asynchronous handling.
           promises.push(() => {
@@ -146,4 +146,18 @@ class RouterStack {
   }
 };
 
+/**
+ * Check if return statement exists. Assumes Promise
+ * since there is no way to detect if a function
+ * returns a Promise event without execution.
+ *
+ * @param {Function} func
+ *   Route/Middleware Function.
+ *
+ * @return {Boolean}
+ */
+function hasReturn(func) {
+  return (func && (typeof func === 'function' && /{\s+return\s+/.test(func.toString())));
+}
+
 module.exports = RouterStack;