refactor: Enhance token refresh mechanism with retry logic

heyverse · Mar 6, 2025 · 18fa7c0 · 18fa7c0
1 parent 63007a0
commit 18fa7c0
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/apps/web/src/helpers/authLink.ts b/apps/web/src/helpers/authLink.ts
@@ -22,7 +22,12 @@ const REFRESH_AUTHENTICATION_MUTATION = `
 
 let refreshPromise: Promise<string> | null = null;
 
-const executeTokenRefresh = async (refreshToken: string): Promise<string> => {
+const MAX_RETRIES = 5;
+
+const executeTokenRefresh = async (
+  refreshToken: string,
+  attempt = 0
+): Promise<string> => {
   try {
     const response = await axios.post(
       LENS_API_URL,
@@ -37,9 +42,14 @@ const executeTokenRefresh = async (refreshToken: string): Promise<string> => {
     const {
       accessToken,
       refreshToken: newRefreshToken,
-      idToken
+      idToken,
+      __typename
     } = response?.data?.data?.refresh ?? {};
 
+    if (__typename !== "AuthenticationTokens" && attempt < MAX_RETRIES) {
+      return executeTokenRefresh(refreshToken, attempt + 1);
+    }
+
     if (!accessToken || !newRefreshToken) {
       signOut();
       throw new Error("Invalid refresh response");
@@ -59,6 +69,7 @@ const refreshTokens = (refreshToken: string): Promise<string> => {
   if (!refreshPromise) {
     refreshPromise = executeTokenRefresh(refreshToken);
   }
+
   return refreshPromise;
 };