kubernetes API

internal/kubernetes/apply.go

@@ -0,0 +1,73 @@
+package kubernetes
+
+import (
+	"context"
+
+	"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
+	"k8s.io/apimachinery/pkg/runtime"
+
+	"sigs.k8s.io/controller-runtime/pkg/client"
+	"sigs.k8s.io/controller-runtime/pkg/client/apiutil"
+)
+
+type Handler struct {
+	cleanup bool
+	client  client.Client
+	owner   Owner
+}
+
+// NewHelper returns an initialized Helper.
+func NewHandler(c client.Client, owner Owner, cleanup bool) (*Handler, error) {
+	return &Handler{
+		cleanup: cleanup,
+		client:  c,
+		owner:   owner,
+	}, nil
+}
+
+func (h Handler) Apply(ctx context.Context, obj client.Object) error {
+	opts := []client.PatchOption{
+		client.ForceOwnership,
+		client.FieldOwner(h.owner.Field),
+	}
+	gvk, err := apiutil.GVKForObject(obj, h.client.Scheme())
+	if err != nil {
+		return err
+	}
+
+	u := &unstructured.Unstructured{}
+	unstructured, err := runtime.DefaultUnstructuredConverter.ToUnstructured(obj)
+	if err != nil {
+		return err
+	}
+	u.Object = unstructured
+	u.SetGroupVersionKind(gvk)
+	u.SetManagedFields(nil)
+	err = h.client.Patch(ctx, u, client.Apply, opts...)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func (h Handler) PatchStatus(ctx context.Context, obj client.Object) error {
+	opts := &client.SubResourcePatchOptions{
+		PatchOptions: client.PatchOptions{
+			FieldManager: h.owner.Field,
+		},
+	}
+	gvk, err := apiutil.GVKForObject(obj, h.client.Scheme())
+	if err != nil {
+		return err
+	}
+
+	u := &unstructured.Unstructured{}
+	unstructured, err := runtime.DefaultUnstructuredConverter.ToUnstructured(obj)
+	if err != nil {
+		return err
+	}
+	u.Object = unstructured
+	u.SetGroupVersionKind(gvk)
+	u.SetManagedFields(nil)
+	return h.client.Status().Patch(ctx, u, client.Apply, opts)
+}

internal/kubernetes/delete.go

@@ -0,0 +1,69 @@
+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
+}

internal/kubernetes/owner.go

@@ -0,0 +1,7 @@
+package kubernetes
+
+// Owner contains options for setting the field manager.
+type Owner struct {
+	// Field sets the field manager name for the given server-side apply patch.
+	Field string
+}