Set $ASSUMED_ROLE in environment

mescam · Feb 8, 2018 · ca1eab4 · ca1eab4
1 parent 4851064
commit ca1eab4
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/main.go b/main.go
@@ -108,11 +108,11 @@ func main() {
 		return
 	}
 
-	err = execWithCredentials(args, creds)
+	err = execWithCredentials(role, args, creds)
 	must(err)
 }
 
-func execWithCredentials(argv []string, creds *credentials.Value) error {
+func execWithCredentials(role string, argv []string, creds *credentials.Value) error {
 	argv0, err := exec.LookPath(argv[0])
 	if err != nil {
 		return err
@@ -122,6 +122,7 @@ func execWithCredentials(argv []string, creds *credentials.Value) error {
 	os.Setenv("AWS_SECRET_ACCESS_KEY", creds.SecretAccessKey)
 	os.Setenv("AWS_SESSION_TOKEN", creds.SessionToken)
 	os.Setenv("AWS_SECURITY_TOKEN", creds.SessionToken)
+	os.Setenv("ASSUMED_ROLE", role)
 
 	env := os.Environ()
 	return syscall.Exec(argv0, argv, env)