diff options
| -rw-r--r-- | cli/clean-drive.js | 56 | ||||
| -rw-r--r-- | package.json | 1 |
2 files changed, 57 insertions, 0 deletions
diff --git a/cli/clean-drive.js b/cli/clean-drive.js new file mode 100644 index 0000000000..f0c23b3b29 --- /dev/null +++ b/cli/clean-drive.js @@ -0,0 +1,56 @@ +const sequential = require('promise-sequential'); +const { default: DriveFile, deleteDriveFile } = require('../built/models/drive-file'); +const { default: Note } = require('../built/models/note'); +const { default: MessagingMessage } = require('../built/models/messaging-message'); +const { default: User } = require('../built/models/user'); + +async function main() { + const promiseGens = []; + + const count = await DriveFile.count({}); + + let prev; + + for (let i = 0; i < count; i++) { + promiseGens.push(() => new Promise(async (res, rej) => { + const file = await DriveFile.findOne(prev ? { + _id: { $gt: prev._id } + } : {}, { + sort: { + _id: 1 + } + }); + + prev = file; + + console.log(`scanning ${file._id}`); + + const attachingUsersCount = await User.count({ + $or: [{ + avatarId: file._id + }, { + bannerId: file._id + }] + }, { limit: 1 }); + if (attachingUsersCount !== 0) return res(); + + const attachingNotesCount = await Note.count({ + mediaIds: file._id + }, { limit: 1 }); + if (attachingNotesCount !== 0) return res(); + + const attachingMessagesCount = await MessagingMessage.count({ + fileId: file._id + }, { limit: 1 }); + if (attachingMessagesCount !== 0) return res(); + + console.log(`deleting ${file._id}`); + + deleteDriveFile(file).then(res).catch(rej); + })); + } + + return await sequential(promiseGens); +} + +main().then(console.dir).catch(console.error); diff --git a/package.json b/package.json index 1c6b41e4c2..6ba6bd298e 100644 --- a/package.json +++ b/package.json @@ -165,6 +165,7 @@ "os-utils": "0.0.14", "progress-bar-webpack-plugin": "1.11.0", "prominence": "0.2.0", + "promise-sequential": "^1.1.1", "pug": "2.0.3", "punycode": "2.1.0", "qrcode": "1.2.0", |