dashboards/apps/customer/src/pages/checkout/CheckoutFlow.tsx

import { Navigate, Outlet, useLocation } from 'react-router-dom'
import { RouteLoader } from '@meshkee/dashboard-ui'
import { useAuth } from '../../context/AuthContext'
import { CheckoutStepper, type CheckoutStepId } from '../../components/checkout/CheckoutStepper'
import { loginPathWithRedirect, safeRedirectPath } from '../../lib/redirect'

/** Cart and checkout login are public. Delivery, payment, and result require auth. */
const PUBLIC_STEPS = new Set<CheckoutStepId>(['cart', 'login'])

function stepFromPath(pathname: string): CheckoutStepId {
  if (pathname.endsWith('/login')) return 'login'
  if (pathname.endsWith('/delivery')) return 'delivery'
  if (pathname.endsWith('/payment')) return 'payment'
  if (
    pathname.endsWith('/success') ||
    pathname.endsWith('/failed') ||
    pathname.endsWith('/result')
  ) {
    return 'payment'
  }
  return 'cart'
}

export function CheckoutFlow() {
  const { user, isLoading } = useAuth()
  const location = useLocation()
  const step = stepFromPath(location.pathname)
  const isAuthenticated = Boolean(user)

  if (isLoading) {
    return <RouteLoader />
  }

  if (!isAuthenticated && !PUBLIC_STEPS.has(step)) {
    const returnTo = `${location.pathname}${location.search}`
    return (
      <Navigate
        to={loginPathWithRedirect(returnTo, '/checkout/login')}
        replace
        state={{ from: returnTo }}
      />
    )
  }

  if (isAuthenticated && step === 'login') {
    const params = new URLSearchParams(location.search)
    const fromQuery = safeRedirectPath(params.get('redirect'), '')
    const fromState = (location.state as { from?: string } | null)?.from
    const destination = fromQuery || fromState || '/checkout/delivery'
    return <Navigate to={destination} replace />
  }

  const hideStepper =
    location.pathname === '/checkout/success' ||
    location.pathname === '/checkout/failed' ||
    location.pathname === '/checkout/result'

  return (
    <>
      {step !== 'login' && !hideStepper && (
        <CheckoutStepper current={step} isAuthenticated={isAuthenticated} />
      )}
      <Outlet />
    </>
  )
}