1 files changed, 104 insertions, 0 deletions
diff --git a/src/api/utils/ThumbUtil.js b/src/api/utils/ThumbUtil.js
new file mode 100644
index 0000000..d08ecab
--- /dev/null
+++ b/src/api/utils/ThumbUtil.js
@@ -0,0 +1,104 @@
+const jetpack = require('fs-jetpack');
+const path = require('path');
+const sharp = require('sharp');
+const ffmpeg = require('fluent-ffmpeg');
+const previewUtil = require('./videoPreview/FragmentPreview');
+
+const log = require('./Log');
+
+class ThumbUtil {
+	static imageExtensions = ['.jpg', '.jpeg', '.gif', '.png', '.webp'];
+	static videoExtensions = ['.webm', '.mp4', '.wmv', '.avi', '.mov'];
+
+	static thumbPath = path.join(__dirname, '../../../', process.env.UPLOAD_FOLDER, 'thumbs');
+	static squareThumbPath = path.join(__dirname, '../../../', process.env.UPLOAD_FOLDER, 'thumbs', 'square');
+	static videoPreviewPath = path.join(__dirname, '../../../', process.env.UPLOAD_FOLDER, 'thumbs', 'preview');
+
+	static generateThumbnails(filename) {
+		const ext = path.extname(filename).toLowerCase();
+		const output = `${filename.slice(0, -ext.length)}.webp`;
+		const previewOutput = `${filename.slice(0, -ext.length)}.webm`;
+
+		// eslint-disable-next-line max-len
+		if (ThumbUtil.imageExtensions.includes(ext)) return ThumbUtil.generateThumbnailForImage(filename, output);
+		// eslint-disable-next-line max-len
+		if (ThumbUtil.videoExtensions.includes(ext)) return ThumbUtil.generateThumbnailForVideo(filename, previewOutput);
+		return null;
+	}
+
+	static async generateThumbnailForImage(filename, output) {
+		const filePath = path.join(__dirname, '../../../', process.env.UPLOAD_FOLDER, filename);
+
+		const file = await jetpack.readAsync(filePath, 'buffer');
+		await sharp(file)
+			.resize(64, 64)
+			.toFormat('webp')
+			.toFile(path.join(ThumbUtil.squareThumbPath, output));
+		await sharp(file)
+			.resize(225, null)
+			.toFormat('webp')
+			.toFile(path.join(ThumbUtil.thumbPath, output));
+	}
+
+	static async generateThumbnailForVideo(filename, output) {
+		const filePath = path.join(__dirname, '../../../', process.env.UPLOAD_FOLDER, filename);
+
+		ffmpeg(filePath)
+			.thumbnail({
+				timestamps: [0],
+				filename: '%b.webp',
+				folder: ThumbUtil.squareThumbPath,
+				size: '64x64'
+			})
+			.on('error', error => log.error(error.message));
+
+		ffmpeg(filePath)
+			.thumbnail({
+				timestamps: [0],
+				filename: '%b.webp',
+				folder: ThumbUtil.thumbPath,
+				size: '150x?'
+			})
+			.on('error', error => log.error(error.message));
+
+		try {
+			await previewUtil({
+				input: filePath,
+				width: 150,
+				output: path.join(ThumbUtil.videoPreviewPath, output)
+			});
+		} catch (e) {
+			log.error(e);
+		}
+	}
+
+	static getFileThumbnail(filename) {
+		if (!filename) return null;
+		const ext = path.extname(filename).toLowerCase();
+
+		const isImage = ThumbUtil.imageExtensions.includes(ext);
+		const isVideo = ThumbUtil.videoExtensions.includes(ext);
+
+		if (isImage) return { thumb: `${filename.slice(0, -ext.length)}.webp` };
+		if (isVideo) {
+			return {
+				thumb: `${filename.slice(0, -ext.length)}.webp`,
+				preview: `${filename.slice(0, -ext.length)}.webm`
+			};
+		}
+
+		return null;
+	}
+
+	static async removeThumbs({ thumb, preview }) {
+		if (thumb) {
+			await jetpack.removeAsync(path.join(ThumbUtil.thumbPath, thumb));
+			await jetpack.removeAsync(path.join(ThumbUtil.squareThumbPath, thumb));
+		}
+		if (preview) {
+			await jetpack.removeAsync(path.join(ThumbUtil.videoPreviewPath, preview));
+		}
+	}
+}
+
+module.exports = ThumbUtil;