feat: Track storage usage

server/api/hooks/query-methods/models/BackgroundImage.js

@@ -7,7 +7,25 @@ const defaultFind = (criteria) => BackgroundImage.find(criteria).sort('id');
 
 /* Query methods */
 
-const createOne = (values) => BackgroundImage.create({ ...values }).fetch();
+const createOne = (values) =>
+  sails.getDatastore().transaction(async (db) => {
+    const backgroundImage = await BackgroundImage.create({ ...values })
+      .fetch()
+      .usingConnection(db);
+
+    const queryResult = await sails
+      .sendNativeQuery(
+        'UPDATE uploaded_file SET references_total = references_total + 1, updated_at = $1 WHERE id = $2 AND references_total IS NOT NULL',
+        [new Date().toISOString(), values.uploadedFileId],
+      )
+      .usingConnection(db);
+
+    if (queryResult.rowCount === 0) {
+      throw 'uploadedFileNotFound';
+    }
+
+    return backgroundImage;
+  });
 
 const getByIds = (ids) => defaultFind(ids);
 
@@ -34,9 +52,99 @@ const getOneById = (id, { projectId } = {}) => {
 };
 
 // eslint-disable-next-line no-underscore-dangle
-const delete_ = (criteria) => BackgroundImage.destroy(criteria).fetch();
+const delete_ = (criteria) =>
+  sails.getDatastore().transaction(async (db) => {
+    const backgroundImages = await BackgroundImage.destroy(criteria).fetch().usingConnection(db);
 
-const deleteOne = (criteria) => BackgroundImage.destroyOne(criteria);
+    let uploadedFiles = [];
+    if (backgroundImages.length > 0) {
+      const backgroundImagesByUploadedFileId = _.groupBy(backgroundImages, 'uploadedFileId');
+
+      const uploadedFileIdsByTotal = Object.entries(backgroundImagesByUploadedFileId).reduce(
+        (result, [uploadedFileId, backgroundImagesItem]) => ({
+          ...result,
+          [backgroundImagesItem.length]: [
+            ...(result[backgroundImagesItem.length] || []),
+            uploadedFileId,
+          ],
+        }),
+        {},
+      );
+
+      const queryValues = [];
+      let query = 'UPDATE uploaded_file SET references_total = CASE WHEN references_total = CASE ';
+
+      Object.entries(uploadedFileIdsByTotal).forEach(([total, uploadedFileIds]) => {
+        const inValues = uploadedFileIds.map((uploadedFileId) => {
+          queryValues.push(uploadedFileId);
+          return `$${queryValues.length}`;
+        });
+
+        queryValues.push(total);
+        query += `WHEN id IN (${inValues.join(', ')}) THEN $${queryValues.length}::int `;
+      });
+
+      query += 'END THEN NULL ELSE references_total - CASE ';
+
+      Object.entries(uploadedFileIdsByTotal).forEach(([total, uploadedFileIds]) => {
+        const inValues = uploadedFileIds.map((uploadedFileId) => {
+          queryValues.push(uploadedFileId);
+          return `$${queryValues.length}`;
+        });
+
+        queryValues.push(total);
+        query += `WHEN id IN (${inValues.join(', ')}) THEN $${queryValues.length}::int `;
+      });
+
+      const inValues = Object.keys(backgroundImagesByUploadedFileId).map((uploadedFileId) => {
+        queryValues.push(uploadedFileId);
+        return `$${queryValues.length}`;
+      });
+
+      queryValues.push(new Date().toISOString());
+      query += `END END, updated_at = $${queryValues.length} WHERE id IN (${inValues.join(', ')}) AND references_total IS NOT NULL RETURNING *`;
+
+      const queryResult = await sails.sendNativeQuery(query, queryValues).usingConnection(db);
+
+      uploadedFiles = queryResult.rows.map((row) => ({
+        id: row.id,
+        type: row.type,
+        mimeType: row.mime_type,
+        size: row.size,
+        referencesTotal: row.references_total,
+        createdAt: row.created_at,
+        updatedAt: row.updated_at,
+      }));
+    }
+
+    return { backgroundImages, uploadedFiles };
+  });
+
+const deleteOne = (criteria) =>
+  sails.getDatastore().transaction(async (db) => {
+    const backgroundImage = await BackgroundImage.destroyOne(criteria).usingConnection(db);
+
+    const queryResult = await sails
+      .sendNativeQuery(
+        'UPDATE uploaded_file SET references_total = CASE WHEN references_total > 1 THEN references_total - 1 END, updated_at = $1 WHERE id = $2 RETURNING *',
+        [new Date().toISOString(), backgroundImage.uploadedFileId],
+      )
+      .usingConnection(db);
+
+    const [row] = queryResult.rows;
+
+    uploadedFile = {
+      id: row.id,
+      type: row.type,
+      mimeType: row.mime_type,
+      size: row.size,
+      referencesTotal: row.references_total,
+      createdAt: row.created_at,
+      updatedAt: row.updated_at,
+    };
+
+    return { backgroundImage, uploadedFile };
+  });
 
 module.exports = {
   createOne,