- register itself to container

jwdeveloper · Jun 5, 2024 · 9a927ee · 9a927ee
1 parent 18a8da9
commit 9a927ee
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/...main/java/io/github/jwdeveloper/dependance/implementation/DependanceContainerBuilder.java b/...main/java/io/github/jwdeveloper/dependance/implementation/DependanceContainerBuilder.java
@@ -45,7 +45,7 @@ public class DependanceContainerBuilder extends ContainerBuilderImpl<DependanceC
 
     private final DecoratorBuilder decoratorBuilder;
     private final JarScannerOptions options;
-    private List<JarScanner> additionalScanners = new ArrayList<>();
+    private final List<JarScanner> additionalScanners = new ArrayList<>();
     private boolean scanEnabled;
 
     @Getter
@@ -109,10 +109,12 @@ public DependanceContainer build() {
         }
         var scanner = new InjectionsScanner(this, options, jarScanner);
         var toInitialize = scanner.scanAndRegister();
+
+        registerSingleton(DependanceContainer.class, DepenDanceContainerImpl::new);
         var container = super.build();
         for (var clazz : toInitialize) {
             container.find(clazz);
         }
-        return new DepenDanceContainerImpl(container);
+        return (DependanceContainer) container.find(DependanceContainer.class);
     }
 }
diff --git a/...io/github/jwdeveloper/dependance/injector/implementation/containers/DefaultContainer.java b/...io/github/jwdeveloper/dependance/injector/implementation/containers/DefaultContainer.java
@@ -2,6 +2,8 @@
 
 import io.github.jwdeveloper.dependance.injector.api.containers.Container;
 import io.github.jwdeveloper.dependance.injector.api.containers.Registrable;
+import io.github.jwdeveloper.dependance.injector.api.enums.LifeTime;
+import io.github.jwdeveloper.dependance.injector.api.enums.RegistrationType;
 import io.github.jwdeveloper.dependance.injector.api.events.EventHandler;
 import io.github.jwdeveloper.dependance.injector.api.events.events.OnInjectionEvent;
 import io.github.jwdeveloper.dependance.injector.api.events.events.OnRegistrationEvent;
@@ -37,7 +39,10 @@ public DefaultContainer(
         this.logger = logger;
         this.injectionInfoFactory = injectionInfoFactory;
         this.injections = new HashMap<>();
-
+        register(new RegistrationInfo(Container.class, this.getClass(), container ->
+        {
+            return this;
+        }, LifeTime.SINGLETON, RegistrationType.InterfaceAndProvider));
         for (var registration : registrationInfos) {
             register(registration);
         }