Save access token to user-specified profile if exists

pingidentity · Nov 12, 2024 · 60c579e · 60c579e
1 parent 503d8df
commit 60c579e
Showing 1 changed file with 14 additions and 6 deletions.
diff --git a/internal/commands/request/request_internal.go b/internal/commands/request/request_internal.go
@@ -242,15 +242,23 @@ func pingoneAuth() (accessToken string, err error) {
 		return "", err
 	}
 
-	// Store access token and expiry
-	profileViper := profiles.GetMainConfig().ActiveProfile().ViperInstance()
-	profileViper.Set(options.RequestAccessTokenOption.ViperKey, pingoneAuthResponse.AccessToken)
-
 	currentTime := time.Now().Unix()
 	tokenExpiry := currentTime + pingoneAuthResponse.ExpiresIn
-	profileViper.Set(options.RequestAccessTokenExpiryOption.ViperKey, tokenExpiry)
 
-	err = profiles.GetMainConfig().SaveProfile(profiles.GetMainConfig().ActiveProfile().Name(), profileViper)
+	// Store access token and expiry
+	pName, err := profiles.GetOptionValue(options.RootProfileOption)
+	if err != nil {
+		return "", err
+	}
+
+	if pName == "" {
+		pName = profiles.GetMainConfig().ActiveProfile().Name()
+	}
+
+	profileViper := profiles.GetMainConfig().ViperInstance().Sub(pName)
+	profileViper.Set(options.RequestAccessTokenOption.ViperKey, pingoneAuthResponse.AccessToken)
+	profileViper.Set(options.RequestAccessTokenExpiryOption.ViperKey, tokenExpiry)
+	err = profiles.GetMainConfig().SaveProfile(pName, profileViper)
 	if err != nil {
 		return "", err
 	}