@nmtjs/core 0.10.3 → 0.10.5

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/dist/container.js CHANGED
@@ -5,6 +5,7 @@ export class Container {
5
5
  resolvers = new Map();
6
6
  injectables = new Set();
7
7
  dependants = new Map();
8
+ disposing = false;
8
9
  constructor(application, scope = Scope.Global, parent) {
9
10
  this.application = application;
10
11
  this.scope = scope;
@@ -34,23 +35,16 @@ export class Container {
34
35
  }
35
36
  async dispose() {
36
37
  this.application.logger.trace("Disposing [%s] scope context...", this.scope);
37
- while (this.instances.size) {
38
- for (const injectable of this.instances.keys()) {
39
- const dependants = this.dependants.get(injectable);
40
- if (!dependants?.size) {
41
- await this.disposeInjectable(injectable);
42
- for (const dependants of this.dependants.values()) {
43
- if (dependants.has(injectable)) {
44
- dependants.delete(injectable);
45
- }
46
- }
47
- }
48
- }
38
+ this.disposing = true;
39
+ const disposalOrder = this.getDisposalOrder();
40
+ for (const injectable of disposalOrder) {
41
+ await this.disposeInjectable(injectable);
49
42
  }
50
43
  this.instances.clear();
51
44
  this.injectables.clear();
52
45
  this.resolvers.clear();
53
46
  this.dependants.clear();
47
+ this.disposing = false;
54
48
  }
55
49
  containsWithinSelf(injectable) {
56
50
  return this.instances.has(injectable) || this.resolvers.has(injectable);
@@ -107,6 +101,9 @@ export class Container {
107
101
  }
108
102
  }
109
103
  resolveInjectable(injectable, dependant) {
104
+ if (this.disposing) {
105
+ throw new Error("Cannot resolve during disposal");
106
+ }
110
107
  if (dependant && compareScope(dependant.scope, "<", injectable.scope)) {
111
108
  throw new Error("Invalid scope: dependant is looser than injectable");
112
109
  }
@@ -116,7 +113,7 @@ export class Container {
116
113
  return this.parent.resolveInjectable(injectable, dependant);
117
114
  } else {
118
115
  const { scope, dependencies, stack, label } = injectable;
119
- if (dependant && compareScope(scope, "=", dependant.scope)) {
116
+ if (dependant) {
120
117
  let dependants = this.dependants.get(injectable);
121
118
  if (!dependants) {
122
119
  this.dependants.set(injectable, dependants = new Set());
@@ -133,42 +130,81 @@ export class Container {
133
130
  const isOptional = isOptionalInjectable(injectable);
134
131
  if (isOptional) return Promise.resolve(undefined);
135
132
  return Promise.reject(new Error(`No instance provided for ${label || "an"} injectable:\n${stack}`));
136
- } else if (isFactoryInjectable(injectable)) {
137
- const resolution = this.createInjectableContext(dependencies, injectable).then((context) => Promise.resolve(injectable.factory(context)).then((instance) => ({
133
+ } else {
134
+ const resolution = this.createResolution(injectable, dependencies, scope);
135
+ if (scope !== Scope.Transient) {
136
+ this.resolvers.set(injectable, resolution);
137
+ }
138
+ return resolution;
139
+ }
140
+ }
141
+ }
142
+ }
143
+ async createResolution(injectable, dependencies, scope) {
144
+ try {
145
+ if (isFactoryInjectable(injectable)) {
146
+ const context = await this.createInjectableContext(dependencies, injectable);
147
+ const instance = await Promise.resolve(injectable.factory(context));
148
+ const picked = injectable.pick(instance);
149
+ if (compareScope(this.scope, ">=", scope)) {
150
+ this.instances.set(injectable, {
138
151
  instance,
152
+ picked,
139
153
  context
140
- }))).then(({ instance, context }) => {
141
- const picked = injectable.pick(instance);
142
- if (compareScope(this.scope, ">=", scope)) this.instances.set(injectable, {
143
- instance,
144
- picked,
145
- context
146
- });
147
- if (scope !== Scope.Transient) this.resolvers.delete(injectable);
148
- return picked;
149
154
  });
150
- if (scope !== Scope.Transient) this.resolvers.set(injectable, resolution);
151
- return resolution;
152
- } else if (isClassInjectable(injectable)) {
153
- const resolution = this.createInjectableContext(dependencies, injectable).then((context) => {
154
- const instance = new injectable(context);
155
- return instance.$onCreate().then(() => instance);
156
- }).then((instance) => {
157
- if (compareScope(this.scope, ">=", scope)) this.instances.set(injectable, {
158
- instance,
159
- picked: instance,
160
- context: undefined
161
- });
162
- if (scope !== Scope.Transient) this.resolvers.delete(injectable);
163
- return instance;
155
+ }
156
+ if (scope !== Scope.Transient) {
157
+ this.resolvers.delete(injectable);
158
+ }
159
+ return picked;
160
+ } else if (isClassInjectable(injectable)) {
161
+ const context = await this.createInjectableContext(dependencies, injectable);
162
+ const instance = new injectable(context);
163
+ await instance.$onCreate();
164
+ if (compareScope(this.scope, ">=", scope)) {
165
+ this.instances.set(injectable, {
166
+ instance,
167
+ picked: instance,
168
+ context: undefined
164
169
  });
165
- if (scope !== Scope.Transient) this.resolvers.set(injectable, resolution);
166
- return resolution;
167
- } else {
168
- throw new Error("Invalid injectable type");
169
170
  }
171
+ if (scope !== Scope.Transient) {
172
+ this.resolvers.delete(injectable);
173
+ }
174
+ return instance;
175
+ } else {
176
+ throw new Error("Invalid injectable type");
177
+ }
178
+ } catch (error) {
179
+ if (scope !== Scope.Transient) {
180
+ this.resolvers.delete(injectable);
181
+ }
182
+ this.instances.delete(injectable);
183
+ throw error;
184
+ }
185
+ }
186
+ getDisposalOrder() {
187
+ const visited = new Set();
188
+ const result = [];
189
+ const visit = (injectable) => {
190
+ if (visited.has(injectable)) return;
191
+ visited.add(injectable);
192
+ const dependants = this.dependants.get(injectable);
193
+ if (dependants) {
194
+ for (const dependant of dependants) {
195
+ if (this.instances.has(dependant)) {
196
+ visit(dependant);
197
+ }
198
+ }
199
+ }
200
+ if (this.instances.has(injectable)) {
201
+ result.push(injectable);
170
202
  }
203
+ };
204
+ for (const injectable of this.instances.keys()) {
205
+ visit(injectable);
171
206
  }
207
+ return result;
172
208
  }
173
209
  async disposeInjectable(injectable) {
174
210
  try {
@@ -192,7 +228,7 @@ export class Container {
192
228
  }
193
229
  export function createInjectFunction(container) {
194
230
  return (injectable, context) => {
195
- const dependencies = {};
231
+ const dependencies = { ...injectable.dependencies };
196
232
  for (const key in context) {
197
233
  const dep = context[key];
198
234
  if (isInjectable(dep) || isOptionalInjectable(dep)) {
@@ -1 +1 @@
1
- {"mappings":"AAAA,SAAS,aAAa,YAAY;AAClC,SAEE,iBACA,cACA,uBAGA,2BACA,mBACA,qBACA,cACA,kBACA,sBACA,yBAEK,kBAAkB;AAIzB,OAAO,MAAM,UAAU;CACrB,AAAS,YAAY,IAAI;CAIzB,AAAiB,YAAY,IAAI;CACjC,AAAiB,cAAc,IAAI;CACnC,AAAiB,aAAa,IAAI;CAElC,YACmBA,aAIDC,QAAyC,MAAM,QAC9CC,QACjB;OANiB;OAID;OACC;AAEjB,MAAK,UAAkB,MAAM,WAAW;AACtC,SAAM,IAAI,MAAM;EACjB;AACD,OAAK,QAAQ,gBAAgB,QAAQ,qBAAqB,KAAK,CAAC;CACjE;CAED,MAAM,OAAO;EACX,MAAM,WAAW,CAACC,iBAA+B;AAC/C,QAAK,MAAM,OAAO,cAAc;IAC9B,MAAM,aAAa,aAAa;IAChC,MAAM,aAAa,0BAA0B,WAAW;AACxD,SAAK,YAAY,IAAI,WAAW;AAChC,aAAS,WAAW,aAAa;GAClC;EACF;AAED,OAAK,MAAM,aAAa,KAAK,YAAY,SAAS,eAAe,EAAE;AACjE,YAAS,UAAU,aAAa;EACjC;EAED,MAAM,cAAc,MAAM,KAAK,KAAK,6BAA6B,CAAC;AAClE,QAAM,QAAQ,IAAI,YAAY,IAAI,CAAC,eAAe,KAAK,QAAQ,WAAW,CAAC,CAAC;CAC7E;CAED,KAAKF,OAAwC;AAC3C,SAAO,IAAI,UAAU,KAAK,aAAa,OAAO;CAC/C;CAED,MAAM,UAAU;AACd,OAAK,YAAY,OAAO,MAAM,mCAAmC,KAAK,MAAM;AAI5E,SAAO,KAAK,UAAU,MAAM;AAC1B,QAAK,MAAM,cAAc,KAAK,UAAU,MAAM,EAAE;IAC9C,MAAM,aAAa,KAAK,WAAW,IAAI,WAAW;AAElD,SAAK,YAAY,MAAM;AACrB,WAAM,KAAK,kBAAkB,WAAW;AACxC,UAAK,MAAM,cAAc,KAAK,WAAW,QAAQ,EAAE;AAEjD,UAAI,WAAW,IAAI,WAAW,EAAE;AAC9B,kBAAW,OAAO,WAAW;MAC9B;KACF;IACF;GACF;EACF;AAED,OAAK,UAAU,OAAO;AACtB,OAAK,YAAY,OAAO;AACxB,OAAK,UAAU,OAAO;AACtB,OAAK,WAAW,OAAO;CACxB;CAED,mBAAmBG,YAA2B;AAC5C,SAAO,KAAK,UAAU,IAAI,WAAW,IAAI,KAAK,UAAU,IAAI,WAAW;CACxE;CAED,SAASA,YAAoC;AAC3C,SACE,KAAK,mBAAmB,WAAW,KAClC,KAAK,QAAQ,SAAS,WAAW,IAAI;CAEzC;CAED,IAA6BC,YAAyC;AACpE,MAAI,KAAK,UAAU,IAAI,WAAW,EAAE;AAClC,UAAO,KAAK,UAAU,IAAI,WAAW,CAAE;EACxC;AAED,MAAI,KAAK,QAAQ,SAAS,WAAW,EAAE;AACrC,UAAO,KAAK,OAAO,IAAI,WAAW;EACnC;AAED,QAAM,IAAI,MAAM;CACjB;CAED,QAAiCA,YAAe;AAC9C,SAAO,KAAK,kBAAkB,WAAW;CAC1C;CAED,MAAM,cAAsCC,cAAiB;AAC3D,SAAO,KAAK,wBAAwB,aAAa;CAClD;CAED,MAAc,wBACZA,cACAC,WACA;EACA,MAAMC,aAAkC,CAAE;EAC1C,MAAM,OAAO,OAAO,QAAQ,aAAa;EACzC,MAAMC,YAA4B,MAAM,KAAK,OAAO;AACpD,OAAK,IAAI,IAAI,GAAG,IAAI,KAAK,QAAQ,KAAK;GACpC,MAAM,CAAC,KAAK,WAAW,GAAG,KAAK;GAC/B,MAAM,aAAa,0BAA0B,WAAW;GACxD,MAAM,WAAW,KAAK,kBAAkB,YAAY,UAAU;AAC9D,aAAU,KAAK,SAAS,KAAK,CAAC,UAAW,WAAW,OAAO,MAAO;EACnE;AACD,QAAM,QAAQ,IAAI,UAAU;AAC5B,SAAO,OAAO,OAAO,WAAW;CACjC;CAED,MAAM,QACJJ,YACAK,UACA;AACA,MAAI,aAAa,WAAW,OAAO,KAAK,KAAK,MAAM,EAAE;AACnD,SAAM,IAAI,MAAM;EACjB;AACD,OAAK,UAAU,IAAI,YAAY;GAC7B;GACA,QAAQ;GACR,SAAS;EACV,EAAC;CACH;CAED,UAAUN,YAA2B;AACnC,SAAO,aAAa,WAAW,OAAO,MAAM,KAAK,MAAM;CACxD;CAED,CAAS,8BAA8B;AACrC,OAAK,MAAM,cAAc,KAAK,aAAa;AACzC,OAAI,WAAW,UAAU,KAAK,OAAO;AACnC,UAAM;GACP;EACF;CACF;CAED,AAAQ,kBACNC,YACAE,WACmC;AACnC,MAAI,aAAa,aAAa,UAAU,OAAO,KAAK,WAAW,MAAM,EAAE;AAErE,SAAM,IAAI,MAAM;EACjB;AAED,MAAI,kBAAkB,WAAW,EAAE;AACjC,UAAO,QAAQ,QAAQ,WAAW,MAAM;EACzC,WACC,KAAK,QAAQ,SAAS,WAAW,IAChC,KAAK,QAAQ,UAAU,WAAW,IACjC,aAAa,KAAK,OAAO,OAAO,KAAK,KAAK,MAAM,EAClD;AACA,UAAO,KAAK,OAAO,kBAAkB,YAAY,UAAU;EAC5D,OAAM;GACL,MAAM,EAAE,OAAO,cAAc,OAAO,OAAO,GAAG;AAE9C,OAAI,aAAa,aAAa,OAAO,KAAK,UAAU,MAAM,EAAE;IAC1D,IAAI,aAAa,KAAK,WAAW,IAAI,WAAW;AAChD,SAAK,YAAY;AACf,UAAK,WAAW,IAAI,YAAa,aAAa,IAAI,MAAO;IAC1D;AACD,eAAW,IAAI,UAAU;GAC1B;AAED,OAAI,KAAK,UAAU,IAAI,WAAW,EAAE;AAClC,WAAO,QAAQ,QAAQ,KAAK,UAAU,IAAI,WAAW,CAAE,OAAO;GAC/D,WAAU,KAAK,UAAU,IAAI,WAAW,EAAE;AACzC,WAAO,KAAK,UAAU,IAAI,WAAW;GACtC,OAAM;IACL,MAAM,SAAS,iBAAiB,WAAW;AAE3C,QAAI,QAAQ;KACV,MAAM,aAAa,qBAAqB,WAAW;AACnD,SAAI,WAAY,QAAO,QAAQ,QAAQ,UAAiB;AACxD,YAAO,QAAQ,OACb,IAAI,OACD,2BAA2B,SAAS,KAAK,gBAAgB,MAAM,GAEnE;IACF,WAAU,oBAAoB,WAAW,EAAE;KAC1C,MAAM,aAAa,KAAK,wBACtB,cACA,WACD,CACE,KAAK,CAAC,YACL,QAAQ,QAAQ,WAAW,QAAQ,QAAQ,CAAC,CAAC,KAAK,CAAC,cAAc;MAC/D;MACA;KACD,GAAE,CACJ,CACA,KAAK,CAAC,EAAE,UAAU,SAAS,KAAK;MAC/B,MAAM,SAAS,WAAW,KAAK,SAAS;AACxC,UAAI,aAAa,KAAK,OAAO,MAAM,MAAM,CACvC,MAAK,UAAU,IAAI,YAAY;OAAE;OAAU;OAAQ;MAAS,EAAC;AAC/D,UAAI,UAAU,MAAM,UAAW,MAAK,UAAU,OAAO,WAAW;AAChE,aAAO;KACR,EAAC;AACJ,SAAI,UAAU,MAAM,UAClB,MAAK,UAAU,IAAI,YAAY,WAAW;AAC5C,YAAO;IACR,WAAU,kBAAkB,WAAW,EAAE;KACxC,MAAM,aAAa,KAAK,wBACtB,cACA,WACD,CACE,KAAK,CAAC,YAAY;MACjB,MAAM,WAAW,IAAI,WAAW;AAChC,aAAO,SAAS,WAAW,CAAC,KAAK,MAAM,SAAS;KACjD,EAAC,CACD,KAAK,CAAC,aAAa;AAElB,UAAI,aAAa,KAAK,OAAO,MAAM,MAAM,CACvC,MAAK,UAAU,IAAI,YAAY;OAC7B;OACA,QAAQ;OACR,SAAS;MACV,EAAC;AACJ,UAAI,UAAU,MAAM,UAAW,MAAK,UAAU,OAAO,WAAW;AAChE,aAAO;KACR,EAAC;AACJ,SAAI,UAAU,MAAM,UAClB,MAAK,UAAU,IAAI,YAAY,WAAW;AAC5C,YAAO;IACR,OAAM;AACL,WAAM,IAAI,MAAM;IACjB;GACF;EACF;CACF;CAED,MAAc,kBAAkBH,YAA2B;AACzD,MAAI;AACF,OAAI,oBAAoB,WAAW,EAAE;IACnC,MAAM,EAAE,SAAS,GAAG;AACpB,QAAI,SAAS;KACX,MAAM,EAAE,UAAU,SAAS,GAAG,KAAK,UAAU,IAAI,WAAW;AAC5D,WAAM,QAAQ,UAAU,QAAQ;IACjC;GACF,WAAU,kBAAkB,WAAW,EAAE;IACxC,MAAM,EAAE,UAAU,GAAG,KAAK,UAAU,IAAI,WAAW;AACnD,UAAM,SAAS,YAAY;GAC5B;EACF,SAAQ,OAAO;GACd,MAAM,QAAQ,IAAI,MAChB,oDACA,EAAE,MAAO;AAEX,QAAK,YAAY,OAAO,MAAM,MAAM;EACrC,UAAS;AACR,QAAK,UAAU,OAAO,WAAW;EAClC;CACF;AACF;AAED,OAAO,SAAS,qBAAqBO,WAAsB;AACzD,QAAO,CACLN,YACAO,YACG;EACH,MAAMT,eAA6B,CAAE;AAErC,OAAK,MAAM,OAAO,SAAS;GACzB,MAAM,MAAM,QAAQ;AACpB,OAAI,aAAa,IAAI,IAAI,qBAAqB,IAAI,EAAE;AAClD,iBAAa,OAAO;GACrB,OAAM;AACL,iBAAa,OAAO,sBAAsB,IAAI;GAC/C;EACF;EAED,MAAM,gBAAgB;GACpB,GAAG;GACH;GACA,OAAO,MAAM;EACd;AAED,SAAO,UAAU,QAAQ,cAAc;CACxC;AACF","names":["application: {\n registry: Registry\n logger: Logger\n }","scope: Exclude<Scope, Scope.Transient>","parent?: Container","dependencies: Dependencies","injectable: AnyInjectable","injectable: T","dependencies: T","dependant?: AnyInjectable","injections: Record<string, any>","resolvers: Promise<any>[]","instance: ResolveInjectableType<T>","container: Container","context: InlineInjectionDependencies<T>"],"sources":["../src/container.ts"],"sourcesContent":["import { Scope } from './enums.ts'\nimport {\n type AnyInjectable,\n CoreInjectables,\n compareScope,\n createValueInjectable,\n type Dependencies,\n type DependencyContext,\n getDepedencencyInjectable,\n isClassInjectable,\n isFactoryInjectable,\n isInjectable,\n isLazyInjectable,\n isOptionalInjectable,\n isValueInjectable,\n type ResolveInjectableType,\n} from './injectables.ts'\nimport type { Logger } from './logger.ts'\nimport type { Registry } from './registry.ts'\n\nexport class Container {\n readonly instances = new Map<\n AnyInjectable,\n { instance: any; picked?: any; context?: any }\n >()\n private readonly resolvers = new Map<AnyInjectable, Promise<any>>()\n private readonly injectables = new Set<AnyInjectable>()\n private readonly dependants = new Map<AnyInjectable, Set<AnyInjectable>>()\n\n constructor(\n private readonly application: {\n registry: Registry\n logger: Logger\n },\n public readonly scope: Exclude<Scope, Scope.Transient> = Scope.Global,\n private readonly parent?: Container,\n ) {\n if ((scope as any) === Scope.Transient) {\n throw new Error('Invalid scope')\n }\n this.provide(CoreInjectables.inject, createInjectFunction(this))\n }\n\n async load() {\n const traverse = (dependencies: Dependencies) => {\n for (const key in dependencies) {\n const dependency = dependencies[key]\n const injectable = getDepedencencyInjectable(dependency)\n this.injectables.add(injectable)\n traverse(injectable.dependencies)\n }\n }\n\n for (const dependant of this.application.registry.getDependants()) {\n traverse(dependant.dependencies)\n }\n\n const injectables = Array.from(this.findCurrentScopeInjectables())\n await Promise.all(injectables.map((injectable) => this.resolve(injectable)))\n }\n\n fork(scope: Exclude<Scope, Scope.Transient>) {\n return new Container(this.application, scope, this)\n }\n\n async dispose() {\n this.application.logger.trace('Disposing [%s] scope context...', this.scope)\n\n // Loop through all instances and dispose them,\n // until there are no more instances left\n while (this.instances.size) {\n for (const injectable of this.instances.keys()) {\n const dependants = this.dependants.get(injectable)\n // Firstly, dispose instances that other injectables don't depend on\n if (!dependants?.size) {\n await this.disposeInjectable(injectable)\n for (const dependants of this.dependants.values()) {\n // Clear current istances as a dependant for other injectables\n if (dependants.has(injectable)) {\n dependants.delete(injectable)\n }\n }\n }\n }\n }\n\n this.instances.clear()\n this.injectables.clear()\n this.resolvers.clear()\n this.dependants.clear()\n }\n\n containsWithinSelf(injectable: AnyInjectable) {\n return this.instances.has(injectable) || this.resolvers.has(injectable)\n }\n\n contains(injectable: AnyInjectable): boolean {\n return (\n this.containsWithinSelf(injectable) ||\n (this.parent?.contains(injectable) ?? false)\n )\n }\n\n get<T extends AnyInjectable>(injectable: T): ResolveInjectableType<T> {\n if (this.instances.has(injectable)) {\n return this.instances.get(injectable)!.instance\n }\n\n if (this.parent?.contains(injectable)) {\n return this.parent.get(injectable)\n }\n\n throw new Error('No instance found')\n }\n\n resolve<T extends AnyInjectable>(injectable: T) {\n return this.resolveInjectable(injectable)\n }\n\n async createContext<T extends Dependencies>(dependencies: T) {\n return this.createInjectableContext(dependencies)\n }\n\n private async createInjectableContext<T extends Dependencies>(\n dependencies: T,\n dependant?: AnyInjectable,\n ) {\n const injections: Record<string, any> = {}\n const deps = Object.entries(dependencies)\n const resolvers: Promise<any>[] = Array(deps.length)\n for (let i = 0; i < deps.length; i++) {\n const [key, dependency] = deps[i]\n const injectable = getDepedencencyInjectable(dependency)\n const resolver = this.resolveInjectable(injectable, dependant)\n resolvers[i] = resolver.then((value) => (injections[key] = value))\n }\n await Promise.all(resolvers)\n return Object.freeze(injections) as DependencyContext<T>\n }\n\n async provide<T extends AnyInjectable>(\n injectable: T,\n instance: ResolveInjectableType<T>,\n ) {\n if (compareScope(injectable.scope, '>', this.scope)) {\n throw new Error('Invalid scope') // TODO: more informative error\n }\n this.instances.set(injectable, {\n instance,\n picked: instance,\n context: undefined,\n })\n }\n\n satisfies(injectable: AnyInjectable) {\n return compareScope(injectable.scope, '<=', this.scope)\n }\n\n private *findCurrentScopeInjectables() {\n for (const injectable of this.injectables) {\n if (injectable.scope === this.scope) {\n yield injectable\n }\n }\n }\n\n private resolveInjectable<T extends AnyInjectable>(\n injectable: T,\n dependant?: AnyInjectable,\n ): Promise<ResolveInjectableType<T>> {\n if (dependant && compareScope(dependant.scope, '<', injectable.scope)) {\n // TODO: more informative error\n throw new Error('Invalid scope: dependant is looser than injectable')\n }\n\n if (isValueInjectable(injectable)) {\n return Promise.resolve(injectable.value)\n } else if (\n this.parent?.contains(injectable) ||\n (this.parent?.satisfies(injectable) &&\n compareScope(this.parent.scope, '<', this.scope))\n ) {\n return this.parent.resolveInjectable(injectable, dependant)\n } else {\n const { scope, dependencies, stack, label } = injectable\n\n if (dependant && compareScope(scope, '=', dependant.scope)) {\n let dependants = this.dependants.get(injectable)\n if (!dependants) {\n this.dependants.set(injectable, (dependants = new Set()))\n }\n dependants.add(dependant)\n }\n\n if (this.instances.has(injectable)) {\n return Promise.resolve(this.instances.get(injectable)!.picked)\n } else if (this.resolvers.has(injectable)) {\n return this.resolvers.get(injectable)!\n } else {\n const isLazy = isLazyInjectable(injectable)\n\n if (isLazy) {\n const isOptional = isOptionalInjectable(injectable)\n if (isOptional) return Promise.resolve(undefined as any)\n return Promise.reject(\n new Error(\n `No instance provided for ${label || 'an'} injectable:\\n${stack}`,\n ),\n )\n } else if (isFactoryInjectable(injectable)) {\n const resolution = this.createInjectableContext(\n dependencies,\n injectable,\n )\n .then((context) =>\n Promise.resolve(injectable.factory(context)).then((instance) => ({\n instance,\n context,\n })),\n )\n .then(({ instance, context }) => {\n const picked = injectable.pick(instance)\n if (compareScope(this.scope, '>=', scope))\n this.instances.set(injectable, { instance, picked, context })\n if (scope !== Scope.Transient) this.resolvers.delete(injectable)\n return picked\n })\n if (scope !== Scope.Transient)\n this.resolvers.set(injectable, resolution)\n return resolution\n } else if (isClassInjectable(injectable)) {\n const resolution = this.createInjectableContext(\n dependencies,\n injectable,\n )\n .then((context) => {\n const instance = new injectable(context)\n return instance.$onCreate().then(() => instance)\n })\n .then((instance) => {\n // const picked = injectable.pick(instance)\n if (compareScope(this.scope, '>=', scope))\n this.instances.set(injectable, {\n instance,\n picked: instance,\n context: undefined,\n })\n if (scope !== Scope.Transient) this.resolvers.delete(injectable)\n return instance\n })\n if (scope !== Scope.Transient)\n this.resolvers.set(injectable, resolution)\n return resolution\n } else {\n throw new Error('Invalid injectable type')\n }\n }\n }\n }\n\n private async disposeInjectable(injectable: AnyInjectable) {\n try {\n if (isFactoryInjectable(injectable)) {\n const { dispose } = injectable\n if (dispose) {\n const { instance, context } = this.instances.get(injectable)!\n await dispose(instance, context)\n }\n } else if (isClassInjectable(injectable)) {\n const { instance } = this.instances.get(injectable)!\n await instance.$onDispose()\n }\n } catch (cause) {\n const error = new Error(\n 'Injectable disposal error. Potential memory leak',\n { cause },\n )\n this.application.logger.error(error)\n } finally {\n this.instances.delete(injectable)\n }\n }\n}\n\nexport function createInjectFunction(container: Container) {\n return <T extends AnyInjectable>(\n injectable: T,\n context: InlineInjectionDependencies<T>,\n ) => {\n const dependencies: Dependencies = {}\n\n for (const key in context) {\n const dep = context[key]\n if (isInjectable(dep) || isOptionalInjectable(dep)) {\n dependencies[key] = dep\n } else {\n dependencies[key] = createValueInjectable(dep)\n }\n }\n\n const newInjectable = {\n ...injectable,\n dependencies,\n scope: Scope.Transient,\n }\n\n return container.resolve(newInjectable)\n }\n}\n\ntype InlineInjectionDependencies<T extends AnyInjectable> = {\n [K in keyof T['dependencies']]:\n | ResolveInjectableType<T['dependencies'][K]>\n | AnyInjectable<ResolveInjectableType<T['dependencies'][K]>>\n}\n\nexport type InjectFn = ReturnType<typeof createInjectFunction>\n"],"version":3,"file":"container.js"}
1
+ {"mappings":"AAAA,SAAS,aAAa,YAAY;AAClC,SAEE,iBACA,cACA,uBAGA,2BACA,mBACA,qBACA,cACA,kBACA,sBACA,yBAEK,kBAAkB;AAIzB,OAAO,MAAM,UAAU;CACrB,AAAS,YAAY,IAAI;CAIzB,AAAiB,YAAY,IAAI;CACjC,AAAiB,cAAc,IAAI;CACnC,AAAiB,aAAa,IAAI;CAClC,AAAQ,YAAY;CAEpB,YACmBA,aAIDC,QAAyC,MAAM,QAC9CC,QACjB;OANiB;OAID;OACC;AAEjB,MAAK,UAAkB,MAAM,WAAW;AACtC,SAAM,IAAI,MAAM;EACjB;AACD,OAAK,QAAQ,gBAAgB,QAAQ,qBAAqB,KAAK,CAAC;CACjE;CAED,MAAM,OAAO;EACX,MAAM,WAAW,CAACC,iBAA+B;AAC/C,QAAK,MAAM,OAAO,cAAc;IAC9B,MAAM,aAAa,aAAa;IAChC,MAAM,aAAa,0BAA0B,WAAW;AACxD,SAAK,YAAY,IAAI,WAAW;AAChC,aAAS,WAAW,aAAa;GAClC;EACF;AAED,OAAK,MAAM,aAAa,KAAK,YAAY,SAAS,eAAe,EAAE;AACjE,YAAS,UAAU,aAAa;EACjC;EAED,MAAM,cAAc,MAAM,KAAK,KAAK,6BAA6B,CAAC;AAClE,QAAM,QAAQ,IAAI,YAAY,IAAI,CAAC,eAAe,KAAK,QAAQ,WAAW,CAAC,CAAC;CAC7E;CAED,KAAKF,OAAwC;AAC3C,SAAO,IAAI,UAAU,KAAK,aAAa,OAAO;CAC/C;CAED,MAAM,UAAU;AACd,OAAK,YAAY,OAAO,MAAM,mCAAmC,KAAK,MAAM;AAG5E,OAAK,YAAY;EAGjB,MAAM,gBAAgB,KAAK,kBAAkB;AAG7C,OAAK,MAAM,cAAc,eAAe;AACtC,SAAM,KAAK,kBAAkB,WAAW;EACzC;AAED,OAAK,UAAU,OAAO;AACtB,OAAK,YAAY,OAAO;AACxB,OAAK,UAAU,OAAO;AACtB,OAAK,WAAW,OAAO;AAEvB,OAAK,YAAY;CAClB;CAED,mBAAmBG,YAA2B;AAC5C,SAAO,KAAK,UAAU,IAAI,WAAW,IAAI,KAAK,UAAU,IAAI,WAAW;CACxE;CAED,SAASA,YAAoC;AAC3C,SACE,KAAK,mBAAmB,WAAW,KAClC,KAAK,QAAQ,SAAS,WAAW,IAAI;CAEzC;CAED,IAA6BC,YAAyC;AACpE,MAAI,KAAK,UAAU,IAAI,WAAW,EAAE;AAClC,UAAO,KAAK,UAAU,IAAI,WAAW,CAAE;EACxC;AAED,MAAI,KAAK,QAAQ,SAAS,WAAW,EAAE;AACrC,UAAO,KAAK,OAAO,IAAI,WAAW;EACnC;AAED,QAAM,IAAI,MAAM;CACjB;CAED,QAAiCA,YAAe;AAC9C,SAAO,KAAK,kBAAkB,WAAW;CAC1C;CAED,MAAM,cAAsCC,cAAiB;AAC3D,SAAO,KAAK,wBAAwB,aAAa;CAClD;CAED,MAAc,wBACZA,cACAC,WACA;EACA,MAAMC,aAAkC,CAAE;EAC1C,MAAM,OAAO,OAAO,QAAQ,aAAa;EACzC,MAAMC,YAA4B,MAAM,KAAK,OAAO;AACpD,OAAK,IAAI,IAAI,GAAG,IAAI,KAAK,QAAQ,KAAK;GACpC,MAAM,CAAC,KAAK,WAAW,GAAG,KAAK;GAC/B,MAAM,aAAa,0BAA0B,WAAW;GACxD,MAAM,WAAW,KAAK,kBAAkB,YAAY,UAAU;AAC9D,aAAU,KAAK,SAAS,KAAK,CAAC,UAAW,WAAW,OAAO,MAAO;EACnE;AACD,QAAM,QAAQ,IAAI,UAAU;AAC5B,SAAO,OAAO,OAAO,WAAW;CACjC;CAED,MAAM,QACJJ,YACAK,UACA;AACA,MAAI,aAAa,WAAW,OAAO,KAAK,KAAK,MAAM,EAAE;AACnD,SAAM,IAAI,MAAM;EACjB;AACD,OAAK,UAAU,IAAI,YAAY;GAC7B;GACA,QAAQ;GACR,SAAS;EACV,EAAC;CACH;CAED,UAAUN,YAA2B;AACnC,SAAO,aAAa,WAAW,OAAO,MAAM,KAAK,MAAM;CACxD;CAED,CAAS,8BAA8B;AACrC,OAAK,MAAM,cAAc,KAAK,aAAa;AACzC,OAAI,WAAW,UAAU,KAAK,OAAO;AACnC,UAAM;GACP;EACF;CACF;CAED,AAAQ,kBACNC,YACAE,WACmC;AACnC,MAAI,KAAK,WAAW;AAClB,SAAM,IAAI,MAAM;EACjB;AAED,MAAI,aAAa,aAAa,UAAU,OAAO,KAAK,WAAW,MAAM,EAAE;AAErE,SAAM,IAAI,MAAM;EACjB;AAED,MAAI,kBAAkB,WAAW,EAAE;AACjC,UAAO,QAAQ,QAAQ,WAAW,MAAM;EACzC,WACC,KAAK,QAAQ,SAAS,WAAW,IAChC,KAAK,QAAQ,UAAU,WAAW,IACjC,aAAa,KAAK,OAAO,OAAO,KAAK,KAAK,MAAM,EAClD;AACA,UAAO,KAAK,OAAO,kBAAkB,YAAY,UAAU;EAC5D,OAAM;GACL,MAAM,EAAE,OAAO,cAAc,OAAO,OAAO,GAAG;AAE9C,OAAI,WAAW;IACb,IAAI,aAAa,KAAK,WAAW,IAAI,WAAW;AAChD,SAAK,YAAY;AACf,UAAK,WAAW,IAAI,YAAa,aAAa,IAAI,MAAO;IAC1D;AACD,eAAW,IAAI,UAAU;GAC1B;AAED,OAAI,KAAK,UAAU,IAAI,WAAW,EAAE;AAClC,WAAO,QAAQ,QAAQ,KAAK,UAAU,IAAI,WAAW,CAAE,OAAO;GAC/D,WAAU,KAAK,UAAU,IAAI,WAAW,EAAE;AACzC,WAAO,KAAK,UAAU,IAAI,WAAW;GACtC,OAAM;IACL,MAAM,SAAS,iBAAiB,WAAW;AAE3C,QAAI,QAAQ;KACV,MAAM,aAAa,qBAAqB,WAAW;AACnD,SAAI,WAAY,QAAO,QAAQ,QAAQ,UAAiB;AACxD,YAAO,QAAQ,OACb,IAAI,OACD,2BAA2B,SAAS,KAAK,gBAAgB,MAAM,GAEnE;IACF,OAAM;KACL,MAAM,aAAa,KAAK,iBACtB,YACA,cACA,MACD;AACD,SAAI,UAAU,MAAM,WAAW;AAC7B,WAAK,UAAU,IAAI,YAAY,WAAW;KAC3C;AACD,YAAO;IACR;GACF;EACF;CACF;CAED,MAAc,iBACZF,YACAF,cACAQ,OACmC;AACnC,MAAI;AACF,OAAI,oBAAoB,WAAW,EAAE;IACnC,MAAM,UAAU,MAAM,KAAK,wBACzB,cACA,WACD;IACD,MAAM,WAAW,MAAM,QAAQ,QAAQ,WAAW,QAAQ,QAAQ,CAAC;IACnE,MAAM,SAAS,WAAW,KAAK,SAAS;AAExC,QAAI,aAAa,KAAK,OAAO,MAAM,MAAM,EAAE;AACzC,UAAK,UAAU,IAAI,YAAY;MAAE;MAAU;MAAQ;KAAS,EAAC;IAC9D;AAED,QAAI,UAAU,MAAM,WAAW;AAC7B,UAAK,UAAU,OAAO,WAAW;IAClC;AAED,WAAO;GACR,WAAU,kBAAkB,WAAW,EAAE;IACxC,MAAM,UAAU,MAAM,KAAK,wBACzB,cACA,WACD;IACD,MAAM,WAAW,IAAI,WAAW;AAChC,UAAM,SAAS,WAAW;AAE1B,QAAI,aAAa,KAAK,OAAO,MAAM,MAAM,EAAE;AACzC,UAAK,UAAU,IAAI,YAAY;MAC7B;MACA,QAAQ;MACR,SAAS;KACV,EAAC;IACH;AAED,QAAI,UAAU,MAAM,WAAW;AAC7B,UAAK,UAAU,OAAO,WAAW;IAClC;AAED,WAAO;GACR,OAAM;AACL,UAAM,IAAI,MAAM;GACjB;EACF,SAAQ,OAAO;AAEd,OAAI,UAAU,MAAM,WAAW;AAC7B,SAAK,UAAU,OAAO,WAAW;GAClC;AACD,QAAK,UAAU,OAAO,WAAW;AACjC,SAAM;EACP;CACF;CAED,AAAQ,mBAAoC;EAC1C,MAAM,UAAU,IAAI;EACpB,MAAMC,SAA0B,CAAE;EAElC,MAAM,QAAQ,CAACR,eAA8B;AAC3C,OAAI,QAAQ,IAAI,WAAW,CAAE;AAC7B,WAAQ,IAAI,WAAW;GAEvB,MAAM,aAAa,KAAK,WAAW,IAAI,WAAW;AAClD,OAAI,YAAY;AACd,SAAK,MAAM,aAAa,YAAY;AAClC,SAAI,KAAK,UAAU,IAAI,UAAU,EAAE;AACjC,YAAM,UAAU;KACjB;IACF;GACF;AAGD,OAAI,KAAK,UAAU,IAAI,WAAW,EAAE;AAClC,WAAO,KAAK,WAAW;GACxB;EACF;AAED,OAAK,MAAM,cAAc,KAAK,UAAU,MAAM,EAAE;AAC9C,SAAM,WAAW;EAClB;AAED,SAAO;CACR;CAED,MAAc,kBAAkBA,YAA2B;AACzD,MAAI;AACF,OAAI,oBAAoB,WAAW,EAAE;IACnC,MAAM,EAAE,SAAS,GAAG;AACpB,QAAI,SAAS;KACX,MAAM,EAAE,UAAU,SAAS,GAAG,KAAK,UAAU,IAAI,WAAW;AAC5D,WAAM,QAAQ,UAAU,QAAQ;IACjC;GACF,WAAU,kBAAkB,WAAW,EAAE;IACxC,MAAM,EAAE,UAAU,GAAG,KAAK,UAAU,IAAI,WAAW;AACnD,UAAM,SAAS,YAAY;GAC5B;EACF,SAAQ,OAAO;GACd,MAAM,QAAQ,IAAI,MAChB,oDACA,EAAE,MAAO;AAEX,QAAK,YAAY,OAAO,MAAM,MAAM;EACrC,UAAS;AACR,QAAK,UAAU,OAAO,WAAW;EAClC;CACF;AACF;AAED,OAAO,SAAS,qBAAqBS,WAAsB;AACzD,QAAO,CACLR,YACAS,YACG;EACH,MAAMX,eAA6B,EACjC,GAAG,WAAW,aACf;AAED,OAAK,MAAM,OAAO,SAAS;GACzB,MAAM,MAAM,QAAQ;AACpB,OAAI,aAAa,IAAI,IAAI,qBAAqB,IAAI,EAAE;AAClD,iBAAa,OAAO;GACrB,OAAM;AACL,iBAAa,OAAO,sBAAsB,IAAI;GAC/C;EACF;EAED,MAAM,gBAAgB;GACpB,GAAG;GACH;GACA,OAAO,MAAM;EACd;AAED,SAAO,UAAU,QAAQ,cAAc;CACxC;AACF","names":["application: {\n registry: Registry\n logger: Logger\n }","scope: Exclude<Scope, Scope.Transient>","parent?: Container","dependencies: Dependencies","injectable: AnyInjectable","injectable: T","dependencies: T","dependant?: AnyInjectable","injections: Record<string, any>","resolvers: Promise<any>[]","instance: ResolveInjectableType<T>","scope: Scope","result: AnyInjectable[]","container: Container","context: InlineInjectionDependencies<T>"],"sources":["../src/container.ts"],"sourcesContent":["import { Scope } from './enums.ts'\nimport {\n type AnyInjectable,\n CoreInjectables,\n compareScope,\n createValueInjectable,\n type Dependencies,\n type DependencyContext,\n getDepedencencyInjectable,\n isClassInjectable,\n isFactoryInjectable,\n isInjectable,\n isLazyInjectable,\n isOptionalInjectable,\n isValueInjectable,\n type ResolveInjectableType,\n} from './injectables.ts'\nimport type { Logger } from './logger.ts'\nimport type { Registry } from './registry.ts'\n\nexport class Container {\n readonly instances = new Map<\n AnyInjectable,\n { instance: any; picked?: any; context?: any }\n >()\n private readonly resolvers = new Map<AnyInjectable, Promise<any>>()\n private readonly injectables = new Set<AnyInjectable>()\n private readonly dependants = new Map<AnyInjectable, Set<AnyInjectable>>()\n private disposing = false\n\n constructor(\n private readonly application: {\n registry: Registry\n logger: Logger\n },\n public readonly scope: Exclude<Scope, Scope.Transient> = Scope.Global,\n private readonly parent?: Container,\n ) {\n if ((scope as any) === Scope.Transient) {\n throw new Error('Invalid scope')\n }\n this.provide(CoreInjectables.inject, createInjectFunction(this))\n }\n\n async load() {\n const traverse = (dependencies: Dependencies) => {\n for (const key in dependencies) {\n const dependency = dependencies[key]\n const injectable = getDepedencencyInjectable(dependency)\n this.injectables.add(injectable)\n traverse(injectable.dependencies)\n }\n }\n\n for (const dependant of this.application.registry.getDependants()) {\n traverse(dependant.dependencies)\n }\n\n const injectables = Array.from(this.findCurrentScopeInjectables())\n await Promise.all(injectables.map((injectable) => this.resolve(injectable)))\n }\n\n fork(scope: Exclude<Scope, Scope.Transient>) {\n return new Container(this.application, scope, this)\n }\n\n async dispose() {\n this.application.logger.trace('Disposing [%s] scope context...', this.scope)\n\n // Prevent new resolutions during disposal\n this.disposing = true\n\n // Get proper disposal order using topological sort\n const disposalOrder = this.getDisposalOrder()\n\n // Dispose in the correct order\n for (const injectable of disposalOrder) {\n await this.disposeInjectable(injectable)\n }\n\n this.instances.clear()\n this.injectables.clear()\n this.resolvers.clear()\n this.dependants.clear()\n\n this.disposing = false\n }\n\n containsWithinSelf(injectable: AnyInjectable) {\n return this.instances.has(injectable) || this.resolvers.has(injectable)\n }\n\n contains(injectable: AnyInjectable): boolean {\n return (\n this.containsWithinSelf(injectable) ||\n (this.parent?.contains(injectable) ?? false)\n )\n }\n\n get<T extends AnyInjectable>(injectable: T): ResolveInjectableType<T> {\n if (this.instances.has(injectable)) {\n return this.instances.get(injectable)!.instance\n }\n\n if (this.parent?.contains(injectable)) {\n return this.parent.get(injectable)\n }\n\n throw new Error('No instance found')\n }\n\n resolve<T extends AnyInjectable>(injectable: T) {\n return this.resolveInjectable(injectable)\n }\n\n async createContext<T extends Dependencies>(dependencies: T) {\n return this.createInjectableContext(dependencies)\n }\n\n private async createInjectableContext<T extends Dependencies>(\n dependencies: T,\n dependant?: AnyInjectable,\n ) {\n const injections: Record<string, any> = {}\n const deps = Object.entries(dependencies)\n const resolvers: Promise<any>[] = Array(deps.length)\n for (let i = 0; i < deps.length; i++) {\n const [key, dependency] = deps[i]\n const injectable = getDepedencencyInjectable(dependency)\n const resolver = this.resolveInjectable(injectable, dependant)\n resolvers[i] = resolver.then((value) => (injections[key] = value))\n }\n await Promise.all(resolvers)\n return Object.freeze(injections) as DependencyContext<T>\n }\n\n async provide<T extends AnyInjectable>(\n injectable: T,\n instance: ResolveInjectableType<T>,\n ) {\n if (compareScope(injectable.scope, '>', this.scope)) {\n throw new Error('Invalid scope') // TODO: more informative error\n }\n this.instances.set(injectable, {\n instance,\n picked: instance,\n context: undefined,\n })\n }\n\n satisfies(injectable: AnyInjectable) {\n return compareScope(injectable.scope, '<=', this.scope)\n }\n\n private *findCurrentScopeInjectables() {\n for (const injectable of this.injectables) {\n if (injectable.scope === this.scope) {\n yield injectable\n }\n }\n }\n\n private resolveInjectable<T extends AnyInjectable>(\n injectable: T,\n dependant?: AnyInjectable,\n ): Promise<ResolveInjectableType<T>> {\n if (this.disposing) {\n throw new Error('Cannot resolve during disposal')\n }\n\n if (dependant && compareScope(dependant.scope, '<', injectable.scope)) {\n // TODO: more informative error\n throw new Error('Invalid scope: dependant is looser than injectable')\n }\n\n if (isValueInjectable(injectable)) {\n return Promise.resolve(injectable.value)\n } else if (\n this.parent?.contains(injectable) ||\n (this.parent?.satisfies(injectable) &&\n compareScope(this.parent.scope, '<', this.scope))\n ) {\n return this.parent.resolveInjectable(injectable, dependant)\n } else {\n const { scope, dependencies, stack, label } = injectable\n\n if (dependant) {\n let dependants = this.dependants.get(injectable)\n if (!dependants) {\n this.dependants.set(injectable, (dependants = new Set()))\n }\n dependants.add(dependant)\n }\n\n if (this.instances.has(injectable)) {\n return Promise.resolve(this.instances.get(injectable)!.picked)\n } else if (this.resolvers.has(injectable)) {\n return this.resolvers.get(injectable)!\n } else {\n const isLazy = isLazyInjectable(injectable)\n\n if (isLazy) {\n const isOptional = isOptionalInjectable(injectable)\n if (isOptional) return Promise.resolve(undefined as any)\n return Promise.reject(\n new Error(\n `No instance provided for ${label || 'an'} injectable:\\n${stack}`,\n ),\n )\n } else {\n const resolution = this.createResolution(\n injectable,\n dependencies,\n scope,\n )\n if (scope !== Scope.Transient) {\n this.resolvers.set(injectable, resolution)\n }\n return resolution\n }\n }\n }\n }\n\n private async createResolution<T extends AnyInjectable>(\n injectable: T,\n dependencies: Dependencies,\n scope: Scope,\n ): Promise<ResolveInjectableType<T>> {\n try {\n if (isFactoryInjectable(injectable)) {\n const context = await this.createInjectableContext(\n dependencies,\n injectable,\n )\n const instance = await Promise.resolve(injectable.factory(context))\n const picked = injectable.pick(instance)\n\n if (compareScope(this.scope, '>=', scope)) {\n this.instances.set(injectable, { instance, picked, context })\n }\n\n if (scope !== Scope.Transient) {\n this.resolvers.delete(injectable)\n }\n\n return picked\n } else if (isClassInjectable(injectable)) {\n const context = await this.createInjectableContext(\n dependencies,\n injectable,\n )\n const instance = new injectable(context)\n await instance.$onCreate()\n\n if (compareScope(this.scope, '>=', scope)) {\n this.instances.set(injectable, {\n instance,\n picked: instance,\n context: undefined,\n })\n }\n\n if (scope !== Scope.Transient) {\n this.resolvers.delete(injectable)\n }\n\n return instance\n } else {\n throw new Error('Invalid injectable type')\n }\n } catch (error) {\n // Clean up on failure to prevent memory leaks\n if (scope !== Scope.Transient) {\n this.resolvers.delete(injectable)\n }\n this.instances.delete(injectable)\n throw error\n }\n }\n\n private getDisposalOrder(): AnyInjectable[] {\n const visited = new Set<AnyInjectable>()\n const result: AnyInjectable[] = []\n\n const visit = (injectable: AnyInjectable) => {\n if (visited.has(injectable)) return\n visited.add(injectable)\n\n const dependants = this.dependants.get(injectable)\n if (dependants) {\n for (const dependant of dependants) {\n if (this.instances.has(dependant)) {\n visit(dependant)\n }\n }\n }\n\n // Only add to result if this container owns the instance\n if (this.instances.has(injectable)) {\n result.push(injectable)\n }\n }\n\n for (const injectable of this.instances.keys()) {\n visit(injectable)\n }\n\n return result\n }\n\n private async disposeInjectable(injectable: AnyInjectable) {\n try {\n if (isFactoryInjectable(injectable)) {\n const { dispose } = injectable\n if (dispose) {\n const { instance, context } = this.instances.get(injectable)!\n await dispose(instance, context)\n }\n } else if (isClassInjectable(injectable)) {\n const { instance } = this.instances.get(injectable)!\n await instance.$onDispose()\n }\n } catch (cause) {\n const error = new Error(\n 'Injectable disposal error. Potential memory leak',\n { cause },\n )\n this.application.logger.error(error)\n } finally {\n this.instances.delete(injectable)\n }\n }\n}\n\nexport function createInjectFunction(container: Container) {\n return <T extends AnyInjectable>(\n injectable: T,\n context: InlineInjectionDependencies<T>,\n ) => {\n const dependencies: Dependencies = {\n ...injectable.dependencies,\n }\n\n for (const key in context) {\n const dep = context[key]\n if (isInjectable(dep) || isOptionalInjectable(dep)) {\n dependencies[key] = dep\n } else {\n dependencies[key] = createValueInjectable(dep)\n }\n }\n\n const newInjectable = {\n ...injectable,\n dependencies,\n scope: Scope.Transient,\n }\n\n return container.resolve(newInjectable)\n }\n}\n\ntype InlineInjectionDependencies<T extends AnyInjectable> = {\n [K in keyof T['dependencies']]?:\n | ResolveInjectableType<T['dependencies'][K]>\n | AnyInjectable<ResolveInjectableType<T['dependencies'][K]>>\n}\n\nexport type InjectFn = ReturnType<typeof createInjectFunction>\n"],"version":3,"file":"container.js"}
package/package.json CHANGED
@@ -10,15 +10,15 @@
10
10
  "peerDependencies": {
11
11
  "pino": "^9.6.0",
12
12
  "pino-pretty": "^13.0.0",
13
- "@nmtjs/common": "0.10.3",
14
- "@nmtjs/type": "0.10.3"
13
+ "@nmtjs/type": "0.10.5",
14
+ "@nmtjs/common": "0.10.5"
15
15
  },
16
16
  "devDependencies": {
17
17
  "@types/node": "^18",
18
18
  "pino": "^9.6.0",
19
19
  "pino-pretty": "^13.0.0",
20
- "@nmtjs/type": "0.10.3",
21
- "@nmtjs/common": "0.10.3"
20
+ "@nmtjs/common": "0.10.5",
21
+ "@nmtjs/type": "0.10.5"
22
22
  },
23
23
  "files": [
24
24
  "src",
@@ -26,7 +26,7 @@
26
26
  "LICENSE.md",
27
27
  "README.md"
28
28
  ],
29
- "version": "0.10.3",
29
+ "version": "0.10.5",
30
30
  "scripts": {
31
31
  "build": "neemata-build --root=./src './**/*.ts'",
32
32
  "type-check": "tsc --noEmit"
package/src/container.ts CHANGED
@@ -26,6 +26,7 @@ export class Container {
26
26
  private readonly resolvers = new Map<AnyInjectable, Promise<any>>()
27
27
  private readonly injectables = new Set<AnyInjectable>()
28
28
  private readonly dependants = new Map<AnyInjectable, Set<AnyInjectable>>()
29
+ private disposing = false
29
30
 
30
31
  constructor(
31
32
  private readonly application: {
@@ -66,28 +67,23 @@ export class Container {
66
67
  async dispose() {
67
68
  this.application.logger.trace('Disposing [%s] scope context...', this.scope)
68
69
 
69
- // Loop through all instances and dispose them,
70
- // until there are no more instances left
71
- while (this.instances.size) {
72
- for (const injectable of this.instances.keys()) {
73
- const dependants = this.dependants.get(injectable)
74
- // Firstly, dispose instances that other injectables don't depend on
75
- if (!dependants?.size) {
76
- await this.disposeInjectable(injectable)
77
- for (const dependants of this.dependants.values()) {
78
- // Clear current istances as a dependant for other injectables
79
- if (dependants.has(injectable)) {
80
- dependants.delete(injectable)
81
- }
82
- }
83
- }
84
- }
70
+ // Prevent new resolutions during disposal
71
+ this.disposing = true
72
+
73
+ // Get proper disposal order using topological sort
74
+ const disposalOrder = this.getDisposalOrder()
75
+
76
+ // Dispose in the correct order
77
+ for (const injectable of disposalOrder) {
78
+ await this.disposeInjectable(injectable)
85
79
  }
86
80
 
87
81
  this.instances.clear()
88
82
  this.injectables.clear()
89
83
  this.resolvers.clear()
90
84
  this.dependants.clear()
85
+
86
+ this.disposing = false
91
87
  }
92
88
 
93
89
  containsWithinSelf(injectable: AnyInjectable) {
@@ -168,6 +164,10 @@ export class Container {
168
164
  injectable: T,
169
165
  dependant?: AnyInjectable,
170
166
  ): Promise<ResolveInjectableType<T>> {
167
+ if (this.disposing) {
168
+ throw new Error('Cannot resolve during disposal')
169
+ }
170
+
171
171
  if (dependant && compareScope(dependant.scope, '<', injectable.scope)) {
172
172
  // TODO: more informative error
173
173
  throw new Error('Invalid scope: dependant is looser than injectable')
@@ -184,7 +184,7 @@ export class Container {
184
184
  } else {
185
185
  const { scope, dependencies, stack, label } = injectable
186
186
 
187
- if (dependant && compareScope(scope, '=', dependant.scope)) {
187
+ if (dependant) {
188
188
  let dependants = this.dependants.get(injectable)
189
189
  if (!dependants) {
190
190
  this.dependants.set(injectable, (dependants = new Set()))
@@ -207,57 +207,108 @@ export class Container {
207
207
  `No instance provided for ${label || 'an'} injectable:\n${stack}`,
208
208
  ),
209
209
  )
210
- } else if (isFactoryInjectable(injectable)) {
211
- const resolution = this.createInjectableContext(
212
- dependencies,
210
+ } else {
211
+ const resolution = this.createResolution(
213
212
  injectable,
214
- )
215
- .then((context) =>
216
- Promise.resolve(injectable.factory(context)).then((instance) => ({
217
- instance,
218
- context,
219
- })),
220
- )
221
- .then(({ instance, context }) => {
222
- const picked = injectable.pick(instance)
223
- if (compareScope(this.scope, '>=', scope))
224
- this.instances.set(injectable, { instance, picked, context })
225
- if (scope !== Scope.Transient) this.resolvers.delete(injectable)
226
- return picked
227
- })
228
- if (scope !== Scope.Transient)
229
- this.resolvers.set(injectable, resolution)
230
- return resolution
231
- } else if (isClassInjectable(injectable)) {
232
- const resolution = this.createInjectableContext(
233
213
  dependencies,
234
- injectable,
214
+ scope,
235
215
  )
236
- .then((context) => {
237
- const instance = new injectable(context)
238
- return instance.$onCreate().then(() => instance)
239
- })
240
- .then((instance) => {
241
- // const picked = injectable.pick(instance)
242
- if (compareScope(this.scope, '>=', scope))
243
- this.instances.set(injectable, {
244
- instance,
245
- picked: instance,
246
- context: undefined,
247
- })
248
- if (scope !== Scope.Transient) this.resolvers.delete(injectable)
249
- return instance
250
- })
251
- if (scope !== Scope.Transient)
216
+ if (scope !== Scope.Transient) {
252
217
  this.resolvers.set(injectable, resolution)
218
+ }
253
219
  return resolution
254
- } else {
255
- throw new Error('Invalid injectable type')
256
220
  }
257
221
  }
258
222
  }
259
223
  }
260
224
 
225
+ private async createResolution<T extends AnyInjectable>(
226
+ injectable: T,
227
+ dependencies: Dependencies,
228
+ scope: Scope,
229
+ ): Promise<ResolveInjectableType<T>> {
230
+ try {
231
+ if (isFactoryInjectable(injectable)) {
232
+ const context = await this.createInjectableContext(
233
+ dependencies,
234
+ injectable,
235
+ )
236
+ const instance = await Promise.resolve(injectable.factory(context))
237
+ const picked = injectable.pick(instance)
238
+
239
+ if (compareScope(this.scope, '>=', scope)) {
240
+ this.instances.set(injectable, { instance, picked, context })
241
+ }
242
+
243
+ if (scope !== Scope.Transient) {
244
+ this.resolvers.delete(injectable)
245
+ }
246
+
247
+ return picked
248
+ } else if (isClassInjectable(injectable)) {
249
+ const context = await this.createInjectableContext(
250
+ dependencies,
251
+ injectable,
252
+ )
253
+ const instance = new injectable(context)
254
+ await instance.$onCreate()
255
+
256
+ if (compareScope(this.scope, '>=', scope)) {
257
+ this.instances.set(injectable, {
258
+ instance,
259
+ picked: instance,
260
+ context: undefined,
261
+ })
262
+ }
263
+
264
+ if (scope !== Scope.Transient) {
265
+ this.resolvers.delete(injectable)
266
+ }
267
+
268
+ return instance
269
+ } else {
270
+ throw new Error('Invalid injectable type')
271
+ }
272
+ } catch (error) {
273
+ // Clean up on failure to prevent memory leaks
274
+ if (scope !== Scope.Transient) {
275
+ this.resolvers.delete(injectable)
276
+ }
277
+ this.instances.delete(injectable)
278
+ throw error
279
+ }
280
+ }
281
+
282
+ private getDisposalOrder(): AnyInjectable[] {
283
+ const visited = new Set<AnyInjectable>()
284
+ const result: AnyInjectable[] = []
285
+
286
+ const visit = (injectable: AnyInjectable) => {
287
+ if (visited.has(injectable)) return
288
+ visited.add(injectable)
289
+
290
+ const dependants = this.dependants.get(injectable)
291
+ if (dependants) {
292
+ for (const dependant of dependants) {
293
+ if (this.instances.has(dependant)) {
294
+ visit(dependant)
295
+ }
296
+ }
297
+ }
298
+
299
+ // Only add to result if this container owns the instance
300
+ if (this.instances.has(injectable)) {
301
+ result.push(injectable)
302
+ }
303
+ }
304
+
305
+ for (const injectable of this.instances.keys()) {
306
+ visit(injectable)
307
+ }
308
+
309
+ return result
310
+ }
311
+
261
312
  private async disposeInjectable(injectable: AnyInjectable) {
262
313
  try {
263
314
  if (isFactoryInjectable(injectable)) {
@@ -287,7 +338,9 @@ export function createInjectFunction(container: Container) {
287
338
  injectable: T,
288
339
  context: InlineInjectionDependencies<T>,
289
340
  ) => {
290
- const dependencies: Dependencies = {}
341
+ const dependencies: Dependencies = {
342
+ ...injectable.dependencies,
343
+ }
291
344
 
292
345
  for (const key in context) {
293
346
  const dep = context[key]
@@ -309,7 +362,7 @@ export function createInjectFunction(container: Container) {
309
362
  }
310
363
 
311
364
  type InlineInjectionDependencies<T extends AnyInjectable> = {
312
- [K in keyof T['dependencies']]:
365
+ [K in keyof T['dependencies']]?:
313
366
  | ResolveInjectableType<T['dependencies'][K]>
314
367
  | AnyInjectable<ResolveInjectableType<T['dependencies'][K]>>
315
368
  }