Change set-labels to be importable (#929)

Author: yuwenma

functions/go/set-labels/run.go

@@ -19,11 +19,12 @@ package main
 import (
 	"context"
 
+	"github.com/GoogleContainerTools/kpt-functions-catalog/functions/go/set-labels/setlabels"
 	"github.com/GoogleContainerTools/kpt-functions-sdk/go/fn"
 )
 
 func NewTransformer() fn.ResourceListProcessor {
-	return fn.WithContext(fn.Context{Context: context.Background()}, &SetLabels{})
+	return fn.WithContext(fn.Context{Context: context.Background()}, &setlabels.SetLabels{})
 }
 
 func run() error {

functions/go/set-labels/run_js.go

@@ -20,6 +20,7 @@ import (
 	"fmt"
 	"syscall/js"
 
+	"github.com/GoogleContainerTools/kpt-functions-catalog/functions/go/set-labels/setlabels"
 	"github.com/GoogleContainerTools/kpt-functions-sdk/go/fn"
 )
 
@@ -33,7 +34,7 @@ func run() error {
 }
 
 func transformLabels(input []byte) ([]byte, error) {
-	runner := fn.WithContext(fn.Context{Context: nil}, &SetLabels{})
+	runner := fn.WithContext(fn.Context{Context: nil}, &setlabels.SetLabels{})
 	return fn.Run(runner, []byte(input))
 }
 

functions/go/set-labels/transformer.go renamed to functions/go/set-labels/setlabels/transformer.go

@@ -1,4 +1,4 @@
-package main
+package setlabels
 
 import (
 	"sort"
@@ -9,9 +9,10 @@ import (
 type FieldPath []string
 
 var (
-	selectorPath    = FieldPath{"selector", "matchLabels"}
 	metaLabelsPath  = FieldPath{"metadata", "labels"}
+	selectorPath    = FieldPath{"selector", "matchLabels"}
 	podSelectorPath = FieldPath{"podSelector", "matchLabels"}
+	labelSelector   = FieldPath{"labelSelector", "matchLabels"}
 )
 
 var _ fn.Runner = &SetLabels{}
@@ -257,8 +258,6 @@ func (p *SetLabels) setLabelsInVolume(volume *fn.SubObject) error {
 
 // setLabelsInPodSpec set label path in PodSpec, that include path under topologySpreadConstraints and affinity
 func (p *SetLabels) setLabelsInPodSpec(podSpec *fn.SubObject) error {
-	labelSelector := FieldPath{"labelSelector", "matchLabels"}
-
 	_, exist, _ := podSpec.NestedSlice("topologySpreadConstraints")
 	if exist {
 		for _, obj := range podSpec.GetSlice("topologySpreadConstraints") {
@@ -306,6 +305,7 @@ func (p *SetLabels) updateLabels(o *fn.SubObject, labelPath FieldPath, labels ma
 		keys = append(keys, k)
 	}
 	sort.Strings(keys)
+
 	for i := 0; i < len(keys); i++ {
 		key := keys[i]
 		val := labels[key]

tests/set-labels/ill-formatted-fn-config/.expected/results.yaml

@@ -8,5 +8,5 @@ items:
     stderr: 'failed to evaluate function: error: function failure'
     exitCode: 1
     results:
-      - message: 'unable to convert object to *main.SetLabels with error: json: cannot unmarshal array into Go struct field SetLabels.labels of type map[string]string'
-        severity: error
+      - message: 'unable to convert object to *setlabels.SetLabels with error: json: cannot unmarshal array into Go struct field SetLabels.labels of type map[string]string'
+        severity: error

tests/set-labels/wrong-config-data-type/.expected/results.yaml

@@ -8,5 +8,5 @@ items:
     stderr: 'failed to evaluate function: error: function failure'
     exitCode: 1
     results:
-      - message: 'unable to convert object to *main.SetLabels with error: json: cannot unmarshal array into Go struct field SetLabels.labels of type map[string]string'
-        severity: error
+      - message: 'unable to convert object to *setlabels.SetLabels with error: json: cannot unmarshal array into Go struct field SetLabels.labels of type map[string]string'
+        severity: error