#455 Store and retrieve snapshot actions in/from the backend

m2ms · Nov 30, 2020 · 95ef8f4 · 95ef8f4
1 parent c2bf2bf
commit 95ef8f4
Show file tree

Hide file tree

Showing 6 changed files with 82 additions and 46 deletions.
diff --git a/js/components/preview/moleculeGroups/redux/dispatchActions.js b/js/components/preview/moleculeGroups/redux/dispatchActions.js
@@ -33,7 +33,7 @@ import { OBJECT_TYPE } from '../../../nglView/constants';
 import { setSortDialogOpen } from '../../molecule/redux/actions';
 import { resetCurrentCompoundsSettings } from '../../compounds/redux/actions';
 import { reloadSession } from '../../../snapshot/redux/dispatchActions';
-import { restoreCurrentActionsList } from '../../../../reducers/tracking/dispatchActions';
+import { resetRestoringState } from '../../../../reducers/tracking/dispatchActions';
 
 export const clearAfterDeselectingMoleculeGroup = ({ molGroupId, currentMolGroup, majorViewStage }) => (
   dispatch,
@@ -251,7 +251,7 @@ export const restoreFromCurrentSnapshot = ({ nglViewList }) => (dispatch, getSta
 };
 
 export const restoreSnapshotActions = ({ nglViewList }) => (dispatch, getState) => {
-  dispatch(restoreCurrentActionsList(nglViewList));
+  dispatch(resetRestoringState(nglViewList));
 };
 
 export const onDeselectMoleculeGroup = ({ moleculeGroup, stageSummaryView, majorViewStage }) => (

diff --git a/js/components/projects/projectPreview/index.js b/js/components/projects/projectPreview/index.js
@@ -21,6 +21,7 @@ export const ProjectPreview = memo(({}) => {
   const currentSnapshotID = useSelector(state => state.projectReducers.currentSnapshot.id);
   const currentProject = useSelector(state => state.projectReducers.currentProject);
   const isActionRestoring = useSelector(state => state.trackingReducers.isActionRestoring);
+  const isActionRestored = useSelector(state => state.trackingReducers.isActionRestored);
 
   useEffect(() => {
     if (!snapshotId && currentSnapshotID === null) {
@@ -58,14 +59,14 @@ export const ProjectPreview = memo(({}) => {
             throw new Error(error);
           });
       } else {
-        if (isActionRestoring === false) {
+        if (isActionRestoring === false && isActionRestored === false) {
           dispatch(restoreCurrentActionsList(nglViewList));
-        } else if (nglViewList && nglViewList.length > 0) {
-          dispatch(restoreAfterTargetActions(nglViewList));
+        } else if (nglViewList && nglViewList.length > 0 && isActionRestored === false) {
+          dispatch(restoreAfterTargetActions(nglViewList, projectId));
         }
       }
     }
-  }, [currentSnapshotID, dispatch, projectId, snapshotId, isActionRestoring, nglViewList, canShow]);
+  }, [currentSnapshotID, dispatch, projectId, snapshotId, isActionRestoring, isActionRestored, nglViewList, canShow]);
 
   if (canShow === false) {
     setSnackBarTitle('Not valid snapshot!');

diff --git a/js/components/target/redux/dispatchActions.js b/js/components/target/redux/dispatchActions.js
@@ -68,19 +68,20 @@ export const updateTarget = ({ target, setIsLoading, targetIdList, projectId })
           let promises = [];
           if (!isActionRestoring || isActionRestoring === false) {
             promises.push(dispatch(setTargetOn(response.data.target.id, true)));
+            promises.push(
+              dispatch(
+                setCurrentProject({
+                  projectID: response.data.id,
+                  authorID: (response.data.author && response.data.author.id) || null,
+                  title: response.data.title,
+                  description: response.data.description,
+                  targetID: response.data.target.id,
+                  tags: JSON.parse(response.data.tags)
+                })
+              )
+            );
           }
-          promises.push(
-            dispatch(
-              setCurrentProject({
-                projectID: response.data.id,
-                authorID: (response.data.author && response.data.author.id) || null,
-                title: response.data.title,
-                description: response.data.description,
-                targetID: response.data.target.id,
-                tags: JSON.parse(response.data.tags)
-              })
-            )
-          );
+
           return Promise.all(promises);
         })
         .finally(() => setIsLoading(false));

diff --git a/js/reducers/tracking/actions.js b/js/reducers/tracking/actions.js
@@ -91,9 +91,10 @@ export const setIsActionsSaving = function(isActionSaving) {
   };
 };
 
-export const setIsActionsRestoring = function(isActionRestoring) {
+export const setIsActionsRestoring = function(isActionRestoring, isActionRestored) {
   return {
     type: constants.SET_IS_ACTIONS_RESTORING,
-    isActionRestoring: isActionRestoring
+    isActionRestoring: isActionRestoring,
+    isActionRestored: isActionRestored
   };
 };
diff --git a/js/reducers/tracking/dispatchActions.js b/js/reducers/tracking/dispatchActions.js
@@ -8,6 +8,7 @@ import { actionType, actionObjectType } from './constants';
 import { VIEWS } from '../../../js/constants/constants';
 import { setCurrentVector, appendToBuyList, removeFromToBuyList } from '../selection/actions';
 import { unmountPreviewComponent, shouldLoadProtein } from '../../components/preview/redux/dispatchActions';
+import { setCurrentProject } from '../../components/projects/redux/actions';
 import {
   selectMoleculeGroup,
   onDeselectMoleculeGroup,
@@ -317,8 +318,12 @@ const getCollectionOfDatasetOfRepresentation = dataList => {
   return list;
 };
 
+export const resetRestoringState = (stages = []) => (dispatch, getState) => {
+  dispatch(setTargetOn(undefined));
+  dispatch(setIsActionsRestoring(false, false));
+};
 export const restoreCurrentActionsList = (stages = []) => (dispatch, getState) => {
-  dispatch(setIsActionsRestoring(true));
+  dispatch(setIsActionsRestoring(true, false));
 
   Promise.resolve(dispatch(restoreTrackingActions())).then(response => {
     dispatch(setIsTrackingMoleculesRestoring(true));
@@ -378,7 +383,7 @@ const restoreTargetActions = (orderedActionList, stages) => (dispatch, getState)
   }
 };
 
-export const restoreAfterTargetActions = stages => (dispatch, getState) => {
+export const restoreAfterTargetActions = (stages, projectId) => async (dispatch, getState) => {
   const state = getState();
 
   const currentActionList = state.trackingReducers.current_actions_list;
@@ -389,9 +394,11 @@ export const restoreAfterTargetActions = stages => (dispatch, getState) => {
     const majorView = stages.find(view => view.id === VIEWS.MAJOR_VIEW);
     const summaryView = stages.find(view => view.id === VIEWS.SUMMARY_VIEW);
 
-    dispatch(shouldLoadProtein({ nglViewList: stages, currentSnapshotID: null, isLoadingCurrentSnapshot: false }));
+    await dispatch(
+      shouldLoadProtein({ nglViewList: stages, currentSnapshotID: null, isLoadingCurrentSnapshot: false })
+    );
 
-    dispatch(
+    await dispatch(
       loadMoleculeGroupsOfTarget({
         summaryView: summaryView.stage,
         isStateLoaded: false,
@@ -402,18 +409,20 @@ export const restoreAfterTargetActions = stages => (dispatch, getState) => {
       .catch(error => {
         throw error;
       })
-      .finally(() => {
-        dispatch(restoreSitesActions(orderedActionList, summaryView));
-        dispatch(loadAllMolecules(orderedActionList, targetId, majorView.stage));
-        dispatch(loadAllDatasets(orderedActionList, targetId, majorView.stage));
-        dispatch(restoreRepresentationActions(orderedActionList, stages));
-      });
+      .finally(() => {});
+
+    await dispatch(restoreSitesActions(orderedActionList, summaryView));
+    await dispatch(loadAllMolecules(orderedActionList, targetId, majorView.stage));
+    await dispatch(loadAllDatasets(orderedActionList, targetId, majorView.stage));
+    await dispatch(restoreRepresentationActions(orderedActionList, stages));
+    await dispatch(restoreProject(projectId));
+    dispatch(setIsActionsRestoring(false, true));
   }
 };
 
-const loadAllDatasets = (orderedActionList, target_on, stage) => (dispatch, getState) => {
+const loadAllDatasets = (orderedActionList, target_on, stage) => async (dispatch, getState) => {
   dispatch(setMoleculeListIsLoading(true));
-  dispatch(loadDataSets(target_on))
+  await dispatch(loadDataSets(target_on))
     .then(results => {
       return dispatch(loadDatasetCompoundsWithScores());
     })
@@ -428,7 +437,7 @@ const loadAllDatasets = (orderedActionList, target_on, stage) => (dispatch, getS
     });
 };
 
-const loadAllMolecules = (orderedActionList, target_on, stage) => (dispatch, getState) => {
+const loadAllMolecules = (orderedActionList, target_on, stage) => async (dispatch, getState) => {
   const state = getState();
   const list_type = listType.MOLECULE;
 
@@ -445,17 +454,18 @@ const loadAllMolecules = (orderedActionList, target_on, stage) => (dispatch, get
       })
     );
   });
-  Promise.all(promises)
-    .then(results => {
-      let listToSet = {};
-      results.forEach(molResult => {
-        listToSet[molResult.mol_group] = molResult.molecules;
-      });
-      dispatch(setAllMolLists(listToSet));
-      dispatch(restoreMoleculesActions(orderedActionList, stage));
-      dispatch(setIsTrackingMoleculesRestoring(false));
-    })
-    .catch(err => console.log(err));
+  try {
+    const results = await Promise.all(promises);
+    let listToSet = {};
+    results.forEach(molResult => {
+      listToSet[molResult.mol_group] = molResult.molecules;
+    });
+    dispatch(setAllMolLists(listToSet));
+    dispatch(restoreMoleculesActions(orderedActionList, stage));
+    dispatch(setIsTrackingMoleculesRestoring(false));
+  } catch (err) {
+    return console.log(err);
+  }
 };
 
 const restoreSitesActions = (orderedActionList, summaryView) => (dispatch, getState) => {
@@ -565,6 +575,27 @@ const restoreRepresentationActions = (moleculesAction, stages) => (dispatch, get
   }
 };
 
+const restoreProject = projectId => (dispatch, getState) => {
+  if (projectId !== undefined) {
+    return api({ url: `${base_url}/api/session-projects/${projectId}/` }).then(response => {
+      let promises = [];
+      promises.push(
+        dispatch(
+          setCurrentProject({
+            projectID: response.data.id,
+            authorID: (response.data.author && response.data.author.id) || null,
+            title: response.data.title,
+            description: response.data.description,
+            targetID: response.data.target.id,
+            tags: JSON.parse(response.data.tags)
+          })
+        )
+      );
+      return Promise.all(promises);
+    });
+  }
+};
+
 const restoreCompoundsActions = (orderedActionList, stage) => (dispatch, getState) => {
   const state = getState();
 

diff --git a/js/reducers/tracking/trackingReducers.js b/js/reducers/tracking/trackingReducers.js
@@ -13,7 +13,8 @@ export const INITIAL_STATE = {
   isActionSaving: false,
   send_actions_list: [],
   project_actions_list: [],
-  isActionRestoring: false
+  isActionRestoring: false,
+  isActionRestored: false
 };
 
 export function trackingReducers(state = INITIAL_STATE, action = {}) {
@@ -85,7 +86,8 @@ export function trackingReducers(state = INITIAL_STATE, action = {}) {
 
     case constants.SET_IS_ACTIONS_RESTORING:
       return Object.assign({}, state, {
-        isActionRestoring: action.isActionRestoring
+        isActionRestoring: action.isActionRestoring,
+        isActionRestored: action.isActionRestored
       });
 
     default: