import { useInject } from '@/lib/di/hooks/useInject';
import { enhanceQueryResult } from '@/lib/di/hooks/useReactQueryWithApiError';
import { SESSION_SERVICE_TOKEN } from '@/lib/di/tokens';
import { ApiError } from '@/lib/gateways/api/base/ApiError';
import { SessionViewModel } from '@/lib/view-models/SessionViewModel';
import { useQuery, UseQueryOptions } from '@tanstack/react-query';

export function useCurrentSession(
  options?: Omit<UseQueryOptions<SessionViewModel | null, ApiError>, 'queryKey' | 'queryFn'> & { initialData?: SessionViewModel | null }
) {
  const sessionService = useInject(SESSION_SERVICE_TOKEN);

  const queryResult = useQuery({
    queryKey: ['currentSession'],
    queryFn: async () => {
      const result = await sessionService.getSession();
      if (result.isErr()) {
        const error = result.getError();
        throw new ApiError(error.message, 'SERVER_ERROR', { timestamp: new Date().toISOString() });
      }
      const session = result.unwrap();
      return session ? new SessionViewModel(session.user as any) : null;
    },
    initialData: options?.initialData,
    retry: false,
    refetchOnWindowFocus: false,
    ...options,
  });

  return enhanceQueryResult(queryResult);
}