From 06c453c3bcc253432501037e59cc1e13d2f10dc1 Mon Sep 17 00:00:00 2001
From: syuilo <syuilotan@yahoo.co.jp>
Date: Fri, 4 May 2018 17:27:14 +0900
Subject: [PATCH] :v:

---
 cli/clean-cached-remote-files.js | 73 ++++++++++++++++++++------------
 cli/clean-unused-drive-files.js  |  8 ++--
 2 files changed, 50 insertions(+), 31 deletions(-)

diff --git a/cli/clean-cached-remote-files.js b/cli/clean-cached-remote-files.js
index a266f0d984..5e359412bf 100644
--- a/cli/clean-cached-remote-files.js
+++ b/cli/clean-cached-remote-files.js
@@ -3,8 +3,6 @@ const log = require('single-line-log').stdout;
 const sequential = require('promise-sequential');
 const { default: DriveFile, DriveFileChunk } = require('../built/models/drive-file');
 const { default: DriveFileThumbnail, DriveFileThumbnailChunk } = require('../built/models/drive-file-thumbnail');
-const { default: Note } = require('../built/models/note');
-const { default: MessagingMessage } = require('../built/models/messaging-message');
 const { default: User } = require('../built/models/user');
 
 const q = {
@@ -33,41 +31,62 @@ async function main() {
 
 				prev = file;
 
-				if (file == null) return res([i, null]);
+				function skip() {
+					res([i, file, false]);
+				}
 
-				// チャンクをすべて削除
-				await DriveFileChunk.remove({
-					files_id: file._id
-				});
+				if (file == null) return skip();
 
-				await DriveFile.update({ _id: file._id }, {
-					$set: {
-						'metadata.deletedAt': new Date(),
-						'metadata.isExpired': true
-					}
-				});
+				log(chalk`{gray ${i}} scanning {bold ${file._id}} ${file.filename} ...`);
 
-				res([i, file]);
+				const attachingUsersCount = await User.count({
+					$or: [{
+						avatarId: file._id
+					}, {
+						bannerId: file._id
+					}]
+				}, { limit: 1 });
+				if (attachingUsersCount !== 0) return skip();
 
-				//#region サムネイルもあれば削除
-				const thumbnail = await DriveFileThumbnail.findOne({
-					'metadata.originalId': file._id
-				});
+				Promise.all([
+					// チャンクをすべて削除
+					DriveFileChunk.remove({
+						files_id: file._id
+					}),
 
-				if (thumbnail) {
-					DriveFileThumbnailChunk.remove({
-						files_id: thumbnail._id
+					DriveFile.update({ _id: file._id }, {
+						$set: {
+							'metadata.deletedAt': new Date(),
+							'metadata.isExpired': true
+						}
+					})
+				]).then(() => {
+					res([i, file, true]);
+
+					//#region サムネイルもあれば削除
+					const thumbnail = await DriveFileThumbnail.findOne({
+						'metadata.originalId': file._id
 					});
 
-					DriveFileThumbnail.remove({ _id: thumbnail._id });
-				}
-				//#endregion
+					if (thumbnail) {
+						DriveFileThumbnailChunk.remove({
+							files_id: thumbnail._id
+						});
+
+						DriveFileThumbnail.remove({ _id: thumbnail._id });
+					}
+					//#endregion
+				});
 			});
 
-			promise.then(([i, file]) => {
-				if (file) {
-					console.log(chalk`{gray ${i}} {green done: {bold ${file._id}} ${file.filename}}`);
+			promise.then(([i, file, deleted]) => {
+				if (deleted) {
+					log(chalk`{gray ${i}} {red deleted: {bold ${file._id}} ${file.filename}}`);
+				} else {
+					log(chalk`{gray ${i}} {green skipped: {bold ${file._id}} ${file.filename}}`);
 				}
+				log.clear();
+				console.log();
 			});
 
 			return promise;
diff --git a/cli/clean-unused-drive-files.js b/cli/clean-unused-drive-files.js
index 1d61c5ccc3..87b158b9ee 100644
--- a/cli/clean-unused-drive-files.js
+++ b/cli/clean-unused-drive-files.js
@@ -16,10 +16,6 @@ async function main() {
 	for (let i = 0; i < count; i++) {
 		promiseGens.push(() => {
 			const promise = new Promise(async (res, rej) => {
-				function skip() {
-					res([i, file, false]);
-				}
-
 				const file = await DriveFile.findOne(prev ? {
 					_id: { $lt: prev._id }
 				} : {}, {
@@ -30,6 +26,10 @@ async function main() {
 
 				prev = file;
 
+				function skip() {
+					res([i, file, false]);
+				}
+
 				if (file == null) return skip();
 
 				log(chalk`{gray ${i}} scanning {bold ${file._id}} ${file.filename} ...`);