From 65cb9bde313b6bd860e9c9d339162262f2970f8a Mon Sep 17 00:00:00 2001
From: Muthu Kumar <muthukumar@thefeathers.in>
Date: Thu, 10 Apr 2025 05:22:58 +0530
Subject: [PATCH] fix: Container breakage after unifying

---
 src/components/Container.tsx | 82 ++++++++++++++++++++------------------------
 src/components/Menu.tsx      |  1 +
 2 files changed, 39 insertions(+), 44 deletions(-)

diff --git a/src/components/Container.tsx b/src/components/Container.tsx
index d697f7d..aaf0b2d 100644
--- a/src/components/Container.tsx
+++ b/src/components/Container.tsx
@@ -5,7 +5,7 @@ import useLocation from "wouter/use-location";
 import { ReactComponent as Logo } from "../assets/logo.svg";
 import { ReactComponent as Right } from "../assets/arrow-right.svg";
 import { get, getTimeout } from "../util";
-import Menu, { MenuEntries } from "./Menu";
+import Menu, { MenuEntries, MenuPaths } from "./Menu";
 import useMediaQuery from "../util/useMediaQuery";
 import { AnimateEntry } from "./AnimateEntry";
 
@@ -28,32 +28,6 @@ const Container: React.FC<{
 
 	const [showMenu, setShowMenu] = useState(false);
 
-	useEffect(() => {
-		// scroll back to top when new page is loaded
-		window.scrollTo({ top: 0 });
-
-		if (highlightCircle.current) {
-			highlightCircle.current.classList.add("highlight");
-			timer(
-				() =>
-					highlightCircle.current &&
-					highlightCircle.current.classList.remove("highlight"),
-				1500,
-			);
-		}
-
-		if (nextBtn.current) {
-			nextBtn.current.style.width = "4rem";
-			timer(
-				() => nextBtn.current && (nextBtn.current.style.right = "10vw"),
-				300,
-			);
-		}
-
-		// cleanup
-		return clear;
-	}, []);
-
 	const handleResize = () => {
 		if (containerChild.current && logoContainer.current)
 			logoContainer.current.style.left = `${
@@ -86,44 +60,64 @@ const Container: React.FC<{
 		} catch {}
 	};
 
-	const current = MenuEntries.findIndex(
-		([, path]) => location === path || location.startsWith(path + "/"),
-	);
-	const next = get.next(MenuEntries, current)[1];
-	const prev = get.prev(MenuEntries, current)[1];
-	const end = current === MenuEntries.length - 1;
-
 	const handlePrev = (e: MouseKb) => {
 		animateArrow(e);
-		timer(() => prev && navigate(prev), 300);
+
+		const current = MenuPaths.findIndex(path => location === path);
+		const index = (current - 1) % MenuPaths.length;
+		const prev = MenuPaths.at(index)!;
+		timer(() => navigate(prev), 300);
 	};
 
 	const handleNext = (e: MouseKb) => {
 		animateArrow(e);
-		timer(() => next && navigate(next), 300);
+
+		const current = MenuPaths.findIndex(path => location === path);
+		const index = (current + 1) % MenuPaths.length;
+		const next = MenuPaths.at(index)!;
+		timer(() => navigate(next), 300);
 	};
 
 	function kbnav(e: KeyboardEvent) {
 		if (e.altKey || e.ctrlKey || e.metaKey || e.shiftKey) return;
 
-		switch (e.key) {
-			case "ArrowLeft":
-				return handlePrev(e);
-			case "ArrowRight":
-				return handleNext(e);
-		}
+		if (e.key === "ArrowLeft") return handlePrev(e);
+		else if (e.key === "ArrowRight") return handleNext(e);
 	}
 
 	useEffect(() => {
+		// scroll back to top when new page is loaded
+		window.scrollTo({ top: 0 });
+
+		if (highlightCircle.current) {
+			highlightCircle.current.classList.add("highlight");
+			timer(
+				() =>
+					highlightCircle.current &&
+					highlightCircle.current.classList.remove("highlight"),
+				1500,
+			);
+		}
+
+		if (nextBtn.current) {
+			nextBtn.current.style.width = "3rem";
+			timer(
+				() => nextBtn.current && (nextBtn.current.style.right = "10vw"),
+				300,
+			);
+		}
+
 		window.addEventListener("keydown", kbnav);
 
 		// cleanup
-		return () => window.removeEventListener("keydown", kbnav);
-	}, []);
+		return () => (window.removeEventListener("keydown", kbnav), clear());
+	}, [location]);
 
 	// on first render
 	useLayoutEffect(handleResize, []);
 
+	const end = location === MenuEntries[MenuEntries.length - 1][1];
+
 	return (
 		<div
 			className={css`
diff --git a/src/components/Menu.tsx b/src/components/Menu.tsx
index 59a48b7..d5c9edb 100644
--- a/src/components/Menu.tsx
+++ b/src/components/Menu.tsx
@@ -12,6 +12,7 @@ export const MENU = {
 } as const;
 
 export const MenuEntries = Object.entries(MENU);
+export const MenuPaths = Object.values(MENU);
 
 const desktopNav = css`
 	float: right;