Fixed updateUnits (but it goes away on refresh)

ScottyLabs · Nov 30, 2024 · b0d47a7 · b0d47a7
1 parent 2416be7
commit b0d47a7
Showing 1 changed file with 23 additions and 9 deletions.
diff --git a/apps/frontend/src/components/ScheduleData.tsx b/apps/frontend/src/components/ScheduleData.tsx
@@ -12,13 +12,16 @@ import { ChevronDownIcon, ChevronUpIcon } from "@heroicons/react/20/solid";
 import { useFetchCourseInfos } from "~/app/api/course";
 import { useFetchFCEInfosByCourse } from "~/app/api/fce";
 import { useAuth } from "@clerk/nextjs";
+import { useQueryClient } from "@tanstack/react-query";
+import { Course } from "~/app/types";
 
 type ScheduleDataProps = {
   scheduled: string[];
 };
 
 const ScheduleData = ({ scheduled }: ScheduleDataProps) => {
   const dispatch = useAppDispatch();
+  const queryClient = useQueryClient();
 
   const active = useAppSelector((state) => state.schedules.active);
   const selected = useAppSelector(selectSelectedCoursesInActiveSchedule);
@@ -80,6 +83,18 @@ const ScheduleData = ({ scheduled }: ScheduleDataProps) => {
       );
   };
 
+  const updateManualUnits = (courseID: string, manualUnits: string) => {
+    queryClient.setQueryData(
+      ["courseInfo", { courseID }],
+      (oldData: Course) => {
+        return {
+          ...oldData,
+          manualUnits,
+        };
+      }
+    );
+  };
+
   return (
     <>
       <div className="flex items-end justify-between">
@@ -162,15 +177,14 @@ const ScheduleData = ({ scheduled }: ScheduleDataProps) => {
                                 ? displayUnits(result.manualUnits)
                                 : displayUnits(result.units)
                             }
-                            // TODO: Fix updateUnits
-                            // onChange={(e) =>
-                            //   dispatch(
-                            //     cacheSlice.actions.updateUnits({
-                            //       courseID: result.courseID,
-                            //       units: e.target.value,
-                            //     })
-                            //   )
-                            // }
+                            onClick={() => {
+                              if (result.manualUnits === undefined) {
+                                updateManualUnits(result.courseID, "");
+                              }
+                            }}
+                            onChange={(e) =>
+                              updateManualUnits(result.courseID, e.target.value)
+                            }
                             placeholder="Units"
                           />
                         ) : (