Fixed SSO login methods (#60)

Co-authored-by: sfc-gh-mobilize-mountain-sa@snowflake.com <sfc-gh-mobilize-mountain-sa>
sfc-gh-klochen · May 29, 2024 · 5015238 · 5015238
1 parent adcd910
commit 5015238
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/SnowflakePS/Commands/Authentication/ConnectAppCommand.cs b/SnowflakePS/Commands/Authentication/ConnectAppCommand.cs
@@ -416,7 +416,7 @@ protected override void ProcessRecord()
                     // Authenticate with SSO
                     int localPort = GetRandomUnusedPort();
 
-                    string ssoLoginLinkResult = SnowflakeDriver.GetSSOLoginLinkForAccountAndUser(appUserContext.AccountUrl, appUserContext.AccountName, appUserContext.UserName, localPort);
+                    string ssoLoginLinkResult = SnowflakeDriver.GetSSOLoginLinkForAccountAndUser(appUserContext.AccountUrl, appUserContext.AccountName, appUserContext.UserName, localPort, appUserContext.Cookies);
                     if (ssoLoginLinkResult.Length == 0)
                     {
                         throw new InvalidCredentialException(String.Format("Invalid response on getting SSO for user {0}@{1}", appUserContext.UserName, appUserContext.AccountName));

diff --git a/SnowflakePS/UtilityDrivers/SnowflakeDriver.cs b/SnowflakePS/UtilityDrivers/SnowflakeDriver.cs
@@ -227,7 +227,7 @@ public static string GetMasterTokenAndSessionTokenFromSSOToken(string accountUrl
                 cookies: cookies, snowflakeContext: String.Empty, referer: String.Empty, classicUIAuthToken: String.Empty);
         }
 
-        public static string GetSSOLoginLinkForAccountAndUser(string accountUrl, string accountName, string userName, int returnRedirectPortNumber)
+        public static string GetSSOLoginLinkForAccountAndUser(string accountUrl, string accountName, string userName, int returnRedirectPortNumber, CookieContainer cookies)
         {
             string requestJSONTemplate =
 @"{{
@@ -248,7 +248,8 @@ public static string GetSSOLoginLinkForAccountAndUser(string accountUrl, string
                 "session/authenticator-request",
                 "application/json",
                 requestBody,
-                "application/json", null);
+                "application/json",
+                cookies:  cookies);
         }
 
         #endregion