send/app.js

68 lines
2 KiB
JavaScript
Raw Normal View History

const express = require('express')
2017-05-30 16:24:16 +00:00
var busboy = require('connect-busboy'); //middleware for form/file upload
var path = require('path'); //used for file path
var fs = require('fs-extra'); //File System - for file manipulation
const app = express()
var redis = require("redis"),
client = redis.createClient();
client.on('error', function(err) {
console.log(err);
})
2017-05-30 16:24:16 +00:00
app.use(busboy());
app.use(express.static(path.join(__dirname, 'public')));
2017-05-30 16:24:16 +00:00
app.get('/', function (req, res) {
res.send('Hello World!')
})
2017-05-30 16:24:16 +00:00
app.get('/download/:id', function(req, res) {
res.sendFile(path.join(__dirname + '/public/download.html'));
});
2017-05-30 16:24:16 +00:00
app.get('/assets/download/:id', function(req, res) {
let id = req.params.id;
2017-05-30 16:24:16 +00:00
client.hget(id, "filename", function(err, reply) { // maybe some expiration logic too
if (!reply) {
res.send('error');
} else {
2017-05-30 16:24:16 +00:00
res.setHeader('Content-Disposition', 'attachment; filename=' + reply);
// res.setHeader('Content-Transfer-Encoding', 'binary');
res.setHeader('Content-Type', 'application/octet-stream');
2017-05-30 16:24:16 +00:00
res.download(__dirname + '/static/' + reply);
}
})
2017-05-30 16:24:16 +00:00
});
2017-05-30 16:24:16 +00:00
app.route('/upload')
.post(function (req, res, next) {
2017-05-30 16:24:16 +00:00
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/' + filename);
file.pipe(fstream);
fstream.on('close', function () {
let id = Math.floor(Math.random()*10000).toString();
client.hset(id, "filename", filename, redis.print);
client.hset(id, "expiration", 0, redis.print);
console.log("Upload Finished of " + filename);
res.send(id); //where to go next
});
});
});
app.listen(3000, function () {
console.log('Example app listening on port 3000!')
})