woomy-v2/bot/index.js

// Copyright 2020 Emily J. / mudkipscience and contributors. Subject to the AGPLv3 license.

const Discord = require('discord.js');
const CommandLoader = require('./util/commandLoader');
const EventLoader = require('./util/eventLoader');
const EventHandler = require('./event_modules/eventHandler');
const Functions = require('./util/functions');
const Database = require('./util/database');
const Logger = require('./util/logger');
const sentry = require('@sentry/node');
const config = require('../botconfig.json');
const version = require('../package.json').version;
class WoomyClient extends Discord.Client {
    constructor (options) {
        super(options);

        // Important information Woomy needs to access
        this.config = config;
        this.path = __dirname;
        this.version = version;

        // Essential modules
        this.logger = Logger;
        this.EmbedBuilder = Discord.EmbedBuilder;
        this.db = new Database(this);
        this.functions = new Functions(this);
        this.commandLoader = new CommandLoader(this);
        this.eventLoader = new EventLoader(this);
        this.eventHandler = new EventHandler(this);

        // Collections to store our successfully loaded events and commands in, as well as cooldowns.
        this.commands = new Discord.Collection();
        this.aliases = new Discord.Collection();
        this.eventModules = new Discord.Collection();
        this.cooldowns = new Discord.Collection();

        // Cache we can store stuff in
        this.cache = new Discord.Collection();
    }

    // Listen for Discord events and pass needed information to the event handler so we can respond to them.
    createEventListeners () {
        this.on('ready', this.runReadyModules);
        this.on('error', this.runErrorModules);
        this.on('interactionCreate', this.runInteractionCreateModules);
        this.on('guildCreate', this.runGuildCreateModules);
        this.on('guildDelete', this.runGuildDeleteModules);
        this.on('guildMemberAdd', this.runGuildMemberAddModules);
        this.on('guildMemberRemove', this.runGuildMemberRemoveModules);
        this.on('voiceStateUpdate', this.runVoiceStateUpdateModules);
    }
    
    // Receives information from the per-event listeners, and passes on needed information to the handler
    mainEventListener (wsEvent, param_1, param_2) {
        try {
            this.eventHandler.handle(wsEvent, param_1, param_2);
        } catch (error) {
            this.logger.error('MODULE_LISTENER_ERROR', error);
        }
    }

    // All the repeated code below just tells the main event listener what information needs to be passed to the event handler
    runReadyModules () {
        this.mainEventListener('ready');
    }

    runErrorModules (error) {
        this.mainEventListener('error', error);
    }

    runInteractionCreateModules (interaction) {
        this.mainEventListener('interactionCreate', interaction);
    }

    runGuildCreateModules (guild) {
        this.mainEventListener('guildCreate', guild);
    }

    runGuildDeleteModules (guild) {
        this.mainEventListener('guildDelete', guild);
    }

    runGuildMemberAddModules (guild, member) {
        this.mainEventListener('guildMemberAdd', guild, member);
    }

    runGuildMemberRemoveModules (guild, member) {
        this.mainEventListener('guildMemberRemove', guild, member);
    }

    runVoiceStateUpdateModules (oldState, newState) {
        this.mainEventListener('voiceStateUpdate', oldState, newState);
    }
}

// Initialize our client
const client = new WoomyClient({ 
    shards: 'auto',
    intents: [
        Discord.GatewayIntentBits.Guilds,
        Discord.GatewayIntentBits.GuildMembers,
        Discord.GatewayIntentBits.GuildEmojisAndStickers,
        Discord.GatewayIntentBits.GuildVoiceStates,
    ],
    allowedMentions: { 
        parse: [
            'users',
            'roles'
        ]
    }
});

// Extensions of native javascript types, *not good practice* but they're useful
require('./util/prototypes');

// Load commands, event modules and create listeners for Discord events (ready, errors, messages, etc)
client.commandLoader.loadCommands();
client.eventLoader.loadEventModules();
client.createEventListeners();

// Development mode is set in botconfig.json
if (client.config.developmentMode === false) {
    try { 
        sentry.init({ dsn: client.config.keys.sentry });
    } catch (err) { 
        client.logger.error('SENTRY_INIT_ERROR', `Sentry failed to initialize: ${err}`);
    }
} else {
    client.logger.warning('DEVELOPMENT_MODE', 'Running in development mode, some features have been disabled.');
}

// Login to Discord
client.login(config.token);

// Process exception/promise rejection listeners
process.on('uncaughtException', (error) => {
    const errorMsg = error.stack.replace(new RegExp(`${client.path}/`, 'g'), './');
    client.logger.error('UNCAUGHT_EXCEPTION_ERROR', errorMsg);
});

process.on('unhandledRejection', err => {
    client.logger.error('UNHANDLED_PROMISE_ERROR', err.stack);
});

// Shut down gracefully when SIGINT is received
process.on('SIGINT', () => {
    client.functions.shutdown();
});