URGENT fix: prevent non-course object from being traversed

nthumodifications · Aug 19, 2024 · 8d66b61 · 8d66b61
1 parent a7ddf70
commit 8d66b61
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/hooks/contexts/useUserTimetable.tsx b/src/hooks/contexts/useUserTimetable.tsx
@@ -114,6 +114,14 @@ const useUserTimetableProvider = (loadCourse = true) => {
 
       const coursesCopy = { ...courses };
 
+      // Check and confirm if coursesCopy is traversable
+      if (Object.keys(coursesCopy).length === 0) return;
+      if (Object.keys(coursesCopy).find((sem) => sem.length !== 5)) return;
+      if (
+        Object.keys(coursesCopy).find((sem) => !Array.isArray(coursesCopy[sem]))
+      )
+        return;
+
       Object.keys(coursesCopy).forEach((sem) => {
         (coursesCopy[sem] ?? []).forEach((courseID, i) => {
           newColorMap[courseID] = newColors[i % newColors.length];