Merge pull request #9 from WaveringAna/expire

Move middlewear to seperate file, Files can expire
This commit is contained in:
Wavering Ana 2022-11-19 08:08:05 -05:00 committed by GitHub
commit 76a3f57736
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
11 changed files with 2047 additions and 377 deletions

31
.eslintrc.json Normal file
View file

@ -0,0 +1,31 @@
{
"env": {
"node": true,
"commonjs": true,
"es2021": true
},
"extends": "eslint:recommended",
"overrides": [
],
"parserOptions": {
"ecmaVersion": "latest"
},
"rules": {
"indent": [
"error",
"tab"
],
"linebreak-style": [
"error",
"unix"
],
"quotes": [
"error",
"double"
],
"semi": [
"error",
"always"
]
}
}

View file

@ -25,7 +25,7 @@ $ npm start
``` ```
Default username is admin with the password being whatever EBPASS is Default username is admin with the password being whatever EBPASS is
ShareX support is enabled at "/upload", requires auth with key ShareX support is enabled at "/upload", requires auth with key, expire key is in days
JSON JSON
``` ```
{ {
@ -39,7 +39,8 @@ JSON
}, },
"Body": "MultipartFormData", "Body": "MultipartFormData",
"Arguments": { "Arguments": {
"fileupload": null "fileupload": null,
"expire": null
}, },
"FileFormName": "fileupload", "FileFormName": "fileupload",
"URL": null, "URL": null,

82
app.js
View file

@ -1,47 +1,51 @@
require('dotenv').config(); require("dotenv").config();
let createError = require('http-errors');
let express = require('express');
let path = require('path');
let cookieParser = require('cookie-parser');
let passport = require('passport');
let session = require('express-session');
let SQLiteStore = require('connect-sqlite3')(session); let express = require("express");
let passport = require("passport");
let session = require("express-session");
let createError = require("http-errors");
let cookieParser = require("cookie-parser");
let SQLiteStore = require("connect-sqlite3")(session);
let indexRouter = require('./routes/index'); let fs = require("fs");
let authRouter = require('./routes/auth'); let path = require("path");
let authRouter = require("./routes/auth");
let indexRouter = require("./routes/index");
let db = require("./db");
let app = express(); let app = express();
app.enable('trust proxy'); app.enable("trust proxy");
// view engine setup // view engine setup
app.set('views', path.join(__dirname, 'views')); app.set("views", path.join(__dirname, "views"));
app.set('view engine', 'ejs'); app.set("view engine", "ejs");
app.use(express.json()); app.use(express.json());
app.use(express.urlencoded({ app.use(express.urlencoded({
extended: false extended: false
})); }));
app.use(cookieParser()); app.use(cookieParser());
app.use(express.static(path.join(__dirname, 'public'))); app.use(express.static(path.join(__dirname, "public")));
app.use(express.static(path.join(__dirname, 'public'))); app.use(express.static(path.join(__dirname, "public")));
app.use(session({ app.use(session({
secret: process.env.EBSECRET || 'pleasechangeme', secret: process.env.EBSECRET || "pleasechangeme",
resave: false, resave: false,
saveUninitialized: false, saveUninitialized: false,
store: new SQLiteStore({ store: new SQLiteStore({
db: 'sessions.db', db: "sessions.db",
dir: './var/db' dir: "./var/db"
}) })
})); }));
app.use(passport.authenticate('session')); app.use(passport.authenticate("session"));
app.use('/', indexRouter); app.use("/", indexRouter);
app.use('/', authRouter); app.use("/", authRouter);
app.use('/uploads', express.static('uploads')) app.use("/uploads", express.static("uploads"));
// catch 404 and forward to error handler // catch 404 and forward to error handler
app.use(function(req, res, next) { app.use(function(req, res, next) {
@ -49,15 +53,41 @@ app.use(function(req, res, next) {
}); });
// error handler // error handler
app.use(function(err, req, res, next) { app.use(function(err, req, res) {
// set locals, only providing error in development // set locals, only providing error in development
console.log(err) console.log(err);
res.locals.message = err.message; res.locals.message = err.message;
res.locals.error = req.app.get('env') === 'development' ? err : {}; res.locals.error = req.app.get("env") === "development" ? err : {};
// render the error page // render the error page
res.status(err.status || 500); res.status(err.status || 500);
res.render('error'); res.render("error");
}); });
function prune () {
console.log("Vacuuming database...");
db.run("VACUUM");
db.all("SELECT * FROM media WHERE expire > ?", [Date.now()], (err, rows) => {
console.log("Expired rows: " + rows);
if (err) return console.error(err);
rows.forEach((row) => {
console.log("Deleting " + row.path);
fs.unlink("uploads/" + row.path, (err) => {
if (err) {
if(err.errno == -4058) return; //file doesn't exist
return console.error(err);
}
console.log("Deleted " + row.path);
});
db.run("DELETE FROM media WHERE expire > ?", [Date.now()], (err) => {
if (err) return console.error(err);
console.log("Deleted " + row.path + " from database");
});
});
});
}
setInterval(prune, 1000 * 60 * 30); //prune every 30 minutes
module.exports = app; module.exports = app;

30
db.js
View file

@ -1,33 +1,33 @@
var sqlite3 = require('sqlite3'); const sqlite3 = require("sqlite3");
var mkdirp = require('mkdirp'); const mkdirp = require("mkdirp");
var crypto = require('crypto'); const crypto = require("crypto");
mkdirp.sync('./var/db'); mkdirp.sync("./var/db");
var db = new sqlite3.Database('./var/db/media.db'); let db = new sqlite3.Database("./var/db/media.db");
db.serialize(function() { db.serialize(function() {
// create the database schema for the todos app // create the database schema for the todos app
db.run("CREATE TABLE IF NOT EXISTS users ( \ db.run("CREATE TABLE IF NOT EXISTS users ( \
id INTEGER PRIMARY KEY, \ id INTEGER PRIMARY KEY, \
username TEXT UNIQUE, \ username TEXT UNIQUE, \
hashed_password BLOB, \ hashed_password BLOB, \
salt BLOB \ salt BLOB \
)"); )");
db.run("CREATE TABLE IF NOT EXISTS media ( \ db.run("CREATE TABLE IF NOT EXISTS media ( \
id INTEGER PRIMARY KEY, \ id INTEGER PRIMARY KEY, \
path TEXT NOT NULL, \ path TEXT NOT NULL, \
expire INTEGER \ expire INTEGER \
)"); )");
// create an initial user (username: alice, password: letmein) // create an initial user (username: alice, password: letmein)
var salt = crypto.randomBytes(16); var salt = crypto.randomBytes(16);
db.run('INSERT OR IGNORE INTO users (username, hashed_password, salt) VALUES (?, ?, ?)', [ db.run("INSERT OR IGNORE INTO users (username, hashed_password, salt) VALUES (?, ?, ?)", [
'admin', "admin",
crypto.pbkdf2Sync(process.env.EBPASS || 'changeme', salt, 310000, 32, 'sha256'), crypto.pbkdf2Sync(process.env.EBPASS || "changeme", salt, 310000, 32, "sha256"),
salt salt
]); ]);
}); });
module.exports = db; module.exports = db;

122
middleware.js Normal file
View file

@ -0,0 +1,122 @@
const ffmpeg = require("fluent-ffmpeg");
const ffmpegpath = require("@ffmpeg-installer/ffmpeg").path;
const ffprobepath = require("@ffprobe-installer/ffprobe").path;
ffmpeg.setFfmpegPath(ffmpegpath);
ffmpeg.setFfprobePath(ffprobepath);
const fs = require("fs");
const process = require("process");
let db = require("./db.js");
function extension(str){
let file = str.split("/").pop();
return [file.substr(0,file.lastIndexOf(".")),file.substr(file.lastIndexOf("."),file.length).toLowerCase()];
}
//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\"}");
}
let 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 (let 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: `<iframe src='${req.protocol}://${req.get("host")}/gifv/${nameAndExtension[0]}${nameAndExtension[1]}'></iframe>`,
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();
}
function handleUpload(req, res, next) {
if (!req.files || Object.keys(req.files).length === 0) {
console.log("No files were uploaded");
return res.status(400).send("No files were uploaded.");
}
for (let file in req.files) {
let currentdate = Date.now();
let expireDate = new Date(currentdate + (req.body.expire * 24 * 60 * 60 * 1000));
console.log(expireDate);
db.run("INSERT INTO media (path, expire) VALUES (?, ?)", [req.files[file].filename, expireDate], function (err) {
if (err) {
console.log(err);
return next(err);
}
console.log("Uploaded " + req.files[file].filename + " to database");
console.log("It will expire in " + req.body.expire + " days");
});
}
next();
}
module.exports = {
checkAuth: checkAuth,
convert: convert,
handleUpload: handleUpload
};

1574
package-lock.json generated

File diff suppressed because it is too large Load diff

View file

@ -39,5 +39,8 @@
"passport": "^0.6.0", "passport": "^0.6.0",
"passport-local": "^1.0.0", "passport-local": "^1.0.0",
"sqlite3": "^5.0.2" "sqlite3": "^5.0.2"
},
"devDependencies": {
"eslint": "^8.28.0"
} }
} }

View file

@ -1,29 +1,31 @@
/* eslint-env browser: true */
function copyURI(evt) { function copyURI(evt) {
evt.preventDefault(); evt.preventDefault();
navigator.clipboard.writeText(absolutePath(evt.target.getAttribute('src'))).then(() => { navigator.clipboard.writeText(absolutePath(evt.target.getAttribute("src"))).then(() => {
/* clipboard successfully set */ /* clipboard successfully set */
console.log("copied") console.log("copied");
}, () => { }, () => {
/* clipboard write failed */ /* clipboard write failed */
console.log("failed") console.log("failed");
}); });
} }
function copyA(evt) { function copyA(evt) {
evt.preventDefault(); evt.preventDefault();
navigator.clipboard.writeText(absolutePath(evt.target.getAttribute('href'))).then(() => { navigator.clipboard.writeText(absolutePath(evt.target.getAttribute("href"))).then(() => {
console.log("copied"); console.log("copied");
}, () => { }, () => {
console.log("failed"); console.log("failed");
}); });
} }
function copyPath(evt) { function copyPath(evt) {
navigator.clipboard.writeText(absolutePath(evt)).then(() => { navigator.clipboard.writeText(absolutePath(evt)).then(() => {
console.log("copied"); console.log("copied");
}, () => { }, () => {
console.log("failed"); console.log("failed");
}); });
} }
function absolutePath (href) { function absolutePath (href) {
@ -33,99 +35,109 @@ function absolutePath (href) {
} }
function extension(string) { function extension(string) {
return string.slice((string.lastIndexOf(".") - 2 >>> 0) + 2); return string.slice((string.lastIndexOf(".") - 2 >>> 0) + 2);
} }
let dropArea = document.getElementById("dropArea"); let dropArea = document.getElementById("dropArea");
;['dragenter', 'dragover', 'dragleave', 'drop'].forEach(eventName => { ["dragenter", "dragover", "dragleave", "drop"].forEach(eventName => {
dropArea.addEventListener(eventName, preventDefaults, false) dropArea.addEventListener(eventName, preventDefaults, false);
}) });
function preventDefaults (e) { function preventDefaults (e) {
e.preventDefault() e.preventDefault();
e.stopPropagation() e.stopPropagation();
} }
;['dragenter', 'dragover'].forEach(eventName => { ["dragenter", "dragover"].forEach(eventName => {
dropArea.addEventListener(eventName, highlight, false) dropArea.addEventListener(eventName, highlight, false);
}) })
;['dragleave', 'drop'].forEach(eventName => { ;["dragleave", "drop"].forEach(eventName => {
dropArea.addEventListener(eventName, unhighlight, false) dropArea.addEventListener(eventName, unhighlight, false);
}) });
function highlight(e) { function highlight(e) {
dropArea.classList.add('highlight') dropArea.classList.add("highlight");
} }
function unhighlight(e) { function unhighlight(e) {
dropArea.classList.remove('highlight') dropArea.classList.remove("highlight");
} }
dropArea.addEventListener('drop', handleDrop, false) dropArea.addEventListener("drop", handleDrop, false);
window.addEventListener('paste', handlePaste); window.addEventListener("paste", handlePaste);
function handleDrop(e) { function handleDrop(e) {
let dt = e.dataTransfer let dt = e.dataTransfer;
let files = dt.files let files = dt.files;
handleFiles(files) handleFiles(files);
} }
function handlePaste(e) { function handlePaste(e) {
// Get the data of clipboard // Get the data of clipboard
const clipboardItems = e.clipboardData.items; const clipboardItems = e.clipboardData.items;
const items = [].slice.call(clipboardItems).filter(function (item) { const items = [].slice.call(clipboardItems).filter(function (item) {
// Filter the image items only // Filter the image items only
return item.type.indexOf('image') !== -1; return item.type.indexOf("image") !== -1;
}); });
if (items.length === 0) { if (items.length === 0) {
return; return;
} }
const item = items[0]; const item = items[0];
// Get the blob of image // Get the blob of image
const blob = item.getAsFile(); const blob = item.getAsFile();
console.log(blob) console.log(blob);
uploadFile(blob) uploadFile(blob);
previewFile(blob); previewFile(blob);
} }
function handleFiles(files) { function handleFiles(files) {
files = [...files] files = [...files];
files.forEach(uploadFile) files.forEach(uploadFile);
files.forEach(previewFile) files.forEach(previewFile);
} }
function previewFile(file) { function previewFile(file) {
let reader = new FileReader() let reader = new FileReader();
reader.readAsDataURL(file) reader.readAsDataURL(file);
reader.onloadend = function() { reader.onloadend = function() {
let img = document.createElement('img'); let img = document.createElement("img");
img.src = reader.result img.src = reader.result;
img.className = "image"; img.className = "image";
document.getElementById('gallery').appendChild(img) document.getElementById("gallery").appendChild(img);
console.log(document.getElementById('fileupload')) console.log(document.getElementById("fileupload"));
document.getElementById('fileupload').src = img.src; document.getElementById("fileupload").src = img.src;
} };
} }
function uploadFile(file) { function uploadFile(file) {
let xhr = new XMLHttpRequest(); let xhr = new XMLHttpRequest();
let formData = new FormData(); let formData = new FormData();
xhr.open('POST', '/', true); let reader = new FileReader();
xhr.open("POST", "/", true);
xhr.addEventListener('readystatechange', function(e) { xhr.addEventListener("readystatechange", function(e) {
if (xhr.readyState == 4 && xhr.status == 200) { if (xhr.readyState == 4 && xhr.status == 200) {
location.reload(); location.reload();
} }
else if (xhr.readyState == 4 && xhr.status != 200) { else if (xhr.readyState == 4 && xhr.status != 200) {
// Error. Inform the user // Error. Inform the user
} }
}) });
formData.append('fileupload', file); if (file == null || file == undefined) {
xhr.send(formData) //file = reader.readAsDataURL(document.getElementById("fileupload").files[0]);
//file = reader.readAsDataURL(document.querySelector("#fileupload").files[0]);
file = document.querySelector("#fileupload").files[0];
}
formData.append("fileupload", file);
formData.append("expire", document.getElementById("expire").value);
console.log(formData);
xhr.send(formData);
} }

View file

@ -1,29 +1,30 @@
let express = require('express'); let crypto = require("crypto");
let passport = require('passport'); let express = require("express");
let LocalStrategy = require('passport-local'); let passport = require("passport");
let crypto = require('crypto'); let LocalStrategy = require("passport-local");
let db = require('../db');
let db = require("../db");
let router = express.Router(); let router = express.Router();
passport.use(new LocalStrategy(function verify(username, password, cb) { passport.use(new LocalStrategy(function verify(username, password, cb) {
db.get('SELECT * FROM users WHERE username = ?', [username], function(err, row) { db.get("SELECT * FROM users WHERE username = ?", [username], function(err, row) {
if (err) { if (err) {
return cb(err); return cb(err);
} }
if (!row) { if (!row) {
return cb(null, false, { return cb(null, false, {
message: 'Incorrect username or password.' message: "Incorrect username or password."
}); });
} }
crypto.pbkdf2(password, row.salt, 310000, 32, 'sha256', function(err, hashedPassword) { crypto.pbkdf2(password, row.salt, 310000, 32, "sha256", function(err, hashedPassword) {
if (err) { if (err) {
return cb(err); return cb(err);
} }
if (!crypto.timingSafeEqual(row.hashed_password, hashedPassword)) { if (!crypto.timingSafeEqual(row.hashed_password, hashedPassword)) {
return cb(null, false, { return cb(null, false, {
message: 'Incorrect username or password.' message: "Incorrect username or password."
}); });
} }
return cb(null, row); return cb(null, row);
@ -46,21 +47,21 @@ passport.deserializeUser(function(user, cb) {
}); });
}); });
router.get('/login', function(req, res, next) { router.get("/login", function(req, res) {
res.render('login'); res.render("login");
}); });
router.post('/login/password', passport.authenticate('local', { router.post("/login/password", passport.authenticate("local", {
successRedirect: '/', successRedirect: "/",
failureRedirect: '/login' failureRedirect: "/login"
})); }));
router.post('/logout', function(req, res, next) { router.post("/logout", function(req, res, next) {
req.logout(function(err) { req.logout(function(err) {
if (err) { if (err) {
return next(err); return next(err);
} }
res.redirect('/'); res.redirect("/");
}); });
}); });

View file

@ -1,268 +1,164 @@
let express = require('express'); let multer = require("multer");
let multer = require('multer'); let express = require("express");
let ffmpegpath = require('@ffmpeg-installer/ffmpeg').path; let ffmpeg = require("fluent-ffmpeg");
let ffprobepath = require('@ffprobe-installer/ffprobe').path; let ffmpegpath = require("@ffmpeg-installer/ffmpeg").path;
let ffmpeg = require('fluent-ffmpeg'); let ffprobepath = require("@ffprobe-installer/ffprobe").path;
ffmpeg.setFfmpegPath(ffmpegpath); ffmpeg.setFfmpegPath(ffmpegpath);
ffmpeg.setFfprobePath(ffprobepath); ffmpeg.setFfprobePath(ffprobepath);
let db = require('../db'); let fs = require("fs");
let fs = require('fs');
let db = require("../db");
let {checkAuth, convert, handleUpload} = require("../middleware");
function extension(str){ function extension(str){
let file = str.split('/').pop(); let file = str.split("/").pop();
return [file.substr(0,file.lastIndexOf('.')),file.substr(file.lastIndexOf('.'),file.length).toLowerCase()] return [file.substr(0,file.lastIndexOf(".")),file.substr(file.lastIndexOf("."),file.length).toLowerCase()];
} }
const storage = multer.diskStorage({ const storage = multer.diskStorage({
destination: function (req, file, cb) { destination: function (req, file, cb) {
cb(null, 'uploads/') cb(null, "uploads/");
}, },
filename : function(req, file, cb) { filename : function(req, file, cb) {
let nameAndExtension = extension(file.originalname); let nameAndExtension = extension(file.originalname);
db.all('SELECT * FROM media WHERE path = ?', [nameAndExtension[0] + nameAndExtension[1]], function (err, exists) { db.all("SELECT * FROM media WHERE path = ?", [nameAndExtension[0] + nameAndExtension[1]], function (err, exists) {
if (exists.length != 0) { if (exists.length != 0) {
let suffix = new Date().getTime() / 1000; let suffix = new Date().getTime() / 1000;
if (req.body.title == '' || req.body.title == null || req.body.title == undefined) if (req.body.title == "" || req.body.title == null || req.body.title == undefined)
cb(null, nameAndExtension[0] + '-' + suffix + nameAndExtension[1]) cb(null, nameAndExtension[0] + "-" + suffix + nameAndExtension[1]);
else else
cb(null, req.body.title + '-' + suffix + nameAndExtension[1]) cb(null, req.body.title + "-" + suffix + nameAndExtension[1]);
} else { } else {
if (req.body.title == '' || req.body.title == null || req.body.title == undefined) if (req.body.title == "" || req.body.title == null || req.body.title == undefined)
cb(null, nameAndExtension[0] + nameAndExtension[1]) cb(null, nameAndExtension[0] + nameAndExtension[1]);
else else
cb(null, req.body.title + nameAndExtension[1]) cb(null, req.body.title + nameAndExtension[1]);
} }
}) });
} }
}); });
let allowedMimeTypes = [ /**let allowedMimeTypes = [
'image/png', "image/png",
'image/jpg', "image/jpg",
'image/jpeg', "image/jpeg",
'image/gif', "image/gif",
'image/webp', "image/webp",
'video/mp4', "video/mp4",
'video/mov', "video/mov",
'video/webm', "video/webm",
'audio/mpeg', "audio/mpeg",
'audio/ogg' "audio/ogg"
] ];
const fileFilter = function(req, file, cb) { const fileFilter = function(req, file, cb) {
if (allowedMimeTypes.includes(file.mimetype)) { if (allowedMimeTypes.includes(file.mimetype)) {
cb(null, true); cb(null, true);
} else { } else {
cb(null, false); cb(null, false);
} }
} };**/
let upload = multer({ storage: storage /**, fileFilter: fileFilter**/ }); //maybe make this a env variable? let upload = multer({ storage: storage /**, fileFilter: fileFilter**/ }); //maybe make this a env variable?
function fetchMedia(req, res, next) { function fetchMedia(req, res, next) {
db.all('SELECT * FROM media', (err, rows) => { db.all("SELECT * FROM media", (err, rows) => {
if (err) return next(err); if (err) return next(err);
let files = rows.map((row)=> { let files = rows.map((row)=> {
return { return {
id: row.id, id: row.id,
path: row.path, path: row.path,
expire: row.expire, expire: row.expire,
url: '/' + row.id url: "/" + row.id
} };
}); });
res.locals.files = files.reverse(); //reverse so newest files appear first res.locals.files = files.reverse(); //reverse so newest files appear first
res.locals.Count = files.length; res.locals.Count = files.length;
next(); 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: "<iframe src='" + req.protocol + "://" + req.get('host') + "/gifv/" + nameAndExtension[0] + "'></iframe>",
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(); let router = express.Router();
router.get('/', function (req, res, next) { router.get("/", function (req, res, next) {
if (!req.user) { return res.render('home'); } if (!req.user) { return res.render("home"); }
next(); next();
}, fetchMedia, function(req, res, next) { }, fetchMedia, function(req, res) {
res.locals.filter = null; res.locals.filter = null;
res.render('index', { user: req.user }); res.render("index", { user: req.user });
}); });
router.get('/gifv/:file', function (req, res, next) { router.get("/gifv/:file", function (req, res, next) {
let url = req.protocol + '://' + req.get('host') + '/uploads/' + req.params.file; let url = req.protocol + "://" + req.get("host") + "/uploads/" + req.params.file;
let width; let height; let width; let height;
nameAndExtension = extension('uploads/' + req.params.file); let nameAndExtension = extension("uploads/" + req.params.file);
if (nameAndExtension[1] == '.mp4') { if (nameAndExtension[1] == ".mp4") {
ffmpeg() ffmpeg()
.input('uploads/' + req.params.file) .input("uploads/" + req.params.file)
.inputFormat('mp4') .inputFormat("mp4")
.ffprobe(function(err, data) { .ffprobe(function(err, data) {
if (err) return next(err); if (err) return next(err);
width = data.streams[0].width; width = data.streams[0].width;
height = data.streams[0].height; height = data.streams[0].height;
console.log(width + 'x' + height); console.log(width + "x" + height);
return res.render('gifv', { url: url, host: req.protocol + '://' + req.get('host'), width: width, height: height }); return res.render("gifv", { url: url, host: req.protocol + "://" + req.get("host"), width: width, height: height });
}); });
} else if (nameAndExtension[1] == '.gif') { } else if (nameAndExtension[1] == ".gif") {
ffmpeg() ffmpeg()
.input('uploads/' + req.params.file) .input("uploads/" + req.params.file)
.inputFormat('gif') .inputFormat("gif")
.ffprobe(function(err, data) { .ffprobe(function(err, data) {
if (err) return next(err); if (err) return next(err);
width = data.streams[0].width; width = data.streams[0].width;
height = data.streams[0].height; height = data.streams[0].height;
console.log(width + 'x' + height); console.log(width + "x" + height);
return res.render('gifv', { url: url, host: req.protocol + '://' + req.get('host'), width: width, height: 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) { router.post("/", [upload.array("fileupload"), convert, handleUpload], function(req, res) {
if (!req.files || Object.keys(req.files).length === 0) { return res.redirect("/");
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) { router.post("/sharex", [checkAuth, upload.array("fileupload"), convert, handleUpload], function(req, res) {
if (!req.files || Object.keys(req.files).length === 0) { return res.send(`${req.protocol}://${req.get("host")}/uploads/${req.files[0].filename}`);
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) { router.post("/:id(\\d+)/delete", function(req, res, next) {
db.all('SELECT path FROM media WHERE id = ?', [ req.params.id ], function(err, path) { db.all("SELECT path FROM media WHERE id = ?", [ req.params.id ], function(err, path) {
if (err) { return next(err); } if (err) { return next(err); }
fs.unlink('uploads/' + path[0].path, (err => { fs.unlink("uploads/" + path[0].path, (err => {
if (err) { if (err) {
console.log(err) console.log(err);
if (err.errno = -4058) { //File just doesnt exist anymore if (err.errno == -4058) { //File just doesnt exist anymore
db.run('DELETE FROM media WHERE id = ?', [ db.run("DELETE FROM media WHERE id = ?", [
req.params.id req.params.id
], function(err) { ], function(err) {
if (err) { return next(err); } if (err) { return next(err); }
return res.redirect('/'); return res.redirect("/");
}); });
} else { } else {
console.log(err) console.log(err);
return res.redirect('/'); return res.redirect("/");
} }
} }
else { else {
console.log(`Deleted ${path}`); console.log(`Deleted ${path}`);
//Callback Hell :D //Callback Hell :D
db.run('DELETE FROM media WHERE id = ?', [ db.run("DELETE FROM media WHERE id = ?", [
req.params.id req.params.id
], function(err) { ], function(err) {
if (err) { return next(err); } if (err) { return next(err); }
return res.redirect('/'); return res.redirect("/");
}); });
} }
})); }));
}); });
}); });
module.exports = router; module.exports = router;

View file

@ -32,11 +32,19 @@ return string.slice((string.lastIndexOf(".") - 2 >>> 0) + 2);
<header class="header"> <header class="header">
<h1>Embedder</h1> <h1>Embedder</h1>
<form action="/" method="post" encType="multipart/form-data"> <form action="/" method="post" encType="multipart/form-data">
<!---->
<div id="dropArea"> <div id="dropArea">
<p class="dragregion">Upload a file, copy paste, or drag n' drop into the dashed region</p> <p class="dragregion">Upload a file, copy paste, or drag n' drop into the dashed region</p>
<div id="gallery"></div> <div id="gallery"></div>
<p class="dragregion"><input class="" type="file" id="fileupload" name="fileupload"><input type="submit" value="Upload"></p> <p class="dragregion"><input class="" type="file" id="fileupload" name="fileupload"><input type="button" value="Upload" id="submit" onclick="uploadFile()"></p>
<br>
<br>
<p class="dragregion">Select file expiration date: <select name="expire" id="expire">
<option value="1">1 day</option>
<option value="7">7 days</option>
<option value="14">14 days</option>
<option value="30">30 days</option>
<option value="">never</option>
</select></p>
<p class="dragregion">Click the file to copy the url</p> <p class="dragregion">Click the file to copy the url</p>
</div> </div>
</form> </form>