embedder/app/routes/index.ts

import type {RequestHandler as Middleware, Router, Request, Response, NextFunction} from 'express';

import multer from "multer";
import express from "express";
import ffmpeg from "fluent-ffmpeg";
import imageProbe from "probe-image-size";
import ffmpegpath from "@ffmpeg-installer/ffmpeg";
// @ts-ignore
import ffprobepath from "@ffprobe-installer/ffprobe";

ffmpeg.setFfmpegPath(ffmpegpath.path);
ffmpeg.setFfprobePath(ffprobepath.path);

import fs from "fs";

import {db, createUser} from "../db";
import {checkAuth, checkSharexAuth, createEmbedData, handleUpload} from "./middleware";
import { MediaRow } from '../types';

function extension(str: String){
	let file = str.split("/").pop();
	return [file.substr(0,file.lastIndexOf(".")),file.substr(file.lastIndexOf("."),file.length).toLowerCase()];
}

const storage = multer.diskStorage({
	destination:  (req, file, cb) => {
		cb(null, "uploads/");
	},
	filename : (req, file, cb) => {
		let nameAndExtension = extension(file.originalname);
		db.all("SELECT * FROM media WHERE path = ?", [nameAndExtension[0] + nameAndExtension[1]],  (err: Error, 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?

const fetchMedia: Middleware = (req, res, next) => {
	db.all("SELECT * FROM media", (err: Error, rows: []) => {
		if (err) return next(err);
		let files = rows.map((row: MediaRow)=> {
			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();
	});
}

let router = express.Router();

router.get("/", (req: Request, res: Response, next: NextFunction) => {
	if (!req.user)
		return res.render("home")
	next();
}, fetchMedia, (req: Request, res: Response) => {
	res.locals.filter = null;
	res.render("index", { user: req.user });
});

router.get("/gifv/:file", async (req, res, next) => {
	let url = `${req.protocol}://${req.get("host")}/uploads/${req.params.file}`;
	let width; let height;

	let nameAndExtension = extension(`uploads/${req.params.file}`);
	if (nameAndExtension[1] == ".mp4" || nameAndExtension[1] == ".mov" || nameAndExtension[1] == ".webm") {
		ffmpeg()
			.input(`uploads/${req.params.file}`)
			.inputFormat(nameAndExtension[1].substring(1))
			.ffprobe((err: Error, data: ffmpeg.FfprobeData) => {
				if (err) return next(err);
				width = data.streams[0].width;
				height = data.streams[0].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((err: Error, data: ffmpeg.FfprobeData) => {
				if (err) return next(err);
				width = data.streams[0].width;
				height = data.streams[0].height;
				return res.render("gifv", { url: url, host: `${req.protocol}://${req.get("host")}`, width: width, height: height });
			});
	} else {
		let imageData = await imageProbe(fs.createReadStream(`uploads/${req.params.file}`));
		return res.render("gifv", { url: url, host: `${req.protocol}://${req.get("host")}`, width: imageData.width, height: imageData.height });
	} 
});

router.post("/", [checkAuth, upload.array("fileupload"), createEmbedData, handleUpload], (req: Request, res: Response) => {
	res.redirect("/")
});

router.post("/sharex", [checkSharexAuth, upload.array("fileupload"), createEmbedData, handleUpload], (req: Request, res: Response) => {
	// @ts-ignore
	return res.send(`${req.protocol}://${req.get("host")}/uploads/${req.files[0].filename}`);
});

router.post("/:id(\\d+)/delete", [checkAuth], (req: Request, res: Response, next: NextFunction) => {
	db.all("SELECT path FROM media WHERE id = ?", [ req.params.id ], (err: Error, path: Array<any>) => {
		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
					], (err: Error) => {
						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
				], (err: Error) => {
					if (err) { return next(err); }
					return res.redirect("/");
				});
			}
		}));
	});
});

export default router;