1 files changed, 62 insertions, 0 deletions
diff --git a/src/api/generateThumbs.js b/src/api/generateThumbs.js
new file mode 100644
index 0000000..8517608
--- /dev/null
+++ b/src/api/generateThumbs.js
@@ -0,0 +1,62 @@
+require('dotenv').config();
+
+const jetpack = require('fs-jetpack');
+const path = require('path');
+const fs = require('fs');
+const log = require('./utils/Log');
+const sharp = require('sharp');
+const ffmpeg = require('fluent-ffmpeg');
+const imageExtensions = ['.jpg', '.jpeg', '.bmp', '.gif', '.png', '.webp'];
+const videoExtensions = ['.webm', '.mp4', '.wmv', '.avi', '.mov'];
+
+class ThumbGenerator {
+	static generateThumbnails(filename) {
+		const ext = path.extname(filename).toLowerCase();
+		const output = `${filename.slice(0, -ext.length)}.png`;
+		if (imageExtensions.includes(ext)) return this.generateThumbnailForImage(filename, output);
+		if (videoExtensions.includes(ext)) return this.generateThumbnailForVideo(filename);
+		return null;
+	}
+
+	static async generateThumbnailForImage(filename, output) {
+		const file = await jetpack.readAsync(path.join(__dirname, '..', '..', process.env.UPLOAD_FOLDER, filename), 'buffer');
+		await sharp(file)
+			.resize(64, 64)
+			.toFormat('png')
+			.toFile(path.join(__dirname, '..', '..', process.env.UPLOAD_FOLDER, 'thumbs', 'square', output));
+		await sharp(file)
+			.resize(225, null)
+			.toFormat('png')
+			.toFile(path.join(__dirname, '..', '..', process.env.UPLOAD_FOLDER, 'thumbs', output));
+	}
+
+	static generateThumbnailForVideo(filename) {
+		ffmpeg(path.join(__dirname, '..', '..', process.env.UPLOAD_FOLDER, filename))
+			.thumbnail({
+				timestamps: [0],
+				filename: '%b.png',
+				folder: path.join(__dirname, '..', '..', process.env.UPLOAD_FOLDER, 'thumbs', 'square'),
+				size: '64x64'
+			})
+			.on('error', error => log.error(error.message));
+		ffmpeg(path.join(__dirname, '..', '..', process.env.UPLOAD_FOLDER, filename))
+			.thumbnail({
+				timestamps: [0],
+				filename: '%b.png',
+				folder: path.join(__dirname, '..', '..', process.env.UPLOAD_FOLDER, 'thumbs'),
+				size: '150x?'
+			})
+			.on('error', error => log.error(error.message));
+	}
+}
+
+
+const start = async () => {
+	const files = fs.readdirSync(path.join(__dirname, '..', '..', process.env.UPLOAD_FOLDER));
+	for (const fileName of files) {
+		console.log(`Generating thumb for '${fileName}`);
+		await ThumbGenerator.generateThumbnails(fileName);
+	}
+}
+
+start();
+\ No newline at end of file