From 5a79523ba0638a6feaf567635f675b756ecea253 Mon Sep 17 00:00:00 2001
From: austinkelsay <austinkelsay@protonmail.com>
Date: Mon, 12 May 2025 09:37:32 -0500
Subject: [PATCH] clean up decryption and retry timers on unmount to prevent
 memory leaks

---
 src/hooks/encryption/useCourseDecryption.js | 20 ++++++++++++++++++--
 1 file changed, 18 insertions(+), 2 deletions(-)

diff --git a/src/hooks/encryption/useCourseDecryption.js b/src/hooks/encryption/useCourseDecryption.js
index e7a06f6..2f57781 100644
--- a/src/hooks/encryption/useCourseDecryption.js
+++ b/src/hooks/encryption/useCourseDecryption.js
@@ -8,6 +8,8 @@ const useCourseDecryption = (session, paidCourse, course, lessons, setLessons, r
   const processingRef = useRef(false);
   const lastLessonIdRef = useRef(null);
   const retryCountRef = useRef({});
+  const retryTimeoutRef = useRef(null);
+  const decryptTimeoutRef = useRef(null);
   const MAX_RETRIES = 3;
   
   // Get the current active lesson using the activeIndex prop instead of router.query
@@ -70,12 +72,12 @@ const useCourseDecryption = (session, paidCourse, course, lessons, setLessons, r
       let timeoutId;
       const timeoutPromise = new Promise((_, reject) => {
         timeoutId = setTimeout(() => {
-          // Cancel the in-flight request when timeout occurs
           if (decryptionPromise.cancel) {
             decryptionPromise.cancel();
           }
           reject(new Error('Decryption timeout'));
         }, 10000);
+        decryptTimeoutRef.current = timeoutId;
       });
       
       // Use a separate try-catch for the race
@@ -89,9 +91,10 @@ const useCourseDecryption = (session, paidCourse, course, lessons, setLessons, r
         
         // Clear the timeout if decryption wins
         clearTimeout(timeoutId);
+        decryptTimeoutRef.current = null;
       } catch (error) {
         // If timeout or network error, schedule a retry
-        setTimeout(() => {
+        retryTimeoutRef.current = setTimeout(() => {
           processingRef.current = false;
           decryptCurrentLesson();
         }, 5000);
@@ -138,6 +141,19 @@ const useCourseDecryption = (session, paidCourse, course, lessons, setLessons, r
     }
   }, [currentLessonId, hasAccess, paidCourse, decryptedLessonIds, decryptCurrentLesson, activeIndex]);
   
+  useEffect(() => {
+    return () => {
+      if (decryptTimeoutRef.current) {
+        clearTimeout(decryptTimeoutRef.current);
+        decryptTimeoutRef.current = null;
+      }
+      if (retryTimeoutRef.current) {
+        clearTimeout(retryTimeoutRef.current);
+        retryTimeoutRef.current = null;
+      }
+    };
+  }, []);
+  
   return {
     decryptionPerformed: isCurrentLessonDecrypted,
     loading,