use old session name for new sessions

felixb · Dec 7, 2017 · 4465e90 · 4465e90
1 parent 64ebc45
commit 4465e90
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/swamp.go b/swamp.go
@@ -173,7 +173,8 @@ func ensureTargetProfile(sess *session.Session, targetProfile, targetRole *strin
 	svc := sts.New(sess)
 
 	userId := getCallerId(svc).Arn
-	roleSessionName := strings.Split(*userId, "/")[1]
+	parts := strings.Split(*userId, "/")
+	roleSessionName := parts[len(parts) -1]
 
 	cred := assumeRole(svc, targetRole, &roleSessionName, duration)
 	writeProfile(cred, targetProfile, sess.Config.Region)