1 files changed, 43 insertions, 4 deletions

diff --git a/app/client/modules/samplernn/samplernn.reducer.js b/app/client/modules/samplernn/samplernn.reducer.js
index fd9dbff..1397a0b 100644
--- a/app/client/modules/samplernn/samplernn.reducer.js
+++ b/app/client/modules/samplernn/samplernn.reducer.js
@@ -144,8 +144,8 @@ const samplernnReducer = (state = samplernnInitialState, action) => {
             output: !action.data.generated ? [] : [action.data.id],
           }
         }
-      console.log('old_dataset', old_dataset_update.name)
-      console.log('new_dataset', new_dataset_update.name)
+        console.log('old_dataset', old_dataset_update.name)
+        console.log('new_dataset', new_dataset_update.name)
         return {
           ...state,
           loading: false,
@@ -169,15 +169,54 @@ const samplernnReducer = (state = samplernnInitialState, action) => {
       return state
 
     case types.file.destroy:
+      console.log(action.data)
       if (action.data.module === 'samplernn') {
-        // delete 
-        const { [action.data.id]: removedFile, ...fileLookup } = state.data.fileLookup
+        // delete the key from fileLookup
+        let fileLookup = {
+          ...state.data.fileLookup
+        }
+        let datasetLookup = {
+          ...state.data.datasetLookup
+        }
+        let folderLookup = {
+          ...state.data.folderLookup
+        }
+        let old_file = fileLookup[action.data.id]
+        if (!old_file) return state
+        let old_dataset = state.data.datasetLookup[old_file.dataset]
+        let dataset
+        console.log(old_file.dataset, old_dataset)
+        if (old_dataset) {
+          dataset = {
+            ...old_dataset,
+            input: old_dataset.input.filter(id => id !== old_file.id),
+            output: old_dataset.output.filter(id => id !== old_file.id),
+          }
+          console.log(dataset.input.length, dataset.checkpoints.length, dataset.output.length)
+          if (! dataset.input.length && ! dataset.checkpoints.length && ! dataset.output.length) {
+            console.log('dataset is empty')
+            delete datasetLookup[old_file.dataset]
+            let old_folder = folderLookup[old_file.folder_id]
+            folderLookup[old_file.folder_id] = {
+              ...old_folder,
+              datasets: old_folder.datasets.filter(name => name !== old_file.dataset)
+            }
+          } else {
+            console.log('dataset just needs sweeping')
+            datasetLookup[old_file.dataset ] = dataset
+          }
+        }
+        delete fileLookup[old_file.id]
+        // console.log(fileLookup)
+
         return {
           ...state,
           loading: false,
           data: {
             ...state.data,
+            folderLookup,
             fileLookup,
+            datasetLookup,
           }
         }
       }