move everything to src
All checks were successful
Build and Publish / build-release (push) Successful in 8m29s
All checks were successful
Build and Publish / build-release (push) Successful in 8m29s
This commit is contained in:
78
internal/controller/machineuser_controller_finalizer.go
Normal file
78
internal/controller/machineuser_controller_finalizer.go
Normal file
@@ -0,0 +1,78 @@
|
||||
package controller
|
||||
|
||||
import (
|
||||
zitadelv1alpha1 "gitea.corredorconect.com/software-engineering/zitadel-resources-operator/api/v1alpha1"
|
||||
"gitea.corredorconect.com/software-engineering/zitadel-resources-operator/pkg/controller/core"
|
||||
|
||||
"context"
|
||||
"fmt"
|
||||
|
||||
clientv2 "github.com/zitadel/zitadel-go/v3/pkg/client"
|
||||
"github.com/zitadel/zitadel-go/v3/pkg/client/zitadel/user/v2"
|
||||
|
||||
"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) core.WrappedCoreFinalizer {
|
||||
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 *clientv2.Client) error {
|
||||
if wf.machineuser.Status.UserId != nil {
|
||||
if _, err := ztdClient.UserServiceV2().DeleteUser(ctx,
|
||||
&user.DeleteUserRequest{
|
||||
UserId: *wf.machineuser.Status.UserId,
|
||||
},
|
||||
); err != nil {
|
||||
return fmt.Errorf("Error deleting organization: %v", 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
|
||||
}
|
||||
Reference in New Issue
Block a user