embedder/routes/index.js

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);
    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(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;