lbry-tipbot/bot/bot.js

'use strict';

// Load up libraries
const Discord = require('discord.js');
// Load config!
let config = require('config');
let botConfig = config.get('bot');
let commands = {};

const bot = new Discord.Client();

bot.on('ready', function () {
    console.log('Logged in! Serving in ' + bot.guilds.array().length + ' servers');
    require('./plugins.js').init();
    console.log('type ' + botConfig.prefix + 'help in Discord for a commands list.');
    bot.user.setActivity(botConfig.prefix + 'tip');
});

bot.on('disconnected', function () {
    console.log('Disconnected!');
    process.exit(1); //exit node.js with an error
});

function checkMessageForCommand(msg) {
    //check if message is a command
    if (msg.author.id !== bot.user.id && msg.content.startsWith(botConfig.prefix)) {
        console.log('treating ' + msg.content + ' from ' + msg.author + ' as command');
        let cmdTxt = msg.content.split(' ')[0].substring(botConfig.prefix.length);
        let suffix = msg.content.substring(cmdTxt.length + botConfig.prefix.length + 1); //add one for the ! and one for the space
        if (msg.isMentioned(bot.user)) {
            try {
                cmdTxt = msg.content.split(' ')[1];
                suffix = msg.content.substring(bot.user.mention().length + cmdTxt.length + botConfig.prefix.length + 1);
            } catch (e) {
                //no command
                return msg.channel.send('Yes?');
            }
        }
        let cmd = commands[cmdTxt];
        if (cmd) {
            // Add permission check here later on ;)
            try {
                cmd.process(bot, msg, suffix);
            } catch (e) {
                let msgTxt = `command ${cmdTxt} failed :(`;
                if (botConfig.debug) {
                    msgTxt += '\n' + e.stack;
                }
                msg.channel.send(msgTxt);
            }
        }
    } else {
        //message isn't a command or is from us
        //drop our own messages to prevent feedback loops
        if (msg.author === bot.user) {
            return;
        }

        if (msg.author !== bot.user && msg.isMentioned(bot.user)) {
            msg.channel.send('yes?'); //using a mention here can lead to looping
        }
    }
}

bot.on('message', msg => checkMessageForCommand(msg));

exports.addCommand = function (commandName, commandObject) {
    try {
        commands[commandName] = commandObject;
    } catch (err) {
        console.log(err);
    }
};
exports.addCustomFunc = function (customFunc) {
    try {
        customFunc(bot);
    } catch (err) {
        console.log(err);
    }
};
exports.commandCount = function () {
    return Object.keys(commands).length;
};

bot.login(botConfig.token);