import ConfigStore from 'sentry/stores/configStore';
import type {Organization} from 'sentry/types/organization';
import type {Region} from 'sentry/types/system';
import {useQuery} from 'sentry/utils/queryClient';
import type RequestError from 'sentry/utils/requestError/requestError';
import useApi from 'sentry/utils/useApi';
import type {OrganizationWithRegion} from 'sentry/views/setupWizard/types';

export function useOrganizationsWithRegion() {
  const api = useApi();

  return useQuery<OrganizationWithRegion[], RequestError>({
    queryKey: ['/organizations/'],
    queryFn: async () => {
      const regions = ConfigStore.get('memberRegions');
      const results = await Promise.all(
        regions.map<Promise<[Region, Organization[]]>>(async region => [
          region,
          await api.requestPromise(`/organizations/`, {
            host: region.url,
            // Authentication errors can happen as we span regions.
            allowAuthError: true,
          }),
        ])
      );
      return results.reduce<OrganizationWithRegion[]>((acc, [region, response]) => {
        // Don't append error results to the org list.
        if (response[0]) {
          acc = acc.concat(response.map(org => ({...org, region})));
        }
        return acc;
      }, []);
    },
    refetchOnWindowFocus: false,
    retry: false,
  });
}