datahub/datahub-web-react/src/app/useBuildEntityRegistry.ts

import { useMemo } from 'react';

import buildEntityRegistry from '@app/buildEntityRegistry';
import buildEntityRegistryV2 from '@app/buildEntityRegistryV2';
import { useIsThemeV2 } from '@app/useIsThemeV2';

export default function useBuildEntityRegistry() {
    const isThemeV2Enabled = useIsThemeV2();
    return useMemo(() => {
        return isThemeV2Enabled ? buildEntityRegistryV2() : buildEntityRegistry();
    }, [isThemeV2Enabled]);
}