From 47216d87a8a7466b1afd63eb3a7b5eeb04a49169 Mon Sep 17 00:00:00 2001
From: "jordan.yordanov" <jordan.yordanov@icloud.com>
Date: Sat, 28 Dec 2024 15:06:25 -0500
Subject: [PATCH] Added UsernamePasswordAsync overload to customize the
 deserialization process in the cases when - during Salesforce downtime -
 response other than json is sometimes returned.

---
 .../AuthenticationClient.cs                   | 22 +++++++++++++++++--
 1 file changed, 20 insertions(+), 2 deletions(-)

diff --git a/src/NetCoreForce.Client/AuthenticationClient.cs b/src/NetCoreForce.Client/AuthenticationClient.cs
index 93a2c51..abcc5ce 100644
--- a/src/NetCoreForce.Client/AuthenticationClient.cs
+++ b/src/NetCoreForce.Client/AuthenticationClient.cs
@@ -119,6 +119,24 @@ public Task UsernamePasswordAsync(string clientId, string clientSecret, string u
         /// <exception cref="ForceAuthException">Thrown if the authentication fails</exception>
         public async Task UsernamePasswordAsync(string clientId, string clientSecret, string username, string password, string tokenRequestEndpointUrl)
         {
+            await UsernamePasswordAsync(clientId, clientSecret, username, password, tokenRequestEndpointUrl,
+                response => JsonConvert.DeserializeObject<AccessTokenResponse>(response),
+                response => JsonConvert.DeserializeObject<AuthErrorResponse>(response));
+        }
+
+        /// <summary>
+        /// Authenticate using the "Username and Password" auth flow
+        /// </summary>
+        /// <param name="clientId">Client ID, a.k.a. Consumer Key</param>
+        /// <param name="clientSecret">Client Secret, a.k.a. Consumer Secret</param>
+        /// <param name="username">Username</param>
+        /// <param name="password">Password</param>
+        /// <param name="tokenRequestEndpointUrl">Token request endpoint URL, e.g. https://login.salesforce.com/services/oauth2/token</param>
+        /// <param name="DeserializeAccessToken">Function to deserialize the access token response</param>
+        /// <param name="DeserializeAuthError">Function to deserialize the auth error response</param>
+        /// <exception cref="ForceAuthException">Thrown if the authentication fails</exception>
+        public async Task UsernamePasswordAsync(string clientId, string clientSecret, string username, string password, string tokenRequestEndpointUrl, Func<string, AccessTokenResponse> DeserializeAccessToken, Func<string, AuthErrorResponse> DeserializeAuthError)
+        {
 #if DEBUG
             Stopwatch sw = new Stopwatch();
             sw.Start();
@@ -153,7 +171,7 @@ public async Task UsernamePasswordAsync(string clientId, string clientSecret, st
 
             if (responseMessage.IsSuccessStatusCode)
             {
-                this.AccessInfo = JsonConvert.DeserializeObject<AccessTokenResponse>(response);
+                this.AccessInfo = DeserializeAccessToken(response);
             }
             else if (responseMessage.StatusCode == HttpStatusCode.NotFound)
             {
@@ -162,7 +180,7 @@ public async Task UsernamePasswordAsync(string clientId, string clientSecret, st
             }
             else
             {
-                var errorResponse = JsonConvert.DeserializeObject<AuthErrorResponse>(response);
+                var errorResponse = DeserializeAuthError(response);
                 throw new ForceAuthException(errorResponse.Error, errorResponse.ErrorDescription, responseMessage.StatusCode);
             }
 #if DEBUG