Added support for postInit

Author: ackava

src/di/di.ts

@@ -18,28 +18,9 @@ const serviceProvider = Symbol("serviceProvider");
 
 const globalServiceProvider = Symbol("globalInstance");
 
-let newServiceTarget = null;
-
-const patched = Symbol("serviceConstructorPatched");
-
-const patchClass = (type) => {
-    if (type[patched]) {
-        return;
-    }
-    const parent = Object.getPrototypeOf(type);
-    if (parent === null || parent === Object || parent === Object.prototype) {
-        // eslint-disable-next-line @typescript-eslint/ban-types
-        const old = type.constructor as Function;
-        type.constructor = function (...a) {
-            this[serviceProvider] = newServiceTarget;
-            return old.apply(this, a);
-        };
-        type[patched] = true;
-        return;
-    }
-    patchClass(parent);
-    type[patched] = true;
-};
+export abstract class ServiceObject {
+    abstract postInit();
+}
 
 export class ServiceProvider implements IDisposable {
 
@@ -190,14 +171,10 @@ export class ServiceProvider implements IDisposable {
     }
 
     private createFromType(type, key = type): any {
-        const old = newServiceTarget;
-        newServiceTarget = this;
         const injectTypes = type[injectServiceTypesSymbol] as any[];
         const injectServices = injectTypes
             ? injectTypes.map((x) => this.resolve(x))
             : [];
-        patchClass(type.prototype);
-        type = type.prototype?.constructor ?? type;
         const instance = new type(... injectServices);
         this.map.set(key, instance);
         instance[serviceProvider] = this;
@@ -207,7 +184,9 @@ export class ServiceProvider implements IDisposable {
         }
         // initialize properties...
         this.resolveProperties(instance, type);
-        newServiceTarget = old;
+        if (instance instanceof ServiceObject) {
+            instance.postInit()?.catch(console.error);
+        }
         return instance;
     }
 
@@ -294,7 +273,7 @@ export default function Inject(target, key, index?: number): any {
     (target[injectServiceKeysSymbol] ??= {})[key] = pType;
     const descriptor = {
         get() {
-            const result = ServiceProvider.resolve(newServiceTarget ?? this, pType);
+            const result = ServiceProvider.resolve(this, pType);
             // get is compatible with AtomWatcher
             // as it will ignore getter and it will
             // not try to set a binding refresher