useTreeMenuClicked.js

import { useCallback, useState } from 'react';
import { useSelector } from 'react-redux';
import { ApplicationStateType } from '../../reducers';
import { filterTestJobsByProject, getTestJobsForSelectedProject } from '../../utils/testJobUtils';

export const useTreeMenuClicked = () => {
  const testJobsLibrary = useSelector((state: ApplicationStateType) => state.testJobsLibrary);
  const { testProjects, testJobs } = testJobsLibrary;
  const testJobsAll = JSON.parse(JSON.stringify(testJobs));
  const [testJobFiltered, setTestJobFiltered] = useState(false);

  const handleTreeMenuClicked: (filterBy: string) => void = useCallback(
    (filterBy) => {
      if (filterBy === 'all') {
        testJobsLibrary.testJobs = testJobsAll;
        setTestJobFiltered(true);
      } else if (filterBy === 'orphaned') {
        const { orphanedTestJobs } = filterTestJobsByProject(testProjects, testJobs);
        testJobsLibrary.testJobs = orphanedTestJobs;
        setTestJobFiltered(true);
      } else if (filterBy === 'projects') {
        const { testJobsWithParent } = filterTestJobsByProject(testProjects, testJobs);
        testJobsLibrary.testJobs = testJobsWithParent;
        setTestJobFiltered(true);
      } else {
        testJobsLibrary.testJobs = getTestJobsForSelectedProject(testProjects, filterBy, testJobs);
        setTestJobFiltered(true);
      }
    },
    [testJobsLibrary]
  );

  return {
    handleTreeMenuClicked,
    setTestJobFiltered,
    testJobFiltered
  };
};