Init commit

src/pkg/controller/core/finalizer.go

@@ -0,0 +1,73 @@
+package core
+
+import (
+	"context"
+	"fmt"
+
+	zitadelv1alpha1 "github.com/HaimKortovich/zitadel-resources-operator/api/v1alpha1"
+	zitadelClient "github.com/HaimKortovich/zitadel-resources-operator/pkg/zitadel"
+	apierrors "k8s.io/apimachinery/pkg/api/errors"
+	"sigs.k8s.io/controller-runtime/pkg/client"
+)
+
+type CoreFinalizer struct {
+	Client      client.Client
+	RefResolver *zitadelv1alpha1.RefResolver
+
+	WrappedFinalizer WrappedCoreFinalizer
+}
+
+func NewCoreFinalizer(client client.Client, wf WrappedCoreFinalizer) Finalizer {
+	return &CoreFinalizer{
+		Client:           client,
+		RefResolver:      zitadelv1alpha1.NewRefResolver(client),
+		WrappedFinalizer: wf,
+	}
+}
+
+func (tf *CoreFinalizer) AddFinalizer(ctx context.Context) error {
+	if tf.WrappedFinalizer.ContainsFinalizer() {
+		return nil
+	}
+	if err := tf.WrappedFinalizer.AddFinalizer(ctx); err != nil {
+		return fmt.Errorf("error adding finalizer in TemplateFinalizer: %v", err)
+	}
+	return nil
+}
+
+func (tf *CoreFinalizer) Finalize(ctx context.Context, resource Resource) error {
+	if !tf.WrappedFinalizer.ContainsFinalizer() {
+		return nil
+	}
+
+	connectionRef, err := resource.ConnectionRef(ctx, tf.RefResolver)
+	if err != nil {
+		return err
+	}
+
+	connection, err := tf.RefResolver.ConnectionRef(ctx, connectionRef, resource.GetNamespace())
+	if err != nil {
+		if apierrors.IsNotFound(err) {
+			if err := tf.WrappedFinalizer.RemoveFinalizer(ctx); err != nil {
+				return fmt.Errorf("error removing %s finalizer: %v", resource.GetName(), err)
+			}
+			return nil
+		}
+		return fmt.Errorf("error getting Cluster: %v", err)
+	}
+
+	ztdClient, err := zitadelClient.NewV2Client(ctx, connection, *tf.RefResolver)
+	if err != nil {
+		return fmt.Errorf("error connecting to Zitadel: %v", err)
+	}
+	defer ztdClient.Close()
+
+	if err := tf.WrappedFinalizer.Reconcile(ctx, ztdClient); err != nil {
+		return fmt.Errorf("error reconciling in TemplateFinalizer: %v", err)
+	}
+
+	if err := tf.WrappedFinalizer.RemoveFinalizer(ctx); err != nil {
+		return fmt.Errorf("error removing finalizer in TemplateFinalizer: %v", err)
+	}
+	return nil
+}