Refactor layout and onboarding components for improved clarity and fu…

apps/web/src/actions/auth/sign-in.ts

@@ -13,6 +13,15 @@ type SignInInput = {
   redirectTo?: string
 }
 
+function getLocaleFromRedirectTo(redirectTo?: string) {
+  if (!redirectTo) return null
+
+  const locale = redirectTo.split('/').filter(Boolean)[0]
+  const isValidLocale = /^[a-z]{2}-[A-Z]{2}$/.test(locale ?? '')
+
+  return isValidLocale ? locale : null
+}
+
 export async function signIn({ login, password, redirectTo }: SignInInput) {
   let token: string | undefined
 
@@ -37,7 +46,9 @@ export async function signIn({ login, password, redirectTo }: SignInInput) {
 
     if (data?.user && !data.user.displayName) {
       const cookieStore = await cookies()
+      const localeFromRedirect = getLocaleFromRedirectTo(redirectTo)
       const lang =
+        localeFromRedirect ||
         cookieStore.get('NEXT_LOCALE')?.value ||
         cookieStore.get('i18next')?.value ||
         'en-US'

apps/web/src/app/[lang]/layout.tsx

@@ -1,4 +1,3 @@
-import { headers } from 'next/headers'
 import { Link } from 'next-view-transitions'
 import type { GetUserPreferences200 } from '@/api/endpoints.schemas'
 import { getUserPreferences } from '@/api/users'
@@ -43,10 +42,6 @@ export default async function RootLayout({
     userPreferences = data?.userPreferences ?? null
   }
 
-  const headersList = await headers()
-  const pathname = headersList.get('x-current-path') || ''
-  const isOnboarding = pathname.includes('/onboarding')
-
   return (
     <ThemeProvider
       attribute="class"
@@ -63,7 +58,6 @@ export default async function RootLayout({
                 <LayoutWrapper
                   header={<Header />}
                   footer={<Footer dictionary={dictionary} language={lang} />}
-                  isOnboarding={isOnboarding}
                   proBadge={
                     session?.user.subscriptionType !== 'PRO' ? (
                       <Link

apps/web/src/app/[lang]/onboarding/_components/steps/onboarding-celebration.tsx

@@ -49,7 +49,7 @@ export function OnboardingCelebration({ lang }: { lang: string }) {
   const subtitle =
     dictionary?.celebration_subtitle ||
     "Your profile is ready. Let's start tracking."
-  const cta = dictionary?.celebration_cta || 'Go to Home'
+  const cta = dictionary?.go_to_profile || 'Go to Profile'
 
   return (
     <div className="flex flex-1 h-full flex-col items-center justify-center px-6">

apps/web/src/components/layout-wrapper.tsx

@@ -1,16 +1,20 @@
+'use client'
+
+import { usePathname } from 'next/navigation'
+
 export function LayoutWrapper({
   header,
   footer,
   children,
   proBadge,
-  isOnboarding,
 }: {
   header: React.ReactNode
   footer: React.ReactNode
   children: React.ReactNode
   proBadge?: React.ReactNode
-  isOnboarding?: boolean
 }) {
+  const pathname = usePathname()
+  const isOnboarding = pathname?.includes('/onboarding') ?? false
 
   return (
     <>