feat: use yaml.v3 to split documents

cmd/gitter/main.go

@@ -20,6 +20,7 @@ import (
 	"bytes"
 	"context"
 	"fmt"
+	"io"
 	"io/ioutil"
 	"log"
 	"net"
@@ -39,6 +40,7 @@ import (
 	"github.com/jackc/pgx/v4/pgxpool"
 	"github.com/pkg/errors"
 	"gopkg.in/square/go-jose.v2/json"
+	yaml "gopkg.in/yaml.v3"
 )
 
 const (
@@ -233,9 +235,28 @@ func splitYAML(greps []git.GrepResult, dir string) map[string][][]byte {
 			log.Printf("failed to read CRD file: %s", res.FileName)
 			continue
 		}
-		// TODO(hasheddan): generalize this replacement
-		b = bytes.ReplaceAll(b, []byte("rw-rw----"), []byte("660"))
-		allCRDs[res.FileName] = bytes.Split(b, []byte("---"))
+
+		decoder := yaml.NewDecoder(bytes.NewReader(b))
+		var yamls [][]byte
+		for {
+			var node yaml.Node
+			err := decoder.Decode(&node)
+			if err == io.EOF {
+				break
+			}
+			if err != nil {
+				log.Printf("failed to decode part of CRD file: %s", res.FileName)
+				continue
+			}
+
+			doc, err := yaml.Marshal(node)
+			if err != nil {
+				log.Printf("failed to decode part of CRD file: %s", res.FileName)
+				continue
+			}
+			yamls = append(yamls, doc)
+		}
+		allCRDs[res.FileName] = yamls
 	}
 	return allCRDs
 }

go.mod

@@ -23,6 +23,7 @@ require (
 	google.golang.org/genproto v0.0.0-20201109203340-2640f1f9cdfb // indirect
 	google.golang.org/grpc v1.33.2 // indirect
 	gopkg.in/square/go-jose.v2 v2.2.2
+	gopkg.in/yaml.v3 v3.0.0-20210107192922-496545a6307b
 	k8s.io/apiextensions-apiserver v0.18.2
 	k8s.io/apimachinery v0.18.2
 	sigs.k8s.io/yaml v1.2.0