@asgardeo/auth-spa 0.2.13 → 0.2.17

package/src/worker/worker-core.ts

@@ -28,6 +28,7 @@ import {
     Store,
     TokenResponse
 } from "@asgardeo/auth-js";
+import { CUSTOM_GRANT_CONFIG } from "../constants";
 import { AsgardeoSPAException } from "../exception";
 import { SPAHelper } from "../helpers";
 import { HttpClient, HttpClientInstance } from "../http-client";
@@ -81,10 +82,6 @@ export const WebWorkerCore = async (
         _httpClient?.setHttpRequestFinishCallback && _httpClient.setHttpRequestFinishCallback(callback);
     };
 
-    const setHttpRequestErrorCallback = (callback: (error: HttpError) => void): void => {
-        _httpClient?.setHttpRequestErrorCallback && _httpClient.setHttpRequestErrorCallback(callback);
-    };
-
     const httpRequest = async (requestConfig: HttpRequestConfig): Promise<HttpResponse> => {
         let matches = false;
         const config = await _dataLayer.getConfigData();
@@ -107,9 +104,8 @@ export const WebWorkerCore = async (
                     return Promise.resolve(response);
                 })
                 .catch((error: HttpError) => {
-                    if (error?.response?.status === 401) {
-                        return _authenticationClient
-                            .refreshAccessToken()
+                    if (error?.response?.status === 401 || !error?.response) {
+                        return refreshAccessToken()
                             .then(() => {
                                 return _httpClient
                                     .request(requestConfig)
@@ -327,6 +323,9 @@ export const WebWorkerCore = async (
             }
         }
 
+        if(config.shouldReplayAfterRefresh) {
+            _dataLayer.setTemporaryDataParameter(CUSTOM_GRANT_CONFIG, JSON.stringify(config));
+        }
         if (useDefaultEndpoint || matches) {
             return _authenticationClient
                 .requestCustomGrant(config)
@@ -361,6 +360,11 @@ export const WebWorkerCore = async (
         return _authenticationClient
             .refreshAccessToken()
             .then(() => {
+                getCustomGrantConfigData().then((customGrantConfig) => {
+                    if(customGrantConfig) {
+                        requestCustomGrant(customGrantConfig)
+                    }
+                });
                 _spaHelper.refreshAccessTokenAutomatically();
 
                 return _authenticationClient.getBasicUserInfo();
@@ -420,6 +424,15 @@ export const WebWorkerCore = async (
         return _dataLayer.getConfigData();
     };
 
+    const getCustomGrantConfigData = async (): Promise<AuthClientConfig<CustomGrantConfig> | null> => {
+        const configString =  await _dataLayer.getTemporaryDataParameter(CUSTOM_GRANT_CONFIG);
+        if(configString) {
+            return JSON.parse(configString as string);
+        } else {
+            return null
+        }
+    };
+
     return {
         disableHttpHandler,
         enableHttpHandler,
@@ -438,7 +451,6 @@ export const WebWorkerCore = async (
         requestAccessToken,
         requestCustomGrant,
         revokeAccessToken,
-        setHttpRequestErrorCallback,
         setHttpRequestFinishCallback,
         setHttpRequestStartCallback,
         setHttpRequestSuccessCallback,