module: initialize hook returns load, resolve

This commit allows the `initialize()` hook to optionally return an
object having the `resolve()` and `load()` hooks as properties.
This allows state passed into `initialize()` to be shared with the
`resolve()` and `load()` hooks either via closure or class
instance.

In addition to developer ergonomics, supporting this model will
make it easier to write tests against a loader module. The
existing design forces state to be shared at the module level
which puts the burden of invalidating the ESM module cache
on anyone hoping to write isolated tests against a loader
module.

Fixes: #50042

Author: ggoodman

lib/internal/modules/esm/hooks.js

@@ -3,6 +3,7 @@
 const {
   ArrayPrototypePush,
   ArrayPrototypePushApply,
+  FunctionPrototypeBind,
   Int32Array,
   ObjectAssign,
   ObjectDefineProperty,
@@ -139,7 +140,7 @@ class Hooks {
    * to the worker.
    * @returns {any | Promise<any>} User data, ignored unless it's a promise, in which case it will be awaited.
    */
-  addCustomLoader(url, exports, data) {
+  async addCustomLoader(url, exports, data) {
     const {
       initialize,
       resolve,
@@ -154,7 +155,30 @@ class Hooks {
       const next = this.#chains.load[this.#chains.load.length - 1];
       ArrayPrototypePush(this.#chains.load, { __proto__: null, fn: load, url, next });
     }
-    return initialize?.(data);
+
+    const hooks = await initialize?.(data);
+
+    if (hooks?.resolve) {
+      if (resolve) {
+        throw new ERR_INTERNAL_ASSERTION(
+          `ESM custom loader '${url}' exposed a 'resolve' hook and returned an object with a 'resolve' hook.`,
+        );
+      }
+      const next = this.#chains.resolve[this.#chains.resolve.length - 1];
+      const boundResolve = FunctionPrototypeBind(hooks.resolve, hooks);
+      ArrayPrototypePush(this.#chains.resolve, { __proto__: null, fn: boundResolve, url, next });
+    }
+
+    if (hooks?.load) {
+      if (load) {
+        throw new ERR_INTERNAL_ASSERTION(
+          `ESM custom loader '${url}' exposed a 'load' hook and returned an object with a 'load' hook.`,
+        );
+      }
+      const next = this.#chains.load[this.#chains.load.length - 1];
+      const boundLoad = FunctionPrototypeBind(hooks.load, hooks);
+      ArrayPrototypePush(this.#chains.load, { __proto__: null, fn: boundLoad, url, next });
+    }
   }
 
   /**