From c4560023d1c61137a903f17d82d7e82ef510d080 Mon Sep 17 00:00:00 2001
From: Moritz <moritz.m@local-it.org>
Date: Thu, 26 Jan 2023 13:08:03 +0100
Subject: [PATCH] add a label to signal that a deploy is a chaos deploy

---
 cli/app/rollback.go    |  1 +
 cli/app/upgrade.go     |  1 +
 cli/internal/deploy.go |  1 +
 pkg/config/app.go      | 12 ++++++++++++
 4 files changed, 15 insertions(+)

diff --git a/cli/app/rollback.go b/cli/app/rollback.go
index cadd4d47..941b2fcd 100644
--- a/cli/app/rollback.go
+++ b/cli/app/rollback.go
@@ -180,6 +180,7 @@ recipes.
 		}
 		config.ExposeAllEnv(stackName, compose, app.Env)
 		config.SetRecipeLabel(compose, stackName, app.Recipe)
+		config.SetChaosLabel(compose, stackName, internal.Chaos)
 
 		if !internal.Force {
 			if err := internal.NewVersionOverview(app, deployedVersion, chosenDowngrade, ""); err != nil {
diff --git a/cli/app/upgrade.go b/cli/app/upgrade.go
index 5037a490..a2138f27 100644
--- a/cli/app/upgrade.go
+++ b/cli/app/upgrade.go
@@ -191,6 +191,7 @@ recipes.
 		}
 		config.ExposeAllEnv(stackName, compose, app.Env)
 		config.SetRecipeLabel(compose, stackName, app.Recipe)
+		config.SetChaosLabel(compose, stackName, internal.Chaos)
 
 		if err := internal.NewVersionOverview(app, deployedVersion, chosenUpgrade, releaseNotes); err != nil {
 			logrus.Fatal(err)
diff --git a/cli/internal/deploy.go b/cli/internal/deploy.go
index 68a007e5..3ae9d8e0 100644
--- a/cli/internal/deploy.go
+++ b/cli/internal/deploy.go
@@ -136,6 +136,7 @@ func DeployAction(c *cli.Context) error {
 	}
 	config.ExposeAllEnv(app.StackName(), compose, app.Env)
 	config.SetRecipeLabel(compose, app.StackName(), app.Recipe)
+	config.SetChaosLabel(compose, app.StackName(), Chaos)
 
 	if err := DeployOverview(app, version, "continue with deployment?"); err != nil {
 		logrus.Fatal(err)
diff --git a/pkg/config/app.go b/pkg/config/app.go
index bb06e10d..f7d338f1 100644
--- a/pkg/config/app.go
+++ b/pkg/config/app.go
@@ -5,6 +5,7 @@ import (
 	"io/ioutil"
 	"os"
 	"path"
+	"strconv"
 	"strings"
 
 	"github.com/schollz/progressbar/v3"
@@ -476,3 +477,14 @@ func SetRecipeLabel(compose *composetypes.Config, stackName string, recipe strin
 		}
 	}
 }
+
+// add a label to signal that a deploy is a chaos deploy
+func SetChaosLabel(compose *composetypes.Config, stackName string, chaos bool) {
+	for _, service := range compose.Services {
+		if service.Name == "app" {
+			logrus.Debugf("set chaos label to %s", chaos)
+			labelKey := fmt.Sprintf("coop-cloud.%s.chaos", stackName)
+			service.Deploy.Labels[labelKey] = strconv.FormatBool(chaos)
+		}
+	}
+}