50 lines
1.1 KiB
JavaScript
50 lines
1.1 KiB
JavaScript
|
const config = require('../config');
|
||
|
|
const storage = require('../storage');
|
||
|
|
const Limiter = require('../limiter');
|
||
|
|
|
||
|
|
function id(user) {
|
||
|
|
return `filelist-${user}`;
|
||
|
|
}
|
||
|
|
|
||
|
|
module.exports = {
|
||
|
|
async get(req, res) {
|
||
|
|
if (!req.user) {
|
||
|
|
return res.sendStatus(401);
|
||
|
|
}
|
||
|
|
try {
|
||
|
|
const fileId = id(req.user);
|
||
|
|
const contentLength = await storage.length(fileId);
|
||
|
|
const fileStream = await storage.get(fileId);
|
||
|
|
res.writeHead(200, {
|
||
|
|
'Content-Type': 'application/octet-stream',
|
||
|
|
'Content-Length': contentLength
|
||
|
|
});
|
||
|
|
fileStream.pipe(res);
|
||
|
|
} catch (e) {
|
||
|
|
res.sendStatus(404);
|
||
|
|
}
|
||
|
|
},
|
||
|
|
|
||
|
|
async post(req, res) {
|
||
|
|
if (!req.user) {
|
||
|
|
return res.sendStatus(401);
|
||
|
|
}
|
||
|
|
try {
|
||
|
|
const limiter = new Limiter(1024 * 1024 * 10);
|
||
|
|
const fileStream = req.pipe(limiter);
|
||
|
|
await storage.set(
|
||
|
|
id(req.user),
|
||
|
|
fileStream,
|
||
|
|
{ n: 'a' }, //TODO
|
||
|
|
config.max_expire_seconds
|
||
|
|
);
|
||
|
|
res.sendStatus(200);
|
||
|
|
} catch (e) {
|
||
|
|
if (e.message === 'limit') {
|
||
|
|
return res.sendStatus(413);
|
||
|
|
}
|
||
|
|
res.sendStatus(500);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
};
|