refactor cmd/config to internal

cmd/config/internal/commands/tree.go

@@ -0,0 +1,187 @@
+// Copyright 2019 The Kubernetes Authors.
+// SPDX-License-Identifier: Apache-2.0
+
+package commands
+
+import (
+	"path/filepath"
+	"strings"
+
+	"sigs.k8s.io/kustomize/cmd/config/internal/generateddocs/commands"
+	"sigs.k8s.io/kustomize/kyaml/kio/filters"
+
+	"github.com/spf13/cobra"
+	"sigs.k8s.io/kustomize/kyaml/kio"
+	"sigs.k8s.io/kustomize/kyaml/yaml"
+)
+
+func GetTreeRunner(name string) *TreeRunner {
+	r := &TreeRunner{}
+	c := &cobra.Command{
+		Use:     "tree DIR",
+		Short:   commands.TreeShort,
+		Long:    commands.TreeLong,
+		Example: commands.TreeExamples,
+		RunE:    r.runE,
+		Args:    cobra.MaximumNArgs(1),
+	}
+	fixDocs(name, c)
+	c.Flags().BoolVar(&r.IncludeSubpackages, "include-subpackages", true,
+		"also print resources from subpackages.")
+
+	// TODO(pwittrock): Figure out if these are the right things to expose, and consider making it
+	// a list of options instead of individual flags
+	c.Flags().BoolVar(&r.name, "name", false, "print name field")
+	c.Flags().BoolVar(&r.resources, "resources", false, "print resources field")
+	c.Flags().BoolVar(&r.ports, "ports", false, "print ports field")
+	c.Flags().BoolVar(&r.images, "image", false, "print image field")
+	c.Flags().BoolVar(&r.replicas, "replicas", false, "print replicas field")
+	c.Flags().BoolVar(&r.args, "args", false, "print args field")
+	c.Flags().BoolVar(&r.cmd, "command", false, "print command field")
+	c.Flags().BoolVar(&r.env, "env", false, "print env field")
+	c.Flags().BoolVar(&r.all, "all", false, "print all field infos")
+	c.Flags().StringSliceVar(&r.fields, "field", []string{}, "print field")
+	c.Flags().BoolVar(&r.includeLocal, "include-local", false,
+		"if true, include local-config in the output.")
+	c.Flags().BoolVar(&r.excludeNonLocal, "exclude-non-local", false,
+		"if true, exclude non-local-config in the output.")
+	c.Flags().StringVar(&r.structure, "graph-structure", "directory",
+		"Graph structure to use for printing the tree.  may be any of: "+
+			strings.Join(kio.GraphStructures, ","))
+
+	r.Command = c
+	return r
+}
+
+func TreeCommand(name string) *cobra.Command {
+	return GetTreeRunner(name).Command
+}
+
+// TreeRunner contains the run function
+type TreeRunner struct {
+	IncludeSubpackages bool
+	Command            *cobra.Command
+	name               bool
+	resources          bool
+	ports              bool
+	images             bool
+	replicas           bool
+	all                bool
+	env                bool
+	args               bool
+	cmd                bool
+	fields             []string
+	includeLocal       bool
+	excludeNonLocal    bool
+	structure          string
+}
+
+func (r *TreeRunner) runE(c *cobra.Command, args []string) error {
+	var input kio.Reader
+	var root = "."
+	if len(args) == 1 {
+		root = filepath.Clean(args[0])
+		input = kio.LocalPackageReader{PackagePath: args[0]}
+	} else {
+		input = &kio.ByteReader{Reader: c.InOrStdin()}
+	}
+
+	var fields []kio.TreeWriterField
+	for _, field := range r.fields {
+		path, err := parseFieldPath(field)
+		if err != nil {
+			return err
+		}
+		fields = append(fields, newField(path...))
+	}
+
+	if r.name || (r.all && !c.Flag("name").Changed) {
+		fields = append(fields,
+			newField("spec", "containers", "[name=.*]", "name"),
+			newField("spec", "template", "spec", "containers", "[name=.*]", "name"),
+		)
+	}
+	if r.images || (r.all && !c.Flag("image").Changed) {
+		fields = append(fields,
+			newField("spec", "containers", "[name=.*]", "image"),
+			newField("spec", "template", "spec", "containers", "[name=.*]", "image"),
+		)
+	}
+
+	if r.cmd || (r.all && !c.Flag("command").Changed) {
+		fields = append(fields,
+			newField("spec", "containers", "[name=.*]", "command"),
+			newField("spec", "template", "spec", "containers", "[name=.*]", "command"),
+		)
+	}
+	if r.args || (r.all && !c.Flag("args").Changed) {
+		fields = append(fields,
+			newField("spec", "containers", "[name=.*]", "args"),
+			newField("spec", "template", "spec", "containers", "[name=.*]", "args"),
+		)
+	}
+	if r.env || (r.all && !c.Flag("env").Changed) {
+		fields = append(fields,
+			newField("spec", "containers", "[name=.*]", "env"),
+			newField("spec", "template", "spec", "containers", "[name=.*]", "env"),
+		)
+	}
+
+	if r.replicas || (r.all && !c.Flag("replicas").Changed) {
+		fields = append(fields,
+			newField("spec", "replicas"),
+		)
+	}
+	if r.resources || (r.all && !c.Flag("resources").Changed) {
+		fields = append(fields,
+			newField("spec", "containers", "[name=.*]", "resources"),
+			newField("spec", "template", "spec", "containers", "[name=.*]", "resources"),
+		)
+	}
+	if r.ports || (r.all && !c.Flag("ports").Changed) {
+		fields = append(fields,
+			newField("spec", "containers", "[name=.*]", "ports"),
+			newField("spec", "template", "spec", "containers", "[name=.*]", "ports"),
+			newField("spec", "ports"),
+		)
+	}
+
+	// show reconcilers in tree
+	fltrs := []kio.Filter{&filters.IsLocalConfig{
+		IncludeLocalConfig:    r.includeLocal,
+		ExcludeNonLocalConfig: r.excludeNonLocal,
+	}}
+
+	return handleError(c, kio.Pipeline{
+		Inputs:  []kio.Reader{input},
+		Filters: fltrs,
+		Outputs: []kio.Writer{kio.TreeWriter{
+			Root:      root,
+			Writer:    c.OutOrStdout(),
+			Fields:    fields,
+			Structure: kio.TreeStructure(r.structure)}},
+	}.Execute())
+}
+
+func newField(val ...string) kio.TreeWriterField {
+	if strings.HasPrefix(strings.Join(val, "."), "spec.template.spec.containers") {
+		return kio.TreeWriterField{
+			Name:        "spec.template.spec.containers",
+			PathMatcher: yaml.PathMatcher{Path: val, StripComments: true},
+			SubName:     val[len(val)-1],
+		}
+	}
+
+	if strings.HasPrefix(strings.Join(val, "."), "spec.containers") {
+		return kio.TreeWriterField{
+			Name:        "spec.containers",
+			PathMatcher: yaml.PathMatcher{Path: val, StripComments: true},
+			SubName:     val[len(val)-1],
+		}
+	}
+
+	return kio.TreeWriterField{
+		Name:        strings.Join(val, "."),
+		PathMatcher: yaml.PathMatcher{Path: val, StripComments: true},
+	}
+}