sentry/static/app/views/ddm/ddmOnboarding/useCurrentProjectState.tsx

import {useEffect, useMemo, useState} from 'react';
import first from 'lodash/first';
import partition from 'lodash/partition';

import {
  customMetricOnboardingPlatforms,
  customMetricPlatforms,
} from 'sentry/data/platformCategories';
import PageFiltersStore from 'sentry/stores/pageFiltersStore';
import {useLegacyStore} from 'sentry/stores/useLegacyStore';
import {Project} from 'sentry/types';
import useProjects from 'sentry/utils/useProjects';

export function useCurrentProjectState({isActive}: {isActive: boolean}) {
  const [currentProject, setCurrentProject] = useState<Project | undefined>(undefined);
  const {projects, initiallyLoaded: projectsLoaded} = useProjects();
  const {selection, isReady} = useLegacyStore(PageFiltersStore);

  const [supportedProjects, unsupportedProjects] = useMemo(() => {
    return partition(projects, p => p.platform && customMetricPlatforms.has(p.platform));
  }, [projects]);

  // Projects where we have the onboarding instructions ready:
  const projectsWithOnboarding = useMemo(
    () =>
      supportedProjects.filter(
        p => p.platform && customMetricOnboardingPlatforms.has(p.platform)
      ),
    [supportedProjects]
  );

  useEffect(() => {
    if (!isActive) {
      setCurrentProject(undefined);
    }
  }, [isActive]);

  useEffect(() => {
    if (currentProject || !projectsLoaded || !projects.length || !isReady || !isActive) {
      return;
    }

    if (!supportedProjects) {
      return;
    }

    if (selection.projects.length) {
      const selectedProjectIds = selection.projects.map(String);
      // If we selected something that has onboarding instructions, pick that first
      const projectWithOnboarding = projectsWithOnboarding.find(p =>
        selectedProjectIds.includes(p.id)
      );
      if (projectWithOnboarding) {
        setCurrentProject(projectWithOnboarding);
        return;
      }

      // If we selected something that supports custom metrics pick that
      const projectSupportsMetrics = supportedProjects.find(p =>
        selectedProjectIds.includes(p.id)
      );
      if (projectSupportsMetrics) {
        setCurrentProject(projectSupportsMetrics);
        return;
      }
      // Else pick the first selected project
      const firstSelectedProject = projects.find(p => selectedProjectIds.includes(p.id));
      setCurrentProject(firstSelectedProject);
    } else {
      setCurrentProject(first(projectsWithOnboarding) || first(supportedProjects));
    }
  }, [
    currentProject,
    projectsLoaded,
    projects,
    isReady,
    isActive,
    selection.projects,
    projectsWithOnboarding,
    supportedProjects,
  ]);

  return {
    projects: supportedProjects,
    hasDocs:
      !!currentProject?.platform &&
      customMetricOnboardingPlatforms.has(currentProject.platform),
    allProjects: projects,
    supportedProjects,
    unsupportedProjects,
    currentProject,
    setCurrentProject,
  };
}