helper methods, types, and unit tests for transformer annotations

Author: natasha41575

api/internal/target/kusttarget.go

@@ -497,7 +497,7 @@ func (kt *KustTarget) accumulateFile(
 		if err != nil {
 			return errors.Wrapf(err, "cannot add path annotation for '%s'", path)
 		}
-		err = resources.AnnotateAll(utils.OriginAnnotation, originAnno)
+		err = resources.AnnotateAll(utils.OriginAnnotationKey, originAnno)
 		if err != nil || originAnno == "" {
 			return errors.Wrapf(err, "cannot add path annotation for '%s'", path)
 		}

api/internal/utils/annotations.go

@@ -3,6 +3,7 @@ package utils
 import "sigs.k8s.io/kustomize/api/konfig"
 
 const (
+	// build annotations
 	BuildAnnotationPreviousKinds      = konfig.ConfigAnnoDomain + "/previousKinds"
 	BuildAnnotationPreviousNames      = konfig.ConfigAnnoDomain + "/previousNames"
 	BuildAnnotationPrefixes           = konfig.ConfigAnnoDomain + "/prefixes"
@@ -17,7 +18,9 @@ const (
 	BuildAnnotationAllowNameChange = konfig.ConfigAnnoDomain + "/allowNameChange"
 	BuildAnnotationAllowKindChange = konfig.ConfigAnnoDomain + "/allowKindChange"
 
-	OriginAnnotation = "config.kubernetes.io/origin"
+	// for keeping track of origin and transformer data
+	OriginAnnotationKey      = "config.kubernetes.io/origin"
+	TransformerAnnotationKey = "config.kubernetes.io/transformations"
 
 	Enabled = "enabled"
 )

api/krusty/buildmetadata_test.go renamed to api/krusty/originannotation_test.go

@@ -11,7 +11,6 @@ import (
 	"github.com/stretchr/testify/assert"
 	"sigs.k8s.io/kustomize/api/internal/utils"
 	"sigs.k8s.io/kustomize/api/krusty"
-	_ "sigs.k8s.io/kustomize/api/krusty"
 	kusttest_test "sigs.k8s.io/kustomize/api/testutils/kusttest"
 	"sigs.k8s.io/kustomize/kyaml/filesys"
 )

api/resmap/resmap.go

@@ -21,6 +21,13 @@ type Transformer interface {
 	Transform(m ResMap) error
 }
 
+// A TransformerWithProperties contains a Transformer and stores
+// some of its properties
+type TransformerWithProperties struct {
+	Transformer
+	Origin *resource.Origin
+}
+
 // A Generator creates an instance of ResMap.
 type Generator interface {
 	Generate() (ResMap, error)
@@ -144,10 +151,24 @@ type ResMap interface {
 	AbsorbAll(ResMap) error
 
 	// AddOriginAnnotation will add the provided origin as
-	// an annotation to all resources in the Resmap, if
+	// an origin annotation to all resources in the ResMap, if
 	// the origin is not nil.
 	AddOriginAnnotation(origin *resource.Origin) error
 
+	// RemoveOriginAnnotation will remove the origin annotation
+	// from all resources in the ResMap
+	RemoveOriginAnnotations() error
+
+	// AddTransformerAnnotation will add the provided origin as
+	// an origin annotation if the resource doesn't have one; a
+	// transformer annotation otherwise; to all resources in
+	// ResMap
+	AddTransformerAnnotation(origin *resource.Origin) error
+
+	// RemoveTransformerAnnotation will remove the transformer annotation
+	// from all resources in the ResMap
+	RemoveTransformerAnnotations() error
+
 	// AnnotateAll annotates all resources in the ResMap with
 	// the provided key value pair.
 	AnnotateAll(key string, value string) error

api/resmap/reswrangler.go

@@ -503,6 +503,50 @@ func (m *resWrangler) AddOriginAnnotation(origin *resource.Origin) error {
 	return nil
 }
 
+// RemoveOriginAnnotation implements ResMap
+func (m *resWrangler) RemoveOriginAnnotations() error {
+	for _, res := range m.rList {
+		if err := res.SetOrigin(nil); err != nil {
+			return err
+		}
+	}
+	return nil
+}
+
+// AddTransformerAnnotation implements ResMap
+func (m *resWrangler) AddTransformerAnnotation(origin *resource.Origin) error {
+	for _, res := range m.rList {
+		or, err := res.GetOrigin()
+		if err != nil {
+			return err
+		}
+		if or == nil {
+			// the resource does not have an origin annotation, so
+			// we assume that the transformer generated the resource
+			// rather than modifying it
+			err = res.SetOrigin(origin)
+		} else {
+			// the resource already has an origin annotation, so we
+			// record the provided origin as a transformation
+			err = res.AddTransformation(origin)
+		}
+		if err != nil {
+			return err
+		}
+	}
+	return nil
+}
+
+// RemoveTransformerAnnotations implements ResMap
+func (m *resWrangler) RemoveTransformerAnnotations() error {
+	for _, res := range m.rList {
+		if err := res.ClearTransformations(); err != nil {
+			return err
+		}
+	}
+	return nil
+}
+
 func (m *resWrangler) appendReplaceOrMerge(res *resource.Resource) error {
 	id := res.CurId()
 	matches := m.GetMatchingResourcesByAnyId(id.Equals)

api/resmap/reswrangler_test.go

@@ -26,6 +26,34 @@ import (
 var depProvider = provider.NewDefaultDepProvider()
 var rf = depProvider.GetResourceFactory()
 var rmF = NewFactory(rf)
+var origin1 = &resource.Origin{
+	Repo:         "github.com/myrepo",
+	Ref:          "master",
+	ConfiguredIn: "config.yaml",
+	ConfiguredBy: yaml.ResourceIdentifier{
+		TypeMeta: yaml.TypeMeta{
+			APIVersion: "builtin",
+			Kind:       "Generator",
+		},
+		NameMeta: yaml.NameMeta{
+			Name:      "my-name",
+			Namespace: "my-namespace",
+		},
+	},
+}
+var origin2 = &resource.Origin{
+	ConfiguredIn: "../base/config.yaml",
+	ConfiguredBy: yaml.ResourceIdentifier{
+		TypeMeta: yaml.TypeMeta{
+			APIVersion: "builtin",
+			Kind:       "Generator",
+		},
+		NameMeta: yaml.NameMeta{
+			Name:      "my-name",
+			Namespace: "my-namespace",
+		},
+	},
+}
 
 func doAppend(t *testing.T, w ResMap, r *resource.Resource) {
 	err := w.Append(r)
@@ -1527,6 +1555,67 @@ $patch: delete
 	}
 }
 
+func TestOriginAnnotations(t *testing.T) {
+	w := New()
+	for i := 0; i < 3; i++ {
+		assert.NoError(t, w.Append(makeCm(i)))
+	}
+	// this should add an origin annotation to every resource
+	assert.NoError(t, w.AddOriginAnnotation(origin1))
+	resources := w.Resources()
+	for _, res := range resources {
+		or, err := res.GetOrigin()
+		assert.NoError(t, err)
+		assert.Equal(t, origin1, or)
+	}
+	// this should not overwrite the existing origin annotations
+	assert.NoError(t, w.AddOriginAnnotation(origin2))
+	for _, res := range resources {
+		or, err := res.GetOrigin()
+		assert.NoError(t, err)
+		assert.Equal(t, origin1, or)
+	}
+	// this should remove origin annotations from all resources
+	assert.NoError(t, w.RemoveOriginAnnotations())
+	for _, res := range resources {
+		or, err := res.GetOrigin()
+		assert.NoError(t, err)
+		assert.Nil(t, or)
+	}
+}
+
+func TestTransformerAnnotations(t *testing.T) {
+	w := New()
+	for i := 0; i < 3; i++ {
+		assert.NoError(t, w.Append(makeCm(i)))
+	}
+	// this should add an origin annotation to every resource
+	assert.NoError(t, w.AddTransformerAnnotation(origin1))
+	resources := w.Resources()
+	for _, res := range resources {
+		or, err := res.GetOrigin()
+		assert.NoError(t, err)
+		assert.Equal(t, origin1, or)
+	}
+	// this should add a transformer annotation to every resource
+	assert.NoError(t, w.AddTransformerAnnotation(origin2))
+	for _, res := range resources {
+		or, err := res.GetOrigin()
+		assert.NoError(t, err)
+		assert.Equal(t, origin1, or)
+		tr, err := res.GetTransformations()
+		assert.NoError(t, err)
+		assert.Equal(t, resource.Transformations{origin2}, tr)
+	}
+	// remove transformer annotations from all resources
+	assert.NoError(t, w.RemoveTransformerAnnotations())
+	for _, res := range resources {
+		tr, err := res.GetTransformations()
+		assert.NoError(t, err)
+		assert.Nil(t, tr)
+	}
+}
+
 // baseResource produces a base object which used to test
 // patch transformation
 // Also the structure is matching the Deployment syntax

api/resource/origin.go

@@ -11,30 +11,31 @@ import (
 	kyaml "sigs.k8s.io/kustomize/kyaml/yaml"
 )
 
-// Origin retains information about where resources in the output
-// of `kustomize build` originated from
+// Origin retains information about the origin of resources and transformer configs
+// that contributed to the output of `kustomize build`
 type Origin struct {
 	// Path is the path to the resource. If a local resource, this path is
 	// rooted from the directory upon which `kustomize build` was invoked. If a
 	// remote resource, this path is rooted from the root of the remote repo.
 	Path string `json:"path,omitempty" yaml:"path,omitempty"`
 
-	// Repo is the remote repository that the resource originated from if it is
+	// Repo is the remote repository that the resource or transformer originated from if it is
 	// not from a local file
 	Repo string `json:"repo,omitempty" yaml:"repo,omitempty"`
 
-	// Ref is the ref of the remote repository that the resource originated from
+	// Ref is the ref of the remote repository that the resource or transformer originated from
 	// if it is not from a local file
 	Ref string `json:"ref,omitempty" yaml:"ref,omitempty"`
 
 	// The following fields only apply to resources that have been
-	// generated by fields other than the `resources` field.
+	// generated by fields other than the `resources` field, or to transformer
+	// configs.
 
-	// ConfiguredIn is the file path to the generator config that created the
+	// ConfiguredIn is the file path to the generator or transformer config that created the
 	// resource
 	ConfiguredIn string `json:"configuredIn,omitempty" yaml:"configuredIn,omitempty"`
 
-	// ConfiguredBy is the ObjectReference of the generator config
+	// ConfiguredBy is the ObjectReference of the generator or transformer config
 	ConfiguredBy kyaml.ResourceIdentifier `json:"configuredBy,omitempty" yaml:"configuredBy,omitempty"`
 }
 
@@ -67,6 +68,17 @@ func (origin *Origin) String() (string, error) {
 	return string(anno), err
 }
 
+// Transformations is a list of Origin
+type Transformations []*Origin
+
+// String returns a string version of Transformations
+func (transformations *Transformations) String() (string, error) {
+	anno, err := kyaml.Marshal(transformations)
+	return string(anno), err
+}
+
+// OriginFromCustomPlugin takes a custom plugin defined as a resource
+// and returns an origin object to describe it
 func OriginFromCustomPlugin(res *Resource) (*Origin, error) {
 	origin, err := res.GetOrigin()
 	if err != nil {

api/resource/origin_test.go

@@ -8,6 +8,7 @@ import (
 
 	"github.com/stretchr/testify/assert"
 	. "sigs.k8s.io/kustomize/api/resource"
+	kyaml "sigs.k8s.io/kustomize/kyaml/yaml"
 )
 
 func TestOriginAppend(t *testing.T) {
@@ -78,6 +79,82 @@ repo: github.com/kubernetes-sigs/kustomize/examples/multibases/dev/
 	for _, test := range tests {
 		actual, err := test.in.String()
 		assert.NoError(t, err)
-		assert.Equal(t, actual, test.expected)
+		assert.Equal(t, test.expected, actual)
+	}
+}
+
+func TestTransformationsString(t *testing.T) {
+	origin1 := &Origin{
+		Repo:         "github.com/myrepo",
+		Ref:          "master",
+		ConfiguredIn: "config.yaml",
+		ConfiguredBy: kyaml.ResourceIdentifier{
+			TypeMeta: kyaml.TypeMeta{
+				APIVersion: "builtin",
+				Kind:       "Generator",
+			},
+			NameMeta: kyaml.NameMeta{
+				Name:      "my-name",
+				Namespace: "my-namespace",
+			},
+		},
+	}
+	origin2 := &Origin{
+		ConfiguredIn: "../base/config.yaml",
+		ConfiguredBy: kyaml.ResourceIdentifier{
+			TypeMeta: kyaml.TypeMeta{
+				APIVersion: "builtin",
+				Kind:       "Generator",
+			},
+			NameMeta: kyaml.NameMeta{
+				Name:      "my-name",
+				Namespace: "my-namespace",
+			},
+		},
+	}
+	tests := []struct {
+		in       Transformations
+		expected string
+	}{
+		{
+			in: Transformations{origin1},
+			expected: `- repo: github.com/myrepo
+  ref: master
+  configuredIn: config.yaml
+  configuredBy:
+    apiVersion: builtin
+    kind: Generator
+    name: my-name
+    namespace: my-namespace
+`,
+		},
+		{
+			in: Transformations{origin1, origin2},
+			expected: `- repo: github.com/myrepo
+  ref: master
+  configuredIn: config.yaml
+  configuredBy:
+    apiVersion: builtin
+    kind: Generator
+    name: my-name
+    namespace: my-namespace
+- configuredIn: ../base/config.yaml
+  configuredBy:
+    apiVersion: builtin
+    kind: Generator
+    name: my-name
+    namespace: my-namespace
+`,
+		},
+		{
+			in: Transformations{},
+			expected: `[]
+`,
+		},
+	}
+	for _, test := range tests {
+		actual, err := test.in.String()
+		assert.NoError(t, err)
+		assert.Equal(t, test.expected, actual)
 	}
 }