From 8ae91ba774e7fb052d884c25a082c3ffa7b73de7 Mon Sep 17 00:00:00 2001
From: Theo Macron <theo.macron0315@gmail.com>
Date: Fri, 30 Aug 2024 16:35:06 +0200
Subject: [PATCH] front: add stop type selector in stdcm

---
 front/public/locales/en/stdcm.json            |  6 +++
 front/public/locales/fr/stdcm.json            |  6 +++
 .../stdcmV2/components/StdcmStopType.tsx      | 33 ++++++++++++++++
 .../stdcmV2/components/StdcmVias.tsx          | 38 ++++++++++++++++---
 front/src/applications/stdcmV2/types.ts       |  6 +++
 front/src/reducers/osrdconf/types.ts          |  3 +-
 6 files changed, 85 insertions(+), 7 deletions(-)
 create mode 100644 front/src/applications/stdcmV2/components/StdcmStopType.tsx

diff --git a/front/public/locales/en/stdcm.json b/front/public/locales/en/stdcm.json
index cd1e8f267af..13a1ab22bad 100644
--- a/front/public/locales/en/stdcm.json
+++ b/front/public/locales/en/stdcm.json
@@ -77,8 +77,14 @@
     "origin": "Origin",
     "preciseTime": "precise time",
     "stopFor": "Minimum stop time",
+    "stopType": {
+      "driverSwitch": "driver switch",
+      "passageTime":"passage time",
+      "serviceStop": "service stop"
+    },
     "time": "Time",
     "tolerance": "Tolerance",
+    "type": "Type",
     "vias": "Intermediate OP"
   }
 }
diff --git a/front/public/locales/fr/stdcm.json b/front/public/locales/fr/stdcm.json
index 5e4363ff9bd..0d5888f8cc5 100644
--- a/front/public/locales/fr/stdcm.json
+++ b/front/public/locales/fr/stdcm.json
@@ -77,8 +77,14 @@
     "origin": "Origine",
     "preciseTime": "horaire précis",
     "stopFor": "Temps d'arrêt minimum",
+    "stopType": {
+      "driverSwitch": "relève conducteur",
+      "passageTime":"passage",
+      "serviceStop": "arrêt de service"
+    },
     "time": "Heure",
     "tolerance": "Tolérance",
+    "type": "Type",
     "vias": "PR intermédiaire"
   }
 }
diff --git a/front/src/applications/stdcmV2/components/StdcmStopType.tsx b/front/src/applications/stdcmV2/components/StdcmStopType.tsx
new file mode 100644
index 00000000000..edbe6a1f910
--- /dev/null
+++ b/front/src/applications/stdcmV2/components/StdcmStopType.tsx
@@ -0,0 +1,33 @@
+import { Select } from '@osrd-project/ui-core';
+import { useTranslation } from 'react-i18next';
+
+import type { StdcmStopTypes } from '../types';
+
+type StdcmStopTypeProps = {
+  stopTypes: StdcmStopTypes;
+  onStdcmStopTypesChange: (arrivalType: StdcmStopTypes) => void;
+};
+
+const StdcmStopType = ({ stopTypes, onStdcmStopTypesChange }: StdcmStopTypeProps) => {
+  const { t } = useTranslation('stdcm');
+
+  return (
+    <div className="stdcm-v2-via-stop-for pl-2">
+      <Select
+        label={t('trainPath.type')}
+        id="type"
+        value={stopTypes}
+        options={['passageTime', 'driverSwitch', 'serviceStop']}
+        onChange={(e) => {
+          if (e) {
+            onStdcmStopTypesChange(e as StdcmStopTypes);
+          }
+        }}
+        getOptionLabel={(option) => t(`trainPath.stopType.${option}`)}
+        getOptionValue={(option) => option}
+      />
+    </div>
+  );
+};
+
+export default StdcmStopType;
diff --git a/front/src/applications/stdcmV2/components/StdcmVias.tsx b/front/src/applications/stdcmV2/components/StdcmVias.tsx
index 115fe6255cd..d66d445ef23 100644
--- a/front/src/applications/stdcmV2/components/StdcmVias.tsx
+++ b/front/src/applications/stdcmV2/components/StdcmVias.tsx
@@ -1,4 +1,4 @@
-import { useEffect, useMemo } from 'react';
+import { useEffect, useMemo, useState } from 'react';
 
 import { Location } from '@osrd-project/ui-icons';
 import { useTranslation } from 'react-i18next';
@@ -17,6 +17,8 @@ import StdcmCard from './StdcmCard';
 import StdcmDefaultCard from './StdcmDefaultCard';
 import StdcmInputVia from './StdcmInputVia';
 import StdcmOperationalPoint from './StdcmOperationalPoint';
+import StdcmStopType from './StdcmStopType';
+import { StdcmStopTypes } from '../types';
 import type { StdcmConfigCardProps } from '../types';
 
 const StdcmVias = ({ disabled = false, setCurrentSimulationInputs }: StdcmConfigCardProps) => {
@@ -27,11 +29,21 @@ const StdcmVias = ({ disabled = false, setCurrentSimulationInputs }: StdcmConfig
     useOsrdConfActions() as StdcmConfSliceActions;
   const pathSteps = useSelector(getPathSteps);
 
+  const [stopType, setStopType] = useState<StdcmStopTypes>(StdcmStopTypes.PASSAGE_TIME);
+
   const intermediatePoints = useMemo(() => pathSteps.slice(1, -1), [pathSteps]);
 
   const updatePathStepsList = (pathStep: PathStep | null, index: number) => {
-    const newPathSteps = replaceElementAtIndex(pathSteps, index, pathStep);
+    if (!pathStep) return;
+    const updatedPathStep = {
+      ...pathStep,
+      stopType,
+    };
+
+    const newPathSteps = replaceElementAtIndex(pathSteps, index, updatedPathStep);
     dispatch(updatePathSteps({ pathSteps: newPathSteps }));
+
+    console.log(pathStep);
   };
 
   const updatePathStepStopTime = (stopTime: string, index: number) => {
@@ -61,6 +73,14 @@ const StdcmVias = ({ disabled = false, setCurrentSimulationInputs }: StdcmConfig
     }));
   }, [pathSteps]);
 
+  // useEffect(() => {
+  //   if (stopType === StdcmStopTypes.DRIVER_SWITCH) {
+  //     setStopTypeTime('PT180S');
+  //   } else if (stopType === StdcmStopTypes.SERVICE_STOP) {
+  //     setStopTypeTime('PT60S');
+  //   }
+  // }, [stopType]);
+
   return (
     <div className="stdcm-v2-vias-list">
       {intermediatePoints.length > 0 &&
@@ -100,10 +120,16 @@ const StdcmVias = ({ disabled = false, setCurrentSimulationInputs }: StdcmConfig
                   />
                 </div>
                 {pathStep && (
-                  <StdcmInputVia
-                    pathStep={pathStep}
-                    updatePathStepStopTime={(e) => updatePathStepStopTime(e, pathStepIndex)}
-                  />
+                  <>
+                    <StdcmStopType stopTypes={stopType} onStdcmStopTypesChange={setStopType} />
+                    {(stopType === StdcmStopTypes.DRIVER_SWITCH ||
+                      stopType === StdcmStopTypes.SERVICE_STOP) && (
+                      <StdcmInputVia
+                        pathStep={pathStep}
+                        updatePathStepStopTime={(e) => updatePathStepStopTime(e, pathStepIndex)}
+                      />
+                    )}
+                  </>
                 )}
               </StdcmCard>
             </>
diff --git a/front/src/applications/stdcmV2/types.ts b/front/src/applications/stdcmV2/types.ts
index ca737de59d6..76eefeee4d3 100644
--- a/front/src/applications/stdcmV2/types.ts
+++ b/front/src/applications/stdcmV2/types.ts
@@ -46,3 +46,9 @@ export type StdcmConfigErrors = {
   errorType: StdcmConfigErrorTypes;
   errorDetails?: { originTime: string; destinationTime: string };
 };
+
+export enum StdcmStopTypes {
+  PASSAGE_TIME = 'passageTime',
+  DRIVER_SWITCH = 'driverSwitch',
+  SERVICE_STOP = 'serviceStop',
+}
diff --git a/front/src/reducers/osrdconf/types.ts b/front/src/reducers/osrdconf/types.ts
index d0676b10dc7..3a1f1211ab3 100644
--- a/front/src/reducers/osrdconf/types.ts
+++ b/front/src/reducers/osrdconf/types.ts
@@ -2,7 +2,7 @@ import type { Feature, Position } from 'geojson';
 
 import type { PowerRestriction } from 'applications/operationalStudies/types';
 import type { AllowanceValue } from 'applications/stdcm/types';
-import type { ArrivalTimeTypes } from 'applications/stdcmV2/types';
+import type { ArrivalTimeTypes, StdcmStopTypes } from 'applications/stdcmV2/types';
 import type { Comfort, Distribution, PathItemLocation } from 'common/api/osrdEditoastApi';
 import type { InfraState } from 'reducers/infra';
 
@@ -61,6 +61,7 @@ export type PathStep = PathItemLocation & {
   arrivalToleranceAfter?: number;
   locked?: boolean;
   stopFor?: string | null;
+  stopType?: StdcmStopTypes;
   theoreticalMargin?: string;
   onStopSignal?: boolean;
   kp?: string;