iam-session and kms-env return exit code of the command

CHANGELOG.md

@@ -1,5 +1,12 @@
 # Changelog
 
+## v6.0.0 (17/12/2018)
+
+**Fix**
+
+* `iam-session` returns the command exit code
+* `kms-env` returns the command exit code
+
 ## v5.11.0 (03/12/2018)
 
 **New**

VERSION

@@ -1 +1 @@
-5.11.0
+6.0.0

common/utils.go

@@ -3,6 +3,8 @@ package common
 import (
 	"log"
 	"os"
+	"os/exec"
+	"syscall"
 )
 
 var (
@@ -18,3 +20,20 @@ func FatalOnError(err error) {
 func Fatalln(message string) {
 	ErrorLog.Fatalln(message)
 }
+
+func GetExitCode(cmd *exec.Cmd, err error) int {
+	// adapted from https://stackoverflow.com/a/40770011
+	if err != nil {
+		// try to get the exit code
+		if exitError, ok := err.(*exec.ExitError); ok {
+			ws := exitError.Sys().(syscall.WaitStatus)
+			return ws.ExitStatus()
+		}
+		return -1
+
+	} else {
+		// success, exitCode should be 0 if go is ok
+		ws := cmd.ProcessState.Sys().(syscall.WaitStatus)
+		return ws.ExitStatus()
+	}
+}

iam/session/main.go

@@ -58,7 +58,7 @@ func main() {
 	}
 
 	if len(*command) > 0 {
-		executeCommand(command, conf, &creds)
+		os.Exit(executeCommand(command, conf, &creds))
 	}
 }
 
@@ -124,7 +124,7 @@ func promptConfirm(text string) bool {
 	return response == "y"
 }
 
-func executeCommand(command *[]string, conf *aws.Config, creds *credentials.Value) {
+func executeCommand(command *[]string, conf *aws.Config, creds *credentials.Value) int {
 	env := os.Environ()
 	var pEnv []string
 	if conf.Credentials != nil {
@@ -152,5 +152,9 @@ func executeCommand(command *[]string, conf *aws.Config, creds *credentials.Valu
 	p.Stdin = os.Stdin
 	p.Stderr = os.Stderr
 	p.Stdout = os.Stdout
-	p.Run()
+	err := p.Run()
+
+	// TODO: When https://github.com/golang/go/commit/be94dac4e945a2921b116761e41f1c22f0af2add is released, replace the below with
+	// os.Exit(p.ProcessState.ExitCode())
+	return common.GetExitCode(p, err)
 }

kms/env/main.go

@@ -60,7 +60,11 @@ func main() {
 	p.Stdin = os.Stdin
 	p.Stderr = os.Stderr
 	p.Stdout = os.Stdout
-	p.Run()
+	err := p.Run()
+
+	// TODO: When https://github.com/golang/go/commit/be94dac4e945a2921b116761e41f1c22f0af2add is released, replace the below with
+	// os.Exit(p.ProcessState.ExitCode())
+	os.Exit(common.GetExitCode(p, err))
 }
 
 func handleEnvVar(kmsClient *kms.KMS, ssmClient *ssm.SSM, secretsManagerClient *secretsmanager.SecretsManager, key, value string) (map[string]string, error) {