feat: Track storage usage

This commit is contained in:
Maksim Eltyshev
2025-08-23 00:03:20 +02:00
parent 2f4bcb0583
commit 4d77a1f596
89 changed files with 1052 additions and 304 deletions

View File

@@ -23,10 +23,12 @@ module.exports = {
inputs.record,
);
const user = await User.qm.deleteOne(inputs.record.id);
const { user, uploadedFile } = await User.qm.deleteOne(inputs.record.id);
if (user) {
sails.helpers.users.removeRelatedFiles(user);
if (uploadedFile) {
sails.helpers.utils.removeUnreferencedUploadedFiles(uploadedFile);
}
const scoper = sails.helpers.users.makeScoper(user);
scoper.boardMemberships = boardMemberships;

View File

@@ -28,9 +28,9 @@ module.exports = {
'termsAcceptedAt',
]),
avatar: inputs.record.avatar && {
url: `${fileManager.buildUrl(`${sails.config.custom.userAvatarsPathSegment}/${inputs.record.avatar.dirname}/original.${inputs.record.avatar.extension}`)}`,
url: `${fileManager.buildUrl(`${sails.config.custom.userAvatarsPathSegment}/${inputs.record.avatar.uploadedFileId}/original.${inputs.record.avatar.extension}`)}`,
thumbnailUrls: {
cover180: `${fileManager.buildUrl(`${sails.config.custom.userAvatarsPathSegment}/${inputs.record.avatar.dirname}/cover-180.${inputs.record.avatar.extension}`)}`,
cover180: `${fileManager.buildUrl(`${sails.config.custom.userAvatarsPathSegment}/${inputs.record.avatar.uploadedFileId}/cover-180.${inputs.record.avatar.extension}`)}`,
},
},
termsType: sails.hooks.terms.getTypeByUserRole(inputs.record.role),

View File

@@ -3,9 +3,9 @@
* Licensed under the Fair Use License: https://github.com/plankanban/planka/blob/master/LICENSE.md
*/
const { v4: uuid } = require('uuid');
const { rimraf } = require('rimraf');
const mime = require('mime');
const { v4: uuid } = require('uuid');
const sharp = require('sharp');
module.exports = {
@@ -32,8 +32,16 @@ module.exports = {
});
let metadata;
let originalBuffer;
try {
metadata = await image.metadata();
if (metadata.orientation && metadata.orientation > 4) {
image = image.rotate();
}
originalBuffer = await image.toBuffer();
} catch (error) {
await rimraf(inputs.file.fd);
throw 'fileIsNotImage';
@@ -41,20 +49,19 @@ module.exports = {
const fileManager = sails.hooks['file-manager'].getInstance();
const dirname = uuid();
const dirPathSegment = `${sails.config.custom.userAvatarsPathSegment}/${dirname}`;
if (metadata.orientation && metadata.orientation > 4) {
image = image.rotate();
}
const extension = metadata.format === 'jpeg' ? 'jpg' : metadata.format;
const size = originalBuffer.length;
const { id: uploadedFileId } = await UploadedFile.qm.createOne({
mimeType,
size,
id: uuid(),
type: UploadedFile.Types.USER_AVATAR,
});
const dirPathSegment = `${sails.config.custom.userAvatarsPathSegment}/${uploadedFileId}`;
let sizeInBytes;
try {
const originalBuffer = await image.toBuffer();
sizeInBytes = originalBuffer.length;
await fileManager.save(
`${dirPathSegment}/original.${extension}`,
originalBuffer,
@@ -81,6 +88,7 @@ module.exports = {
await fileManager.deleteDir(dirPathSegment);
await rimraf(inputs.file.fd);
await UploadedFile.qm.deleteOne(uploadedFileId);
throw 'fileIsNotImage';
}
@@ -88,9 +96,9 @@ module.exports = {
await rimraf(inputs.file.fd);
return {
dirname,
uploadedFileId,
extension,
sizeInBytes,
size,
};
},
};

View File

@@ -69,8 +69,10 @@ module.exports = {
}
let user;
let uploadedFile;
try {
user = await User.qm.updateOne(inputs.record.id, values);
({ user, uploadedFile } = await User.qm.updateOne(inputs.record.id, values));
} catch (error) {
if (error.code === 'E_UNIQUE') {
throw 'emailAlreadyInUse';
@@ -91,10 +93,8 @@ module.exports = {
}
if (user) {
if (inputs.record.avatar) {
if (!user.avatar || user.avatar.dirname !== inputs.record.avatar.dirname) {
sails.helpers.users.removeRelatedFiles(inputs.record);
}
if (uploadedFile) {
sails.helpers.utils.removeUnreferencedUploadedFiles(uploadedFile);
}
if (!_.isUndefined(values.password) || isDeactivatedChangeToTrue) {