✨ 리프레시 훅 적용

epigram5-9 · Aug 6, 2024 · a29dcb9 · a29dcb9
1 parent b39ad83
commit a29dcb9
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 6 deletions.
diff --git a/src/pages/auth/SignIn.tsx b/src/pages/auth/SignIn.tsx
@@ -9,17 +9,21 @@ import { zodResolver } from '@hookform/resolvers/zod';
 import { Form, FormControl, FormField, FormItem, FormMessage } from '@/components/ui/form';
 import { PostSigninRequest, PostSigninRequestType } from '@/schema/auth';
 import useSigninMutation from '@/hooks/useSignInMutation';
+import useRefreshToken from '@/hooks/useRefreshToken';
 
 export default function SignIn() {
   const mutationSignin = useSigninMutation();
+  const { mutate: refreshAccessToken } = useRefreshToken();
   const router = useRouter();
 
   useEffect(() => {
-    const accessToken = localStorage.getItem('accessToken');
-    if (accessToken) {
+    const refreshToken = typeof window !== 'undefined' ? localStorage.getItem('refreshToken') : null;
+
+    if (refreshToken) {
+      refreshAccessToken({ refreshToken });
       router.push('/epigrams');
     }
-  }, [router]);
+  }, [refreshAccessToken]);
 
   // 폼 정의
   const form = useForm<PostSigninRequestType>({

diff --git a/src/pages/auth/SignUp.tsx b/src/pages/auth/SignUp.tsx
@@ -9,17 +9,21 @@ import { Input } from '@/components/ui/input';
 import { Button } from '@/components/ui/button';
 import { Form, FormControl, FormField, FormItem, FormLabel, FormMessage } from '@/components/ui/form';
 import useRegisterMutation from '@/hooks/useRegisterMutation';
+import useRefreshToken from '@/hooks/useRefreshToken';
 
 export default function SignUp() {
   const [focusedField, setFocusedField] = useState<string | null>(null);
+  const { mutate: refreshAccessToken } = useRefreshToken();
   const router = useRouter();
 
   useEffect(() => {
-    const accessToken = localStorage.getItem('accessToken');
-    if (accessToken) {
+    const refreshToken = typeof window !== 'undefined' ? localStorage.getItem('refreshToken') : null;
+
+    if (refreshToken) {
+      refreshAccessToken({ refreshToken });
       router.push('/epigrams');
     }
-  }, [router]);
+  }, [refreshAccessToken]);
 
   const form = useForm<PostSignUpRequestType>({
     resolver: zodResolver(PostSignUpRequest),