Browse Source

Moved thumbnail processing to it's own utils file

Pitu 7 years ago
parent
commit
8a96a0df1d
1 changed files with 64 additions and 0 deletions
  1. 64 0
      controllers/utilsController.js

+ 64 - 0
controllers/utilsController.js

@@ -0,0 +1,64 @@
+const path = require('path')
+const config = require('../config.js')
+const fs = require('fs')
+const gm = require('gm')
+const ffmpeg = require('fluent-ffmpeg')
+
+const utilsController = {}
+
+utilsController.getPrettyDate = function(date) {
+	return date.getFullYear() + '-'
+		+ (date.getMonth() + 1) + '-'
+		+ date.getDate() + ' '
+		+ (date.getHours() < 10 ? '0' : '')
+		+ date.getHours() + ':'
+		+ (date.getMinutes() < 10 ? '0' : '')
+		+ date.getMinutes() + ':'
+		+ (date.getSeconds() < 10 ? '0' : '')
+		+ date.getSeconds()
+}
+
+utilsController.generateThumbs = function(file, basedomain) {
+	if (config.uploads.generateThumbnails !== true) return
+
+	let extensions = ['.jpg', '.jpeg', '.bmp', '.gif', '.png', '.webm', '.mp4']
+	for (let ext of extensions) {
+		if (path.extname(file.name).toLowerCase() === ext) {
+			let thumbname = path.join(__dirname, '..', config.uploads.folder, 'thumbs', file.name.slice(0, -ext.length) + '.png')
+			fs.access(thumbname, function(err) {
+				if (err && err.code === 'ENOENT') {
+					// File doesnt exist
+
+					if (ext === '.webm' || ext === '.mp4') {
+						ffmpeg(path.join(__dirname, '..', config.uploads.folder, file.name))
+							.thumbnail({
+								timestamps: [0],
+								filename: '%b.png',
+								folder: path.join(__dirname, '..', config.uploads.folder, 'thumbs'),
+								size: '200x?'
+							})
+							.on('error', (error) => {
+								console.log('Error - ', error.message)
+							})
+					} else {
+						let size = {
+							width: 200,
+							height: 200
+						}
+
+						gm(path.join(__dirname, '..', config.uploads.folder, file.name))
+							.resize(size.width, size.height + '>')
+							.gravity('Center')
+							.extent(size.width, size.height)
+							.background('transparent')
+							.write(thumbname, (error) => {
+								if (error) console.log('Error - ', error)
+							})
+					}
+				}
+			})
+		}
+	}
+}
+
+module.exports = utilsController