fixup! front: add waypoint menu in manchette

OpenRailAssociation · Dec 12, 2024 · 4f87f08 · 4f87f08
1 parent e0b511e
commit 4f87f08
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/...s/simulationResult/components/ManchetteWithSpaceTimeChart/ManchetteWithSpaceTimeChart.tsx b/...s/simulationResult/components/ManchetteWithSpaceTimeChart/ManchetteWithSpaceTimeChart.tsx
@@ -181,7 +181,11 @@ const ManchetteWithSpaceTimeChartWrapper = ({
       color: ASPECT_LABELS_COLORS[block.aspect_label as AspectLabel],
     }));
   });
-  const waypointMenuData = useWaypointMenu(waypointsPanelData?.setFilteredWaypoints);
+
+  const waypointMenuData = useWaypointMenu(
+    waypointsPanelData?.filteredWaypoints,
+    waypointsPanelData?.setFilteredWaypoints
+  );
 
   const manchettePropsWithWaypointMenu = useMemo(
     () => ({

diff --git a/front/src/modules/simulationResult/components/SpaceTimeChart/useWaypointMenu.tsx b/front/src/modules/simulationResult/components/SpaceTimeChart/useWaypointMenu.tsx
@@ -8,7 +8,7 @@ import type { OSRDMenuItem } from 'common/OSRDMenu';
 import useModalFocusTrap from 'utils/hooks/useModalFocusTrap';
 
 const useWaypointMenu = (
-  filteredWaypoints: OperationalPoint[],
+  filteredWaypoints?: OperationalPoint[],
   setFilteredWaypoints?: Dispatch<SetStateAction<OperationalPoint[]>>
 ) => {
   const { t } = useTranslation('simulation');
@@ -49,9 +49,10 @@ const useWaypointMenu = (
     {
       title: t('waypointMenu.hide'),
       icon: <EyeClosed />,
+      disabled: filteredWaypoints ? filteredWaypoints.length <= 2 : false,
       onClick: () => {
         closeMenu();
-        if (filteredWaypoints.length > 2 && setFilteredWaypoints) {
+        if (setFilteredWaypoints) {
           setFilteredWaypoints((prevFilteredWaypoints) =>
             prevFilteredWaypoints.filter((waypoint) => waypoint.id !== activeWaypointId)
           );