mirror of
https://github.com/arthur-pbty/xiao.git
synced 2026-06-03 23:36:43 +02:00
42 lines
1.1 KiB
JavaScript
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;
|
|
}
|
|
};
|