Create MessageAttachment class

This commit is contained in:
Amish Shah
2016-08-29 15:35:39 +01:00
parent 5d06be6333
commit f6c95de85d
2 changed files with 67 additions and 4 deletions

View File

@@ -1,4 +1,5 @@
const Collection = require('../util/Collection');
const Attachment = require('./MessageAttachment');
/**
* Represents a Message on Discord
*/
@@ -76,10 +77,13 @@ class Message {
*/
this.embeds = data.embeds;
/**
* A list of attachments in the message - e.g. Pictures
* @type {Array<Object>}
* A collection of attachments in the message - e.g. Pictures - mapped by their ID.
* @type {Collection<String, MessageAttachment>}
*/
this.attachments = data.attachments;
this.attachments = new Collection();
for (const attachment of data.attachments) {
this.attachments.set(attachment.id, new Attachment(this, attachment));
}
/**
* An object containing a further users, roles or channels collections
* @type {Object}
@@ -155,7 +159,10 @@ class Message {
this.embeds = data.embeds;
}
if (data.attachments) {
this.attachments = data.attachments;
this.attachments = new Collection();
for (const attachment of data.attachments) {
this.attachments.set(attachment.id, new Attachment(this, attachment));
}
}
if (data.mentions) {
for (const mention of data.mentions) {

View File

@@ -0,0 +1,56 @@
/**
* Represents an Attachment in a Message
*/
class MessageAttachment {
constructor(message, data) {
/**
* The Client that instantiated this Message.
*/
this.client = message.client;
/**
* The message this attachment is part of.
*/
this.message = message;
this.setup(data);
}
setup(data) {
/**
* The ID of this attachment
* @type {String}
*/
this.id = data.id;
/**
* The file name of this attachment
* @type {String}
*/
this.filename = data.filename;
/**
* The size of this attachment in bytes
* @type {Number}
*/
this.filesize = data.size;
/**
* The URL to this attachment
* @type {String}
*/
this.url = data.url;
/**
* The Proxy URL to this attachment
* @type {String}
*/
this.proxyURL = data.url;
/**
* The height of this attachment (if an image)
* @type {?Number}
*/
this.height = data.height;
/**
* The width of this attachment (if an image)
* @type {?Number}
*/
this.width = data.width;
}
}
module.exports = MessageAttachment;