feat: annotation reports-created by reports-server (kyverno#154)

Signed-off-by: Vishal Choudhary <[email protected]>
Signed-off-by: Zach Stone <[email protected]>

pkg/api/annotate.go

@@ -0,0 +1,14 @@
+package api
+
+const (
+	ServedByReportsServerAnnotation = "kyverno.reports-server.io/served-by"
+	ServedByReportsServerValue      = "reports-server"
+)
+
+func labelReports(obj map[string]string) map[string]string {
+	if obj == nil {
+		obj = make(map[string]string)
+	}
+	obj[ServedByReportsServerAnnotation] = ServedByReportsServerValue
+	return obj
+}

pkg/api/cephr.go

@@ -128,6 +128,7 @@ func (c *cephrStore) Create(ctx context.Context, obj runtime.Object, createValid
 		cephr.Name = nameGenerator.GenerateName(cephr.GenerateName)
 	}
 
+	cephr.Annotations = labelReports(cephr.Annotations)
 	klog.Infof("creating cluster ephemeral reports name=%s", cephr.Name)
 	if !isDryRun {
 		r, err := c.createCephr(cephr)
@@ -186,6 +187,7 @@ func (c *cephrStore) Update(ctx context.Context, name string, objInfo rest.Updat
 		return nil, false, errors.NewBadRequest("failed to validate cluster ephemeral report")
 	}
 
+	cephr.Annotations = labelReports(cephr.Annotations)
 	klog.Infof("updating cluster ephemeral reports name=%s", cephr.Name)
 	if !isDryRun {
 		r, err := c.updateCephr(cephr, oldObj)

pkg/api/cpolr.go

@@ -128,6 +128,7 @@ func (c *cpolrStore) Create(ctx context.Context, obj runtime.Object, createValid
 		cpolr.Name = nameGenerator.GenerateName(cpolr.GenerateName)
 	}
 
+	cpolr.Annotations = labelReports(cpolr.Annotations)
 	klog.Infof("creating cluster policy report name=%s", cpolr.Name)
 	if !isDryRun {
 		r, err := c.createCpolr(cpolr)
@@ -184,7 +185,7 @@ func (c *cpolrStore) Update(ctx context.Context, name string, objInfo rest.Updat
 	if !ok {
 		return nil, false, errors.NewBadRequest("failed to validate cluster policy report")
 	}
-
+	cpolr.Annotations = labelReports(cpolr.Annotations)
 	klog.Infof("updating cluster policy report name=%s", cpolr.Name)
 	if !isDryRun {
 		r, err := c.updateCpolr(cpolr, oldObj)

pkg/api/ephr.go

@@ -139,6 +139,7 @@ func (p *ephrStore) Create(ctx context.Context, obj runtime.Object, createValida
 		ephr.Namespace = namespace
 	}
 
+	ephr.Annotations = labelReports(ephr.Annotations)
 	klog.Infof("creating ephemeral reports name=%s namespace=%s", ephr.Name, ephr.Namespace)
 	if !isDryRun {
 		r, err := p.createEphr(ephr)
@@ -202,6 +203,7 @@ func (p *ephrStore) Update(ctx context.Context, name string, objInfo rest.Update
 		ephr.Namespace = namespace
 	}
 
+	ephr.Annotations = labelReports(ephr.Annotations)
 	klog.Infof("updating ephemeral reports name=%s namespace=%s", ephr.Name, ephr.Namespace)
 	if !isDryRun {
 		r, err := p.updateEphr(ephr, oldObj)

pkg/api/polr.go

@@ -139,6 +139,7 @@ func (p *polrStore) Create(ctx context.Context, obj runtime.Object, createValida
 		polr.Name = nameGenerator.GenerateName(polr.GenerateName)
 	}
 
+	polr.Annotations = labelReports(polr.Annotations)
 	klog.Infof("creating policy reports name=%s namespace=%s", polr.Name, polr.Namespace)
 	if !isDryRun {
 		r, err := p.createPolr(polr)
@@ -203,6 +204,7 @@ func (p *polrStore) Update(ctx context.Context, name string, objInfo rest.Update
 		polr.Namespace = namespace
 	}
 
+	polr.Annotations = labelReports(polr.Annotations)
 	klog.Infof("updating policy reports name=%s namespace=%s", polr.Name, polr.Namespace)
 	if !isDryRun {
 		r, err := p.updatePolr(polr, oldObj)