fix: create useLazyPathname hook

components/Current/Header/LoginButton.tsx

@@ -1,6 +1,5 @@
 "use client"
 
-import { usePathname } from "next/navigation"
 import { PropsWithChildren, useEffect } from "react"
 
 import { login } from "@/constants/routes/handleAuth"
@@ -9,6 +8,7 @@ import Link from "@/components/TempDesignSystem/Link"
 import { LinkProps } from "@/components/TempDesignSystem/Link/link"
 import useLang from "@/hooks/useLang"
 import { trackLoginClick } from "@/utils/tracking"
+import { useLazyPathname } from "@/utils/url"
 
 import { TrackingPosition } from "@/types/components/tracking"
 
@@ -25,7 +25,11 @@ export default function LoginButton({
   color?: LinkProps["color"]
 }>) {
   const lang = useLang()
-  const pathName = usePathname()
+  const pathName = useLazyPathname()
+
+  const href = pathName
+    ? `${login[lang]}?redirectTo=${encodeURIComponent(pathName)}`
+    : login[lang]
 
   useEffect(() => {
     document
@@ -43,7 +47,7 @@ export default function LoginButton({
       className={className}
       id={trackingId}
       color={color}
-      href={`${login[lang]}?redirectTo=${encodeURIComponent(pathName)}`}
+      href={href}
       prefetch={false}
     >
       {children}

utils/url.ts

@@ -1,3 +1,6 @@
+import { usePathname } from "next/navigation"
+import { useEffect, useState } from "react"
+
 export function removeMultipleSlashes(str: string) {
   return str.replaceAll(/\/\/+/g, "/")
 }
@@ -9,3 +12,16 @@ export function removeTrailingSlash(pathname: string) {
   }
   return pathname
 }
+
+/*** This hook is used to get the current pathname (as reflected in window.location.href) of the page. During ssr, the value from usePathname()
+ *  is the value return from NextResponse.rewrite() (e.g. the path from the app directory) instead of the actual pathname from the URL.
+ */
+export function useLazyPathname() {
+  const pathName = usePathname()
+  const [updatedPathName, setUpdatedPathName] = useState<string | null>(null)
+
+  useEffect(() => {
+    setUpdatedPathName(pathName)
+  }, [pathName])
+  return updatedPathName ? updatedPathName : null
+}