front: fix infra lock status

Author: Akctarus

front/src/applications/editor/Editor.tsx

@@ -19,15 +19,15 @@ import { useSwitchTypes } from 'applications/editor/tools/switchEdition/types';
 import type { switchProps } from 'applications/editor/tools/switchProps';
 import type { CommonToolState } from 'applications/editor/tools/types';
 import { centerMapOnObject, selectEntities } from 'applications/editor/tools/utils';
-import type { ObjectType } from 'common/api/osrdEditoastApi';
+import { osrdEditoastApi, type ObjectType } from 'common/api/osrdEditoastApi';
 import { useModal } from 'common/BootstrapSNCF/ModalSNCF';
 import { LoaderState } from 'common/Loaders';
 import MapButtons from 'common/Map/Buttons/MapButtons';
 import MapSearch from 'common/Map/Search/MapSearch';
 import { useInfraID, useOsrdActions } from 'common/osrdContext';
 import Tipped from 'common/Tipped';
 import type { EditorSliceActions } from 'reducers/editor';
-import { getEditorState, getInfraLockStatus } from 'reducers/editor/selectors';
+import { getEditorState } from 'reducers/editor/selectors';
 import { loadDataModel, updateTotalsIssue } from 'reducers/editor/thunkActions';
 import { setFailure } from 'reducers/main';
 import { getIsLoading } from 'reducers/main/mainSelector';
@@ -49,7 +49,6 @@ const Editor = () => {
   const { urlInfra } = useParams();
   const infraID = useInfraID();
   const [searchParams, setSearchParams] = useSearchParams();
-  const isLocked = useSelector(getInfraLockStatus);
   const isLoading = useSelector(getIsLoading);
   const editorState = useSelector(getEditorState);
   const switchTypes = useSwitchTypes(infraID);
@@ -65,6 +64,17 @@ const Editor = () => {
   }, [setRenderingFingerprint]);
 
   const [isFormSubmited, setIsFormSubmited] = useState(false);
+  const [getInfraById, { data: infra }] = osrdEditoastApi.endpoints.getInfraById.useLazyQuery({});
+
+  /**
+   * When infra id changes
+   * => fetch it
+   */
+  useEffect(() => {
+    if (infraID) {
+      getInfraById({ id: infraID });
+    }
+  }, [infraID, getInfraById]);
 
   const switchTool = useCallback(
     ({ toolType, toolState }: switchProps) => {
@@ -147,7 +157,6 @@ const Editor = () => {
       dispatch,
       editorState,
       infraID,
-      isInfraLocked: isLocked,
       isLoading,
       isFormSubmited,
       setIsFormSubmited,
@@ -166,7 +175,6 @@ const Editor = () => {
       switchTypes,
       viewport,
       isLoading,
-      isLocked,
       isFormSubmited,
       setIsFormSubmited,
     ]
@@ -387,7 +395,7 @@ const Editor = () => {
             })}
           </div>
           <div>
-            {isLocked && (
+            {infra?.locked && (
               <div className="infra-locked bg-yellow">{t('Editor.infra-errors.infra-locked')}</div>
             )}
             {toolAndState.tool.leftPanelComponent && (