Store Channel Messages in Maps

src/client/actions/MessageDelete.js

@@ -12,10 +12,10 @@ class MessageDeleteAction extends Action {
     const client = this.client;
     const channel = client.store.get('channels', data.channel_id);
     if (channel) {
-      let message = channel.store.get('messages', data.id);
+      let message = channel.messages.get(data.id);
 
       if (message) {
-        channel.store.remove('messages', message.id);
+        channel.messages.delete(message.id);
         this.deleted[channel.id + message.id] = message;
         this.scheduleForDeletion(channel.id, message.id);
       } else if (this.deleted[channel.id + data.id]) {

src/client/actions/MessageUpdate.js

@@ -9,7 +9,7 @@ class MessageUpdateAction extends Action {
     const channel = client.store.get('channels', data.channel_id);
 
     if (channel) {
-      const message = channel.store.get('messages', data.id);
+      const message = channel.messages.get(data.id);
       if (message && !message.equals(data, true)) {
         const oldMessage = cloneObject(message);
         message.patch(data);

src/structures/DMChannel.js

@@ -1,7 +1,6 @@
 const Channel = require('./Channel');
 const TextBasedChannel = require('./interface/TextBasedChannel');
 const User = require('./User');
-const TextChannelDataStore = require('./datastore/TextChannelDataStore');
 
 /**
  * Represents a Direct Message Channel between two users.
@@ -11,22 +10,23 @@ const TextChannelDataStore = require('./datastore/TextChannelDataStore');
 class DMChannel extends Channel {
   constructor(client, data) {
     super(client, data);
-    this.store = new TextChannelDataStore();
+    this.messages = new Map();
   }
 
   _cacheMessage(message) {
     const maxSize = this.client.options.max_message_cache;
     if (maxSize === 0) {
       // saves on performance
-      return;
+      return null;
     }
 
-    const storeKeys = Object.keys(this.store);
-    if (storeKeys.length >= maxSize) {
-      this.store.remove(storeKeys[0]);
+    if (this.messages.size >= maxSize) {
+      this.messages.delete(Array.from(this.messages.keys())[0]);
     }
 
-    this.store.add('messages', message);
+    this.messages.set(message.id, message);
+
+    return message;
   }
 
   setup(data) {

src/structures/TextChannel.js

@@ -1,5 +1,4 @@
 const GuildChannel = require('./GuildChannel');
-const TextChannelDataStore = require('./datastore/TextChannelDataStore');
 const TextBasedChannel = require('./interface/TextBasedChannel');
 
 /**
@@ -11,7 +10,7 @@ class TextChannel extends GuildChannel {
 
   constructor(guild, data) {
     super(guild, data);
-    this.store = new TextChannelDataStore();
+    this.messages = new Map();
   }
 
   _cacheMessage(message) {
@@ -21,12 +20,13 @@ class TextChannel extends GuildChannel {
       return null;
     }
 
-    const storeKeys = Object.keys(this.store);
-    if (storeKeys.length >= maxSize) {
-      this.store.remove(storeKeys[0]);
+    if (this.messages.size >= maxSize) {
+      this.messages.delete(Array.from(this.messages.keys())[0]);
     }
 
-    return this.store.add('messages', message);
+    this.messages.set(message.id, message);
+
+    return message;
   }
 
   sendMessage() {

src/structures/interface/TextBasedChannel.js

@@ -3,6 +3,14 @@
  * @interface
  */
 class TextBasedChannel {
+
+  constructor() {
+    /**
+     * A Map containing the messages sent to this channel.
+     * @type {Map<String, Message>}
+     */
+    this.messages = new Map();
+  }
   /**
    * Send a message to this channel
    * @param {String} content the content to send