mirror of
https://iceshrimp.dev/crimekillz/iceshrimp-161sh.git
synced 2024-11-24 21:19:07 +01:00
c54f1928be
* Introduce per-instance chart * Implement chart view in client * Handle note deleting * More chart srcs * Add drive stats * Improve drive stats * Fix bug * Add icon
100 lines
3.4 KiB
TypeScript
100 lines
3.4 KiB
TypeScript
import * as Minio from 'minio';
|
|
import DriveFile, { DriveFileChunk, IDriveFile } from '../../models/drive-file';
|
|
import DriveFileThumbnail, { DriveFileThumbnailChunk } from '../../models/drive-file-thumbnail';
|
|
import config from '../../config';
|
|
import driveChart from '../../services/chart/drive';
|
|
import perUserDriveChart from '../../services/chart/per-user-drive';
|
|
import instanceChart from '../../services/chart/instance';
|
|
import DriveFileWebpublic, { DriveFileWebpublicChunk } from '../../models/drive-file-webpublic';
|
|
import Instance from '../../models/instance';
|
|
import { isRemoteUser } from '../../models/user';
|
|
|
|
export default async function(file: IDriveFile, isExpired = false) {
|
|
if (file.metadata.storage == 'minio') {
|
|
const minio = new Minio.Client(config.drive.config);
|
|
|
|
// 後方互換性のため、file.metadata.storageProps.key があるかどうかチェックしています。
|
|
// 将来的には const obj = file.metadata.storageProps.key; とします。
|
|
const obj = file.metadata.storageProps.key ? file.metadata.storageProps.key : `${config.drive.prefix}/${file.metadata.storageProps.id}`;
|
|
await minio.removeObject(config.drive.bucket, obj);
|
|
|
|
if (file.metadata.thumbnailUrl) {
|
|
// 後方互換性のため、file.metadata.storageProps.thumbnailKey があるかどうかチェックしています。
|
|
// 将来的には const thumbnailObj = file.metadata.storageProps.thumbnailKey; とします。
|
|
const thumbnailObj = file.metadata.storageProps.thumbnailKey ? file.metadata.storageProps.thumbnailKey : `${config.drive.prefix}/${file.metadata.storageProps.id}-thumbnail`;
|
|
await minio.removeObject(config.drive.bucket, thumbnailObj);
|
|
}
|
|
|
|
if (file.metadata.webpublicUrl) {
|
|
const webpublicObj = file.metadata.storageProps.webpublicKey ? file.metadata.storageProps.webpublicKey : `${config.drive.prefix}/${file.metadata.storageProps.id}-original`;
|
|
await minio.removeObject(config.drive.bucket, webpublicObj);
|
|
}
|
|
}
|
|
|
|
// チャンクをすべて削除
|
|
await DriveFileChunk.remove({
|
|
files_id: file._id
|
|
});
|
|
|
|
const set = {
|
|
metadata: {
|
|
deletedAt: new Date(),
|
|
isExpired: isExpired
|
|
}
|
|
} as any;
|
|
|
|
// リモートファイル期限切れ削除後は直リンクにする
|
|
if (isExpired && file.metadata && file.metadata._user && file.metadata._user.host != null) {
|
|
set.metadata.withoutChunks = true;
|
|
set.metadata.isRemote = true;
|
|
set.metadata.url = file.metadata.uri;
|
|
set.metadata.thumbnailUrl = undefined;
|
|
set.metadata.webpublicUrl = undefined;
|
|
}
|
|
|
|
await DriveFile.update({ _id: file._id }, {
|
|
$set: set
|
|
});
|
|
|
|
//#region サムネイルもあれば削除
|
|
const thumbnail = await DriveFileThumbnail.findOne({
|
|
'metadata.originalId': file._id
|
|
});
|
|
|
|
if (thumbnail) {
|
|
await DriveFileThumbnailChunk.remove({
|
|
files_id: thumbnail._id
|
|
});
|
|
|
|
await DriveFileThumbnail.remove({ _id: thumbnail._id });
|
|
}
|
|
//#endregion
|
|
|
|
//#region Web公開用もあれば削除
|
|
const webpublic = await DriveFileWebpublic.findOne({
|
|
'metadata.originalId': file._id
|
|
});
|
|
|
|
if (webpublic) {
|
|
await DriveFileWebpublicChunk.remove({
|
|
files_id: webpublic._id
|
|
});
|
|
|
|
await DriveFileWebpublic.remove({ _id: webpublic._id });
|
|
}
|
|
//#endregion
|
|
|
|
// 統計を更新
|
|
driveChart.update(file, false);
|
|
perUserDriveChart.update(file, false);
|
|
if (isRemoteUser(file.metadata._user)) {
|
|
instanceChart.updateDrive(file, false);
|
|
Instance.update({ host: file.metadata._user.host }, {
|
|
$inc: {
|
|
driveUsage: -file.length,
|
|
driveFiles: -1
|
|
}
|
|
});
|
|
}
|
|
}
|