fix(api): delete orphaned db records when deleting training folder / …

…files
jakowenko · Oct 7, 2021 · f031545 · f031545
1 parent 233b74e
commit f031545
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/api/src/controllers/fs.controller.js b/api/src/controllers/fs.controller.js
@@ -1,5 +1,6 @@
 const fs = require('fs');
 const filesystem = require('../util/fs.util');
+const database = require('../util/db.util');
 const { resync } = require('../util/db.util');
 const { STORAGE } = require('../constants')();
 
@@ -18,6 +19,9 @@ module.exports.folders = {
       fs.rmdirSync(`${STORAGE.PATH}/train/${name}`, { recursive: true });
       await resync.files();
     }
+    const db = database.connect();
+    db.prepare('DELETE FROM file WHERE name = ?').run(name);
+    db.prepare('DELETE FROM train WHERE name = ?').run(name);
     res.send({ success: true });
   },
 };
diff --git a/api/src/controllers/storage.controller.js b/api/src/controllers/storage.controller.js
@@ -116,6 +116,9 @@ module.exports.delete = async (req, res) => {
     db.prepare(
       `DELETE FROM file WHERE id IN (${files.map((obj) => `'${obj.id}'`).join(',')})`
     ).run();
+    db.prepare(
+      `DELETE FROM train WHERE fileId IN (${files.map((obj) => `'${obj.id}'`).join(',')})`
+    ).run();
     files.forEach((obj) => {
       filesystem.delete(`${PATH}/${obj.key}`);
     });