add generic kubernetest resource management api

Author: reshnm

pkg/resources/clusterrole.go

@@ -0,0 +1,38 @@
+package resources
+
+import (
+	"fmt"
+
+	v1 "k8s.io/api/rbac/v1"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+)
+
+type ClusterRoleMutator struct {
+	Name   string
+	Labels map[string]string
+	Rules  []v1.PolicyRule
+}
+
+var _ Mutator[*v1.ClusterRole] = &ClusterRoleMutator{}
+
+func (m *ClusterRoleMutator) String() string {
+	return fmt.Sprintf("clusterrole %s", m.Name)
+}
+
+func (m *ClusterRoleMutator) Empty() *v1.ClusterRole {
+	return &v1.ClusterRole{
+		TypeMeta: metav1.TypeMeta{
+			APIVersion: "rbac.authorization.k8s.io/v1",
+			Kind:       "ClusterRole",
+		},
+		ObjectMeta: metav1.ObjectMeta{
+			Name: m.Name,
+		},
+	}
+}
+
+func (m *ClusterRoleMutator) Mutate(r *v1.ClusterRole) error {
+	r.ObjectMeta.Labels = m.Labels
+	r.Rules = m.Rules
+	return nil
+}

pkg/resources/clusterrole_test.go

@@ -0,0 +1 @@
+package resources

pkg/resources/clusterrolebinding.go

@@ -0,0 +1,51 @@
+package resources
+
+import (
+	"fmt"
+
+	v1 "k8s.io/api/rbac/v1"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+)
+
+type ClusterRoleBindingMutator struct {
+	ClusterRoleBindingName  string
+	ClusterRoleName         string
+	ServiceAccountName      string
+	ServiceAccountNamespace string
+	Labels                  map[string]string
+}
+
+var _ Mutator[*v1.ClusterRoleBinding] = &ClusterRoleBindingMutator{}
+
+func (m *ClusterRoleBindingMutator) String() string {
+	return fmt.Sprintf("clusterrolebinding %s", m.ClusterRoleBindingName)
+}
+
+func (m *ClusterRoleBindingMutator) Empty() *v1.ClusterRoleBinding {
+	return &v1.ClusterRoleBinding{
+		TypeMeta: metav1.TypeMeta{
+			APIVersion: "rbac.authorization.k8s.io/v1",
+			Kind:       "ClusterRoleBinding",
+		},
+		ObjectMeta: metav1.ObjectMeta{
+			Name: m.ClusterRoleBindingName,
+		},
+	}
+}
+
+func (m *ClusterRoleBindingMutator) Mutate(r *v1.ClusterRoleBinding) error {
+	r.ObjectMeta.Labels = m.Labels
+	r.RoleRef = v1.RoleRef{
+		APIGroup: "rbac.authorization.k8s.io",
+		Kind:     "ClusterRole",
+		Name:     m.ClusterRoleName,
+	}
+	r.Subjects = []v1.Subject{
+		{
+			Kind:      "ServiceAccount",
+			Name:      m.ServiceAccountName,
+			Namespace: m.ServiceAccountNamespace,
+		},
+	}
+	return nil
+}

pkg/resources/clusterrolebinding_test.go

@@ -0,0 +1 @@
+package resources

pkg/resources/configmap.go

@@ -0,0 +1 @@
+package resources

pkg/resources/configmap_test.go

@@ -0,0 +1 @@
+package resources

pkg/resources/crd.go

@@ -0,0 +1,39 @@
+package resources
+
+import (
+	"fmt"
+
+	apiextv1 "k8s.io/apiextensions-apiserver/pkg/apis/apiextensions/v1"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+)
+
+type crdMutator struct {
+	crd *apiextv1.CustomResourceDefinition
+}
+
+var _ Mutator[*apiextv1.CustomResourceDefinition] = &crdMutator{}
+
+func NewCRDMutator(crd *apiextv1.CustomResourceDefinition) Mutator[*apiextv1.CustomResourceDefinition] {
+	return &crdMutator{crd: crd}
+}
+
+func (m *crdMutator) String() string {
+	return fmt.Sprintf("crd %s", m.crd.Name)
+}
+
+func (m *crdMutator) Empty() *apiextv1.CustomResourceDefinition {
+	return &apiextv1.CustomResourceDefinition{
+		TypeMeta: metav1.TypeMeta{
+			APIVersion: "apiextensions.k8s.io/v1",
+			Kind:       "CustomResourceDefinition",
+		},
+		ObjectMeta: metav1.ObjectMeta{
+			Name: m.crd.Name,
+		},
+	}
+}
+
+func (m *crdMutator) Mutate(r *apiextv1.CustomResourceDefinition) error {
+	m.crd.Spec.DeepCopyInto(&r.Spec)
+	return nil
+}

pkg/resources/crd_test.go

@@ -0,0 +1 @@
+package resources

pkg/resources/metadata.go

@@ -0,0 +1,20 @@
+package resources
+
+import (
+	v1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"sigs.k8s.io/controller-runtime/pkg/client"
+)
+
+type baseMutator struct {
+	Labels      map[string]string
+	Annotations map[string]string
+}
+
+var _ Mutator[client.Object] = &baseMutator{}
+
+func NewBaseMutator(labels, annotations map[string]string) Mutator[client.Object] {
+	return &baseMutator{
+		Labels:      labels,
+		Annotations: annotations,
+	}
+}

pkg/resources/mutator.go

@@ -0,0 +1,42 @@
+package resources
+
+import (
+	"context"
+	"fmt"
+
+	"sigs.k8s.io/controller-runtime/pkg/client"
+	"sigs.k8s.io/controller-runtime/pkg/controller/controllerutil"
+)
+
+type Mutator[K client.Object] interface {
+	Empty() K
+	Mutate(res K) error
+	String() string
+}
+
+func GetResource[K client.Object](ctx context.Context, clt client.Client, m Mutator[K]) (K, error) {
+	res := m.Empty()
+	if err := clt.Get(ctx, client.ObjectKeyFromObject(res), res); err != nil {
+		return res, fmt.Errorf("failed to get %s: %w", m.String(), err)
+	}
+	return res, nil
+}
+
+func CreateOrUpdateResource[K client.Object](ctx context.Context, clt client.Client, m Mutator[K]) error {
+	res := m.Empty()
+	_, err := controllerutil.CreateOrUpdate(ctx, clt, res, func() error {
+		return m.Mutate(res)
+	})
+	if err != nil {
+		return fmt.Errorf("failed to create or update %s: %w", m.String(), err)
+	}
+	return nil
+}
+
+func DeleteResource[K client.Object](ctx context.Context, clt client.Client, m Mutator[K]) error {
+	res := m.Empty()
+	if err := clt.Delete(ctx, res); client.IgnoreNotFound(err) != nil {
+		return fmt.Errorf("failed to delete %s: %w", m.String(), err)
+	}
+	return nil
+}