discordbot.js/bot.js

const Discord = require("discord.js"),
    fs = require('fs'),
    logger = require('./lib/logging').getLogger(),
    cmd = require("./lib/cmd"),
    guilding = require('./lib/guilding'),
    utils = require('./lib/utils'),
    config = require('./config.json'),
    client = new Discord.Client(),
    args = require('args-parser')(process.argv),
    authToken = args.token || config.token,
    prefix = args.prefix || config.prefix,
    gamepresence = args.game || config.presence;

let presences = [],     // loaded from presences.txt file if the file exists
    rotator = null;     // an interval id to stop presence duration if needed

function main() {
    utils.Cleanup(() => {
        guilding.destroyAll();
        client.destroy();
    });
    cmd.setLogger(logger);
    guilding.setLogger(logger);
    cmd.init(prefix);
    registerCommands();

    utils.dirExistence('./data', () => {
        fs.exists('./data/presences.txt', (exist) => {
            if (exist) {
                logger.debug('Loading presences from file...');
                let lineReader = require('readline').createInterface({
                    input: require('fs').createReadStream('./data/presences.txt')
                });
                lineReader.on('line', (line) => {
                    presences.push(line);
                });
                rotator = client.setInterval(() => rotatePresence(), config.presence_duration);
            }
        })
    });
    registerCallbacks();

    client.login(authToken).then(() => {
        logger.debug("Logged in");
    });
}

/**
 * registeres global commands
 */
function registerCommands() {
    // useless test command
    cmd.createGlobalCommand(prefix + 'repeatafterme', (msg, argv, args) => {
        return args.join(' ');
    }, [], "Repeats what you say");

    // adds a presence that will be saved in the presence file and added to the rotation
    cmd.createGlobalCommand(prefix + 'addpresence', (msg, argv, args) => {
        let p = args.join(' ');
        presences.push(p);
        fs.writeFile('./data/presences.txt', presences.join('\n'), (err) => {
        });
        return `Added Presence \`${p}\``;
    }, [], "Adds a presence to the rotation.", 'owner');

    // shuts down the bot after destroying the client
    cmd.createGlobalCommand(prefix + 'shutdown', (msg) => {
        msg.reply('Shutting down...').finally(() => {
            logger.debug('Destroying client...');
            client.destroy().finally(() => {
                logger.debug(`Exiting Process...`);
                process.exit(0);
            });
        });
    }, [], "Shuts the bot down.", 'owner');

    // forces a presence rotation
    cmd.createGlobalCommand(prefix + 'rotate', () => {
        try {
            client.clearInterval(rotator);
            rotatePresence();
            rotator = client.setInterval(() => rotatePresence(), config.presence_duration);
        } catch (error) {
            logger.warn(JSON.stringify(error));
        }
    }, [], 'Force presence rotation', 'owner');

    // ping command that returns the ping attribute of the client
    cmd.createGlobalCommand(prefix + 'ping', () => {
        return `Current average ping: \`${client.ping} ms\``;
    }, [], 'Returns the current average ping', 'owner');

    // returns the time the bot is running
    cmd.createGlobalCommand(prefix + 'uptime', () => {
        return `Uptime: \`${client.uptime / 1000} s\``
    }, [], 'Returns the uptime of the bot', 'owner');

    // returns the numbe of guilds, the bot has joined
    cmd.createGlobalCommand(prefix + 'guilds', () => {
        return `Number of guilds: \`${client.guilds.size}\``
    }, [], 'Returns the number of guilds the bot has joined', 'owner');
}

function rotatePresence() {
    let pr = presences.shift();
    presences.push(pr);
    client.user.setPresence({game: {name: `${gamepresence} | ${pr}`, type: "PLAYING"}, status: 'online'});
    logger.debug(`Presence rotation to ${pr}`);
}

/**
 * Sends the answer recieved from the commands callback.
 * Handles the sending differently depending on the type of the callback return
 * @param msg
 * @param answer
 */
function answerMessage(msg, answer) {
    if (answer instanceof Promise || answer) {
        if (answer instanceof Discord.RichEmbed) {
            (this.mention)? msg.reply('', answer) : msg.channel.send('', answer);
        } else if (answer instanceof Promise) {
            answer
                .then((answer) => answerMessage(msg, answer))
                .catch((error) => answerMessage(msg, error));
        } else {
            (this.mention)? msg.reply(answer) : msg.channel.send(answer);
        }
    } else {
        logger.warn(`Empty answer won't be send.`);
    }
}

/**
 * Registeres callbacks for client events
 */
function registerCallbacks() {
    client.on('error', (err) => {
        logger.error(err.message);
    });

    client.on('ready', () => {
        logger.info(`logged in as ${client.user.tag}!`);
        client.user.setPresence({game: {name: gamepresence, type: "PLAYING"}, status: 'online'});
    });

    client.on('message', msg => {
        try {
            if (msg.author === client.user) {
                logger.verbose(`ME: ${msg.content}`);
                return;
            }
            logger.verbose(`<${msg.author.tag}>: ${msg.content}`);
            if (!msg.guild) {
                let reply = cmd.parseMessage(msg);
                answerMessage(msg, reply);
            } else {
                guilding.getHandler(msg.guild, prefix).handleMessage(msg);
            }
        } catch (err) {
            logger.error(err.stack);
        }
    });
}

// Executing the main function
if (typeof require !== 'undefined' && require.main === module) {
    logger.info("Starting up... ");  // log the current date so that the logfile is better to read.
    main();
}