chore: type env name in strict type operations (#183)

Signed-off-by: Akshay Patidar <[email protected]>
dream11 · Mar 18, 2023 · afeec36 · afeec36
1 parent 786c1cb
commit afeec36
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 8 deletions.
diff --git a/internal/command/commands/component.go b/internal/command/commands/component.go
@@ -48,16 +48,15 @@ func (c *Component) Run(args []string) int {
 				return 1
 			}
 			if envTypeResp.Strict {
-				consentMessage := "\nYou are executing the above command on production environment. Are you sure? Enter Y/n: "
-				allowedInputs := map[string]struct{}{"Y": {}, "n": {}}
-				val, err := c.Input.AskWithConstraints(consentMessage, allowedInputs)
+				consentMessage := fmt.Sprintf("\nYou are executing the above command on a restricted environment. Are you sure? Enter \033[1m%s\033[0m to continue:", *envName)
+				val, err := c.Input.Ask(consentMessage)
 
 				if err != nil {
 					c.Logger.Error(err.Error())
 					return 1
 				}
 
-				if val != "Y" {
+				if val != *envName {
 					c.Logger.Info("Aborting the operation")
 					return 1
 				}

diff --git a/internal/command/commands/service.go b/internal/command/commands/service.go
@@ -455,16 +455,15 @@ func (s *Service) askForConsent(envName *string) int {
 		return 1
 	}
 	if envTypeResp.Strict {
-		consentMessage := "\nYou are executing the above command on production environment. Are you sure? Enter Y/n:"
-		allowedInputs := map[string]struct{}{"Y": {}, "n": {}}
-		val, err := s.Input.AskWithConstraints(consentMessage, allowedInputs)
+		consentMessage := fmt.Sprintf("\nYou are executing the above command on a restricted environment. Are you sure? Enter \033[1m%s\033[0m to continue:", *envName)
+		val, err := s.Input.Ask(consentMessage)
 
 		if err != nil {
 			s.Logger.Error(err.Error())
 			return 1
 		}
 
-		if val != "Y" {
+		if val != *envName {
 			s.Logger.Info("Aborting the operation")
 			return 1
 		}