Add machine user

[ZITADOPER-1]

src/internal/controller/machineuser_controller_finalizer.go

@@ -0,0 +1,90 @@
+package controller
+
+import (
+	"strings"
+
+	zitadelv1alpha1 "bitbucket.org/topmanage-software-engineering/zitadel-k8s-operator/src/api/v1alpha1"
+	"bitbucket.org/topmanage-software-engineering/zitadel-k8s-operator/src/pkg/controller/zitadel"
+
+	"context"
+	"fmt"
+
+	"github.com/zitadel/zitadel-go/v2/pkg/client/management"
+	pb "github.com/zitadel/zitadel-go/v2/pkg/client/zitadel/management"
+	"sigs.k8s.io/controller-runtime/pkg/client"
+	ctrlClient "sigs.k8s.io/controller-runtime/pkg/client"
+	"sigs.k8s.io/controller-runtime/pkg/controller/controllerutil"
+)
+
+const (
+	machineuserFinalizerName = "machineuser.zitadel.topmanage.com/machineuser"
+)
+
+type wrappedMachineUserFinalizer struct {
+	client.Client
+	machineuser *zitadelv1alpha1.MachineUser
+}
+
+func newWrappedMachineUserFinalizer(client client.Client, machineuser *zitadelv1alpha1.MachineUser) zitadel.WrappedFinalizer {
+	return &wrappedMachineUserFinalizer{
+		Client:      client,
+		machineuser: machineuser,
+	}
+}
+
+func (wf *wrappedMachineUserFinalizer) AddFinalizer(ctx context.Context) error {
+	if wf.ContainsFinalizer() {
+		return nil
+	}
+	return wf.patch(ctx, wf.machineuser, func(machineuser *zitadelv1alpha1.MachineUser) {
+		controllerutil.AddFinalizer(machineuser, machineuserFinalizerName)
+	})
+}
+
+func (wf *wrappedMachineUserFinalizer) RemoveFinalizer(ctx context.Context) error {
+	if !wf.ContainsFinalizer() {
+		return nil
+	}
+	return wf.patch(ctx, wf.machineuser, func(machineuser *zitadelv1alpha1.MachineUser) {
+		controllerutil.RemoveFinalizer(wf.machineuser, machineuserFinalizerName)
+	})
+}
+
+func (wr *wrappedMachineUserFinalizer) ContainsFinalizer() bool {
+	return controllerutil.ContainsFinalizer(wr.machineuser, machineuserFinalizerName)
+}
+
+func (wf *wrappedMachineUserFinalizer) Reconcile(ctx context.Context, ztdClient *management.Client) error {
+	if wf.machineuser.Status.UserId == "" {
+		return nil
+	}
+	{
+		_, err := ztdClient.GetUserByID(ctx, &pb.GetUserByIDRequest{
+			Id: wf.machineuser.Status.UserId,
+		})
+		if err != nil {
+			if strings.Contains(err.Error(), `User doesn't exist`) {
+				return nil
+			}
+			return err
+		}
+	}
+	_, err := ztdClient.RemoveUser(ctx, &pb.RemoveUserRequest{
+		Id: wf.machineuser.Status.UserId,
+	})
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func (wr *wrappedMachineUserFinalizer) patch(ctx context.Context, machineuser *zitadelv1alpha1.MachineUser,
+	patchFn func(*zitadelv1alpha1.MachineUser)) error {
+	patch := ctrlClient.MergeFrom(machineuser.DeepCopy())
+	patchFn(machineuser)
+
+	if err := wr.Client.Patch(ctx, machineuser, patch); err != nil {
+		return fmt.Errorf("error patching MachineUser finalizer: %v", err)
+	}
+	return nil
+}