fix: 시간표 수정하면 메인시간표로만 돌아가는 현상 수정

BCSDLab · Nov 30, 2024 · e463a3d · e463a3d
1 parent 651800a
commit e463a3d
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 12 deletions.
diff --git a/src/pages/TimetablePage/MainTimetablePage/index.tsx b/src/pages/TimetablePage/MainTimetablePage/index.tsx
@@ -5,6 +5,7 @@ import useTimetableFrameList from 'pages/TimetablePage/hooks/useTimetableFrameLi
 import useMediaQuery from 'utils/hooks/layout/useMediaQuery';
 import useScrollToTop from 'utils/hooks/ui/useScrollToTop';
 import useTokenState from 'utils/hooks/state/useTokenState';
+import { useLocation } from 'react-router-dom';
 import DefaultPage from './DefaultPage';
 import styles from './TimetablePage.module.scss';
 
@@ -13,21 +14,25 @@ function TimetablePage() {
   useScrollToTop();
   const token = useTokenState();
   const semester = useSemester();
-  const [currentFrameIndex, setCurrentFrameIndex] = React.useState(0);
+  const location = useLocation();
   const { data: timetableFrameList } = useTimetableFrameList(token, semester);
-  sessionStorage.setItem('enterTimetablePage', new Date().getTime().toString());
+  const mainFrame = timetableFrameList.find(
+    (frame) => frame.is_main === true,
+  );
+  const [currentFrameIndex, setCurrentFrameIndex] = React.useState(
+    mainFrame?.id ? mainFrame.id : 0,
+  );
 
   React.useEffect(() => {
-    if (timetableFrameList) {
-      const mainFrame = timetableFrameList.find(
-        (frame) => frame.is_main === true,
-      );
-      if (mainFrame && mainFrame.id) {
-        setCurrentFrameIndex(mainFrame.id);
-      }
+    if (location.state?.frameId) {
+      setCurrentFrameIndex(Number(location.state?.frameId));
+    } else {
+      setCurrentFrameIndex(mainFrame?.id ? mainFrame.id : 0);
     }
-    // eslint-disable-next-line react-hooks/exhaustive-deps
-  }, [semester]);
+  // eslint-disable-next-line react-hooks/exhaustive-deps
+  }, []);
+
+  sessionStorage.setItem('enterTimetablePage', new Date().getTime().toString());
 
   return (
     <div className={styles.page}>

diff --git a/src/pages/TimetablePage/ModifyTimetablePage/DefaultPage/index.tsx b/src/pages/TimetablePage/ModifyTimetablePage/DefaultPage/index.tsx
@@ -99,7 +99,7 @@ export default function DefaultPage({ frameId }: { frameId: string | undefined }
               <button
                 type="button"
                 className={styles['page__save-button']}
-                onClick={() => navigate('/timetable')}
+                onClick={() => navigate('/timetable', { state: { frameId } })}
               >
                 <div className={styles['page__pen-icon']}>
                   <PenIcon />