101 lines
3.9 KiB
JavaScript
101 lines
3.9 KiB
JavaScript
const logger = require('winston');
|
|
const { getClaimByClaimId, getClaimByName } = require('../controllers/serveController.js');
|
|
const { postToStats, sendGoogleAnalytics } = require('../controllers/statsController.js');
|
|
const errorHandlers = require('../helpers/libraries/errorHandlers.js');
|
|
|
|
function serveFile ({ fileName, fileType, filePath }, res) {
|
|
logger.info(`serving file ${fileName}`);
|
|
// set default options
|
|
let options = {
|
|
headers: {
|
|
'X-Content-Type-Options': 'nosniff',
|
|
'Content-Type' : fileType,
|
|
},
|
|
};
|
|
// adjust default options as needed
|
|
switch (fileType) {
|
|
case 'image/jpeg':
|
|
break;
|
|
case 'image/gif':
|
|
break;
|
|
case 'image/png':
|
|
break;
|
|
case 'video/mp4':
|
|
break;
|
|
default:
|
|
logger.warn('sending file with unknown type as .jpeg');
|
|
options['headers']['Content-Type'] = 'image/jpeg';
|
|
break;
|
|
}
|
|
// send file
|
|
res.status(200).sendFile(filePath, options);
|
|
}
|
|
|
|
function sendAnalyticsAndLog (headers, ip, originalUrl) {
|
|
// google analytics
|
|
sendGoogleAnalytics('serve', headers, ip, originalUrl);
|
|
}
|
|
|
|
module.exports = (app) => {
|
|
// route to serve a specific asset
|
|
app.get('/:name/:claim_id', ({ headers, ip, originalUrl, params }, res) => {
|
|
sendAnalyticsAndLog(headers, ip, originalUrl);
|
|
// begin image-serve processes
|
|
getClaimByClaimId(params.name, params.claim_id)
|
|
.then(fileInfo => {
|
|
// check to make sure a file was found
|
|
if (!fileInfo) {
|
|
res.status(307).render('noClaims');
|
|
return;
|
|
}
|
|
// serve the file or the show route
|
|
if (headers['accept']) { // note: added b/c some requests errored out due to no accept param in header
|
|
const mimetypes = headers['accept'].split(',');
|
|
if (mimetypes.includes('text/html')) {
|
|
postToStats('show', originalUrl, ip, fileInfo.name, fileInfo.claimId, fileInfo.fileName, fileInfo.fileType, fileInfo.nsfw, 'success');
|
|
res.status(200).render('showLite', { fileInfo });
|
|
} else {
|
|
postToStats('serve', originalUrl, ip, fileInfo.name, fileInfo.claimId, fileInfo.fileName, fileInfo.fileType, fileInfo.nsfw, 'success');
|
|
serveFile(fileInfo, res);
|
|
}
|
|
} else {
|
|
postToStats('serve', originalUrl, ip, fileInfo.name, fileInfo.claimId, fileInfo.fileName, fileInfo.fileType, fileInfo.nsfw, 'success');
|
|
serveFile(fileInfo, res);
|
|
}
|
|
})
|
|
.catch(error => {
|
|
errorHandlers.handleRequestError('serve', originalUrl, ip, error, res);
|
|
});
|
|
});
|
|
// route to serve the winning claim
|
|
app.get('/:name', ({ headers, ip, originalUrl, params }, res) => {
|
|
sendAnalyticsAndLog(headers, ip, originalUrl);
|
|
// begin image-serve processes
|
|
getClaimByName(params.name)
|
|
.then(fileInfo => {
|
|
// check to make sure a file was found
|
|
if (!fileInfo) {
|
|
res.status(307).render('noClaims');
|
|
return;
|
|
}
|
|
// serve the file or the show route
|
|
if (headers['accept']) { // note: added b/c some requests errored out due to no accept param in header
|
|
const mimetypes = headers['accept'].split(',');
|
|
if (mimetypes.includes('text/html')) {
|
|
postToStats('show', originalUrl, ip, fileInfo.name, fileInfo.claimId, fileInfo.fileName, fileInfo.fileType, fileInfo.nsfw, 'success');
|
|
res.status(200).render('showLite', { fileInfo });
|
|
} else {
|
|
postToStats('serve', originalUrl, ip, fileInfo.name, fileInfo.claimId, fileInfo.fileName, fileInfo.fileType, fileInfo.nsfw, 'success');
|
|
serveFile(fileInfo, res);
|
|
}
|
|
} else {
|
|
postToStats('serve', originalUrl, ip, fileInfo.name, fileInfo.claimId, fileInfo.fileName, fileInfo.fileType, fileInfo.nsfw, 'success');
|
|
serveFile(fileInfo, res);
|
|
}
|
|
})
|
|
.catch(error => {
|
|
errorHandlers.handleRequestError('serve', originalUrl, ip, error, res);
|
|
});
|
|
});
|
|
};
|