kyverno/pkg/backward_compatibility/add_labels.go

package backwardcompatibility

import (
	"context"
	"fmt"
	"strings"

	kyvernoclient "github.com/kyverno/kyverno/pkg/client/clientset/versioned"
	kyvernoinformer "github.com/kyverno/kyverno/pkg/client/informers/externalversions/kyverno/v1"
	"github.com/kyverno/kyverno/pkg/config"
	dclient "github.com/kyverno/kyverno/pkg/dclient"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	"k8s.io/apimachinery/pkg/labels"
)

// AddLabels - adds labels to all the existing generate requests
func AddLabels(client *kyvernoclient.Clientset, grInformer kyvernoinformer.GenerateRequestInformer) {
	// Get all the GR's that are existing
	// Extract and Update all of them with the with the labels
	grList, err := grInformer.Lister().List(labels.NewSelector())
	if err != nil {
		// throw some error!
		fmt.Println("error occurred while getting gr list")
		fmt.Println(err)
	}

	for _, gr := range grList {

		grLabels := gr.Labels
		if grLabels == nil || len(grLabels) == 0 {
			grLabels = make(map[string]string)
		}
		grLabels["generate.kyverno.io/policy-name"] = gr.Spec.Policy
		grLabels["generate.kyverno.io/resource-name"] = gr.Spec.Resource.Name
		grLabels["generate.kyverno.io/resource-kind"] = gr.Spec.Resource.Kind
		grLabels["generate.kyverno.io/resource-namespace"] = gr.Spec.Resource.Namespace

		gr.SetLabels(grLabels)

		_, err = client.KyvernoV1().GenerateRequests(config.KyvernoNamespace).Update(context.TODO(), gr, metav1.UpdateOptions{})
		if err != nil {
			fmt.Println("error occured while updating gr", gr.Name)
			fmt.Println(err)
		}
	}
}

// AddCloneLabel - add label to the source resource about the new clone
func AddCloneLabel(client *dclient.Client, pInformer kyvernoinformer.ClusterPolicyInformer) {
	// Get all the Generate Policies which has clone
	// Get the resource with Kind, NameSpace, Name
	// Add Policy name if label not found
	policies, err := pInformer.Lister().List(labels.NewSelector())
	if err != nil {
		fmt.Println("error occurred while getting policy list")
		fmt.Println(err)
	}

	for _, policy := range policies {
		// policyHasClone := false
		for _, rule := range policy.Spec.Rules {
			if rule.HasGenerate() {
				clone := rule.Generation.Clone
				if clone.Name != "" {
					namespace := clone.Namespace
					name := clone.Name
					kind := rule.Generation.Kind
					obj, err := client.GetResource("", kind, namespace, name)

					if err != nil {
						fmt.Println("error occured while getting resource")
						fmt.Println(err)
					}
					updateSource := true

					// add label
					label := obj.GetLabels()
					if len(label) == 0 {
						label = make(map[string]string)
						label["generate.kyverno.io/clone-policy-name"] = policy.GetName()
					} else {
						if label["generate.kyverno.io/clone-policy-name"] != "" {
							policyNames := label["generate.kyverno.io/clone-policy-name"]
							if !strings.Contains(policyNames, policy.GetName()) {
								policyNames = policyNames + "," + policy.GetName()
								label["generate.kyverno.io/clone-policy-name"] = policyNames
							} else {
								updateSource = false
							}
						} else {
							label["generate.kyverno.io/clone-policy-name"] = policy.GetName()
						}
					}

					if updateSource {
						fmt.Println("updating existing clone source")
						obj.SetLabels(label)
						_, err = client.UpdateResource(obj.GetAPIVersion(), kind, namespace, obj, false)
						if err != nil {
							fmt.Printf("failed to update source  name:%v namespace:%v kind:%v\n", obj.GetName(), obj.GetNamespace(), obj.GetKind())
							return
						}
						fmt.Printf("updated source  name:%v namespace:%v kind:%v\n", obj.GetName(), obj.GetNamespace(), obj.GetKind())
					}

					// fmt.Println("-------------------------------------------------------------------------")
					// fmt.Println("policy name: ", policy.Name)
					// fmt.Println("rule name: ", rule.Name)
					// fmt.Println("namespace: ", namespace)
					// fmt.Println("name       ", name)
					// fmt.Println("kind:      ", kind)
					// b, _ := json.Marshal(obj)
					// fmt.Println("Cloned resource: \n", string(b))
					// fmt.Println("-------------------------------------------------------------------------")

				}
			}
		}
	}
}