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.";