From 82c930ebfcbb7355b849ada07d52aec603d7451e Mon Sep 17 00:00:00 2001
From: shuting <shuting@nirmata.com>
Date: Mon, 3 Jun 2019 16:10:18 -0700
Subject: [PATCH] parse multiple documents defined in one yaml file

---
 pkg/kyverno/apply/apply.go | 47 +++++++++++++++++++++++---------------
 1 file changed, 28 insertions(+), 19 deletions(-)

diff --git a/pkg/kyverno/apply/apply.go b/pkg/kyverno/apply/apply.go
index 575b4d47dc..4b930bfa93 100644
--- a/pkg/kyverno/apply/apply.go
+++ b/pkg/kyverno/apply/apply.go
@@ -1,6 +1,7 @@
 package apply
 
 import (
+	"bytes"
 	"encoding/json"
 	"fmt"
 	"io"
@@ -22,11 +23,15 @@ import (
 	"k8s.io/client-go/restmapper"
 )
 
-const applyExample = `  # Apply a policy to the resource.
+const (
+	applyExample = `  # Apply a policy to the resource.
   kyverno apply @policy.yaml @resource.yaml
   kyverno apply @policy.yaml @resourceDir/
   kyverno apply @policy.yaml @resource.yaml --kubeconfig=$PATH_TO_KUBECONFIG_FILE`
 
+	defaultYamlSeparator = "---"
+)
+
 // NewCmdApply returns the apply command for kyverno
 func NewCmdApply(in io.Reader, out, errout io.Writer) *cobra.Command {
 	var kubeconfig string
@@ -153,27 +158,31 @@ func extractResource(fileDir, kubeconfig string) ([]*resourceInfo, error) {
 			continue
 		}
 
-		decode := scheme.Codecs.UniversalDeserializer().Decode
-		obj, gvk, err := decode([]byte(data), nil, nil)
-		if err != nil {
-			fmt.Printf("Warning: error while decoding YAML object, err: %s\n", err)
-			continue
-		}
+		dd := bytes.Split(data, []byte(defaultYamlSeparator))
 
-		actualObj, err := convertToActualObject(kubeconfig, gvk, obj)
-		if err != nil {
-			fmt.Printf("Warning: error while converting resource %s to actual k8s object, err: %v\n", gvk.Kind, err)
-			fmt.Printf("Apply policy on raw resource.\n")
-		}
+		for _, d := range dd {
+			decode := scheme.Codecs.UniversalDeserializer().Decode
+			obj, gvk, err := decode([]byte(d), nil, nil)
+			if err != nil {
+				fmt.Printf("Warning: error while decoding YAML object, err: %s\n", err)
+				continue
+			}
 
-		raw, err := json.Marshal(actualObj)
-		if err != nil {
-			fmt.Printf("Warning: error while marshalling manifest, err: %v\n", err)
-			continue
-		}
+			actualObj, err := convertToActualObject(kubeconfig, gvk, obj)
+			if err != nil {
+				fmt.Printf("Warning: error while converting resource %s to actual k8s object, err: %v\n", gvk.Kind, err)
+				fmt.Printf("Apply policy on raw resource.\n")
+			}
 
-		gvkInfo := &metav1.GroupVersionKind{Group: gvk.Group, Version: gvk.Version, Kind: gvk.Kind}
-		resources = append(resources, &resourceInfo{rawResource: raw, gvk: gvkInfo})
+			raw, err := json.Marshal(actualObj)
+			if err != nil {
+				fmt.Printf("Warning: error while marshalling manifest, err: %v\n", err)
+				continue
+			}
+
+			gvkInfo := &metav1.GroupVersionKind{Group: gvk.Group, Version: gvk.Version, Kind: gvk.Kind}
+			resources = append(resources, &resourceInfo{rawResource: raw, gvk: gvkInfo})
+		}
 	}
 
 	return resources, err