'use client';

import { QueryClient, QueryClientProvider } from '@tanstack/react-query';
import { ReactQueryDevtools } from '@tanstack/react-query-devtools';
import { useEffect } from 'react';
import { Toaster } from 'sonner';
import { useAuthStore } from '@/store/auth.store';
import { useMe } from '@/hooks/useAuth';

// ============================================================
// Session Recovery — Runs on initial load if token exists
// ============================================================
function SessionRecovery({ children }: { children: React.ReactNode }) {
  const { isAuthenticated, setInitialized, logout } = useAuthStore();
  const { isError } = useMe();

  useEffect(() => {
    // If no token, mark as initialized immediately
    if (!isAuthenticated) {
      setInitialized(true);
    }
  }, [isAuthenticated, setInitialized]);

  useEffect(() => {
    // If /auth/me fails (invalid/expired token), clean up
    if (isError) {
      logout();
      setInitialized(true);
    }
  }, [isError, logout, setInitialized]);

  return <>{children}</>;
}

// ============================================================
// React Query Provider
// ============================================================
function makeQueryClient() {
  return new QueryClient({
    defaultOptions: {
      queries: {
        staleTime: 2 * 60 * 1000, // 2 minute
        // eslint-disable-next-line @typescript-eslint/no-explicit-any
        retry: (failureCount, error: any) => {
          // Don't retry on 401/403/404
          if ([401, 403, 404].includes(error?.response?.status)) return false;
          return failureCount < 2;
        },
      },
      mutations: {
        retry: false,
      },
    },
  });
}

let browserQueryClient: QueryClient | undefined;

function getQueryClient() {
  if (typeof window === 'undefined') {
    return makeQueryClient();
  }
  if (!browserQueryClient) {
    browserQueryClient = makeQueryClient();
  }
  return browserQueryClient;
}

// ============================================================
// Combined Providers
// ============================================================
export function Providers({ children }: { children: React.ReactNode }) {
  const queryClient = getQueryClient();

  return (
    <QueryClientProvider client={queryClient}>
      <SessionRecovery>
        {children}
        <Toaster
          position="top-right"
          richColors
          toastOptions={{
            duration: 1000,
          }}
        />
      </SessionRecovery>
      <ReactQueryDevtools initialIsOpen={false} />
    </QueryClientProvider>
  );
}