SessionCheck.tsx

'use client'

import { usePathname, useRouter, useSearchParams } from 'next/navigation'

import { useEffect } from 'react'
import { useSession } from 'next-auth/react'

const SessionCheck = ({
  children,
}: {
  children: React.ReactNode
}): JSX.Element | null => {
  const { data: session, status } = useSession()
  const router = useRouter()
  const searchParams = useSearchParams()
  const pathname = usePathname()

  const redirectTo = searchParams.get('redirectTo') ?? '/dashboard'

  const preventRedirectOnPaths = [
    '/organizations/create-organization',
    '/organizations/agent-config',
    '/organizations/dashboard',
  ]

  useEffect(() => {
    if (status === 'loading') {
      return
    }

    const isOnRestrictedPage = preventRedirectOnPaths.some((page) =>
      pathname.startsWith(page),
    )

    if (session && redirectTo && !isOnRestrictedPage) {
      router.push(redirectTo)
    }

    if (session === null) {
      localStorage.removeItem('persist:root')
    }
  }, [session, status, redirectTo, router, pathname])

  if (status === 'loading') {
    return <div>Loading session...</div>
  }

  return <>{children}</>
}

export default SessionCheck