feat(runtime): allow onload hook to return wrapped module (#1987)

Author: ScriptedAlchemy

.changeset/fast-sheep-melt.md

@@ -0,0 +1,5 @@
+---
+'@module-federation/runtime': patch
+---
+
+onLoad hook will allow you to return a custom module factory or proxy

packages/runtime/src/core.ts

@@ -571,7 +571,7 @@ export class FederationHost {
       const { pkgNameOrAlias, remote, expose, id: idRes } = remoteMatchInfo;
       const moduleOrFactory = (await module.get(expose, options)) as T;
 
-      await this.hooks.lifecycle.onLoad.emit({
+      const moduleWrapper = await this.hooks.lifecycle.onLoad.emit({
         id: idRes,
         pkgNameOrAlias,
         expose,
@@ -583,6 +583,10 @@ export class FederationHost {
         origin: this,
       });
 
+      if (typeof moduleWrapper === 'function') {
+        return moduleWrapper as T;
+      }
+
       return moduleOrFactory;
     } catch (error) {
       const { from = 'runtime' } = options || { from: 'runtime' };