ngrok-operator/internal/controller/util.go

package controller

import (
	"context"

	"sigs.k8s.io/controller-runtime/pkg/client"
	"sigs.k8s.io/controller-runtime/pkg/controller/controllerutil"
)

const (
	FinalizerName = "k8s.ngrok.com/finalizer"
)

func IsUpsert(o client.Object) bool {
	return o.GetDeletionTimestamp().IsZero()
}

func IsDelete(o client.Object) bool {
	return !o.GetDeletionTimestamp().IsZero()
}

func HasFinalizer(o client.Object) bool {
	return controllerutil.ContainsFinalizer(o, FinalizerName)
}

func AddFinalizer(o client.Object) bool {
	return controllerutil.AddFinalizer(o, FinalizerName)
}

func RemoveFinalizer(o client.Object) bool {
	return controllerutil.RemoveFinalizer(o, FinalizerName)
}

func RegisterAndSyncFinalizer(ctx context.Context, c client.Writer, o client.Object) error {
	if !HasFinalizer(o) {
		AddFinalizer(o)
		return c.Update(ctx, o)
	}
	return nil
}

func RemoveAndSyncFinalizer(ctx context.Context, c client.Writer, o client.Object) error {
	RemoveFinalizer(o)
	return c.Update(ctx, o)
}

func AddAnnotations(o client.Object, annotations map[string]string) {
	if o == nil || annotations == nil {
		return
	}

	existing := o.GetAnnotations()
	if existing == nil {
		existing = make(map[string]string)
	}

	for k, v := range annotations {
		existing[k] = v
	}

	o.SetAnnotations(existing)
}