From c70c4c171ec91a152fc7c3652f03790130d46bef Mon Sep 17 00:00:00 2001
From: Fabricio Aguiar <fabricio.aguiar@gmail.com>
Date: Mon, 27 Nov 2023 14:06:16 +0000
Subject: [PATCH] Deduplicate related images

Ensuring controlplane and dataplane operator share the same hash when the related image has the same name

Signed-off-by: Fabricio Aguiar <fabricio.aguiar@gmail.com>
---
 cmd/csv-merger/csv-merger.go | 50 +++++++++++++++++++++++++++++++-----
 1 file changed, 43 insertions(+), 7 deletions(-)

diff --git a/cmd/csv-merger/csv-merger.go b/cmd/csv-merger/csv-merger.go
index 10263cae5..707bd73cf 100644
--- a/cmd/csv-merger/csv-merger.go
+++ b/cmd/csv-merger/csv-merger.go
@@ -27,6 +27,7 @@ import (
 	"github.com/ghodss/yaml"
 	"github.com/imdario/mergo"
 	"github.com/openstack-k8s-operators/openstack-operator/pkg/util"
+	"golang.org/x/exp/maps"
 
 	semver "github.com/blang/semver/v4"
 	"github.com/operator-framework/api/pkg/lib/version"
@@ -155,14 +156,31 @@ func main() {
 		Status:     csvBase.Status}
 
 	installStrategyBase := csvBase.Spec.InstallStrategy.StrategySpec
-	csvNew.Spec.RelatedImages = csvBase.Spec.RelatedImages
 
-	envVarList := []v1.EnvVar{}
+	uniqueImages := make(map[string]csvv1alpha1.RelatedImage)
+	uniqueEnv := make(map[string]v1.EnvVar)
 	if len(*importEnvFiles) > 0 {
 		for _, filename := range strings.Split(*importEnvFiles, ",") {
-			envVarList = append(envVarList, getEnvsFromFile(filename)...)
+			envVarList := getEnvsFromFile(filename)
+			for _, env := range envVarList {
+				uniqueEnv[env.Name] = env
+				if strings.HasPrefix(env.Name, "RELATED_IMAGE_") {
+					imgName := strings.Replace(strings.ToLower(env.Name), "related_image_", "", -1)
+					uniqueImages[imgName] = csvv1alpha1.RelatedImage{
+						Name:  imgName,
+						Image: env.Value,
+					}
+					imgNameDash := strings.Replace(imgName, "_", "-", -1)
+					uniqueImages[imgNameDash] = csvv1alpha1.RelatedImage{
+						Name:  imgNameDash,
+						Image: env.Value,
+					}
+				}
+			}
+
 		}
 	}
+
 	for _, csvFile := range csvs {
 		if csvFile != "" {
 			csvBytes, err := os.ReadFile(csvFile)
@@ -182,14 +200,31 @@ func main() {
 			for _, container := range csvStruct.Spec.InstallStrategy.StrategySpec.DeploymentSpecs[0].Spec.Template.Spec.Containers {
 				// Copy env vars from the Service Operator into the OpenStack Operator
 				if container.Name == "manager" {
-					envVarList = append(envVarList, container.Env...)
+					for idx, env := range container.Env {
+						_, ok := uniqueEnv[env.Name]
+						// Ensuring dataplane gets the same hash from service operators
+						if ok && strings.HasPrefix(csvStruct.Name, "dataplane") {
+							container.Env[idx] = uniqueEnv[env.Name]
+							continue
+						}
+						uniqueEnv[env.Name] = env
+					}
 				}
 			}
 
 			installStrategyBase.DeploymentSpecs = append(installStrategyBase.DeploymentSpecs, csvStruct.Spec.InstallStrategy.StrategySpec.DeploymentSpecs...)
 			installStrategyBase.ClusterPermissions = append(installStrategyBase.ClusterPermissions, csvStruct.Spec.InstallStrategy.StrategySpec.ClusterPermissions...)
 			installStrategyBase.Permissions = append(installStrategyBase.Permissions, csvStruct.Spec.InstallStrategy.StrategySpec.Permissions...)
-			csvNew.Spec.RelatedImages = append(csvNew.Spec.RelatedImages, csvStruct.Spec.RelatedImages...)
+			for _, img := range csvStruct.Spec.RelatedImages {
+				_, ok := uniqueImages[img.Name]
+				// Ensuring dataplane doesn't overwrite the service operators related images
+				if ok && strings.HasPrefix(csvStruct.Name, "dataplane") {
+					continue
+				}
+				if img.Name != "manager" {
+					uniqueImages[img.Name] = img
+				}
+			}
 
 			for _, owned := range csvStruct.Spec.CustomResourceDefinitions.Owned {
 				csvNew.Spec.CustomResourceDefinitions.Owned = append(
@@ -227,14 +262,15 @@ func main() {
 		}
 
 	}
+	csvNew.Spec.RelatedImages = append(csvBase.Spec.RelatedImages, maps.Values(uniqueImages)...)
 	if len(*exportEnvFile) > 0 {
-		writeEnvToYaml(*exportEnvFile, envVarList)
+		writeEnvToYaml(*exportEnvFile, maps.Values(uniqueEnv))
 	} else {
 		installStrategyBase.DeploymentSpecs[0].Spec.Template.Spec.Containers[1].Env = append(
 			// OpenStack Operator controller-manager container env vars
 			installStrategyBase.DeploymentSpecs[0].Spec.Template.Spec.Containers[1].Env,
 			// Service Operator controller-manager container env vars
-			envVarList...,
+			maps.Values(uniqueEnv)...,
 		)
 	}