Add session saving and restoring to the provenance store

package.json

@@ -21,7 +21,7 @@
     "d3": "7.8.0",
     "eslint-import-resolver-alias": "^1.1.2",
     "lineupjs": "^4.2.0",
-    "multinet": "^0.21.11",
+    "multinet": "^0.21.13",
     "multinet-components": "^0.0.6",
     "pinia": "^2.0.28",
     "react": "^18.2.0",

src/App.vue

@@ -3,7 +3,7 @@
 import ProvVis from '@/components/ProvVis.vue';
 import ControlPanel from '@/components/ControlPanel.vue';
 import MultiMatrix from '@/components/MultiMatrix.vue';
 import { getUrlVars } from '@/lib/utils';
 import { useStore } from '@/store';
 import 'multinet-components/dist/style.css';
 import { storeToRefs } from 'pinia';
@@ -22,11 +22,7 @@
   userInfo,
 } = storeToRefs(store);
 
-const urlVars = getUrlVars();
-store.fetchNetwork(
-  urlVars.workspace,
-  urlVars.network,
-);
+store.fetchNetwork();
 
 // Set up provenance undo and redo, provenance is not a ref here
 const { provenance } = store;

src/store/index.ts

@@ -37,10 +37,10 @@ export const useStore = defineStore('store', () => {
     slicingConfig,
     sliceIndex,
     sortBy,
+    workspaceName,
+    networkName,
   } = storeToRefs(provStore);
 
-  const workspaceName = ref('');
-  const networkName = ref('');
   const loadError = ref<LoadError>({
     message: '',
     href: '',
@@ -290,10 +290,14 @@ export const useStore = defineStore('store', () => {
     });
   }
 
-  async function fetchNetwork(workspaceNameLocal: string, networkNameLocal: string) {
-    workspaceName.value = workspaceNameLocal;
-    networkName.value = networkNameLocal;
-
+  async function fetchNetwork() {
+    if (workspaceName.value === '' || networkName.value === '') {
+      loadError.value = {
+        message: 'Workspace and/or network were not defined in the url',
+        href: 'https://multinet.app',
+      };
+      return;
+    }
     let networkLocal: NetworkSpec | undefined;
 
     // Get all table names
@@ -302,17 +306,10 @@ export const useStore = defineStore('store', () => {
       // eslint-disable-next-line @typescript-eslint/no-explicit-any
     } catch (error: any) {
       if (error.status === 404) {
-        if (workspaceName.value === undefined || networkName.value === undefined) {
-          loadError.value = {
-            message: 'Workspace and/or network were not defined in the url',
-            href: 'https://multinet.app',
-          };
-        } else {
-          loadError.value = {
-            message: error.statusText,
-            href: 'https://multinet.app',
-          };
-        }
+        loadError.value = {
+          message: error.statusText,
+          href: 'https://multinet.app',
+        };
       } else if (error.status === 401) {
         loadError.value = {
           message: 'You are not authorized to view this workspace',

src/store/provenance.ts

@@ -1,4 +1,6 @@
+import api from '@/api';
 import { findDifferencesInPrimitiveStates, getTrrackLabel, isArray } from '@/lib/provenanceUtils';
+import { getUrlVars } from '@/lib/utils';
 import { Cell, ProvState, SlicingConfig } from '@/types';
 import { initializeTrrack, Registry } from '@trrack/core';
 import { defineStore } from 'pinia';
@@ -112,6 +114,30 @@ export const useProvenanceStore = defineStore('provenance', () => {
     }
   });
 
+  // Variables to help with session management (this is not tracked in trrack)
+  const { sessionId, workspace, network } = getUrlVars();
+  const workspaceName = ref(workspace || '');
+  const networkName = ref(network || '');
+
+  // Update the session when the provenance changes
+  provenance.currentChange(() => {
+    api.updateSession(workspaceName.value, sessionId || '', 'network', provenance.export());
+  });
+
+  // Attempt to restore session on first load
+  async function restoreSession() {
+    if (sessionId) {
+      const session = await api.getSession(workspaceName.value, sessionId, 'network');
+
+      // If the session is empty, the API will be an empty object
+      // Only attempt to import if we have a string
+      if (typeof session.state === 'string') {
+        provenance.import(session.state);
+      }
+    }
+  }
+  restoreSession();
+
   return {
     provenance,
     selectNeighbors,
@@ -126,5 +152,7 @@ export const useProvenanceStore = defineStore('provenance', () => {
     slicingConfig,
     sliceIndex,
     sortBy,
+    workspaceName,
+    networkName,
   };
 });

yarn.lock

@@ -3267,10 +3267,10 @@ multinet-components@^0.0.6:
   resolved "https://registry.yarnpkg.com/multinet-components/-/multinet-components-0.0.6.tgz#7b5b62de7a3504c0fe8b5928aea048506b209802"
   integrity sha512-PkAEWTtk7/7lwZhYlHfMeYFo8zutvVJ/DxMsuN4wZCddyseBAP70vTnKkLcSK3WsS96XsibWjCDl1AHz2pACIg==
 
-multinet@^0.21.11:
-  version "0.21.11"
-  resolved "https://registry.yarnpkg.com/multinet/-/multinet-0.21.11.tgz#040209b42a7b5dc4122c30f346109094ba1e0199"
-  integrity sha512-7MYTnqv4Gbr+AcQB2fwkA03F5nNqBrr/mNu/sSeIUcFd6cC/nZRQyRwPa40jTjFV80HahYeHvqQZCpMvdY/PYA==
+multinet@^0.21.13:
+  version "0.21.13"
+  resolved "https://registry.yarnpkg.com/multinet/-/multinet-0.21.13.tgz#a4c440b05369bcb2001a728c9ecf9cc2b86a298c"
+  integrity sha512-UI++fPe3nr1ueEO5VSqGtL1rb6iV8Qt7ACE2Srw2HqcHt+jdhXmaf439XZtrRHl/TqcBE6xlnsObDSnbyiHbgQ==
   dependencies:
     axios "^0.21.1"
     django-s3-file-field "^0.1.2"