Attempt to refresh access token upon token validation

Author: acoumb

src/Umbraco.Cms.Integrations.Crm.Hubspot/Api/Management/Controllers/ValidateAccessTokenController.cs

@@ -15,15 +15,18 @@ public class ValidateAccessTokenController : HubspotFormsControllerBase
     {
         private readonly IHttpClientFactory _httpClientFactory;
         private readonly ITokenService _tokenService;
+        private readonly IHubspotAuthorizationService _authorizationService;
 
         public ValidateAccessTokenController(
             IOptions<HubspotSettings> settingsOptions,
             IHttpClientFactory httpClientFactory,
-            ITokenService tokenService) 
+            ITokenService tokenService,
+            IHubspotAuthorizationService authorizationService)
             : base(settingsOptions)
         {
             _httpClientFactory = httpClientFactory;
             _tokenService = tokenService;
+            _authorizationService = authorizationService;
         }
 
         [HttpGet("validate", Name = Constants.OperationIdentifiers.ValidateAccessToken)]
@@ -41,14 +44,47 @@ public async Task<IActionResult> ValidateAccessToken()
                     Error = Constants.ErrorMessages.OAuthInvalidToken
                 });
 
-            var requestMessage = new HttpRequestMessage
+            using var requestMessage = new HttpRequestMessage
             {
                 Method = HttpMethod.Get,
                 RequestUri = new Uri(HubspotFormsApiEndpoint)
             };
             requestMessage.Headers.Authorization = new AuthenticationHeaderValue("Bearer", accessToken);
 
             var response = await client.SendAsync(requestMessage);
+            if (response.StatusCode == HttpStatusCode.Unauthorized)
+            {
+                // Attempt to refresh the access token
+                var refreshAccessTokenResponse = await _authorizationService.RefreshAccessTokenAsync();
+                if (string.IsNullOrEmpty(refreshAccessTokenResponse)
+                    || refreshAccessTokenResponse.StartsWith(BaseAuthorizationService.ErrorPrefix))
+                {
+                    return Ok(new ResponseDto
+                    {
+                        IsValid = false,
+                        Error = Constants.ErrorMessages.OAuthInvalidToken
+                    });
+                }
+
+                if (!_tokenService.TryGetParameters(Constants.AccessTokenDbKey, out string newAccessToken)
+                    || string.IsNullOrEmpty(newAccessToken))
+                {
+                    return Ok(new ResponseDto
+                    {
+                        IsValid = false,
+                        Error = Constants.ErrorMessages.OAuthInvalidToken
+                    });
+                }
+
+                // Retry the request with the new access token
+                using var newRequestMessage = new HttpRequestMessage
+                {
+                    Method = HttpMethod.Get,
+                    RequestUri = new Uri(HubspotFormsApiEndpoint)
+                };
+                newRequestMessage.Headers.Authorization = new AuthenticationHeaderValue("Bearer", newAccessToken);
+                response = await client.SendAsync(newRequestMessage);
+            }
 
             return Ok(new ResponseDto
             {

src/Umbraco.Cms.Integrations.Crm.Hubspot/Services/AuthorizationService.cs

@@ -61,10 +61,10 @@ public async Task<string> GetAccessTokenAsync(string code)
                 var errorResult = await response.Content.ReadAsStringAsync();
                 var errorDto = JsonSerializer.Deserialize<ErrorDto>(errorResult);
 
-                return "Error: " + errorDto.Message;
+                return string.Format("{0}: {1}", ErrorPrefix, errorDto.Message);
             }
 
-            return "Error: An unexpected error occurred.";
+            return string.Format("{0}: An unexpected error occurred.", ErrorPrefix);
         }
 
         public string RefreshAccessToken() =>
@@ -102,7 +102,7 @@ public async Task<string> RefreshAccessTokenAsync()
                 return result;
             }
 
-            return "error";
+            return ErrorPrefix;
         }
     }
 }

src/Umbraco.Cms.Integrations.Crm.Hubspot/Services/BaseAuthorizationService.cs

@@ -15,6 +15,8 @@ public class BaseAuthorizationService
 
         protected readonly ITokenService TokenService;
 
+        public const string ErrorPrefix = "Error";
+
         public BaseAuthorizationService(ITokenService tokenService)
         {
             TokenService = tokenService;

src/Umbraco.Cms.Integrations.Crm.Hubspot/Services/UmbracoAuthorizationService.cs

@@ -1,10 +1,6 @@
 using Microsoft.Extensions.Options;
-using System;
-using System.Collections.Generic;
 using System.Net;
-using System.Net.Http;
 using System.Text.Json;
-using System.Threading.Tasks;
 using Umbraco.Cms.Integrations.Crm.Hubspot.Configuration;
 using Umbraco.Cms.Integrations.Crm.Hubspot.Models.Dtos;
 
@@ -74,10 +70,10 @@ public async Task<string> GetAccessTokenAsync(string code)
                 var errorResult = await response.Content.ReadAsStringAsync();
                 var errorDto = JsonSerializer.Deserialize<ErrorDto>(errorResult);
 
-                return "Error: " + errorDto.Message;
+                return string.Format("{0}: {1}", ErrorPrefix, errorDto.Message);
             }
 
-            return "Error: An unexpected error occurred.";
+            return string.Format("{0}: An unexpected error occurred.", ErrorPrefix);
         }
 
         public string RefreshAccessToken() =>
@@ -115,7 +111,7 @@ public async Task<string> RefreshAccessTokenAsync()
                 return result;
             }
 
-            return "error";
+            return ErrorPrefix;
         }
     }
 }

src/Umbraco.Cms.Integrations.Crm.Hubspot/Umbraco.Cms.Integrations.Crm.Hubspot.csproj

@@ -16,7 +16,7 @@
 		<PackageIconUrl></PackageIconUrl>
 		<PackageProjectUrl>https://github.com/umbraco/Umbraco.Cms.Integrations/blob/main-v16/src/Umbraco.Cms.Integrations.Crm.Hubspot</PackageProjectUrl>
 		<RepositoryUrl>https://github.com/umbraco/Umbraco.Cms.Integrations</RepositoryUrl>
-		<Version>7.0.0</Version>
+		<Version>7.0.1</Version>
 		<Authors>Umbraco HQ</Authors>
 		<Company>Umbraco</Company>
 		<PackageTags>Umbraco;Umbraco-Marketplace</PackageTags>