WIP: change thenable detection

nodejs · Jun 12, 2022 · 8425665 · 8425665
1 parent 8bf7965
commit 8425665
Showing 1 changed file with 6 additions and 17 deletions.
diff --git a/lib/internal/modules/esm/loader.js b/lib/internal/modules/esm/loader.js
@@ -15,6 +15,8 @@ const {
   ObjectDefineProperty,
   ObjectSetPrototypeOf,
   PromiseAll,
+  PromiseResolve,
+  PromisePrototypeThen,
   ReflectApply,
   RegExpPrototypeExec,
   SafeArrayIterator,
@@ -158,28 +160,15 @@ function nextHookFactory(chain, meta, validate) {
       if (generatedHookIndex === 0) { meta.chainFinished = true; }
 
       ArrayPrototypePush(args, nextNextHook);
-      const output = ReflectApply(hook, undefined, args);
+      let output = ReflectApply(hook, undefined, args);
 
       function checkShortCircuited(output) {
         if (output?.shortCircuit === true) { meta.shortCircuited = true; }
       }
 
-      const then = output?.then;
-      if (typeof then === 'function') {
-        if (!meta.isChainAsync) {
-          throw ERR_INVALID_RETURN_VALUE(
-            'an object',
-            // MUST use generatedHookIndex because the chain has already advanced,
-            // causing meta.hookIndex to advance
-            `${chain[generatedHookIndex].url} '${hookName}' hook's ${nextHookName}()`,
-            output,
-          );
-        }
-
-        ReflectApply(then, output, [
-          checkShortCircuited,
-          // TODO: handle error case
-        ]);
+      if (meta.isChainAsync) {
+        output = PromiseResolve(output);
+        PromisePrototypeThen(output, checkShortCircuited);
       } else {
         checkShortCircuited(output);
       }