fox-send/server/routes/download.js

55 lines
1.3 KiB
JavaScript
Raw Normal View History

const storage = require('../storage');
const mozlog = require('../log');
const log = mozlog('send.download');
2019-02-12 19:50:06 +00:00
const { statDownloadEvent } = require('../amplitude');
module.exports = async function(req, res) {
const id = req.params.id;
try {
2018-02-06 22:31:18 +00:00
const meta = req.meta;
2020-07-25 18:22:57 +00:00
const contentLength = await storage.length(id);
2018-08-07 22:40:17 +00:00
const fileStream = await storage.get(id);
let cancelled = false;
2020-04-30 00:33:12 +00:00
req.on('aborted', () => {
2018-07-12 23:07:18 +00:00
cancelled = true;
2018-08-07 22:40:17 +00:00
fileStream.destroy();
2018-07-12 23:07:18 +00:00
});
2020-07-25 18:22:57 +00:00
res.writeHead(200, {
'Content-Type': 'application/octet-stream',
'Content-Length': contentLength
});
2018-11-16 21:33:40 +00:00
fileStream.pipe(res).on('finish', async () => {
if (cancelled) {
2018-06-05 22:26:24 +00:00
return;
}
2018-07-17 16:48:47 +00:00
2018-02-06 22:31:18 +00:00
const dl = meta.dl + 1;
const dlimit = meta.dlimit;
2019-02-12 19:50:06 +00:00
const ttl = await storage.ttl(id);
statDownloadEvent({
id,
ip: req.ip,
country: req.geo.country,
state: req.geo.state,
2019-02-12 19:50:06 +00:00
owner: meta.owner,
download_count: dl,
2019-05-03 16:25:12 +00:00
ttl,
agent: req.ua.browser.name || req.ua.ua.substring(0, 6)
2019-02-12 19:50:06 +00:00
});
try {
2017-11-30 21:41:09 +00:00
if (dl >= dlimit) {
await storage.kill(id);
2017-11-30 21:41:09 +00:00
} else {
await storage.incrementField(id, 'dl');
2017-11-30 21:41:09 +00:00
}
} catch (e) {
2017-11-30 21:41:09 +00:00
log.info('StorageError:', id);
}
});
} catch (e) {
res.sendStatus(404);
}
};