import { useQuery } from "@tanstack/react-query";
import { authApi } from "@/api/auth";
import { primaryAgentApi } from "@/api/primaryAgent";
import { queryKeys } from "@/lib/queryKeys";

export function usePrimaryAgent(companyId: string | null | undefined) {
  const session = useQuery({ queryKey: queryKeys.auth.session, queryFn: () => authApi.getSession() });
  const userId = session.data?.user?.id ?? session.data?.session?.userId ?? "local-board";
  const query = useQuery({
    queryKey: queryKeys.primaryAgent.mine(companyId ?? "__none__", userId),
    queryFn: () => primaryAgentApi.get(companyId!),
    enabled: !!companyId && session.isSuccess,
  });
  return {
    ...query,
    userId,
    loading: !!companyId && (session.isPending || query.isPending && !session.error),
    error: session.error ?? query.error,
    retry: () => session.isError ? session.refetch() : query.refetch(),
  };
}