From e2d90ec8c29bd9eb27991d2f0e7238d3e57bc42e Mon Sep 17 00:00:00 2001
From: Javier Marcos <javier.marcos@gmail.com>
Date: Sat, 11 Apr 2020 09:52:48 -0700
Subject: [PATCH] Refresh flags on environment updates

---
 cli/environment.go | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/cli/environment.go b/cli/environment.go
index 7add50d7..92f71dca 100644
--- a/cli/environment.go
+++ b/cli/environment.go
@@ -87,6 +87,15 @@ func updateEnvironment(c *cli.Context) error {
 	if err := envs.Update(env); err != nil {
 		return err
 	}
+	// Make sure flags are up to date
+	flags, err := environments.GenerateFlags(env, "", "")
+	if err != nil {
+		return err
+	}
+	// Update flags in the newly created environment
+	if err := envs.UpdateFlags(envName, flags); err != nil {
+		return err
+	}
 	return nil
 }