Enhances login to use the key expiration returned from API result

pkg/config/profile.go

@@ -48,9 +48,6 @@ const (
 	// DateStringFormat is the format for expiredAt date
 	DateStringFormat = "2006-01-02"
 
-	// KeyValidInDays is the number of days the API key is valid for
-	KeyValidInDays = 90
-
 	// KeyManagementService is the key management service name
 	KeyManagementService = "Stripe CLI"
 )

pkg/login/client_login.go

@@ -4,7 +4,9 @@ import (
 	"context"
 	"fmt"
 	"github.com/stripe/stripe-cli/pkg/config"
+	"math"
 	"os"
+	"time"
 
 	"github.com/briandowns/spinner"
 
@@ -90,7 +92,15 @@ func (a *Authenticator) Login(ctx context.Context, links *Links) error {
 			} else {
 				ansi.StopSpinner(s, message, os.Stdout)
 			}
-			fmt.Println(ansi.Italic("Please note: this key will expire after 90 days, at which point you'll need to re-authenticate."))
+
+			keyValidityDurationDays := 90
+			if !res.KeyExpiration.IsZero() {
+				keyValidityDurationHours := res.KeyExpiration.Sub(time.Now()).Hours()
+				keyValidityDurationDays = int(math.Round(keyValidityDurationHours / 24.0))
+			}
+
+			keyDurationCourtesyMessage := fmt.Sprintf("Please note: this key will expire after %d days, at which point you'll need to re-authenticate.", keyValidityDurationDays)
+			fmt.Println(ansi.Italic(keyDurationCourtesyMessage))
 			return nil
 		}
 	}

pkg/login/keys/keytransfer.go

@@ -11,6 +11,7 @@ import (
 type AsyncPollResult struct {
 	TestModeAPIKey string
 	Account        *acct.Account
+	KeyExpiration  time.Time
 	Err            error
 }
 
@@ -57,6 +58,7 @@ func (rt *RAKTransfer) AsyncPollKey(ctx context.Context, pollURL string, interva
 
 	ch <- AsyncPollResult{
 		TestModeAPIKey: response.TestModeAPIKey,
+		KeyExpiration:  time.Unix(response.TestModeAPIKeyExpiry, 0),
 		Account:        account,
 		Err:            nil,
 	}

pkg/login/keys/polling.go

@@ -23,8 +23,10 @@ type PollAPIKeyResponse struct {
 	AccountID              string `json:"account_id"`
 	AccountDisplayName     string `json:"account_display_name"`
 	LiveModeAPIKey         string `json:"livemode_key_secret"`
+	LiveModeAPIKeyExpiry   int64  `json:"livemode_key_expiry"`
 	LiveModePublishableKey string `json:"livemode_key_publishable"`
 	TestModeAPIKey         string `json:"testmode_key_secret"`
+	TestModeAPIKeyExpiry   int64  `json:"testmode_key_expiry"`
 	TestModePublishableKey string `json:"testmode_key_publishable"`
 }