const express = require("express") const busboy = require("connect-busboy"); const path = require("path"); const fs = require("fs-extra"); const bodyParser = require('body-parser'); const app = express() const redis = require("redis"), client = redis.createClient(); client.on("error", function(err) { console.log(err); }) app.use(busboy()); app.use(bodyParser.json()); app.use(express.static(path.join(__dirname, "public"))); app.get("/download/:id", function(req, res) { res.sendFile(path.join(__dirname + "/public/download.html")); }); app.get("/assets/download/:id", function(req, res) { let id = req.params.id; client.hget(id, "filename", function(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, function(err) { if (!err) { client.del(id); fs.unlinkSync(__dirname + "/static/" + id); } }); } }) }); app.post("/delete/:id", function(req, res) { let id = req.params.id; let delete_token = req.body.delete_token; if (!delete_token){ res.sendStatus(404); } client.hget(id, "delete", function(err, reply) { if (!reply) { res.sendStatus(404); } else { client.del(id); fs.unlinkSync(__dirname + "/static/" + id); res.sendStatus(200); } }) }); app.post("/upload/:id", function (req, res, next) { var fstream; req.pipe(req.busboy); req.busboy.on("file", function (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", function () { let id = req.params.id; let uuid = Math.floor(Math.random() * 10000000).toString() + Math.floor(Math.random() * 10000000).toString() + Math.floor(Math.random() * 10000000).toString(); client.hset(id, "filename", filename); client.hset(id, "expiration", 0); client.hset(id, "delete", uuid); // delete the file off the server in 24 hours setTimeout(function() { fs.unlinkSync(__dirname + "/static/" + id); }, 86400000); client.expire(id, 86400000); console.log("Upload Finished of " + filename); res.send(uuid); }); }); }); app.listen(3000, function () { console.log("Portal app listening on port 3000!") })