Files
xiao/types/image.js
T
2021-01-30 16:30:13 -05:00

42 lines
1.1 KiB
JavaScript

const { ArgumentType } = require('discord.js-commando');
const fileTypeRe = /\.(jpe?g|png|gif|jfif|bmp)(\?.+)?$/i;
const request = require('node-superfetch');
const validURL = require('valid-url');
module.exports = class ImageArgumentType extends ArgumentType {
constructor(client) {
super(client, 'image');
}
async validate(value, msg, arg, currentMsg) {
const attachment = currentMsg.attachments.first();
if (attachment) {
if (attachment.size > 8e+6) return 'Please provide an image under 8 MB.';
if (!fileTypeRe.test(attachment.name)) return 'Please only send PNG, JPG, BMP, or GIF format images.';
return true;
}
if (fileTypeRe.test(value.toLowerCase())) {
if (!validURL.isWebUri(value)) return false;
try {
await request.get(value);
return true;
} catch {
return false;
}
}
return false;
}
parse(value, msg, arg, currentMsg) {
const attachment = currentMsg.attachments.first();
if (attachment) return attachment.url;
if (fileTypeRe.test(value.toLowerCase())) return value;
return null;
}
isEmpty(value, msg, arg, currentMsg) {
if (currentMsg.attachments.size) return false;
return !value;
}
};