From 7e7e9790cc6f8a174c770a0b55f8c68abf7a3ae4 Mon Sep 17 00:00:00 2001
From: yeyounging <133792082+yeyounging@users.noreply.github.com>
Date: Thu, 28 Nov 2024 23:09:17 +0900
Subject: [PATCH] HOTFIX

---
 src/app/auth/callback/page.tsx       |  1 +
 src/app/home/components/TimeCard.tsx |  2 +-
 src/app/page.tsx                     |  1 -
 src/app/start/api/api.ts             |  2 ++
 src/app/start/components/Step3.tsx   |  2 +-
 src/app/start/components/Step4.tsx   |  4 ----
 src/app/start/page.tsx               |  2 +-
 src/middleware.ts                    | 31 ++++++++++++++++++++++++++++
 8 files changed, 37 insertions(+), 8 deletions(-)
 create mode 100644 src/middleware.ts

diff --git a/src/app/auth/callback/page.tsx b/src/app/auth/callback/page.tsx
index fb5137c..1cf36de 100644
--- a/src/app/auth/callback/page.tsx
+++ b/src/app/auth/callback/page.tsx
@@ -43,6 +43,7 @@ function LoginCheck() {
 
       const data = await res.json()
       Cookies.set('accessToken', data.data.accessToken)
+      Cookies.set('role', data.data.userInfo.role)
       setUserInfo({
         ...data.data.userInfo,
         profileImage:
diff --git a/src/app/home/components/TimeCard.tsx b/src/app/home/components/TimeCard.tsx
index 2cbbcc8..1750037 100644
--- a/src/app/home/components/TimeCard.tsx
+++ b/src/app/home/components/TimeCard.tsx
@@ -1,5 +1,5 @@
 import { categoryLabels } from '@/app/archive/api/types'
-import { Badge, Cup, Div } from '@/components'
+import { Badge, Div } from '@/components'
 import Image from 'next/image'
 
 interface TimeCardProps {
diff --git a/src/app/page.tsx b/src/app/page.tsx
index 589e28d..69cb02e 100644
--- a/src/app/page.tsx
+++ b/src/app/page.tsx
@@ -12,7 +12,6 @@ export default function Home() {
   const [logoColor, setlogoColor] = useState('white')
 
   useEffect(() => {
-
     const timer = setTimeout(() => {
       setIsSplash(false)
     }, 1200)
diff --git a/src/app/start/api/api.ts b/src/app/start/api/api.ts
index f4950ed..07e6c9a 100644
--- a/src/app/start/api/api.ts
+++ b/src/app/start/api/api.ts
@@ -1,3 +1,4 @@
+import Cookies from 'js-cookie'
 import { http } from '@/api'
 import useUserInfo, { UserInfo } from '@/store/useUserInfo'
 import { useMutation, useSuspenseQuery } from '@tanstack/react-query'
@@ -24,6 +25,7 @@ export const usePostOnboard = () => {
     onSuccess: ({ data }) => {
       const updatedUserInfo = { ...data, role: 'MEMBER' as const }
       setUserInfo(updatedUserInfo)
+      Cookies.set('role', 'MEMBER')
       router.push('/home')
     },
     onError: (error) => {
diff --git a/src/app/start/components/Step3.tsx b/src/app/start/components/Step3.tsx
index fb59140..4a739c5 100644
--- a/src/app/start/components/Step3.tsx
+++ b/src/app/start/components/Step3.tsx
@@ -8,7 +8,7 @@ export default function Step3() {
     useProfileSelector()
 
   return (
-    <div className=" px-20 w-full flex flex-col relative]">
+    <div className="px-20 w-full flex flex-col relative">
       <h1 className="title !mb-6">어떤 프로필로 함께 하시겠어요?</h1>
       <h2 className="text-primary_foundation_50 text-sm font-medium leading-snug">
         프로필은 나중에 바꿀 수 있어요
diff --git a/src/app/start/components/Step4.tsx b/src/app/start/components/Step4.tsx
index 2c52326..64e219b 100644
--- a/src/app/start/components/Step4.tsx
+++ b/src/app/start/components/Step4.tsx
@@ -6,10 +6,6 @@ import Image from 'next/image'
 export default function Step4() {
   const { userInfo } = useUserInfo()
 
-  if (!userInfo.nickname) {
-    return <div>loading...</div>
-  }
-
   return (
     <div className="relative h-screen">
       <div className="absolute w-screen h-full bg-gradient-to-b from-white via-[#fffbfb] to-[#ffa89c]" />
diff --git a/src/app/start/page.tsx b/src/app/start/page.tsx
index 2d2d170..f54bfe7 100644
--- a/src/app/start/page.tsx
+++ b/src/app/start/page.tsx
@@ -73,7 +73,7 @@ export default function Start() {
           </If>
         </div>
 
-        <div className="w-full py-4 flex justify-center">
+        <div className="w-full pt-10 flex justify-center relative">
           <Button
             className="w-[90%] mx-auto"
             disabled={!!error}
diff --git a/src/middleware.ts b/src/middleware.ts
new file mode 100644
index 0000000..f6d742f
--- /dev/null
+++ b/src/middleware.ts
@@ -0,0 +1,31 @@
+import { NextResponse } from 'next/server'
+import type { NextRequest } from 'next/server'
+
+export function middleware(req: NextRequest) {
+  const token = req.cookies.get('accessToken')?.value
+  const role = req.cookies.get('role')?.value
+  const response = NextResponse.next()
+  const { pathname } = req.nextUrl
+
+  if (
+    pathname.startsWith('/_next') ||
+    pathname.startsWith('/static') ||
+    pathname.includes('.')
+  ) {
+    return response
+  }
+
+  if (pathname !== '/') {
+    if (!token) {
+      return NextResponse.redirect('/')
+    }
+  }
+
+  if (pathname !== '/start' && role === 'MEMBER') {
+    return NextResponse.redirect(new URL('/start', req.url))
+  }
+
+  return NextResponse.next()
+}
+
+export const config = {}