kyverno/cmd/cleanup-controller/utils/utils.go

package utils

import (
	"encoding/json"
	"fmt"

	kyvernov1alpha1 "github.com/kyverno/kyverno/api/kyverno/v1alpha1"
	admissionv1 "k8s.io/api/admission/v1"
)

func UnmarshalCleanupPolicy(kind string, raw []byte) (kyvernov1alpha1.CleanupPolicyInterface, error) {
	var policy kyvernov1alpha1.CleanupPolicyInterface
	if kind == "CleanupPolicy" {
		if err := json.Unmarshal(raw, &policy); err != nil {
			return policy, err
		}
		return policy, nil
	}
	return policy, fmt.Errorf("admission request does not contain a cleanuppolicy")
}

func GetCleanupPolicy(request *admissionv1.AdmissionRequest) (kyvernov1alpha1.CleanupPolicyInterface, error) {
	return UnmarshalCleanupPolicy(request.Kind.Kind, request.Object.Raw)
}

func GetCleanupPolicies(request *admissionv1.AdmissionRequest) (kyvernov1alpha1.CleanupPolicyInterface, kyvernov1alpha1.CleanupPolicyInterface, error) {
	var emptypolicy kyvernov1alpha1.CleanupPolicyInterface
	policy, err := UnmarshalCleanupPolicy(request.Kind.Kind, request.Object.Raw)
	if err != nil {
		return policy, emptypolicy, err
	}
	if request.Operation == admissionv1.Update {
		oldPolicy, err := UnmarshalCleanupPolicy(request.Kind.Kind, request.OldObject.Raw)
		return policy, oldPolicy, err
	}
	return policy, emptypolicy, nil
}

func GetResourceName(request *admissionv1.AdmissionRequest) string {
	resourceName := request.Kind.Kind + "/" + request.Name
	if request.Namespace != "" {
		resourceName = request.Namespace + "/" + resourceName
	}
	return resourceName
}