enable all other resources
This commit is contained in:
76
src/internal/controller/project_controller_finalizer.go
Normal file
76
src/internal/controller/project_controller_finalizer.go
Normal file
@@ -0,0 +1,76 @@
|
||||
package controller
|
||||
|
||||
import (
|
||||
zitadelv1alpha1 "github.com/HaimKortovich/zitadel-resources-operator/api/v1alpha1"
|
||||
"github.com/HaimKortovich/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/project/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 (
|
||||
projectFinalizerName = "project.zitadel.github.com/project"
|
||||
)
|
||||
|
||||
type wrappedProjectFinalizer struct {
|
||||
client.Client
|
||||
project *zitadelv1alpha1.Project
|
||||
refresolver *zitadelv1alpha1.RefResolver
|
||||
}
|
||||
|
||||
func newWrappedProjectFinalizer(client client.Client, project *zitadelv1alpha1.Project, refresolver *zitadelv1alpha1.RefResolver) core.WrappedCoreFinalizer {
|
||||
return &wrappedProjectFinalizer{
|
||||
Client: client,
|
||||
project: project,
|
||||
refresolver: refresolver,
|
||||
}
|
||||
}
|
||||
|
||||
func (wf *wrappedProjectFinalizer) AddFinalizer(ctx context.Context) error {
|
||||
if wf.ContainsFinalizer() {
|
||||
return nil
|
||||
}
|
||||
return wf.patch(ctx, wf.project, func(project *zitadelv1alpha1.Project) {
|
||||
controllerutil.AddFinalizer(project, projectFinalizerName)
|
||||
})
|
||||
}
|
||||
|
||||
func (wf *wrappedProjectFinalizer) RemoveFinalizer(ctx context.Context) error {
|
||||
if !wf.ContainsFinalizer() {
|
||||
return nil
|
||||
}
|
||||
return wf.patch(ctx, wf.project, func(project *zitadelv1alpha1.Project) {
|
||||
controllerutil.RemoveFinalizer(wf.project, projectFinalizerName)
|
||||
})
|
||||
}
|
||||
|
||||
func (wr *wrappedProjectFinalizer) ContainsFinalizer() bool {
|
||||
return controllerutil.ContainsFinalizer(wr.project, projectFinalizerName)
|
||||
}
|
||||
|
||||
func (wf *wrappedProjectFinalizer) Reconcile(ctx context.Context, ztdClient *clientv2.Client) error {
|
||||
if wf.project.Status.ProjectId != nil {
|
||||
_, err := ztdClient.ProjectServiceV2().DeleteProject(ctx, &project.DeleteProjectRequest{ProjectId: *wf.project.Status.ProjectId})
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (wr *wrappedProjectFinalizer) patch(ctx context.Context, project *zitadelv1alpha1.Project,
|
||||
patchFn func(*zitadelv1alpha1.Project)) error {
|
||||
patch := ctrlClient.MergeFrom(project.DeepCopy())
|
||||
patchFn(project)
|
||||
|
||||
if err := wr.Client.Patch(ctx, project, patch); err != nil {
|
||||
return fmt.Errorf("error patching Project finalizer: %v", err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
Reference in New Issue
Block a user