fix: 뒤로가기 오류 수정

i-Tracker · Jul 19, 2024 · d0cd41a · d0cd41a
1 parent 6c0080a
commit d0cd41a
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/src/shared/components/Header/HeaderWithBackRoute/index.tsx b/src/shared/components/Header/HeaderWithBackRoute/index.tsx
@@ -1,16 +1,19 @@
 import { ArrowBack } from '@/shared/assets/Icons';
-import { useRouter } from 'next/navigation';
+import { useRouter, usePathname } from 'next/navigation';
 
 const HeaderWithBackRoute = () => {
   const router = useRouter();
+  const pathName = usePathname();
+
+  const getProductCategory = () => {
+    const parts = pathName.split('/');
+    return parts.length > 2 ? parts[2] : '';
+  };
 
   const goToBackPage = () => {
-    if (window.history.length > 1) {
-      router.back();
-    } else {
-      router.push('/category/macbook_air');
-    }
+    router.push(`/category/${getProductCategory()}`);
   };
+
   return (
     <header className="w-full flex items-center">
       <nav className="w-[1080px] h-[70px] flex items-center justify-between mx-auto border-b">