const express = require("express") const busboy = require("connect-busboy"); const path = require("path"); const fs = require("fs-extra"); const bodyParser = require("body-parser"); const crypto = require("crypto"); const app = express() const redis = require("redis"), client = redis.createClient(); client.on("error", (err) => { console.log(err); }) app.use(busboy()); app.use(bodyParser.json()); app.use(express.static(path.join(__dirname, "../public"))); app.get("/download/:id", (req, res) => { res.sendFile(path.join(__dirname + "/../public/download.html")); }); app.get("/assets/download/:id", (req, res) => { let id = req.params.id; if (!validateID(id)){ res.send(404); return; } client.hget(id, "filename", (err, reply) => { // maybe some expiration logic too if (!reply) { res.sendStatus(404); } else { res.setHeader("Content-Disposition", "attachment; filename=" + reply); res.setHeader("Content-Type", "application/octet-stream"); res.download(__dirname + "/../static/" + id, reply, (err) => { if (!err) { client.del(id); fs.unlinkSync(__dirname + "/../static/" + id); } }); } }) }); app.post("/delete/:id", (req, res) => { let id = req.params.id; if (!validateID(id)){ res.send(404); return; } let delete_token = req.body.delete_token; if (!delete_token){ res.sendStatus(404); } client.hget(id, "delete", (err, reply) => { if (!reply) { res.sendStatus(404); } else { client.del(id); fs.unlinkSync(__dirname + "/../static/" + id); res.sendStatus(200); } }) }); app.post("/upload/:id", (req, res, next) => { if (!validateID(req.params.id)){ res.send(404); return; } let fstream; req.pipe(req.busboy); req.busboy.on("file", (fieldname, file, filename) => { console.log("Uploading: " + filename); //Path where image will be uploaded fstream = fs.createWriteStream(__dirname + "/../static/" + req.params.id); file.pipe(fstream); fstream.on("close", () => { let id = req.params.id; let uuid = crypto.randomBytes(10).toString('hex'); client.hmset([id, "filename", filename, "delete", uuid]); // delete the file off the server in 24 hours // setTimeout(() => { // fs.unlinkSync(__dirname + "/static/" + id); // }, 86400000); client.expire(id, 86400000); console.log("Upload Finished of " + filename); res.send(uuid); }); }); }); app.listen(3000, () => { console.log("Portal app listening on port 3000!") }) let validateID = (route_id) => { return route_id.match(/^[0-9a-fA-F]{32}$/) !== null; }