Convert k8s metadata to slice

aquasecurity · Sep 12, 2024 · 7cf7654 · 7cf7654
1 parent 7a1e8b8
commit 7cf7654
Show file tree

Hide file tree

Showing 2 changed files with 55 additions and 36 deletions.
diff --git a/pkg/k8s/report/report.go b/pkg/k8s/report/report.go
@@ -57,9 +57,9 @@ type Resource struct {
 	Namespace string `json:",omitempty"`
 	Kind      string
 	Name      string
-	Metadata  types.Metadata `json:",omitempty"`
-	Results   types.Results  `json:",omitempty"`
-	Error     string         `json:",omitempty"`
+	Metadata  []types.Metadata `json:",omitempty"`
+	Results   types.Results    `json:",omitempty"`
+	Error     string           `json:",omitempty"`
 
 	// original report
 	Report types.Report `json:"-"`
@@ -99,11 +99,15 @@ func (r Report) consolidate() ConsolidatedReport {
 		key := v.fullname()
 
 		if res, ok := index[key]; ok {
+			// Combine metadata
+			metadata := lo.UniqBy(append(res.Metadata, v.Metadata...), func(x types.Metadata) string {
+				return x.ImageID
+			})
 			index[key] = Resource{
 				Namespace: res.Namespace,
 				Kind:      res.Kind,
 				Name:      res.Name,
-				Metadata:  res.Metadata,
+				Metadata:  metadata,
 				Results:   append(res.Results, v.Results...),
 				Error:     res.Error,
 			}
@@ -214,7 +218,7 @@ func infraResource(misConfig Resource) bool {
 func CreateResource(artifact *artifacts.Artifact, report types.Report, err error) Resource {
 	r := createK8sResource(artifact, report.Results)
 
-	r.Metadata = report.Metadata
+	r.Metadata = []types.Metadata{report.Metadata}
 	r.Report = report
 	// if there was any error during the scan
 	if err != nil {
@@ -244,7 +248,7 @@ func createK8sResource(artifact *artifacts.Artifact, scanResults types.Results)
 		Namespace: artifact.Namespace,
 		Kind:      artifact.Kind,
 		Name:      artifact.Name,
-		Metadata:  types.Metadata{},
+		Metadata:  []types.Metadata{},
 		Results:   results,
 		Report: types.Report{
 			Results:      results,

diff --git a/pkg/k8s/report/report_test.go b/pkg/k8s/report/report_test.go
@@ -14,12 +14,15 @@ var (
 		Namespace: "default",
 		Kind:      "Deploy",
 		Name:      "orion",
-		Metadata: types.Metadata{
-			RepoTags: []string{
-				"alpine:3.14",
-			},
-			RepoDigests: []string{
-				"alpine:3.14@sha256:8fe1727132b2506c17ba0e1f6a6ed8a016bb1f5735e43b2738cd3fd1979b6260",
+		Metadata: []types.Metadata{
+			{
+				ImageID: "123",
+				RepoTags: []string{
+					"alpine:3.14",
+				},
+				RepoDigests: []string{
+					"alpine:3.14@sha256:8fe1727132b2506c17ba0e1f6a6ed8a016bb1f5735e43b2738cd3fd1979b6260",
+				},
 			},
 		},
 		Results: types.Results{
@@ -69,12 +72,15 @@ var (
 		Namespace: "default",
 		Kind:      "Deploy",
 		Name:      "orion",
-		Metadata: types.Metadata{
-			RepoTags: []string{
-				"alpine:3.14",
-			},
-			RepoDigests: []string{
-				"alpine:3.14@sha256:8fe1727132b2506c17ba0e1f6a6ed8a016bb1f5735e43b2738cd3fd1979b6260",
+		Metadata: []types.Metadata{
+			{
+				ImageID: "123",
+				RepoTags: []string{
+					"alpine:3.14",
+				},
+				RepoDigests: []string{
+					"alpine:3.14@sha256:8fe1727132b2506c17ba0e1f6a6ed8a016bb1f5735e43b2738cd3fd1979b6260",
+				},
 			},
 		},
 		Results: types.Results{
@@ -117,12 +123,15 @@ var (
 		Namespace: "default",
 		Kind:      "Deploy",
 		Name:      "orion",
-		Metadata: types.Metadata{
-			RepoTags: []string{
-				"alpine:3.14",
-			},
-			RepoDigests: []string{
-				"alpine:3.14@sha256:8fe1727132b2506c17ba0e1f6a6ed8a016bb1f5735e43b2738cd3fd1979b6260",
+		Metadata: []types.Metadata{
+			{
+				ImageID: "123",
+				RepoTags: []string{
+					"alpine:3.14",
+				},
+				RepoDigests: []string{
+					"alpine:3.14@sha256:8fe1727132b2506c17ba0e1f6a6ed8a016bb1f5735e43b2738cd3fd1979b6260",
+				},
 			},
 		},
 		Results: types.Results{
@@ -204,12 +213,15 @@ var (
 		Namespace: "default",
 		Kind:      "Cronjob",
 		Name:      "hello",
-		Metadata: types.Metadata{
-			RepoTags: []string{
-				"alpine:3.14",
-			},
-			RepoDigests: []string{
-				"alpine:3.14@sha256:8fe1727132b2506c17ba0e1f6a6ed8a016bb1f5735e43b2738cd3fd1979b6260",
+		Metadata: []types.Metadata{
+			{
+				ImageID: "123",
+				RepoTags: []string{
+					"alpine:3.14",
+				},
+				RepoDigests: []string{
+					"alpine:3.14@sha256:8fe1727132b2506c17ba0e1f6a6ed8a016bb1f5735e43b2738cd3fd1979b6260",
+				},
 			},
 		},
 		Results: types.Results{
@@ -221,12 +233,15 @@ var (
 		Namespace: "default",
 		Kind:      "Pod",
 		Name:      "prometheus",
-		Metadata: types.Metadata{
-			RepoTags: []string{
-				"alpine:3.14",
-			},
-			RepoDigests: []string{
-				"alpine:3.14@sha256:8fe1727132b2506c17ba0e1f6a6ed8a016bb1f5735e43b2738cd3fd1979b6260",
+		Metadata: []types.Metadata{
+			{
+				ImageID: "123",
+				RepoTags: []string{
+					"alpine:3.14",
+				},
+				RepoDigests: []string{
+					"alpine:3.14@sha256:8fe1727132b2506c17ba0e1f6a6ed8a016bb1f5735e43b2738cd3fd1979b6260",
+				},
 			},
 		},
 		Results: types.Results{