const path = require('path') const config = require('../config.js') const fs = require('fs') const gm = require('gm') const ffmpeg = require('fluent-ffmpeg') const db = require('knex')(config.database) const utilsController = {} utilsController.imageExtensions = ['.jpg', '.jpeg', '.bmp', '.gif', '.png'] utilsController.videoExtensions = ['.webm', '.mp4', '.wmv', '.avi', '.mov'] 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.authorize = async (req, res) => { const token = req.headers.token if (token === undefined) return res.status(401).json({ success: false, description: 'No token provided.' }) const user = await db.table('users').where('token', token).first() if (!user) return res.status(401).json({ success: false, description: 'Invalid token.' }) return user } utilsController.generateThumbs = function (file, basedomain) { const ext = path.extname(file.name).toLowerCase() const isVideoExt = utilsController.videoExtensions.includes(ext) const isImageExt = utilsController.imageExtensions.includes(ext) if (!isVideoExt && !isImageExt) return if (isVideoExt && config.uploads.generateThumbnails.video !== true) return if (isImageExt && config.uploads.generateThumbnails.image !== true) return let thumbname = path.join(__dirname, '..', config.uploads.folder, 'thumbs', file.name.slice(0, -ext.length) + '.png') fs.access(thumbname, err => { if (err && err.code === 'ENOENT') { if (isVideoExt) { 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 if (isImageExt) { 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