discord.js/src/structures/datastore/ClientDataStore.js

const AbstractDataStore = require('./AbstractDataStore');
const Constants = require('../../util/Constants');
const cloneObject = require('../../util/CloneObject');
const Guild = require('../Guild');
const User = require('../User');
const DMChannel = require('../DMChannel');
const TextChannel = require('../TextChannel');
const VoiceChannel = require('../VoiceChannel');
const GuildChannel = require('../GuildChannel');

class ClientDataStore extends AbstractDataStore {
  constructor(client) {
    super();

    this.client = client;
    this.token = null;
    this.session = null;
    this.user = null;
    this.email = null;
    this.password = null;

    this.register('users');
    this.register('guilds');
    this.register('channels');
  }

  get pastReady() {
    return this.client.ws.status === Constants.Status.READY;
  }

  newGuild(data) {
    const already = this.get('guilds', data.id);
    const guild = this.add('guilds', new Guild(this.client, data));
    if (this.pastReady && !already) {
      this.client.emit(Constants.Events.GUILD_CREATE, guild);
    }

    return guild;
  }

  newUser(data) {
    return this.add('users', new User(this.client, data));
  }

  newChannel(data, $guild) {
    let guild = $guild;
    const already = this.get('channels', data.id);
    let channel;
    if (data.type === Constants.ChannelTypes.DM) {
      channel = new DMChannel(this.client, data);
    } else {
      guild = guild || this.get('guilds', data.guild_id);
      if (guild) {
        if (data.type === Constants.ChannelTypes.text) {
          channel = new TextChannel(guild, data);
          guild.channels.set(channel.id, channel);
        } else if (data.type === Constants.ChannelTypes.voice) {
          channel = new VoiceChannel(guild, data);
          guild.channels.set(channel.id, channel);
        }
      }
    }

    if (channel) {
      if (this.pastReady && !already) {
        this.client.emit(Constants.Events.CHANNEL_CREATE, channel);
      }

      return this.add('channels', channel);
    }
    return null;
  }

  killGuild(guild) {
    const already = this.get('guilds', guild.id);
    this.remove('guilds', guild);
    if (already && this.pastReady) {
      this.client.emit(Constants.Events.GUILD_DELETE, guild);
    }
  }

  killUser(user) {
    this.remove('users', user);
  }

  killChannel(channel) {
    this.remove('channels', channel);
    if (channel instanceof GuildChannel) {
      channel.guild.channels.delete(channel.id);
    }
  }

  updateGuild(currentGuild, newData) {
    const oldGuild = cloneObject(currentGuild);
    currentGuild.setup(newData);
    if (this.pastReady) {
      this.client.emit(Constants.Events.GUILD_UPDATE, oldGuild, currentGuild);
    }
  }

  updateChannel(currentChannel, newData) {
    currentChannel.setup(newData);
  }
}

module.exports = ClientDataStore;