mirror of https://gitgud.io/fatchan/jschan.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
69 lines
1.5 KiB
69 lines
1.5 KiB
'use strict';
|
|
|
|
const FileType = require('file-type')
|
|
, config = require(__dirname+'/../misc/config.js')
|
|
, { addCallback } = require(__dirname+'/../redis/redis.js')
|
|
, image = new Set([
|
|
'image/jpeg',
|
|
'image/pjpeg',
|
|
'image/png',
|
|
'image/bmp',
|
|
])
|
|
, animatedImage = new Set([
|
|
'image/gif',
|
|
'image/webp',
|
|
'image/apng',
|
|
])
|
|
, video = new Set([
|
|
'video/mpeg',
|
|
'video/quicktime',
|
|
'video/mp4',
|
|
'video/webm',
|
|
'video/x-matroska',
|
|
'video/ogg',
|
|
])
|
|
, audio = new Set([
|
|
'audio/flac',
|
|
'audio/mp3',
|
|
'audio/mpeg',
|
|
'audio/ogg',
|
|
'audio/wave',
|
|
'audio/wav',
|
|
]);
|
|
|
|
let other;
|
|
|
|
const updateOtherMimes = () => {
|
|
other = new Set(config.get.otherMimeTypes);
|
|
};
|
|
|
|
updateOtherMimes();
|
|
addCallback('config', updateOtherMimes);
|
|
|
|
module.exports = {
|
|
|
|
allowed: (mimetype, options) => {
|
|
return (options.image && image.has(mimetype)) ||
|
|
(options.animatedImage && animatedImage.has(mimetype)) ||
|
|
(options.video && video.has(mimetype)) ||
|
|
(options.audio && audio.has(mimetype)) ||
|
|
(options.other && other.has(mimetype));
|
|
},
|
|
|
|
realMimeCheck: async (file) => {
|
|
const supposedMimeType = file.mimetype;
|
|
const realMimeType = await FileType.fromFile(file.tempFilePath);
|
|
if (realMimeType) {
|
|
//note the correct file extension in case it is incorrect/missing
|
|
file.extension = `.${realMimeType.ext}`;
|
|
file.realMimetype = realMimeType.mime;
|
|
return supposedMimeType === realMimeType.mime;
|
|
}
|
|
return config.get.allowMimeNoMatch;
|
|
},
|
|
|
|
getOther: () => other,
|
|
|
|
image, animatedImage, video, audio
|
|
|
|
};
|
|
|