Catch edge case where an ARN isn't an ARN..

glassechidna · Aug 13, 2018 · 0325025 · 0325025
1 parent 32c5633
commit 0325025
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/pkg/lastkeypair/ssh.go b/pkg/lastkeypair/ssh.go
@@ -84,11 +84,14 @@ func NewReifiedLoginWithCmd(cmd *cobra.Command, args []string) *ReifiedLogin {
 	kmsKeyId := viper.GetString("kms-key")
 	instanceArn, _ := cmd.PersistentFlags().GetString("instance-arn")
 	username, _ := cmd.PersistentFlags().GetString("ssh-username")
+	region, _ := cmd.PersistentFlags().GetString("region")
 	vouchers, _ := cmd.PersistentFlags().GetStringSlice("voucher")
 
 	instanceArnParts := strings.Split(instanceArn, ":")
-	instanceRegion := instanceArnParts[3]
-	sess := ClientAwsSession(profile, instanceRegion)
+	if len(instanceArnParts) > 3 {
+		region = instanceArnParts[3]
+	}
+	sess := ClientAwsSession(profile, region)
 
 	return &ReifiedLogin{
 		sess:            sess,