update e2econtainerconfig

cmd/config/internal/commands/e2e/e2econtainerconfig/main.go

@@ -6,54 +6,65 @@ package main
 import (
 	"fmt"
 	"os"
+	"strconv"
 
 	"sigs.k8s.io/kustomize/kyaml/fn/framework"
+	"sigs.k8s.io/kustomize/kyaml/fn/framework/command"
+	"sigs.k8s.io/kustomize/kyaml/kio"
 	"sigs.k8s.io/kustomize/kyaml/yaml"
 )
 
-// Data contains the items
-type Data struct {
-	StringValue string `yaml:"stringValue,omitempty"`
-
-	IntValue int `yaml:"intValue,omitempty"`
-
-	BoolValue bool `yaml:"boolValue,omitempty"`
-}
-
-// Example defines the ResourceList.functionConfig schema.
-type Example struct {
+// Config defines the ResourceList.functionConfig schema.
+type Config struct {
 	// Data contains configuration data for the Example
 	// Nest values under Data so that the function can accept a ConfigMap as its
 	// functionConfig (`run` generates a ConfigMap for the functionConfig when run with --)
 	// e.g. `config run DIR/ --image my-image -- a-string-value=foo` will create the input
 	// with ResourceList.functionConfig.data.a-string-value=foo
-	Data Data `yaml:"data,omitempty"`
+	Data map[string]string `yaml:"data,omitempty"`
 }
 
 func main() {
-	functionConfig := &Example{}
-	resourceList := &framework.ResourceList{FunctionConfig: functionConfig}
+	config := &Config{}
 
-	cmd := framework.Command(resourceList, func() error {
-		for i := range resourceList.Items {
-			if err := resourceList.Items[i].PipeE(yaml.SetAnnotation("a-string-value",
-				functionConfig.Data.StringValue)); err != nil {
-				return err
+	fn := func(items []*yaml.RNode) ([]*yaml.RNode, error) {
+		var newNodes []*yaml.RNode
+		for i := range items {
+			// set the annotation on each resource item
+			if err := items[i].PipeE(yaml.SetAnnotation("a-string-value", config.Data["stringValue"])); err != nil {
+				return nil, fmt.Errorf("%w", err)
+			}
+			intValue, err := strconv.Atoi(config.Data["intValue"])
+			if config.Data["intValue"] == "" {
+				// default value
+				intValue = 0
+			} else if err != nil {
+				return nil, fmt.Errorf("intValue convert error: %w", err)
+
+			}
+			if err := items[i].PipeE(yaml.SetAnnotation("a-int-value", strconv.Itoa(intValue))); err != nil {
+				return nil, fmt.Errorf("%w", err)
+			}
+			boolValue, err := strconv.ParseBool(config.Data["boolValue"])
+			if config.Data["boolValue"] == "" {
+				// default value
+				boolValue = false
+			} else if err != nil {
+				return nil, fmt.Errorf("boolValue convert error: %w", err)
+
+			}
+			if err := items[i].PipeE(yaml.SetAnnotation("a-bool-value", strconv.FormatBool(boolValue))); err != nil {
+				return nil, fmt.Errorf("%w", err)
 			}
 
-			if err := resourceList.Items[i].PipeE(yaml.SetAnnotation("a-int-value",
-				fmt.Sprintf("%v", functionConfig.Data.IntValue))); err != nil {
-				return err
-			}
-
-			if err := resourceList.Items[i].PipeE(yaml.SetAnnotation("a-bool-value",
-				fmt.Sprintf("%v", functionConfig.Data.BoolValue))); err != nil {
-				return err
-			}
+			newNodes = append(newNodes, items[i])
 		}
-		return nil
-	})
+		return newNodes, nil
+	}
 
+	p := framework.SimpleProcessor{Config: config, Filter: kio.FilterFunc(fn)}
+	cmd := command.Build(p, command.StandaloneDisabled, false)
+	command.AddGenerateDockerfile(cmd)
 	if err := cmd.Execute(); err != nil {
 		os.Exit(1)
 	}