1
0
mirror of https://github.com/Foltik/Shimapan synced 2024-11-28 05:26:44 -05:00
shimapan/app/routes/view.js

43 lines
1.3 KiB
JavaScript

const express = require('express');
const router = express.Router();
const config = require('config');
const ModelPath = '../models/';
const Upload = require(ModelPath + 'Upload.js');
const wrap = require('../util/wrap.js');
const incrementViews = async id =>
Upload.updateOne({id: id}, {$inc: {views: 1}});
router.get('/:id', wrap(async (req, res) => {
const upload = await Upload.findOne({id: req.params.id});
if (!upload)
return res.status(404).json({message: 'File not found.'});
// Increment the file's view counter
await incrementViews(req.params.id);
// Whether the file should be an attachment or displayed inline on the page
let inline = false;
const mimetype = upload.file.mimetype.split('/');
const inlineMimeTypes = config.get('View.inlineMimeTypes').map(type => type.split('/'));
for (let type in inlineMimeTypes)
if (mimetype[0] === type[0])
if (mimetype[1] === type[1] || type[1] === '*')
inline = true;
res.set({
'Content-Disposition': inline ? 'inline' : 'attachment; filename="' + upload.file.originalname + '"',
'Content-Type': upload.file.mimetype
});
fs.createReadStream(upload.file.path)
.pipe(res);
}));
module.exports = router;