Past container to PostInjection

package.json

@@ -1,7 +1,7 @@
 {
     "name": "@technove/inject",
     "description": "Dependency injection for TypeScript",
-    "version": "0.1.11",
+    "version": "0.1.12",
     "author": "PaulBGD",
     "license": "MIT",
     "scripts": {

src/container.spec.ts

@@ -212,13 +212,13 @@ describe("Container", () => {
             private dep!: Dependency;
 
             @PostInjection()
-            private afterInit() {
-                retrievedValue = this.dep.val;
+            private afterInit(container: Container) {
+                retrievedValue = container.get(Dependency).val + this.dep.val;
             }
         }
 
         container.get(B);
-        expect(retrievedValue).to.be.equal(5);
+        expect(retrievedValue).to.be.equal(10);
     });
 
     it("handles after initialization with promise", async () => {

src/container.ts

@@ -132,7 +132,7 @@ export class Container {
                 let anyPromises = false;
 
                 const mapped = injectData.postInjection.map((func) => {
-                    const val = func.call(service);
+                    const val = func.call(service, this);
                     anyPromises ||= val instanceof Promise;
                     return val;
                 });

src/decorators/post-injection.ts

@@ -14,8 +14,8 @@ export const PostInjection: () => MethodDecorator =
             throw new Error("Expected function");
         }
 
-        if (func.length !== 0) {
-            throw new Error("Expected function to have 0 parameters");
+        if (func.length !== 0 && func.length !== 1) {
+            throw new Error("Expected function to have 0-1 parameters");
         }
 
         data.postInjection.push(func);

src/injector.ts

@@ -1,9 +1,10 @@
 import { ParameterProvider, Provider } from "./types";
+import { Container } from "./container";
 
 export interface InjectedData {
     properties: Provider[];
     parameters: ParameterProvider<unknown>[];
-    postInjection: (() => Promise<void> | unknown)[];
+    postInjection: ((container: Container) => Promise<void> | unknown)[];
 }
 
 const INJECT_KEY = "__inject__";