From 5dc58ffa42be10b4d1bc542540e9307ff0bfeeb9 Mon Sep 17 00:00:00 2001
From: Ryan <ryantl66@gmail.com>
Date: Fri, 4 Apr 2025 02:29:27 -0400
Subject: [PATCH] loadCsrfTokenWithRetry

---
 main.js | 21 +++++++++++++++------
 1 file changed, 15 insertions(+), 6 deletions(-)

diff --git a/main.js b/main.js
index b9b9e96..0fb6cf9 100644
--- a/main.js
+++ b/main.js
@@ -20,9 +20,14 @@ import { setupTrashRestoreDelete } from './trashRestoreDelete.js';
 import { initDragAndDrop, loadSidebarOrder, loadHeaderOrder } from './dragAndDrop.js';
 import { initTagSearch, openTagModal, filterFilesByTag } from './fileTags.js';
 
-function loadCsrfToken() {
+function loadCsrfTokenWithRetry(retries = 3, delay = 1000) {
   return fetch('token.php', { credentials: 'include' })
-    .then(response => response.json())
+    .then(response => {
+      if (!response.ok) {
+        throw new Error("Token fetch failed with status: " + response.status);
+      }
+      return response.json();
+    })
     .then(data => {
       // Set global variables.
       window.csrfToken = data.csrf_token;
@@ -49,7 +54,12 @@ function loadCsrfToken() {
       return data;
     })
     .catch(error => {
-      console.error("Error loading CSRF token and share URL:", error);
+      if (retries > 0) {
+        console.warn(`CSRF token load failed. Retrying in ${delay}ms... (${retries} retries left)`, error);
+        return new Promise(resolve => setTimeout(resolve, delay))
+          .then(() => loadCsrfTokenWithRetry(retries - 1, delay * 2));
+      }
+      console.error("Failed to load CSRF token after retries.", error);
       throw error;
     });
 }
@@ -66,8 +76,8 @@ window.renameFile = renameFile;
 window.currentFolder = "root";
 
 document.addEventListener("DOMContentLoaded", function () {
-  // First, load the CSRF token.
-  loadCsrfToken().then(() => {
+  // First, load the CSRF token (with retry).
+  loadCsrfTokenWithRetry().then(() => {
     // Once CSRF token is loaded, initialize authentication.
     initAuth();
 
@@ -163,7 +173,6 @@ document.addEventListener("DOMContentLoaded", function () {
       sessionStorage.removeItem("welcomeMessage");
     }
   }).catch(error => {
-    // If the CSRF token fails to load, you might choose to show an error or retry.
     console.error("Initialization halted due to CSRF token load failure.", error);
   });