From 8f8a608a25f212f5eec5ca33011a0e9fd839b71f Mon Sep 17 00:00:00 2001
From: Paul Filimon <113104137+SharglutDev@users.noreply.github.com>
Date: Tue, 25 Apr 2023 16:50:35 +0200
Subject: [PATCH] Pfn/front/electrification error handler (#3935)

close #3835

This PR adds an error message when selecting a power restriction code
than cannot be applied by the rolling stock given the electrification on
the chosen path.
---
 .../manageTrainSchedule.json                  |   2 +
 .../DisplayItinerary/Destination.tsx          |   7 +-
 .../Itinerary/DisplayItinerary/Origin.tsx     |   2 +
 .../PowerRestrictionSelector.tsx              | 135 +++++++++++++-----
 .../RollingStockCardButtons.jsx               |   2 +
 front/src/reducers/osrdconf/index.ts          |   2 +-
 6 files changed, 108 insertions(+), 42 deletions(-)

diff --git a/front/public/locales/fr/operationalStudies/manageTrainSchedule.json b/front/public/locales/fr/operationalStudies/manageTrainSchedule.json
index 82fb12385f6..6e7d5e2651b 100644
--- a/front/public/locales/fr/operationalStudies/manageTrainSchedule.json
+++ b/front/public/locales/fr/operationalStudies/manageTrainSchedule.json
@@ -12,6 +12,7 @@
   "deleteVias": "Supprimer les étapes",
   "destination": "Destination",
   "errorMessages": {
+    "error": "Une erreur est survenue",
     "mandatoryField": "Champ obligatoire",
     "noDelta": "L'intervalle ente deux trains doit être d'au moins une minute",
     "noDeltaInput": "Doit être > 0",
@@ -26,6 +27,7 @@
     "noTrainCountInput": "Doit être > 0",
     "noTrainStep": "Le pas d'incrément du nom du train doit être supérieur à zéro",
     "noTrainStepInput": "Doit être > 0",
+    "powerRestrictionInvalidCombination": "Code {{powerRestrictionCode}} incompatible avec l'électrification à {{electrification}} V de l'itinéraire",
     "stdcmError": "Erreur de calcul Sillon de dernière minute",
     "The requested train couldn't reach its destination": "Le train ne peut pas atteindre sa destination",
     "trainScheduleTitle": "Une erreur est survenue",
diff --git a/front/src/applications/operationalStudies/components/ManageTrainSchedule/Itinerary/DisplayItinerary/Destination.tsx b/front/src/applications/operationalStudies/components/ManageTrainSchedule/Itinerary/DisplayItinerary/Destination.tsx
index e6dac8400af..99801e4f03b 100644
--- a/front/src/applications/operationalStudies/components/ManageTrainSchedule/Itinerary/DisplayItinerary/Destination.tsx
+++ b/front/src/applications/operationalStudies/components/ManageTrainSchedule/Itinerary/DisplayItinerary/Destination.tsx
@@ -15,12 +15,12 @@ import {
   updateDestination,
   updateDestinationDate,
   updateDestinationTime,
+  updatePathfindingID,
   updateStdcmMode,
 } from 'reducers/osrdconf';
 import { makeEnumBooleans } from 'utils/constants';
 
 import InputSNCF from 'common/BootstrapSNCF/InputSNCF';
-import { store } from 'Store';
 import { MODES, STDCM_MODES } from 'applications/operationalStudies/consts';
 
 interface DestinationProps {
@@ -61,7 +61,10 @@ function Destination(props: DestinationProps) {
             <button
               className="btn btn-sm btn-only-icon btn-white"
               type="button"
-              onClick={() => store.dispatch(updateDestination(undefined))}
+              onClick={() => {
+                dispatch(updateDestination(undefined));
+                dispatch(updatePathfindingID(undefined));
+              }}
             >
               <i className="icons-circle-delete" />
               <span className="sr-only" aria-hidden="true">
diff --git a/front/src/applications/operationalStudies/components/ManageTrainSchedule/Itinerary/DisplayItinerary/Origin.tsx b/front/src/applications/operationalStudies/components/ManageTrainSchedule/Itinerary/DisplayItinerary/Origin.tsx
index 042fce1aa7e..56bd0b97882 100644
--- a/front/src/applications/operationalStudies/components/ManageTrainSchedule/Itinerary/DisplayItinerary/Origin.tsx
+++ b/front/src/applications/operationalStudies/components/ManageTrainSchedule/Itinerary/DisplayItinerary/Origin.tsx
@@ -13,6 +13,7 @@ import {
   updateOriginUpperBoundTime,
   updateStdcmMode,
   toggleOriginLinkedBounds,
+  updatePathfindingID,
 } from 'reducers/osrdconf';
 import {
   getStdcmMode,
@@ -105,6 +106,7 @@ function Origin(props: OriginProps) {
               type="button"
               onClick={() => {
                 dispatch(updateOrigin(undefined));
+                dispatch(updatePathfindingID(undefined));
               }}
             >
               <i className="icons-circle-delete" />
diff --git a/front/src/applications/operationalStudies/components/ManageTrainSchedule/PowerRestrictionSelector.tsx b/front/src/applications/operationalStudies/components/ManageTrainSchedule/PowerRestrictionSelector.tsx
index 7add2d37056..9b2272311e2 100644
--- a/front/src/applications/operationalStudies/components/ManageTrainSchedule/PowerRestrictionSelector.tsx
+++ b/front/src/applications/operationalStudies/components/ManageTrainSchedule/PowerRestrictionSelector.tsx
@@ -1,5 +1,4 @@
-/* eslint-disable no-restricted-syntax */
-import React, { useState, useEffect } from 'react';
+import React, { useEffect, useMemo } from 'react';
 import { useDispatch, useSelector } from 'react-redux';
 import { useTranslation } from 'react-i18next';
 import { updatePowerRestriction } from 'reducers/osrdconf';
@@ -10,67 +9,125 @@ import {
   getPowerRestriction,
   getPathfindingID,
 } from 'reducers/osrdconf/selectors';
-import { osrdEditoastApi, LightRollingStock } from 'common/api/osrdEditoastApi';
+import { osrdEditoastApi, RollingStock } from 'common/api/osrdEditoastApi';
 import { osrdMiddlewareApi, PowerRestrictionRange } from 'common/api/osrdMiddlewareApi';
 import { lengthFromLineCoordinates } from 'utils/geometry';
+import { setFailure } from 'reducers/main';
+import { compact, reduce, uniq } from 'lodash';
 
 type selectorOption = { key: string | undefined; value: string | undefined };
 
+type ElectrificationPR = {
+  [key: string]: string[] | string[];
+};
+
 export default function PowerRestrictionSelector() {
   const { t } = useTranslation(['operationalStudies/manageTrainSchedule']);
   const dispatch = useDispatch();
   const rollingStockID: number | undefined = useSelector(getRollingStockID);
   const pathFindingID = useSelector(getPathfindingID);
   const powerRestriction = useSelector(getPowerRestriction);
-  const [powerRestrictions, setPowerRestrictions] = useState<selectorOption[] | undefined>();
-  const [getRollingStockById] = osrdEditoastApi.endpoints.getLightRollingStockById.useLazyQuery({});
-  const [getPathFindingById] = osrdMiddlewareApi.endpoints.getPathfindingById.useLazyQuery({});
 
-  const getPowerRestrictionsList = async (rollingStockInfo: LightRollingStock) => {
-    if (rollingStockInfo.power_restrictions) {
-      const powerRestrictionsList = Object.entries(rollingStockInfo.power_restrictions).map(
+  const { data: pathFinding } = osrdMiddlewareApi.useGetPathfindingByIdQuery(
+    { id: pathFindingID as number },
+    { skip: !pathFindingID }
+  );
+
+  const { data: pathWithCatenaries } = osrdEditoastApi.useGetPathfindingByPathIdCatenariesQuery(
+    { pathId: pathFindingID as number },
+    { skip: !pathFindingID }
+  );
+
+  const { data: rollingStock } = osrdEditoastApi.useGetRollingStockByIdQuery(
+    { id: rollingStockID as number },
+    { skip: !rollingStockID }
+  );
+
+  const powerRestrictions = useMemo(() => {
+    let powerRestrictionsList: selectorOption[] = [];
+    if (rollingStock) {
+      powerRestrictionsList = Object.entries(rollingStock.power_restrictions).map(
         ([key, _]: [string | undefined, string]) => ({ key, value: key })
       );
-      if (powerRestrictionsList.length > 0) {
-        powerRestrictionsList.unshift({ key: undefined, value: t('noPowerRestriction') });
-      }
-      setPowerRestrictions(powerRestrictionsList);
+      powerRestrictionsList.unshift({ key: undefined, value: t('noPowerRestriction') });
     }
+    return powerRestrictionsList;
+  }, [rollingStock]);
+
+  // Extract unique rollingstock's power restriction codes allowed by each electrification mode
+  const cleanConditionalEffortCurves = (rollingStockToClean: RollingStock) => {
+    const curvesMode = rollingStockToClean.effort_curves.modes;
+    const curvesModesKey = Object.keys(curvesMode);
+
+    const parsedElectrification: ElectrificationPR = reduce(
+      curvesModesKey,
+      (result, mode) => {
+        const powerCodes = curvesMode[mode].curves.map(
+          (curve) => curve.cond?.power_restriction_code
+        );
+        compact(uniq(powerCodes));
+        return {
+          ...result,
+          [mode]: powerCodes,
+        };
+      },
+      {}
+    );
+
+    return parsedElectrification;
   };
 
-  const definePowerRestrictionRange = (powerRestrictionCode?: string) => {
-    if (powerRestrictionCode && pathFindingID) {
-      getPathFindingById({ id: pathFindingID })
-        .unwrap()
-        .then((pathFinding) => {
-          const pathLength = Math.round(
-            lengthFromLineCoordinates(pathFinding?.geographic?.coordinates) * 1000
-          );
-          const powerRestrictionRange: PowerRestrictionRange[] = [
-            {
-              begin_position: 0,
-              end_position: pathLength,
-              power_restriction_code: powerRestrictionCode,
-            },
-          ];
-          dispatch(updatePowerRestriction(powerRestrictionRange));
-        });
+  const definePowerRestrictionRange = async (powerRestrictionCode?: string) => {
+    if (powerRestrictionCode && pathFinding) {
+      const pathLength = Math.round(
+        lengthFromLineCoordinates(pathFinding.geographic?.coordinates) * 1000
+      );
+      const powerRestrictionRange: PowerRestrictionRange = {
+        begin_position: 0,
+        end_position: pathLength,
+        power_restriction_code: powerRestrictionCode,
+      };
+
+      dispatch(updatePowerRestriction(powerRestrictionRange));
     } else dispatch(updatePowerRestriction(undefined));
   };
 
   useEffect(() => {
-    if (rollingStockID) {
-      dispatch(updatePowerRestriction(undefined));
-      setPowerRestrictions(undefined);
-      getRollingStockById({ id: rollingStockID })
-        .unwrap()
-        .then((rollingStockInfo) => {
-          getPowerRestrictionsList(rollingStockInfo);
+    if (powerRestriction && rollingStock && pathWithCatenaries) {
+      const parsedElectrification = cleanConditionalEffortCurves(rollingStock);
+
+      const powerRestrictionCode = powerRestriction.power_restriction_code;
+      const pathCatenaryRanges = pathWithCatenaries.catenary_ranges;
+
+      if (pathCatenaryRanges && powerRestrictionCode) {
+        // Extract path electrification mode and check compatibility
+        const pathElectrification = compact(pathCatenaryRanges.map((range) => range.mode));
+
+        // Display an error when the first incompatibility is encountered
+        pathElectrification.some((electrification) => {
+          const isInvalid =
+            !parsedElectrification[electrification as keyof ElectrificationPR].includes(
+              powerRestrictionCode
+            );
+          if (isInvalid) {
+            dispatch(
+              setFailure({
+                name: t('errorMessages.error'),
+                message: t('errorMessages.powerRestrictionInvalidCombination', {
+                  powerRestrictionCode,
+                  electrification,
+                }),
+              })
+            );
+          }
+
+          return isInvalid;
         });
+      }
     }
-  }, [rollingStockID]);
+  }, [powerRestriction]);
 
-  return powerRestrictions && powerRestrictions.length > 0 && pathFindingID ? (
+  return powerRestrictions.length > 1 && pathFindingID ? (
     <div className="osrd-config-item mb-2">
       <div className="osrd-config-item-container">
         <img width="32px" className="mr-2" src={icon} alt="PowerRestrictionIcon" />
diff --git a/front/src/common/RollingStockSelector/RollingStockCardButtons.jsx b/front/src/common/RollingStockSelector/RollingStockCardButtons.jsx
index 92a0785805c..58ba8033172 100644
--- a/front/src/common/RollingStockSelector/RollingStockCardButtons.jsx
+++ b/front/src/common/RollingStockSelector/RollingStockCardButtons.jsx
@@ -7,6 +7,7 @@ import {
   updateRollingStockComfort,
   updateRollingStockID,
   updateShouldRunPathfinding,
+  updatePowerRestriction,
 } from 'reducers/osrdconf';
 import { getRollingStockComfort } from 'reducers/osrdconf/selectors';
 import { ModalContext } from 'common/BootstrapSNCF/ModalSNCF/ModalProvider';
@@ -36,6 +37,7 @@ export default function RollingStockCardButtons(props) {
     dispatch(updateRollingStockComfort(comfort));
     dispatch(updateShouldRunPathfinding(true));
     dispatch(updateRollingStockID(id));
+    dispatch(updatePowerRestriction(undefined));
     closeModal();
   };
 
diff --git a/front/src/reducers/osrdconf/index.ts b/front/src/reducers/osrdconf/index.ts
index 225cc2ea134..a3beb3932ac 100644
--- a/front/src/reducers/osrdconf/index.ts
+++ b/front/src/reducers/osrdconf/index.ts
@@ -669,7 +669,7 @@ export function deleteItinerary() {
     });
   };
 }
-export function updatePowerRestriction(powerRestriction?: PowerRestrictionRange[]) {
+export function updatePowerRestriction(powerRestriction?: PowerRestrictionRange) {
   return (dispatch: Dispatch) => {
     dispatch({
       type: UPDATE_POWER_RESTRICTION,