Fix Ask comfirmation while destroying infrastructure

Author: razzkumar

cli/internals/destroy/destroy.go

@@ -3,7 +3,6 @@ package destroy
 import (
 	"errors"
 	"path/filepath"
-	"strings"
 
 	"github.com/AlecAivazis/survey/v2"
 	"github.com/leapfrogtechnology/shift/core/services/storage"
@@ -12,10 +11,10 @@ import (
 	"github.com/leapfrogtechnology/shift/infrastructure/internals/terraform"
 )
 
-func askConfirmation(environment, projectName string) string {
-	confirmation := ""
-	prompt := &survey.Input{
-		Message: "Are you sure you want to destroy " + environment + " environment from " + projectName + " ?(Y/N): ",
+func askConfirmation(environment, projectName string) bool {
+	confirmation := false
+	prompt := &survey.Confirm{
+		Message: "Are you sure you want to destroy " + environment + " environment from " + projectName + " ?",
 	}
 	survey.AskOne(prompt, &confirmation)
 
@@ -32,24 +31,22 @@ func Run(environment string) {
 		exit.Error(errors.New(message+"'"+environment+"'"), "Error")
 	}
 
-	confirmation := askConfirmation(environment, project.Name)
+	confirm := askConfirmation(environment, project.Name)
 
-	if strings.EqualFold(confirmation, "Y") || strings.EqualFold(confirmation, "yes") {
-
-		workspaceRoot := "/tmp"
-		workspaceDir := filepath.Join(workspaceRoot, project.Name, project.Type, environment)
-		terraformFile := workspaceDir + "/infrastructure.tf"
+	if !confirm {
+		const message = "Operation aborted"
+		exit.Error(errors.New(message), "Cancelled")
+	}
 
-		exists := file.Exists(terraformFile)
+	workspaceRoot := "/tmp"
+	workspaceDir := filepath.Join(workspaceRoot, project.Name, project.Type, environment)
+	terraformFile := workspaceDir + "/infrastructure.tf"
 
-		if exists {
-			terraform.DestroyInfrastructure(workspaceDir)
-		} else {
-			terraform.MakeTempAndDestroy(project, environment, workspaceDir)
-		}
+	exists := file.Exists(terraformFile)
 
+	if exists {
+		terraform.DestroyInfrastructure(workspaceDir)
 	} else {
-		const message = "Operation aborted"
-		exit.Error(errors.New(message), "Cancelled")
+		terraform.MakeTempAndDestroy(project, environment, workspaceDir)
 	}
 }