feat: allow to inject container itself

Author: 4strodev

pkg/container/container.go

@@ -16,12 +16,21 @@ type Container struct {
 	connected  bool
 }
 
+// Retuns a new container and sets a default dependency that allows
+// to inject this container as a parameter on the resolvers
 func New() *Container {
-	return &Container{
+	container := &Container{
 		Graph:      graph.NewGraph[resolver.DependencyResolver[any]](),
 		typeIndex:  make(map[reflect.Type]*graph.Node[resolver.DependencyResolver[any]]),
 		tokenIndex: make(map[string]*graph.Node[resolver.DependencyResolver[any]]),
 	}
+
+	// Allow resolvers to inject container
+	container.AddDependency(func() *Container {
+		return container
+	})
+
+	return container
 }
 
 func (c *Container) AddDependency(res any) error {

pkg/container/container_test.go

@@ -99,3 +99,25 @@ func TestResolveToken_WithDependencies(t *testing.T) {
 	require.NoError(t, err)
 	require.NotNil(t, buf)
 }
+
+func TestContainerDependency(t *testing.T) {
+	cont := container.New()
+
+	cont.AddDependency(testutils.NewService)
+	cont.AddTokenDependency("buffer", func() *bytes.Buffer {
+		return bytes.NewBuffer([]byte{})
+	})
+
+	cont.AddDependency(func(c *container.Container) testutils.MyDeps {
+		deps := testutils.MyDeps{}
+		err := c.Fill(&deps)
+		require.NoError(t, err)
+
+		return deps
+	})
+
+	deps, err := container.Resolve[testutils.MyDeps](cont)
+	require.NoError(t, err)
+	err = deps.CheckResolvedDependencies()
+	require.NoError(t, err)
+}