120 lines
2.8 KiB
JavaScript
120 lines
2.8 KiB
JavaScript
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 conf = require('./config.js');
|
|
const stream = require('stream');
|
|
const fetch = require('node-fetch');
|
|
const storage = require('./storage.js');
|
|
|
|
let isProduction =
|
|
conf.env === 'production' &&
|
|
conf.s3_bucket !== 'localhost' &&
|
|
conf.bitly_key !== 'localhost';
|
|
|
|
const AWS = require('aws-sdk');
|
|
const s3 = new AWS.S3();
|
|
|
|
const app = express();
|
|
const redis = require('redis');
|
|
const redis_client = redis.createClient();
|
|
|
|
redis_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.sendStatus(404);
|
|
return;
|
|
}
|
|
|
|
redis_client.hget(id, 'filename', (err, reply) => {
|
|
if (!reply) {
|
|
res.sendStatus(404);
|
|
} else {
|
|
storage.length(id).then(contentLength => {
|
|
res.writeHead(200, {
|
|
'Content-Disposition': 'attachment; filename=' + reply,
|
|
'Content-Type': 'application/octet-stream',
|
|
'Content-Length': contentLength
|
|
});
|
|
});
|
|
|
|
let file_stream = storage.get(id);
|
|
|
|
file_stream.on('close', () => {
|
|
storage.forceDelete(id, redis_client).then(err => {
|
|
if (!err) {
|
|
console.log('Deleted.');
|
|
}
|
|
});
|
|
});
|
|
|
|
file_stream.pipe(res);
|
|
}
|
|
});
|
|
});
|
|
|
|
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);
|
|
}
|
|
|
|
storage
|
|
.delete(id, redis_client, delete_token)
|
|
.then(err => {
|
|
if (!err) {
|
|
console.log('Deleted off s3.');
|
|
}
|
|
})
|
|
.catch(err => res.sendStatus(404));
|
|
});
|
|
|
|
app.post('/upload/:id', (req, res, next) => {
|
|
if (!validateID(req.params.id)) {
|
|
res.send(404);
|
|
return;
|
|
}
|
|
|
|
req.pipe(req.busboy);
|
|
req.busboy.on('file', (fieldname, file, filename) => {
|
|
console.log('Uploading: ' + filename);
|
|
let url = `${req.protocol}://${req.get('host')}/download/${req.params.id}/`;
|
|
|
|
storage
|
|
.set(req.params.id, file, filename, redis_client, url)
|
|
.then(linkAndID => {
|
|
res.json(linkAndID);
|
|
});
|
|
});
|
|
});
|
|
|
|
let server = app.listen(conf.listen_port, () => {
|
|
console.log(`Portal app listening on port ${conf.listen_port}!`);
|
|
});
|
|
|
|
let validateID = route_id => {
|
|
return route_id.match(/^[0-9a-fA-F]{32}$/) !== null;
|
|
};
|