Add duration flags for login, increase max assumeRoleDuration to AWS …

…limit

cmd/login.go

@@ -16,9 +16,10 @@ import (
 
 // loginCmd represents the login command
 var loginCmd = &cobra.Command{
-	Use:   "login <profile>",
-	Short: "login will authenticate you through okta and allow you to access your AWS environment through a browser",
-	RunE:  loginRun,
+	Use:    "login <profile>",
+	Short:  "login will authenticate you through okta and allow you to access your AWS environment through a browser",
+	RunE:   loginRun,
+	PreRun: loginPre,
 }
 
 // Stdout is the bool for -stdout
@@ -29,6 +30,16 @@ func init() {
 	loginCmd.Flags().BoolVarP(&Stdout, "stdout", "", false, "Print login URL to stdout instead of opening in default browser")
 }
 
+func loginPre(cmd *cobra.Command, args []string) {
+	if err := loadDurationFlagFromEnv(cmd, "session-ttl", "AWS_SESSION_TTL", &sessionTTL); err != nil {
+		fmt.Fprintln(os.Stderr, "warning: failed to parse duration from AWS_SESSION_TTL")
+	}
+
+	if err := loadDurationFlagFromEnv(cmd, "assume-role-ttl", "AWS_ASSUME_ROLE_TTL", &assumeRoleTTL); err != nil {
+		fmt.Fprintln(os.Stderr, "warning: failed to parse duration from AWS_ASSUME_ROLE_TTL")
+	}
+}
+
 func loginRun(cmd *cobra.Command, args []string) error {
 	if len(args) < 1 {
 		return ErrTooFewArguments

lib/provider.go

@@ -18,7 +18,7 @@ const (
 	MaxSessionDuration    = time.Hour * 36
 	MinSessionDuration    = time.Minute * 15
 	MinAssumeRoleDuration = time.Minute * 15
-	MaxAssumeRoleDuration = time.Hour
+	MaxAssumeRoleDuration = time.Hour * 12
 
 	DefaultSessionDuration    = time.Hour * 4
 	DefaultAssumeRoleDuration = time.Minute * 15