diff --git a/types/image-or-avatar.js b/types/image-or-avatar.js index 20d4fcb5..889785ce 100644 --- a/types/image-or-avatar.js +++ b/types/image-or-avatar.js @@ -5,21 +5,21 @@ module.exports = class ImageOrAvatarArgumentType extends ArgumentType { super(client, 'image-or-avatar'); } - async validate(value, msg, arg) { - const image = await this.client.registry.types.get('image').validate(value, msg, arg); + async validate(value, msg, arg, currentMsg) { + const image = await this.client.registry.types.get('image').validate(value, msg, arg, currentMsg); if (image) return image; - return this.client.registry.types.get('user').validate(value, msg, arg); + return this.client.registry.types.get('user').validate(value, msg, arg, currentMsg); } async parse(value, msg, arg) { - const image = this.client.registry.types.get('image').parse(value, msg, arg); + const image = this.client.registry.types.get('image').parse(value, msg, arg, currentMsg); if (image) return image; - const user = await this.client.registry.types.get('user').parse(value, msg, arg); + const user = await this.client.registry.types.get('user').parse(value, msg, arg, currentMsg); return user.displayAvatarURL({ format: 'png', size: 512 }); } isEmpty(value, msg, arg) { - return this.client.registry.types.get('image').isEmpty(value, msg, arg) - && this.client.registry.types.get('user').isEmpty(value, msg, arg); + return this.client.registry.types.get('image').isEmpty(value, msg, arg, currentMsg) + && this.client.registry.types.get('user').isEmpty(value, msg, arg, currentMsg); } }; diff --git a/types/image.js b/types/image.js index 49a5c948..60b3b6c7 100644 --- a/types/image.js +++ b/types/image.js @@ -8,8 +8,8 @@ module.exports = class ImageArgumentType extends ArgumentType { super(client, 'image'); } - async validate(value, msg) { - const attachment = msg.attachments.first(); + 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.'; @@ -27,15 +27,15 @@ module.exports = class ImageArgumentType extends ArgumentType { return false; } - parse(value, msg) { - const attachment = msg.attachments.first(); + 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) { - if (msg.attachments.size) return false; - return true; + isEmpty(value, msg, arg, currentMsg) { + if (currentMsg.attachments.size) return false; + return !value; } };