murm/mrmBot-Matrix
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Re-add a form of size limit

Browse source
This commit is contained in:
TheEssem 2021-05-06 22:01:30 -05:00
parent 364d8bf006
commit c4b57ac617
2 changed files with 8 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
classes/imageCommand.js
View file

@ -57,6 +57,10 @@ class ImageCommand extends Command {
collections.runningCommands.delete(this.message.author.id); collections.runningCommands.delete(this.message.author.id);
return this.constructor.noImage; return this.constructor.noImage;
} }
if (image.type === "large") {
collections.runningCommands.delete(this.message.author.id);
return `${this.message.author.mention}, that image is too large!`;
}
magickParams.path = image.path; magickParams.path = image.path;
magickParams.type = image.type; magickParams.type = image.type;
magickParams.url = image.url; // technically not required but can be useful for text filtering magickParams.url = image.url; // technically not required but can be useful for text filtering

4
utils/image.js
View file

@ -210,6 +210,10 @@ exports.getType = async (image) => {
"Range": "bytes=0-1023" "Range": "bytes=0-1023"
}}); }});
clearTimeout(timeout); clearTimeout(timeout);
if (parseInt(imageRequest.headers.get("Content-Range").split("/")[1]) > 20971520) {
type = "large";
return type;
}
const imageBuffer = await imageRequest.buffer(); const imageBuffer = await imageRequest.buffer();
const imageType = await fileType.fromBuffer(imageBuffer); const imageType = await fileType.fromBuffer(imageBuffer);
if (imageType && formats.includes(imageType.mime)) { if (imageType && formats.includes(imageType.mime)) {