From c7e7614868b3addcac349f45d8737bcf9ca9c9c2 Mon Sep 17 00:00:00 2001
From: romainvalls <romain.valls95@gmail.com>
Date: Wed, 4 Dec 2024 15:14:54 +0100
Subject: [PATCH] front: cancel fetch request when user cancels simulation in
 lmr

Signed-off-by: romainvalls <romain.valls95@gmail.com>
---
 .../src/applications/stdcm/hooks/useStdcm.ts  | 38 ++++++++++---------
 .../applications/stdcm/views/StdcmView.tsx    |  7 ++++
 2 files changed, 28 insertions(+), 17 deletions(-)

diff --git a/front/src/applications/stdcm/hooks/useStdcm.ts b/front/src/applications/stdcm/hooks/useStdcm.ts
index f4841167c50..97a5ee24506 100644
--- a/front/src/applications/stdcm/hooks/useStdcm.ts
+++ b/front/src/applications/stdcm/hooks/useStdcm.ts
@@ -1,4 +1,4 @@
-import { useState } from 'react';
+import { useRef, useState } from 'react';
 
 import { useTranslation } from 'react-i18next';
 import nextId from 'react-id-generator';
@@ -52,6 +52,7 @@ const useStdcm = ({
   const { getConf, getTimetableID } = useOsrdConfSelectors();
   const osrdconf = useSelector(getConf) as OsrdStdcmConfState;
   const timetableId = useSelector(getTimetableID);
+  const requestPromise = useRef<ReturnType<typeof postTimetableByIdStdcm>>();
 
   const stdcmResults = useStdcmResults(stdcmResponse, stdcmTrainResult, setPathProperties);
 
@@ -65,9 +66,6 @@ const useStdcm = ({
       { skip: !osrdconf.rollingStockID }
     );
 
-  // https://developer.mozilla.org/en-US/docs/Web/API/AbortController
-  const controller = new AbortController();
-
   const { speedLimitByTag } = useStoreDataForSpeedLimitByTagSelector({ isStdcm: true });
 
   const triggerShowFailureNotification = (error: Error) => {
@@ -85,7 +83,10 @@ const useStdcm = ({
     if (validConfig) {
       const payload = formatStdcmPayload(validConfig);
       try {
-        const response = await postTimetableByIdStdcm(payload).unwrap();
+        const promise = postTimetableByIdStdcm(payload);
+        requestPromise.current = promise;
+
+        const response = await promise.unwrap();
 
         if (
           response.status === 'success' &&
@@ -132,28 +133,30 @@ const useStdcm = ({
           });
         }
       } catch (e) {
-        setCurrentStdcmRequestStatus(STDCM_REQUEST_STATUS.rejected);
-        triggerShowFailureNotification(
-          castErrorToFailure(e, {
-            name: t('stdcm:stdcmErrors.requestFailed'),
-            message: t('translation:common.error'),
-          })
-        );
+        if ((e as Error).name !== 'AbortError') {
+          setCurrentStdcmRequestStatus(STDCM_REQUEST_STATUS.rejected);
+          triggerShowFailureNotification(
+            castErrorToFailure(e, {
+              name: t('stdcm:stdcmErrors.requestFailed'),
+              message: t('translation:common.error'),
+            })
+          );
+        }
       }
-    } else {
-      setCurrentStdcmRequestStatus(STDCM_REQUEST_STATUS.rejected);
     }
   };
 
   const cancelStdcmRequest = () => {
-    // when http ready https://axios-http.com/docs/cancellation
-
-    controller.abort();
+    if (typeof requestPromise.current?.abort === 'function') {
+      requestPromise.current.abort();
+    }
+    requestPromise.current = undefined;
     setCurrentStdcmRequestStatus(STDCM_REQUEST_STATUS.canceled);
   };
 
   const isPending = currentStdcmRequestStatus === STDCM_REQUEST_STATUS.pending;
   const isRejected = currentStdcmRequestStatus === STDCM_REQUEST_STATUS.rejected;
+  const isCanceled = currentStdcmRequestStatus === STDCM_REQUEST_STATUS.canceled;
   const hasConflicts = (stdcmTrainConflicts?.length ?? 0) > 0;
 
   return {
@@ -164,6 +167,7 @@ const useStdcm = ({
     setPathProperties,
     isPending,
     isRejected,
+    isCanceled,
     stdcmTrainConflicts,
     hasConflicts,
     isCalculationFailed: isRejected,
diff --git a/front/src/applications/stdcm/views/StdcmView.tsx b/front/src/applications/stdcm/views/StdcmView.tsx
index d864cde91b8..9464b883cc2 100644
--- a/front/src/applications/stdcm/views/StdcmView.tsx
+++ b/front/src/applications/stdcm/views/StdcmView.tsx
@@ -35,6 +35,7 @@ const StdcmView = () => {
     cancelStdcmRequest,
     isPending,
     isRejected,
+    isCanceled,
     stdcmResults,
     pathProperties,
     stdcmTrainConflicts,
@@ -107,6 +108,12 @@ const StdcmView = () => {
     }
   }, [isDebugMode]);
 
+  useEffect(() => {
+    if (isCanceled) {
+      setShowBtnToLaunchSimulation(true);
+    }
+  }, [isCanceled]);
+
   useEffect(() => {
     /*
      * Due to frequent re-renders and the fact that "speedSpaceChartData" is initially null before