@illuma/core 1.3.1 → 1.4.0

package/dist/index.js

@@ -228,6 +228,7 @@ var Illuma = class _Illuma {
   }
   static _diagnostics = [];
   static _scanners = [];
+  static _middlewares = [];
   /** @internal */
   static get contextScanners() {
     return _Illuma._scanners;
@@ -250,6 +251,20 @@ var Illuma = class _Illuma {
   static extendContextScanner(scanner) {
     _Illuma._scanners.push(scanner);
   }
+  /**
+  * Registers a global middleware to be applied during instance creation.
+  * Typically used for cross-cutting concerns like logging, profiling, or custom instantiation logic.
+  * Function should accept instantiation parameters and a `next` function to proceed with the next middleware or actual instantiation.
+  *
+  * @param m - The middleware function to register
+  */
+  static registerGlobalMiddleware(m) {
+    _Illuma._middlewares.push(m);
+  }
+  middlewares = [];
+  registerMiddleware(m) {
+    this.middlewares.push(m);
+  }
   static onReport(report) {
     for (const diag of _Illuma._diagnostics) diag.onReport(report);
   }
@@ -454,7 +469,34 @@ function isNotTransparentProto(proto) {
 }
 __name(isNotTransparentProto, "isNotTransparentProto");
 
+// src/lib/utils/injector.ts
+var InjectorImpl = class {
+  static {
+    __name(this, "InjectorImpl");
+  }
+  container;
+  constructor(container) {
+    this.container = container;
+  }
+  get(token) {
+    return this.container.get(token);
+  }
+  produce(fn) {
+    return this.container.produce(fn);
+  }
+};
+var Injector = new NodeToken("Injector");
+
 // src/lib/provider/tree-node.ts
+function runMiddlewares(middlewares, params) {
+  const ms = middlewares;
+  const next = /* @__PURE__ */ __name((i, current) => {
+    if (i >= ms.length) return current.factory();
+    return ms[i](current, (nextParams) => next(i + 1, nextParams));
+  }, "next");
+  return next(0, params);
+}
+__name(runMiddlewares, "runMiddlewares");
 function retrieverFactory(node, deps, transparentDeps) {
   return (token, optional) => {
     const depNode = deps.get(token);
@@ -472,10 +514,12 @@ var TreeRootNode = class {
     __name(this, "TreeRootNode");
   }
   instant;
+  middlewares;
   _deps = /* @__PURE__ */ new Set();
   _treePool = /* @__PURE__ */ new WeakMap();
-  constructor(instant = true) {
+  constructor(instant = true, middlewares = []) {
     this.instant = instant;
+    this.middlewares = middlewares;
   }
   get dependencies() {
     return this._deps;
@@ -486,14 +530,14 @@ var TreeRootNode = class {
   build() {
     for (const dep of this._deps) {
       if ("token" in dep.proto) this._treePool.set(dep.proto.token, dep);
-      if (this.instant) dep.instantiate(this._treePool);
+      if (this.instant) dep.instantiate(this._treePool, this.middlewares);
       else dep.collectPool(this._treePool);
     }
   }
   find(token) {
     const node = this._treePool.get(token);
     if (!node) return null;
-    if (!this.instant) node.instantiate(this._treePool);
+    if (!this.instant) node.instantiate(this._treePool, this.middlewares);
     return node;
   }
   toString() {
@@ -518,6 +562,10 @@ var TreeNodeSingle = class {
   }
   constructor(proto) {
     this.proto = proto;
+    if (proto.token === Injector) {
+      this._instance = proto.factory();
+      this._resolved = true;
+    }
   }
   addDependency(node) {
     if (node instanceof TreeNodeTransparent) this._transparent.add(node);
@@ -529,14 +577,18 @@ var TreeNodeSingle = class {
     for (const dep of this._transparent) dep.collectPool(pool);
     pool.set(this.proto.token, this);
   }
-  instantiate(pool) {
+  instantiate(pool, middlewares = []) {
     if (this._resolved) return;
-    for (const node of this._deps.values()) node.instantiate(pool);
-    for (const dep of this._transparent) dep.instantiate(pool);
+    for (const node of this._deps.values()) node.instantiate(pool, middlewares);
+    for (const dep of this._transparent) dep.instantiate(pool, middlewares);
     const retriever = retrieverFactory(this.proto.token, this._deps, this._transparent);
     const factory = this.proto.factory ?? this.proto.token.opts?.factory;
     if (!factory) throw InjectionError.notFound(this.proto.token);
-    this._instance = InjectionContext.instantiate(factory, retriever);
+    const contextFactory = /* @__PURE__ */ __name(() => InjectionContext.instantiate(factory, retriever), "contextFactory");
+    this._instance = runMiddlewares(middlewares, {
+      token: this.proto.token,
+      factory: contextFactory
+    });
     this._resolved = true;
     if (pool) pool.set(this.proto.token, this);
   }
@@ -570,12 +622,16 @@ var TreeNodeTransparent = class _TreeNodeTransparent {
     for (const node of this._deps.values()) node.collectPool(pool);
     for (const dep of this._transparent) dep.collectPool(pool);
   }
-  instantiate(pool) {
+  instantiate(pool, middlewares = []) {
     if (this._resolved) return;
-    for (const dep of this._transparent) dep.instantiate(pool);
-    for (const node of this._deps.values()) node.instantiate(pool);
+    for (const dep of this._transparent) dep.instantiate(pool, middlewares);
+    for (const node of this._deps.values()) node.instantiate(pool, middlewares);
     const retriever = retrieverFactory(this.proto.parent.token, this._deps, this._transparent);
-    this._instance = InjectionContext.instantiate(this.proto.factory, retriever);
+    const refFactory = /* @__PURE__ */ __name(() => InjectionContext.instantiate(this.proto.factory, retriever), "refFactory");
+    this._instance = runMiddlewares(middlewares, {
+      token: this.proto.parent.token,
+      factory: refFactory
+    });
     this._resolved = true;
   }
   toString() {
@@ -598,10 +654,10 @@ var TreeNodeMulti = class _TreeNodeMulti {
     for (const dep of this._deps) dep.collectPool(pool);
     pool.set(this.proto.token, this);
   }
-  instantiate(pool) {
+  instantiate(pool, middlewares = []) {
     if (this._resolved) return;
     for (const dep of this._deps) {
-      dep.instantiate(pool);
+      dep.instantiate(pool, middlewares);
       if (dep instanceof TreeNodeSingle) {
         this.instance.push(dep.instance);
       } else if (dep instanceof _TreeNodeMulti) {
@@ -756,24 +812,6 @@ function resolveTreeNode(rootProto, cache, singleNodes, multiNodes, upstreamGett
 }
 __name(resolveTreeNode, "resolveTreeNode");
 
-// src/lib/utils/injector.ts
-var InjectorImpl = class {
-  static {
-    __name(this, "InjectorImpl");
-  }
-  container;
-  constructor(container) {
-    this.container = container;
-  }
-  get(token) {
-    return this.container.get(token);
-  }
-  produce(fn) {
-    return this.container.produce(fn);
-  }
-};
-var Injector = new NodeToken("Injector");
-
 // src/lib/utils/defer.ts
 function injectDefer(provider, options) {
   const injector = nodeInject(Injector);
@@ -999,7 +1037,11 @@ var NodeContainer = class extends Illuma {
     return parentNode.findNode(token);
   }
   _buildInjectionTree() {
-    const root = new TreeRootNode(this._opts?.instant);
+    const middlewares = [
+      ...Illuma._middlewares,
+      ...this.collectMiddlewares()
+    ];
+    const root = new TreeRootNode(this._opts?.instant, middlewares);
     const cache = /* @__PURE__ */ new Map();
     const nodes = [
       ...this._protoNodes.values(),
@@ -1016,6 +1058,12 @@ var NodeContainer = class extends Illuma {
     this._multiProtoNodes.clear();
     return root;
   }
+  collectMiddlewares() {
+    return [
+      ...this._parent && "collectMiddlewares" in this._parent && typeof this._parent.collectMiddlewares === "function" ? this._parent.collectMiddlewares() : [],
+      ...this.middlewares
+    ];
+  }
   /**
   * Bootstraps the container by resolving the dependency trees and instantiating all providers.
   * This must be called after all providers are registered and before calling {@link get}.