feat: add reading crds from stdin (#172)

Signed-off-by: Gergely Brautigam <[email protected]>

Author: Skarlso

cmd/crd.go

@@ -128,6 +128,8 @@ func constructHandler(args *rootArgs) (Handler, error) {
 	var crdHandler Handler
 
 	switch {
+	case args.stdin:
+		crdHandler = &StdInHandler{}
 	case args.fileLocation != "":
 		crdHandler = &FileHandler{location: args.fileLocation}
 	case args.folderLocation != "":

cmd/generate.go

@@ -17,6 +17,7 @@ type rootArgs struct {
 	privSSHKey         string
 	useSSHAgent        bool
 	gitURL             string
+	stdin              bool
 }
 
 var (
@@ -33,6 +34,7 @@ func init() {
 	rootCmd.AddCommand(generateCmd)
 	// using persistent flags so all flags will be available for all sub commands.
 	f := generateCmd.PersistentFlags()
+	f.BoolVarP(&args.stdin, "stdin", "i", false, "Take CRD content from stdin.")
 	f.StringVarP(&args.fileLocation, "crd", "c", "", "The CRD file to generate a yaml from.")
 	f.StringVarP(&args.folderLocation, "folder", "r", "", "A folder from which to parse a series of CRDs.")
 	f.StringVarP(&args.url, "url", "u", "", "If provided, will use this URL to fetch CRD YAML content from.")

cmd/stdin_handler.go

@@ -0,0 +1,49 @@
+package cmd
+
+import (
+	"fmt"
+	"io"
+	"os"
+
+	"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
+	"k8s.io/apimachinery/pkg/util/yaml"
+
+	"github.com/Skarlso/crd-to-sample-yaml/pkg"
+	"github.com/Skarlso/crd-to-sample-yaml/pkg/sanitize"
+)
+
+type StdInHandler struct {
+	group string
+}
+
+func (s *StdInHandler) CRDs() ([]*pkg.SchemaType, error) {
+	content, err := io.ReadAll(os.Stdin)
+	if err != nil {
+		return nil, fmt.Errorf("failed to read file: %w", err)
+	}
+
+	content, err = sanitize.Sanitize(content)
+	if err != nil {
+		return nil, fmt.Errorf("failed to sanitize content: %w", err)
+	}
+
+	crd := &unstructured.Unstructured{}
+	if err := yaml.Unmarshal(content, crd); err != nil {
+		return nil, fmt.Errorf("failed to unmarshal into custom resource definition: %w", err)
+	}
+
+	schemaType, err := pkg.ExtractSchemaType(crd)
+	if err != nil {
+		return nil, fmt.Errorf("failed to extract schema type: %w", err)
+	}
+
+	if schemaType == nil {
+		return nil, nil
+	}
+
+	if s.group != "" {
+		schemaType.Rendering = pkg.Rendering{Group: s.group}
+	}
+
+	return []*pkg.SchemaType{schemaType}, nil
+}