From 71803c46d81160b065a2f7ce5d6906404d4e69ff Mon Sep 17 00:00:00 2001
From: Vyankatesh Kudtarkar <vyankateshkd@gmail.com>
Date: Thu, 12 Aug 2021 23:26:35 +0530
Subject: [PATCH] GVK format

---
 pkg/policymutation/policymutation.go | 47 ++++++++++++++++++++++++++++
 1 file changed, 47 insertions(+)

diff --git a/pkg/policymutation/policymutation.go b/pkg/policymutation/policymutation.go
index a5d83a7b26..ebdd7f31ad 100644
--- a/pkg/policymutation/policymutation.go
+++ b/pkg/policymutation/policymutation.go
@@ -60,6 +60,17 @@ func GenerateJSONPatchesForDefaults(policy *kyverno.ClusterPolicy, log logr.Logg
 	}
 	patches = append(patches, convertPatch...)
 
+	formatedGVK, errs := checkForGVKFormatPatch(policy, log)
+	if len(errs) > 0 {
+		var errMsgs []string
+		for _, err := range errs {
+			errMsgs = append(errMsgs, err.Error())
+			log.Error(err, "failed to format the kind")
+		}
+		updateMsgs = append(updateMsgs, strings.Join(errMsgs, ";"))
+	}
+	patches = append(patches, formatedGVK...)
+
 	overlaySMPPatches, errs := convertOverlayToStrategicMerge(policy, log)
 	if len(errs) > 0 {
 		var errMsgs []string
@@ -74,6 +85,42 @@ func GenerateJSONPatchesForDefaults(policy *kyverno.ClusterPolicy, log logr.Logg
 	return utils.JoinPatches(patches), updateMsgs
 }
 
+func checkForGVKFormatPatch(policy *kyverno.ClusterPolicy, log logr.Logger) (patches [][]byte, errs []error) {
+	patches = make([][]byte, 0)
+	for i, rule := range policy.Spec.Rules {
+		kindList := []string{}
+		for _, k := range rule.MatchResources.Kinds {
+			kindList = append(kindList, getFormatedKind(k))
+		}
+		jsonPatch := struct {
+			Path  string   `json:"path"`
+			Op    string   `json:"op"`
+			Value []string `json:"value"`
+		}{
+			fmt.Sprintf("/spec/rules/%s/match/resources/kinds", strconv.Itoa(i)),
+			"replace",
+			kindList,
+		}
+		patchByte, err := json.Marshal(jsonPatch)
+		if err != nil {
+			errs = append(errs, fmt.Errorf("failed to convert policy '%s': %v", policy.Name, err))
+		}
+		patches = append(patches, patchByte)
+	}
+	return patches, errs
+}
+
+func getFormatedKind(str string) (kind string) {
+	if strings.Count(str, "/") == 0 {
+		return strings.Title(strings.ToLower(str))
+	}
+	splitString := strings.Split(str, "/")
+	if strings.Count(str, "/") == 1 {
+		return splitString[0] + "/" + strings.Title(strings.ToLower(splitString[1]))
+	}
+	return splitString[0] + "/" + splitString[1] + "/" + strings.Title(strings.ToLower(splitString[2]))
+}
+
 func convertPatchToJSON6902(policy *kyverno.ClusterPolicy, log logr.Logger) (patches [][]byte, errs []error) {
 	patches = make([][]byte, 0)