Update set-namespace function (#123)

update set-namespace function

Author: Shell32-Natsu

functions/go/Makefile

@@ -0,0 +1,34 @@
+GOPATH := $(shell go env GOPATH)
+
+# Edit this list to contain all go functions
+FUNCTIONS := \
+	set-namespace
+
+CURRENT_FUNCTION := unknown
+
+.PHONY: fix vet fmt test lint all
+.PHONY: $(FUNCTIONS)
+
+all: $(FUNCTIONS)
+
+$(FUNCTIONS):
+	$(MAKE) CURRENT_FUNCTION=$@ function
+
+function: fix vet fmt test lint
+
+fix:
+	cd $(CURRENT_FUNCTION) && go fix ./...
+
+fmt:
+	cd $(CURRENT_FUNCTION) && go fmt ./...
+
+lint:
+	(which $(GOPATH)/bin/golangci-lint || \
+		curl -sSfL https://raw.githubusercontent.com/golangci/golangci-lint/master/install.sh | sh -s -- -b $(GOPATH)/bin v1.32.2)
+	cd $(CURRENT_FUNCTION) && $(GOPATH)/bin/golangci-lint run ./...
+
+test:
+	cd $(CURRENT_FUNCTION) && go test -cover ./...
+
+vet:
+	cd $(CURRENT_FUNCTION) && go vet ./...

functions/go/publish-functions.sh

@@ -25,13 +25,15 @@ image_tag=${TAG:-latest}
 prefix="release-go-functions-"
 [[ "${image_tag}" = "${prefix}"* ]] && image_tag="${image_tag#$prefix}"
 
+# TODO: Put build and publish commands in Makefile
+make
+
 # iterate over each subdir, build and push Docker images.
 for dir in  */
 do
   image_name=gcr.io/kpt-functions/"${dir%/}"
   image="${image_name}":"${image_tag}"
   set -x
-  ( cd "${dir}" && make )
   docker build -t "${image}" -t "${image_name}" -f "${dir}"/Dockerfile "${dir}"
   if [ -z "${BUILDONLY}" ]; then
     docker push "${image_name}"

functions/go/set-namespace/Dockerfile

@@ -1,9 +1,8 @@
-FROM golang:1.14
+FROM golang:1.15
 ENV CGO_ENABLED=0
 WORKDIR /go/src/
 COPY . .
-RUN go build -v -o /usr/local/bin/set-namespace ./
-
+RUN go build -v -o /usr/local/bin/function ./
 FROM alpine:latest
-COPY --from=0 /usr/local/bin/set-namespace /usr/local/bin/set-namespace
-CMD ["set-namespace"]
+COPY --from=0 /usr/local/bin/function /usr/local/bin/function
+CMD ["function"]

functions/go/set-namespace/Makefile

@@ -0,0 +1,49 @@
+// Copyright 2019 The Kubernetes Authors.
+// SPDX-License-Identifier: Apache-2.0
+
+//go:generate pluginator
+package main
+
+import (
+	"sigs.k8s.io/kustomize/api/filters/namespace"
+	"sigs.k8s.io/kustomize/api/resmap"
+	"sigs.k8s.io/kustomize/api/types"
+	"sigs.k8s.io/kustomize/kyaml/filtersutil"
+	"sigs.k8s.io/yaml"
+)
+
+// Change or set the namespace of non-cluster level resources.
+type plugin struct {
+	Namespace  string            `json:"namespace,omitempty" yaml:"namespace,omitempty"`
+	FieldSpecs []types.FieldSpec `json:"fieldSpecs,omitempty" yaml:"fieldSpecs,omitempty"`
+}
+
+//noinspection GoUnusedGlobalVariable
+var KustomizePlugin plugin
+
+func (p *plugin) Config(
+	_ *resmap.PluginHelpers, c []byte) (err error) {
+	p.Namespace = ""
+	p.FieldSpecs = nil
+	return yaml.Unmarshal(c, p)
+}
+
+func (p *plugin) Transform(m resmap.ResMap) error {
+	if len(p.Namespace) == 0 {
+		return nil
+	}
+	for _, r := range m.Resources() {
+		if r.IsEmpty() {
+			// Don't mutate empty objects?
+			continue
+		}
+		err := filtersutil.ApplyToJSON(namespace.Filter{
+			Namespace: p.Namespace,
+			FsSlice:   p.FieldSpecs,
+		}, r)
+		if err != nil {
+			return err
+		}
+	}
+	return nil
+}