Handle gracefull shutdown

Author: user-cube

cmd/root.go

@@ -16,6 +16,8 @@ var rootCmd = &cobra.Command{
 	Use:   "aws-cli-manager",
 	Short: "AWS CLI Manager is a tool to manage AWS CLI profiles",
 	Run: func(cmd *cobra.Command, args []string) {
+		// SelectProfile now returns a boolean indicating whether a profile was selected
+		// We don't need to do anything with the return value here
 		profile.SelectProfile()
 	},
 }

cmd/select.go

@@ -13,7 +13,12 @@ var selectCmd = &cobra.Command{
 	Run: func(cmd *cobra.Command, args []string) {
 		if len(args) == 0 {
 			// If no profile is specified, show the selection menu
-			profile.SelectProfile()
+			// SelectProfile now returns a boolean indicating whether a profile was selected
+			selected := profile.SelectProfile()
+			if !selected {
+				// User cancelled the selection with Ctrl+C
+				return
+			}
 		} else {
 			// If a profile is specified, select it directly
 			profileName := args[0]

pkg/profile/profile.go

@@ -8,14 +8,15 @@ import (
 	"sort"
 	"strings"
 
+	"github.com/manifoldco/promptui"
 	"github.com/user-cube/aws-cli-manager/v2/pkg/models"
 	"github.com/user-cube/aws-cli-manager/v2/pkg/settings"
 	"github.com/user-cube/aws-cli-manager/v2/pkg/sharedModules"
 	"github.com/user-cube/aws-cli-manager/v2/pkg/ui"
 	"gopkg.in/yaml.v2"
 )
 
-func SelectProfile() {
+func SelectProfile() bool {
 	// Get current profile first
 	currentProfile := GetCurrentProfile()
 	profileNames, awsProfiles := GetProfiles()
@@ -52,13 +53,20 @@ func SelectProfile() {
 
 	index, _, err := prompt.Run()
 	if err != nil {
+		// Check if it's an interrupt error (Ctrl+C)
+		if err == promptui.ErrInterrupt {
+			fmt.Println("\nProfile selection cancelled")
+			return false
+		}
+		// Handle other errors
 		ui.PrintError("Error selecting profile: %v", err)
 		os.Exit(1)
 	}
 
 	// Get the selected profile name
 	selectedProfile := sortedNames[index]
 	setProfile(selectedProfile, awsProfiles)
+	return true
 }
 
 func GetProfiles() (profileNames []string, awsProfiles models.AwsProfile) {
@@ -203,6 +211,12 @@ func PromptProfileName() string {
 
 		result, err := prompt.Run()
 		if err != nil {
+			// Check if it's an interrupt error (Ctrl+C)
+			if err == promptui.ErrInterrupt {
+				fmt.Println("\nProfile creation cancelled")
+				os.Exit(0)
+			}
+			// Handle other errors
 			ui.PrintError("Failed to get profile name: %v", err)
 			os.Exit(1)
 		}
@@ -231,6 +245,11 @@ func PromptProfileDetails() models.ProfileDetails {
 
 	region, err := regionPrompt.Run()
 	if err != nil {
+		// Check if it's an interrupt error (Ctrl+C)
+		if err == promptui.ErrInterrupt {
+			fmt.Println("\nProfile creation cancelled")
+			os.Exit(0)
+		}
 		ui.PrintError("Failed to get region: %v", err)
 		os.Exit(1)
 	}
@@ -244,6 +263,11 @@ func PromptProfileDetails() models.ProfileDetails {
 
 	ssoIndex, _, err := ssoPrompt.Run()
 	if err != nil {
+		// Check if it's an interrupt error (Ctrl+C)
+		if err == promptui.ErrInterrupt {
+			fmt.Println("\nProfile creation cancelled")
+			os.Exit(0)
+		}
 		ui.PrintError("Failed to get SSO information: %v", err)
 		os.Exit(1)
 	}
@@ -257,6 +281,11 @@ func PromptProfileDetails() models.ProfileDetails {
 
 	config, err := configPrompt.Run()
 	if err != nil {
+		// Check if it's an interrupt error (Ctrl+C)
+		if err == promptui.ErrInterrupt {
+			fmt.Println("\nProfile creation cancelled")
+			os.Exit(0)
+		}
 		ui.PrintError("Failed to get config file information: %v", err)
 		os.Exit(1)
 	}
@@ -275,6 +304,11 @@ func PromptProfileDetails() models.ProfileDetails {
 
 	credentials, err := credentialsPrompt.Run()
 	if err != nil {
+		// Check if it's an interrupt error (Ctrl+C)
+		if err == promptui.ErrInterrupt {
+			fmt.Println("\nProfile creation cancelled")
+			os.Exit(0)
+		}
 		ui.PrintError("Failed to get credentials file information: %v", err)
 		os.Exit(1)
 	}