Setter Definitions read and write

kyaml/setters2/setterdefinition.go

@@ -0,0 +1,56 @@
+// Copyright 2019 The Kubernetes Authors.
+// SPDX-License-Identifier: Apache-2.0
+
+package setters2
+
+import (
+	"io/ioutil"
+
+	"sigs.k8s.io/kustomize/kyaml/yaml"
+)
+
+const DefinitionPrefix = "io.k8s.cli.setters."
+
+type SetterDefinition struct {
+	Name  string
+	Value string
+}
+
+func (sd SetterDefinition) AddSetterToFile(path string) error {
+	b, err := ioutil.ReadFile(path)
+	if err != nil {
+		return err
+	}
+	y, err := yaml.Parse(string(b))
+	if err != nil {
+		return err
+	}
+	if err := y.PipeE(sd); err != nil {
+		return err
+	}
+	out, err := y.String()
+	if err != nil {
+		return err
+	}
+	if err := ioutil.WriteFile(path, []byte(out), 0600); err != nil {
+		return err
+	}
+	return nil
+}
+
+func (sd SetterDefinition) Filter(object *yaml.RNode) (*yaml.RNode, error) {
+	key := DefinitionPrefix + sd.Name
+
+	def, err := object.Pipe(yaml.LookupCreate(
+		yaml.MappingNode, "openAPI", "definitions", key, "x-k8s-cli", "setter"))
+	if err != nil {
+		return nil, err
+	}
+	if err := def.PipeE(yaml.FieldSetter{Name: "name", StringValue: sd.Name}); err != nil {
+		return nil, err
+	}
+	if err := def.PipeE(yaml.FieldSetter{Name: "value", StringValue: sd.Value}); err != nil {
+		return nil, err
+	}
+	return object, nil
+}

kyaml/setters2/setterdefinition_test.go

@@ -0,0 +1,88 @@
+// Copyright 2019 The Kubernetes Authors.
+// SPDX-License-Identifier: Apache-2.0
+
+package setters2
+
+import (
+	"io/ioutil"
+	"os"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+var kptfile = `apiVersion: kpt.dev/v1alpha1
+kind: Kptfile
+metadata:
+    name: hello-world-set
+upstream:
+    type: git
+    git:
+        commit: 5c1c019b59299a4f6c7edd1ff5ff54d720621bbe
+        repo: git@github.com:GoogleContainerTools/kpt
+        directory: /package-examples/helloworld-set
+        ref: v0.1.0
+packageMetadata:
+    shortDescription: example package using setters`
+
+func TestAddUpdateSetter(t *testing.T) {
+	path := os.TempDir() + "Kptfile"
+
+	//write initial kptfile to temp path
+	err := ioutil.WriteFile(path, []byte(kptfile), 0600)
+	if !assert.NoError(t, err) {
+		t.FailNow()
+	}
+
+	//add a setter definition
+	sd := SetterDefinition{
+		Name:  "no-match-1",
+		Value: "1",
+	}
+
+	err = sd.AddSetterToFile(path)
+
+	if !assert.NoError(t, err) {
+		t.FailNow()
+	}
+
+	// update setter definition
+	sd2 := SetterDefinition{
+		Name:  "no-match-1",
+		Value: "2",
+	}
+
+	err = sd2.AddSetterToFile(path)
+
+	if !assert.NoError(t, err) {
+		t.FailNow()
+	}
+
+	b, err := ioutil.ReadFile(path)
+	if err != nil {
+		t.FailNow()
+	}
+
+	expected := `apiVersion: kpt.dev/v1alpha1
+kind: Kptfile
+metadata:
+  name: hello-world-set
+upstream:
+  type: git
+  git:
+    commit: 5c1c019b59299a4f6c7edd1ff5ff54d720621bbe
+    repo: git@github.com:GoogleContainerTools/kpt
+    directory: /package-examples/helloworld-set
+    ref: v0.1.0
+packageMetadata:
+  shortDescription: example package using setters
+openAPI:
+  definitions:
+    io.k8s.cli.setters.no-match-1:
+      x-k8s-cli:
+        setter:
+          name: no-match-1
+          value: 2
+`
+	assert.Equal(t, expected, string(b))
+}