pull request changes

This commit is contained in:
Abhinav Adduri 2017-05-31 09:57:02 -07:00
parent 83f0cd82a8
commit ab86ecf90e
1 changed files with 8 additions and 3 deletions

9
app.js
View File

@ -26,12 +26,17 @@ app.get('/assets/download/:id', function(req, res) {
let id = req.params.id; let id = req.params.id;
client.hget(id, "filename", function(err, reply) { // maybe some expiration logic too client.hget(id, "filename", function(err, reply) { // maybe some expiration logic too
if (!reply) { if (!reply) {
res.send('This link has expired!'); res.sendStatus(404);
} else { } else {
res.setHeader('Content-Disposition', 'attachment; filename=' + reply); res.setHeader('Content-Disposition', 'attachment; filename=' + reply);
res.setHeader('Content-Type', 'application/octet-stream'); res.setHeader('Content-Type', 'application/octet-stream');
res.download(__dirname + '/static/' + id, reply, function(err) {
if (!err) {
client.del(id); client.del(id);
res.download(__dirname + '/static/' + id, reply); fs.unlink(__dirname + '/static/' + id);
}
});
} }
}) })