107 lines
No EOL
4 KiB
TypeScript
107 lines
No EOL
4 KiB
TypeScript
import type {RequestHandler as Middleware, 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";
|
|
import ffprobepath from "@ffprobe-installer/ffprobe";
|
|
|
|
ffmpeg.setFfmpegPath(ffmpegpath.path);
|
|
ffmpeg.setFfprobePath(ffprobepath.path);
|
|
|
|
import fs from "fs";
|
|
|
|
import {extension, videoExtensions} from "../lib/lib";
|
|
import {db, MediaRow, getPath, deleteId} from "../lib/db";
|
|
import {fileStorage} from "../lib/multer";
|
|
import {checkAuth, checkSharexAuth, convertTo720p, createEmbedData, handleUpload} from "../lib/middleware";
|
|
|
|
const upload = multer({ storage: fileStorage /**, fileFilter: fileFilter**/ }); //maybe make this a env variable?
|
|
/**Middleware to grab media from media database */
|
|
const fetchMedia: Middleware = (req, res, next) => {
|
|
const admin: boolean = req.user.username == "admin" ? true : false;
|
|
/**Check if the user is an admin, if so, show all posts from all users */
|
|
const query: string = admin == true ? "SELECT * FROM media" : `SELECT * FROM media WHERE username = '${req.user.username}'`;
|
|
|
|
db.all(query, (err:Error, rows: []) => {
|
|
if (err) return next(err);
|
|
const files = rows.map((row: MediaRow)=> {
|
|
return {
|
|
id: row.id,
|
|
path: row.path,
|
|
expire: row.expire,
|
|
username: row.username,
|
|
url: "/" + row.id
|
|
};
|
|
});
|
|
res.locals.files = files.reverse(); //reverse so newest files appear first
|
|
res.locals.Count = files.length;
|
|
next();
|
|
});
|
|
};
|
|
|
|
const 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: Request, res: Response, next: NextFunction) => {
|
|
const url = `${req.protocol}://${req.get("host")}/uploads/${req.params.file}`;
|
|
let width; let height;
|
|
|
|
const nameAndExtension = extension(`uploads/${req.params.file}`);
|
|
if (nameAndExtension[1] == ".mp4" || nameAndExtension[1] == ".mov" || nameAndExtension[1] == ".webm" || nameAndExtension[1] == ".gif") {
|
|
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 {
|
|
const 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"), convertTo720p, createEmbedData, handleUpload], (req: Request, res: Response) => {
|
|
res.redirect("/");
|
|
});
|
|
|
|
router.post("/sharex", [checkSharexAuth, upload.single("fileupload"), createEmbedData, handleUpload], (req: Request, res: Response) => {
|
|
return res.send(`${req.protocol}://${req.get("host")}/uploads/${req.file.filename}`);
|
|
});
|
|
|
|
router.post("/:id(\\d+)/delete", [checkAuth], async (req: Request, res: Response) => {
|
|
const path: any = await getPath(req.params.id);
|
|
|
|
const nameAndExtension = extension(path.path);
|
|
|
|
const filesToDelete = [path.path, "oembed-" + path.path + ".json"];
|
|
|
|
if (videoExtensions.includes(nameAndExtension[1]) || nameAndExtension[1] == ".gif") {
|
|
filesToDelete.push("720p-" + path.path);
|
|
}
|
|
|
|
filesToDelete.forEach(path => {
|
|
fs.unlink(path, async (err) => {
|
|
console.log(`Deleting ${path}`);
|
|
if (err && err.errno == -4058) {
|
|
await deleteId("media", req.params.id);
|
|
}
|
|
await deleteId("media", req.params.id);
|
|
});
|
|
});
|
|
|
|
return res.redirect("/");
|
|
});
|
|
|
|
export default router; |