diff --git a/src/components/Navbar.jsx b/src/components/Navbar.jsx index 1e72bcc..63888f4 100644 --- a/src/components/Navbar.jsx +++ b/src/components/Navbar.jsx @@ -7,6 +7,7 @@ import { GiHamburgerMenu } from "react-icons/gi"; import { ThemeContext } from '../themeContext'; const Navbar = () => { + const canteenId = localStorage.getItem("canteenId"); const { theme, toggleTheme } = useContext(ThemeContext); const [isOpen, setIsOpen] = useState(false); const { scrollYProgress } = useScroll(); @@ -37,6 +38,13 @@ const Navbar = () => { }>About }>News }>RateUs + {/* Conditionally render "My Canteen" button */} + {canteenId && ( + }>My Canteen + )} + + +
}>Home @@ -94,6 +102,38 @@ const Navbar = () => { News Contact + + + {isOpen && ( + +
+ Home + About us + News + Contact + RateUs + {/* Conditionally render "My Canteen" button */} + {canteenId && ( + My Canteen + )} + + + +
+
+ )} +
+ + Rateus @@ -146,5 +186,6 @@ const IconHome = () => 🏠; const IconAbout = () => ℹī¸; const IconNews = () => 📰; const IconRateUs = () => ⭐; +const IconCanteen = () => đŸĨ— export default Navbar; diff --git a/src/pages/Login.jsx b/src/pages/Login.jsx index 5374fd8..8a0cc25 100644 --- a/src/pages/Login.jsx +++ b/src/pages/Login.jsx @@ -67,9 +67,12 @@ function Login() { if (formData.accountType === "User") { navigate("/home"); } else { + navigate("/home"); + localStorage.setItem("canteenId",response.data.cantId); localStorage.setItem("token",response.data.token); navigate(`/section/${response.data.cantId}`); + } } catch (error) { toast.error("Failed to login"); diff --git a/src/pages/Signup.jsx b/src/pages/Signup.jsx index 67493be..6f87557 100644 --- a/src/pages/Signup.jsx +++ b/src/pages/Signup.jsx @@ -77,7 +77,12 @@ function Signup() { const token = response.data.token; localStorage.setItem("token", token); localStorage.setItem("canteenId", response.data.cantId); + navigate("/home"); + } else { + navigate("/"); + navigate(`/section/${response.data.cantId}`); + } } catch (error) { const errorMessage = error.response?.data?.message || "Failed to create account. Please try again.";