irsa-manager/internal/kubernetes/delete.go

package kubernetes

import (
	"context"
	"fmt"

	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"

	"k8s.io/apimachinery/pkg/api/errors"
	"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
	"k8s.io/apimachinery/pkg/labels"

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

// DeleteOptions contains options for delete requests.
type DeleteOptions struct {
	// DeletionPropagation decides how the garbage collector will handle the propagation.
	DeletionPropagation metav1.DeletionPropagation

	// Inclusions determines which in-cluster objects are subject to deletion
	// based on the labels.
	// A nil Inclusions map means all objects are subject to deletion
	Inclusions map[string]string
}

func (h *Handler) DeleteAll(ctx context.Context, resources []*unstructured.Unstructured, opts DeleteOptions) error {
	if !h.cleanup {
		return nil
	}
	for _, r := range resources {
		err := h.Delete(ctx, r, opts)
		if err != nil {
			return err
		}
	}
	return nil
}

// Delete deletes the given object (not found errors are ignored).
func (h *Handler) Delete(ctx context.Context, object *unstructured.Unstructured, opts DeleteOptions) error {
	if !h.cleanup {
		return nil
	}
	existingObject := &unstructured.Unstructured{}
	existingObject.SetGroupVersionKind(object.GroupVersionKind())
	err := h.client.Get(ctx, client.ObjectKeyFromObject(object), existingObject)
	if err != nil {
		if !errors.IsNotFound(err) {
			return fmt.Errorf("failed to delete: %w", err)
		}
		return nil // already deleted
	}

	sel, err := metav1.LabelSelectorAsSelector(&metav1.LabelSelector{MatchLabels: opts.Inclusions})
	if err != nil {
		return fmt.Errorf("label selector failed: %w", err)
	}

	if !sel.Matches(labels.Set(existingObject.GetLabels())) {
		return nil
	}

	if err := h.client.Delete(ctx, existingObject, client.PropagationPolicy(opts.DeletionPropagation)); err != nil {
		return fmt.Errorf("delete failed: %w", err)
	}

	return nil
}