'use client'

import { useEffect, useState, useRef } from 'react'

import { useAuthContext } from '@/contexts/AuthContext'

import { RedirectToRegister } from './auth/redirect-to-register'

interface AuthGuardProps {
  children: React.ReactNode
  fallback?: React.ReactNode
}

export function AuthGuard({ children, fallback }: AuthGuardProps) {
  const { isAuthenticated, isLoading, checkAuth, user } = useAuthContext()
  const [isChecking, setIsChecking] = useState(true)
  const initRef = useRef(false) // Защита от повторных инициализаций

  useEffect(() => {
    const initAuth = async () => {
      if (initRef.current) {
        return
      }

      initRef.current = true
      await checkAuth()
      setIsChecking(false)
    }

    initAuth()
  }, []) // Запускаем только один раз при монтировании


  // Показываем лоадер пока проверяем авторизацию
  if (isChecking || isLoading) {
    return (
      <div className="min-h-screen flex items-center justify-center">
        <div className="text-center text-white">
          <div className="animate-spin rounded-full h-16 w-16 border-4 border-white border-t-transparent mx-auto mb-4"></div>
          <p className="text-white/80">Проверяем авторизацию...</p>
        </div>
      </div>
    )
  }

  // Если не авторизован ИЛИ нет организации (незавершенная регистрация), перенаправляем на /register
  if (!isAuthenticated || (isAuthenticated && user && !user.organization)) {
    console.warn('🔒 AuthGuard - Unauthorized access, redirecting to register')
    return fallback || <RedirectToRegister />
  }

  // Если авторизован И у пользователя есть организация, показываем защищенный контент
  return <>{children}</>
}
