notabot.js

var Discord = require('discord.js');
var logger = require('winston');
var auth = require('./notabotauth.json');
var fs = require('fs');
var rita = require('rita');

var chance = 0.2; // pourcentage de chance

// Configure logger settings
logger.remove(logger.transports.Console);
logger.add(logger.transports.Console, {
    colorize: true
});
logger.level = 'debug';

// INIT MARKOV CHAIN
var markov = new rita.RiMarkov(3,true,true);
markov.loadFrom('markov_lexicon.txt');
markov.minSentenceLength = 3;

// Initialize Discord Bot
var bot = new Discord.Client();
bot.on('ready', () => {
    logger.info('Connected');
    logger.info('Logged in as: ');
    logger.info(bot.user.username + ' - (' + bot.user.id + ')');
});

bot.on('error',(error) => { logger.error(error.message); });

bot.on('message', (message) => {
    if(message.author.username != 'notabot'){
        // clean message
        var clean = message.cleanContent;
        clean = clean.replace(new RegExp(/@/,'g'), '');
        clean = clean.replace(new RegExp(/[`]{1,3}.+?[`]{1,3}/,'g'), '');
        // remove urls
        clean = clean.replace(/(?:https?|ftp):\/\/[\n\S]+/g, '');
        clean = clean.trim();
        var last = clean.slice(-1);
        if(!(last == '?' || last == '!' || last == '.')){
            clean = clean + '. '; 
        }else{
            clean = clean + ' '; 
        }
        // capitalize
        clean = clean.charAt(0).toUpperCase() + clean.slice(1);

        if(!message.isMentioned(bot.user) && message.channel.id != 389449583421423627){ // not #botfun
            // learn it only if it is not a mention to you
            markov.loadText(clean);

            // store it for next time
            fs.appendFileSync('markov_lexicon.txt', clean);
        }

        // say something ?
        if(Math.random()*100 < chance || message.isMentioned(bot.user)) {

            // get key words in the questions
            // by removing not significant words
            var cleaner = clean.replace(new RegExp(/le|lui|la|elle|il|nous|nos|vous|vos|tu|ta|tes|les|je|j'|l'|ton|mon|ma|mes|des|du|on|ne|se|sa|son|ses|ces|ce|cette|cettes/,'g'), '');
            var keywords = cleaner.replace(new RegExp(/\s+/,'g'),' ').split(" ");

            var nb = Math.floor(Math.random() * 2.7) + 1;
            for (let index = 0; index < nb; index++) {
                // generate 10 sentences
                let sentences = markov.generateSentences(10);
                let maxRelevance = 0;
                let sentenceToSay = sentences[0];
                // for each sentence compute relevance
                sentences.forEach(s => {
                    let relevance = 0;
                    // relevance is the number of keyword present in the sentence
                    keywords.forEach(word => {
                        if(s.includes(word)){
                            relevance = relevance + 1;
                        }
                    });
                    // store the sentence with best relevance
                    if(relevance > maxRelevance){
                        maxRelevance = relevance;
                        sentenceToSay = s;
                    }
                });
                // send the most relevant sentence (or at least the first sentence by default)
                message.channel.send(sentenceToSay);
            }// do it nb times
            
        }
    }
});

bot.login(auth.token);