From 6e04bbcedba4c2a11793bc09150439a23a926ab2 Mon Sep 17 00:00:00 2001
From: Kevin Hermawan <kevin@hyperjump.tech>
Date: Wed, 17 Jul 2024 15:02:56 +0700
Subject: [PATCH] fix: set pathname when the page completely loaded

---
 packages/react-next-pathname-nextjs/src/index.tsx | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/packages/react-next-pathname-nextjs/src/index.tsx b/packages/react-next-pathname-nextjs/src/index.tsx
index 59bee39..3b3c06c 100644
--- a/packages/react-next-pathname-nextjs/src/index.tsx
+++ b/packages/react-next-pathname-nextjs/src/index.tsx
@@ -19,10 +19,16 @@ export function NextPathnameProvider({ children }: { children: ReactNode }) {
   const [nextPathname, setNextPathname] = useState(router.pathname);
 
   useEffect(() => {
-    router.events.on("routeChangeStart", setNextPathname);
+    const handleRouteChange = (url: string) => {
+      setNextPathname(url);
+    };
+
+    router.events.on("routeChangeStart", handleRouteChange);
+    router.events.on("routeChangeComplete", handleRouteChange);
 
     return () => {
-      router.events.off("routeChangeStart", setNextPathname);
+      router.events.off("routeChangeStart", handleRouteChange);
+      router.events.off("routeChangeComplete", handleRouteChange);
     };
   }, [router.events]);