let express = require('express'); let multer = require('multer'); let ffmpegpath = require('@ffmpeg-installer/ffmpeg').path; let ffprobepath = require('@ffprobe-installer/ffprobe').path; let ffmpeg = require('fluent-ffmpeg'); ffmpeg.setFfmpegPath(ffmpegpath); ffmpeg.setFfprobePath(ffprobepath); let db = require('../db'); let fs = require('fs'); function extension(str){ let file = str.split('/').pop(); return [file.substr(0,file.lastIndexOf('.')),file.substr(file.lastIndexOf('.'),file.length).toLowerCase()] } const storage = multer.diskStorage({ destination: function (req, file, cb) { cb(null, 'uploads/') }, filename : function(req, file, cb) { let nameAndExtension = extension(file.originalname); db.all('SELECT * FROM media WHERE path = ?', [nameAndExtension[0] + nameAndExtension[1]], function (err, exists) { if (exists.length != 0) { let suffix = new Date().getTime() / 1000; if (req.body.title == '' || req.body.title == null || req.body.title == undefined) cb(null, nameAndExtension[0] + '-' + suffix + nameAndExtension[1]) else cb(null, req.body.title + '-' + suffix + nameAndExtension[1]) } else { if (req.body.title == '' || req.body.title == null || req.body.title == undefined) cb(null, nameAndExtension[0] + nameAndExtension[1]) else cb(null, req.body.title + nameAndExtension[1]) } }) } }); let allowedMimeTypes = [ 'image/png', 'image/jpg', 'image/jpeg', 'image/gif', 'image/webp', 'video/mp4', 'video/mov', 'video/webm', 'audio/mpeg', 'audio/ogg' ] const fileFilter = function(req, file, cb) { if (allowedMimeTypes.includes(file.mimetype)) { cb(null, true); } else { cb(null, false); } } let upload = multer({ storage: storage /**, fileFilter: fileFilter**/ }); //maybe make this a env variable? function fetchMedia(req, res, next) { db.all('SELECT * FROM media', (err, rows) => { if (err) return next(err); let files = rows.map((row)=> { return { id: row.id, path: row.path, expire: row.expire, url: '/' + row.id } }); res.locals.files = files.reverse(); //reverse so newest files appear first res.locals.Count = files.length; next(); }); } //middleware //Checks ShareX key function checkAuth(req, res, next) { let auth = process.env.EBAPI_KEY || process.env.EBPASS || 'pleaseSetAPI_KEY'; let key = null; if (req.headers['key']) { key = req.headers['key']; } else { return res.status(400).send('{success: false, message: "No key provided", fix: "Provide a key"}'); } if (auth != key) { return res.status(401).send('{success: false, message: "Invalid key", fix: "Provide a valid key"}'); } shortKey = key.substr(0, 3) + '...'; console.log('Authenicated user with key: ' + shortKey); next(); } //Converts mp4 to gif and vice versa with ffmpeg function convert(req, res, next) { for (file in req.files) { let nameAndExtension = extension(req.files[file].originalname); let oembed = { type: "video", version: "1.0", provider_name: "embedder", provider_url: "https://github.com/WaveringAna/embedder", cache_age: 86400, html: "", width: 640, height: 360 }; fs.writeFile('uploads/oembed-' + nameAndExtension[0] + '.json', JSON.stringify(oembed), function (err) { if (err) return next(err); console.log('oembed file created ' + nameAndExtension[0] + '.json'); }); if (nameAndExtension[1] == '.mp4') { console.log('Converting ' + nameAndExtension[0] + nameAndExtension[1] + ' to gif'); console.log(nameAndExtension[0] + nameAndExtension[1]); ffmpeg() .input('uploads/' + req.files[file].originalname) .inputFormat('mp4') .outputFormat('gif') .output('uploads/' + nameAndExtension[0] + '.gif') .on('end', function() { console.log('Conversion complete'); console.log('Uploaded to uploads/' + nameAndExtension[0] + '.gif'); }) .on('error', (e) => console.log(e)) .run(); } else if (nameAndExtension[1] == '.gif') { console.log('Converting ' + nameAndExtension[0] + nameAndExtension[1] + ' to mp4'); ffmpeg('uploads/' + req.files[file].originalname) .inputFormat('gif') .outputFormat('mp4') .outputOptions([ '-pix_fmt yuv420p', '-c:v libx264', '-movflags +faststart' ]) .noAudio() .output('uploads/' + nameAndExtension[0] + '.mp4') .on('end', function() { console.log('Conversion complete'); console.log('Uploaded to uploads/' + nameAndExtension[0] + '.mp4'); }) .run(); } } next(); }; let router = express.Router(); router.get('/', function (req, res, next) { if (!req.user) { return res.render('home'); } next(); }, fetchMedia, function(req, res, next) { res.locals.filter = null; res.render('index', { user: req.user }); }); router.get('/gifv/:file', function (req, res, next) { let url = req.protocol + '://' + req.get('host') + '/uploads/' + req.params.file; let width; let height; nameAndExtension = extension('uploads/' + req.params.file); if (nameAndExtension[1] == '.mp4') { ffmpeg() .input('uploads/' + req.params.file) .inputFormat('mp4') .ffprobe(function(err, data) { if (err) return next(err); width = data.streams[0].width; height = data.streams[0].height; console.log(width + 'x' + height); return res.render('gifv', { url: url, host: req.protocol + '://' + req.get('host'), width: width, height: height }); }); } else if (nameAndExtension[1] == '.gif') { ffmpeg() .input('uploads/' + req.params.file) .inputFormat('gif') .ffprobe(function(err, data) { if (err) return next(err); width = data.streams[0].width; height = data.streams[0].height; console.log(width + 'x' + height); return res.render('gifv', { url: url, host: req.protocol + '://' + req.get('host'), width: width, height: height }); }); } }); router.post('/', [upload.array('fileupload'), convert], function(req, res, next) { if (!req.files || Object.keys(req.files).length === 0) { console.log(req) return res.status(400).send('No files were uploaded.'); } for (file in req.files) { db.run('INSERT INTO media (path) VALUES (?)', [req.files[file].filename], function (err) { if (err) { console.log(err); return next(err); } return res.redirect('/'); }) } }); router.post('/sharex', [checkAuth, upload.array('fileupload')], function(req, res, next) { if (!req.files || Object.keys(req.files).length === 0) { console.log(req); return res.status(400).send('No files were uploaded.'); } for (file in req.files) { db.run('INSERT INTO media (path) VALUES (?)', [req.files[file].filename], function (err) { if (err) { console.log(err); return next(err); } console.log(req.protocol + '://' + req.get('host') + '/uploads/' + req.files[file].filename); return res.send(req.protocol + '://' + req.get('host') + '/uploads/' + req.files[file].filename); }); } }); router.post('/:id(\\d+)/delete', function(req, res, next) { db.all('SELECT path FROM media WHERE id = ?', [ req.params.id ], function(err, path) { if (err) { return next(err); } fs.unlink('uploads/' + path[0].path, (err => { if (err) { console.log(err) if (err.errno = -4058) { //File just doesnt exist anymore db.run('DELETE FROM media WHERE id = ?', [ req.params.id ], function(err) { if (err) { return next(err); } return res.redirect('/'); }); } else { console.log(err) return res.redirect('/'); } } else { console.log(`Deleted ${path}`); //Callback Hell :D db.run('DELETE FROM media WHERE id = ?', [ req.params.id ], function(err) { if (err) { return next(err); } return res.redirect('/'); }); } })); }); }); module.exports = router;