v1.0.1

joanbono · Mar 11, 2021 · 9a51f6d · 9a51f6d
1 parent 1a7dfa7
commit 9a51f6d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/modules/rotate/rotate.go b/modules/rotate/rotate.go
@@ -143,7 +143,7 @@ func CheckIAMErr(err error, profile, username string) {
 				fmt.Printf("%v Unknown error %v\n", red.Sprintf("[-]"), bold.Sprintf(`¯\_(ツ)_/¯`))
 			default:
 				if aerr.Error()[:21] == "NoCredentialProviders" {
-					fmt.Printf("[-] Profile %s not found\n", profile)
+					fmt.Printf("%v Profile %v not found\n", red.Sprintf("[-]"), bold.Sprintf(profile))
 				} else if aerr.Error()[:12] == "AccessDenied" {
 					if username == "" {
 						username = GetUsername(profile)
@@ -156,7 +156,7 @@ func CheckIAMErr(err error, profile, username string) {
 		} else {
 			fmt.Printf("%v Generic error: %v\n", red.Sprintf("[-]"), err.Error())
 		}
-		os.Exit(1)
+		os.Exit(3)
 	}
 
 }
diff --git a/modules/writer/writer.go b/modules/writer/writer.go
@@ -53,7 +53,10 @@ func Reader() string {
 //keys and replace them with the new ones
 func UpdateCredentials(awsCredentials, oldKey, newKey string) {
 	read, err := ioutil.ReadFile(awsCredentials)
-	CheckErr(err)
+	if err != nil {
+		fmt.Printf("%v Error reading %v\n", red.Sprintf("[-]"), bold.Sprintf(awsCredentials))
+		os.Exit(2)
+	}
 
 	newKeys := strings.Replace(string(read), oldKey, newKey, -1)