delete dataset if all the files go away

1 files changed, 41 insertions, 8 deletions

diff --git a/public/bundle.js b/public/bundle.js
index 462988b..309dae6 100644
--- a/public/bundle.js
+++ b/public/bundle.js
@@ -2902,6 +2902,7 @@ var fetchURL = exports.fetchURL = function fetchURL(module, folder, url) {
       _actions2.default.file.create({
         folder_id: folder.id,
         module: module.name,
+        dataset: media.token,
         activity: 'url',
         duration: parseInt(media.duration) || 0,
         epoch: 0,
@@ -6090,8 +6091,6 @@ var _types2 = _interopRequireDefault(_types);
 
 function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
 
-function _objectWithoutProperties(obj, keys) { var target = {}; for (var i in obj) { if (keys.indexOf(i) >= 0) continue; if (!Object.prototype.hasOwnProperty.call(obj, i)) continue; target[i] = obj[i]; } return target; }
-
 function _defineProperty(obj, key, value) { if (key in obj) { Object.defineProperty(obj, key, { value: value, enumerable: true, configurable: true, writable: true }); } else { obj[key] = value; } return obj; }
 
 var samplernnInitialState = {
@@ -6233,16 +6232,50 @@ var samplernnReducer = function samplernnReducer() {
       return state;
 
     case _types2.default.file.destroy:
+      console.log(action.data);
       if (action.data.module === 'samplernn') {
-        // delete 
-        var _state$data$fileLooku = state.data.fileLookup,
-            removedFile = _state$data$fileLooku[action.data.id],
-            fileLookup = _objectWithoutProperties(_state$data$fileLooku, [action.data.id]);
+        // delete the key from fileLookup
+        var fileLookup = _extends({}, state.data.fileLookup);
+        var datasetLookup = _extends({}, state.data.datasetLookup);
+        var folderLookup = _extends({}, state.data.folderLookup);
+        var _old_file = fileLookup[action.data.id];
+        if (!_old_file) return state;
+        var _old_dataset2 = state.data.datasetLookup[_old_file.dataset];
+        var _dataset = void 0;
+        console.log(_old_file.dataset, _old_dataset2);
+        if (_old_dataset2) {
+          _dataset = _extends({}, _old_dataset2, {
+            input: _old_dataset2.input.filter(function (id) {
+              return id !== _old_file.id;
+            }),
+            output: _old_dataset2.output.filter(function (id) {
+              return 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];
+            var _old_folder = folderLookup[_old_file.folder_id];
+            folderLookup[_old_file.folder_id] = _extends({}, _old_folder, {
+              datasets: _old_folder.datasets.filter(function (name) {
+                return 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 _extends({}, state, {
           loading: false,
           data: _extends({}, state.data, {
-            fileLookup: fileLookup
+            folderLookup: folderLookup,
+            fileLookup: fileLookup,
+            datasetLookup: datasetLookup
           })
         });
       }
@@ -6613,7 +6646,7 @@ var SampleRNNShow = function (_Component) {
 
       var input = fileLookup[dataset.input[0]];
       if (!input) return null;
-      if (input.name.match(/(gif|jpe?g|png)$/i)) return null;
+      if (input.name && input.name.match(/(gif|jpe?g|png)$/i)) return null;
       return (0, _preact.h)(
         'div',
         null,