From 5c2a4603a5bb8b76ab54678a59a776232a21295c Mon Sep 17 00:00:00 2001
From: romainvalls <romain.valls95@gmail.com>
Date: Wed, 8 Jan 2025 12:07:00 +0100
Subject: [PATCH] front: fix nge update after importing or deleting multiple
 trains

---
 .../operationalStudies/components/Scenario/ScenarioContent.tsx | 1 +
 .../operationalStudies/views/ImportTrainSchedule.tsx           | 3 +++
 .../ImportTrainSchedule/ImportTrainScheduleTrainsList.tsx      | 3 +++
 .../modules/trainschedule/components/Timetable/Timetable.tsx   | 1 +
 4 files changed, 8 insertions(+)

diff --git a/front/src/applications/operationalStudies/components/Scenario/ScenarioContent.tsx b/front/src/applications/operationalStudies/components/Scenario/ScenarioContent.tsx
index 2a5a788c8da..48a81b07da4 100644
--- a/front/src/applications/operationalStudies/components/Scenario/ScenarioContent.tsx
+++ b/front/src/applications/operationalStudies/components/Scenario/ScenarioContent.tsx
@@ -212,6 +212,7 @@ const ScenarioContent = ({
               <ImportTrainSchedule
                 timetableId={scenario.timetable_id}
                 upsertTrainSchedules={upsertTrainSchedules}
+                dtoImport={dtoImport}
               />
             </div>
           )}
diff --git a/front/src/applications/operationalStudies/views/ImportTrainSchedule.tsx b/front/src/applications/operationalStudies/views/ImportTrainSchedule.tsx
index 0c35ec4a3ad..5addb7f027e 100644
--- a/front/src/applications/operationalStudies/views/ImportTrainSchedule.tsx
+++ b/front/src/applications/operationalStudies/views/ImportTrainSchedule.tsx
@@ -14,9 +14,11 @@ import { useAppDispatch } from 'store';
 const ImportTrainSchedule = ({
   timetableId,
   upsertTrainSchedules,
+  dtoImport,
 }: {
   timetableId: number;
   upsertTrainSchedules: (trainSchedules: TrainScheduleResult[]) => void;
+  dtoImport: () => void;
 }) => {
   const dispatch = useAppDispatch();
   const { t } = useTranslation(['rollingstock']);
@@ -57,6 +59,7 @@ const ImportTrainSchedule = ({
         trainsJsonData={trainsJsonData}
         trainsXmlData={trainsXmlData}
         upsertTrainSchedules={upsertTrainSchedules}
+        dtoImport={dtoImport}
       />
     </main>
   ) : (
diff --git a/front/src/modules/trainschedule/components/ImportTrainSchedule/ImportTrainScheduleTrainsList.tsx b/front/src/modules/trainschedule/components/ImportTrainSchedule/ImportTrainScheduleTrainsList.tsx
index 59977dcef39..ff3c6b25cf1 100644
--- a/front/src/modules/trainschedule/components/ImportTrainSchedule/ImportTrainScheduleTrainsList.tsx
+++ b/front/src/modules/trainschedule/components/ImportTrainSchedule/ImportTrainScheduleTrainsList.tsx
@@ -38,6 +38,7 @@ type ImportTrainScheduleTrainsListProps = {
   trainsJsonData: TrainScheduleBase[];
   trainsXmlData: ImportedTrainSchedule[];
   upsertTrainSchedules: (trainSchedules: TrainScheduleResult[]) => void;
+  dtoImport: () => void;
 };
 
 const ImportTrainScheduleTrainsList = ({
@@ -48,6 +49,7 @@ const ImportTrainScheduleTrainsList = ({
   trainsJsonData,
   trainsXmlData,
   upsertTrainSchedules,
+  dtoImport,
 }: ImportTrainScheduleTrainsListProps) => {
   const { t } = useTranslation(['operationalStudies/importTrainSchedule']);
   const rollingStockDict = useMemo(
@@ -91,6 +93,7 @@ const ImportTrainScheduleTrainsList = ({
 
       const trainSchedules = await postTrainSchedule({ id: timetableId, body: payloads }).unwrap();
       upsertTrainSchedules(trainSchedules);
+      dtoImport();
       dispatch(
         setSuccess({
           title: t('success'),
diff --git a/front/src/modules/trainschedule/components/Timetable/Timetable.tsx b/front/src/modules/trainschedule/components/Timetable/Timetable.tsx
index 1d2579aa573..adcf528bd99 100644
--- a/front/src/modules/trainschedule/components/Timetable/Timetable.tsx
+++ b/front/src/modules/trainschedule/components/Timetable/Timetable.tsx
@@ -64,6 +64,7 @@ const Timetable = ({
   const removeAndUnselectTrains = (trainIds: number[]) => {
     removeTrains(trainIds);
     setSelectedTrainIds([]);
+    dtoImport();
   };
 
   const toggleConflictsListExpanded = () => {