ad wasm functionality to apply-setters and rewrite to new go kpt fn sdk

Author: wmuizelaar

functions/go/apply-setters/applysetters/apply_setters.go

@@ -6,50 +6,14 @@ import (
 	"strings"
 
 	"sigs.k8s.io/kustomize/kyaml/errors"
-	"sigs.k8s.io/kustomize/kyaml/kio"
 	"sigs.k8s.io/kustomize/kyaml/kio/kioutil"
 	"sigs.k8s.io/kustomize/kyaml/yaml"
 )
 
 const SetterCommentIdentifier = "# kpt-set: "
 
-var _ kio.Filter = &ApplySetters{}
-
-// ApplySetters applies the setter values to the resource fields which are tagged
-// by the setter reference comments
-type ApplySetters struct {
-	// Setters holds the user provided values for all the setters
-	Setters []Setter
-
-	// Results are the results of applying setter values
-	Results []*Result
-
-	// filePath file path of resource
-	filePath string
-}
-
-type Setter struct {
-	// Name is the name of the setter
-	Name string
-
-	// Value is the input value for setter
-	Value string
-}
-
-// Result holds result of search and replace operation
-type Result struct {
-	// FilePath is the file path of the matching field
-	FilePath string
-
-	// FieldPath is field path of the matching field
-	FieldPath string
-
-	// Value of the matching field
-	Value string
-}
-
 // Filter implements Set as a yaml.Filter
-func (as *ApplySetters) Filter(nodes []*yaml.RNode) ([]*yaml.RNode, error) {
+func (as *Setters) Filter(nodes []*yaml.RNode) ([]*yaml.RNode, error) {
 	for i := range nodes {
 		filePath, _, err := kioutil.GetFileAnnotations(nodes[i])
 		if err != nil {
@@ -77,14 +41,13 @@ environments: # kpt-set: ${env}
 - dev
 - stage
 
-For input ApplySetters [name: env, value: "[stage, prod]"], qthe yaml node is transformed to
+For input Setters [name: env, value: "[stage, prod]"], qthe yaml node is transformed to
 
 environments: # kpt-set: ${env}
 - stage
 - prod
-
 */
-func (as *ApplySetters) visitMapping(object *yaml.RNode, path string) error {
+func (as *Setters) visitMapping(object *yaml.RNode, path string) error {
 	return object.VisitFields(func(node *yaml.MapNode) error {
 		if node == nil || node.Key.IsNil() || node.Value.IsNil() {
 			// don't do IsNilOrEmpty check as empty sequences are allowed
@@ -180,17 +143,18 @@ e.g.for input of scalar node 'nginx:1.7.1 # kpt-set: ${image}:${tag}' in the yam
 
 apiVersion: v1
 ...
-  image: nginx:1.7.1 # kpt-set: ${image}:${tag}
 
-and for input ApplySetters [[name: image, value: ubuntu], [name: tag, value: 1.8.0]]
+	image: nginx:1.7.1 # kpt-set: ${image}:${tag}
+
+and for input Setters [[name: image, value: ubuntu], [name: tag, value: 1.8.0]]
 The yaml node is transformed to
 
 apiVersion: v1
 ...
-  image: ubuntu:1.8.0 # kpt-set: ${image}:${tag}
 
+	image: ubuntu:1.8.0 # kpt-set: ${image}:${tag}
 */
-func (as *ApplySetters) visitScalar(object *yaml.RNode, path string) error {
+func (as *Setters) visitScalar(object *yaml.RNode, path string) error {
 	if object.IsNil() {
 		return nil
 	}
@@ -358,7 +322,7 @@ func clean(input string) string {
 }
 
 // Decode decodes the input yaml node into Set struct
-func Decode(rn *yaml.RNode, fcd *ApplySetters) {
+func Decode(rn *yaml.RNode, fcd *Setters) {
 	for k, v := range rn.GetDataMap() {
 		fcd.Setters = append(fcd.Setters, Setter{Name: k, Value: v})
 	}

functions/go/apply-setters/applysetters/apply_setters_test.go

@@ -454,7 +454,7 @@ roles: # kpt-set: ${roles}
 				t.FailNow()
 			}
 
-			s := &ApplySetters{}
+			s := &Setters{}
 			node, err := kyaml.Parse(test.config)
 			if !assert.NoError(t, err) {
 				t.FailNow()

functions/go/apply-setters/applysetters/setters.go

@@ -0,0 +1,108 @@
+package applysetters
+
+import (
+	"fmt"
+	"sigs.k8s.io/kustomize/kyaml/yaml"
+
+	"github.com/GoogleContainerTools/kpt-functions-sdk/go/fn"
+)
+
+const fnConfigKind = "ApplySetters"
+const fnConfigApiVersion = "fn.kpt.dev/v1alpha1"
+
+func ApplySetters(rl *fn.ResourceList) (bool, error) {
+	r := Setters{}
+	return r.Process(rl)
+}
+
+type Setters struct {
+	// Setters holds the user provided values for all the setters
+	Setters []Setter `yaml:"setters,omitempty" json:"setters,omitempty"`
+
+	// Results are the results of applying setter values
+	Results []*Result
+
+	// filePath file path of resource
+	filePath string
+}
+
+type Setter struct {
+	// Name is the name of the setter
+	Name string `yaml:"name" json:"name"`
+
+	// Value is the input value for setter
+	Value string `yaml:"value" json:"value"`
+}
+
+// Result holds result of search and replace operation
+type Result struct {
+	// FilePath is the file path of the matching field
+	FilePath string
+
+	// FieldPath is field path of the matching field
+	FieldPath string
+
+	// Value of the matching field
+	Value string
+}
+
+// Config initializes Setters from a functionConfig fn.KubeObject
+func (r *Setters) Config(functionConfig *fn.KubeObject) error {
+	if functionConfig.IsEmpty() {
+		return fmt.Errorf("FunctionConfig is missing. Expect `ApplySetters`")
+	}
+	if functionConfig.GetKind() != fnConfigKind || functionConfig.GetAPIVersion() != fnConfigApiVersion {
+		return fmt.Errorf("received functionConfig of kind %s and apiVersion %s, "+
+			"only functionConfig of kind %s and apiVersion %s is supported",
+			functionConfig.GetKind(), functionConfig.GetAPIVersion(), fnConfigKind, fnConfigApiVersion)
+	}
+	r.Setters = []Setter{}
+	if err := functionConfig.As(r); err != nil {
+		return fmt.Errorf("unable to convert functionConfig to %s:\n%w",
+			"setters", err)
+	}
+	return nil
+}
+
+// Process configures the setters and transforms them.
+func (r *Setters) Process(rl *fn.ResourceList) (bool, error) {
+	if err := r.Config(rl.FunctionConfig); err != nil {
+		rl.LogResult(err)
+		return false, nil
+	}
+	transformedItems, err := r.Transform(rl.Items)
+	if err != nil {
+		rl.LogResult(err)
+		return false, nil
+	}
+	rl.Items = transformedItems
+	return true, nil
+}
+
+// Transform runs the setters filter in order to apply the setters - this
+// does the actual work.
+func (r *Setters) Transform(items []*fn.KubeObject) ([]*fn.KubeObject, error) {
+	var transformedItems []*fn.KubeObject
+	var nodes []*yaml.RNode
+
+	for _, obj := range items {
+		objRN, err := yaml.Parse(obj.String())
+		if err != nil {
+			return nil, err
+		}
+		nodes = append(nodes, objRN)
+	}
+
+	transformedNodes, err := r.Filter(nodes)
+	if err != nil {
+		return nil, err
+	}
+	for _, n := range transformedNodes {
+		obj, err := fn.ParseKubeObject([]byte(n.MustString()))
+		if err != nil {
+			return nil, err
+		}
+		transformedItems = append(transformedItems, obj)
+	}
+	return transformedItems, nil
+}

functions/go/apply-setters/go.mod

@@ -3,28 +3,43 @@ module github.com/GoogleContainerTools/kpt-functions-catalog/functions/go/apply-
 go 1.18
 
 require (
-	github.com/stretchr/testify v1.6.1
-	sigs.k8s.io/kustomize/kyaml v0.10.21
+	github.com/GoogleContainerTools/kpt-functions-sdk/go/fn v0.0.0-20220723081830-33aee7fe8a2e
+	github.com/stretchr/testify v1.7.1
+	sigs.k8s.io/kustomize/api v0.12.1
+	sigs.k8s.io/kustomize/kyaml v0.13.9
 )
 
 require (
+	github.com/GoogleContainerTools/kpt-functions-sdk/go/api v0.0.0-20220720212527-133180134b93 // indirect
 	github.com/PuerkitoBio/purell v1.1.1 // indirect
 	github.com/PuerkitoBio/urlesc v0.0.0-20170810143723-de5bf2ad4578 // indirect
+	github.com/asaskevich/govalidator v0.0.0-20190424111038-f61b66f89f4a // indirect
 	github.com/davecgh/go-spew v1.1.1 // indirect
 	github.com/go-errors/errors v1.0.1 // indirect
-	github.com/go-openapi/jsonpointer v0.19.3 // indirect
-	github.com/go-openapi/jsonreference v0.19.3 // indirect
-	github.com/go-openapi/swag v0.19.5 // indirect
+	github.com/go-logr/logr v1.2.0 // indirect
+	github.com/go-openapi/jsonpointer v0.19.5 // indirect
+	github.com/go-openapi/jsonreference v0.19.6 // indirect
+	github.com/go-openapi/swag v0.21.1 // indirect
+	github.com/golang/protobuf v1.5.2 // indirect
+	github.com/google/gnostic v0.5.7-v3refs // indirect
 	github.com/inconshreveable/mousetrap v1.0.0 // indirect
-	github.com/mailru/easyjson v0.7.0 // indirect
+	github.com/josharian/intern v1.0.0 // indirect
+	github.com/mailru/easyjson v0.7.7 // indirect
+	github.com/mitchellh/mapstructure v1.4.1 // indirect
 	github.com/monochromegane/go-gitignore v0.0.0-20200626010858-205db1a8cc00 // indirect
+	github.com/pkg/errors v0.9.1 // indirect
 	github.com/pmezard/go-difflib v1.0.0 // indirect
-	github.com/spf13/cobra v1.0.0 // indirect
+	github.com/spf13/cobra v1.4.0 // indirect
 	github.com/spf13/pflag v1.0.5 // indirect
-	github.com/xlab/treeprint v0.0.0-20181112141820-a009c3971eca // indirect
-	golang.org/x/net v0.0.0-20200324143707-d3edc9973b7e // indirect
-	golang.org/x/text v0.3.2 // indirect
+	github.com/xlab/treeprint v1.1.0 // indirect
+	golang.org/x/net v0.0.0-20220412020605-290c469a71a5 // indirect
+	golang.org/x/sys v0.0.0-20211216021012-1d35b9e2eb4e // indirect
+	golang.org/x/text v0.3.7 // indirect
+	google.golang.org/protobuf v1.28.0 // indirect
 	gopkg.in/yaml.v2 v2.4.0 // indirect
 	gopkg.in/yaml.v3 v3.0.0-20210107192922-496545a6307b // indirect
-	k8s.io/kube-openapi v0.0.0-20210421082810-95288971da7e // indirect
+	k8s.io/klog/v2 v2.60.1 // indirect
+	k8s.io/kube-openapi v0.0.0-20220401212409-b28bf2818661 // indirect
+	k8s.io/utils v0.0.0-20220210201930-3a6ce19ff2f9 // indirect
+	sigs.k8s.io/yaml v1.3.0 // indirect
 )