asdqwex/discobot

'use strict'

// Libraries!
const DiscordClient = require('discord.io')
const glob = require('glob')
const Sifter = require('sifter')
const schedule = require('node-schedule')
import Random from 'random-js'
const random = new Random(Random.engines.mt19937().autoSeed())
const fs = require('fs')
import { LOG } from './logger.js'

let announceGuard = 0
let configBlock = {}
let help_text = []

const dependencies = {
  random
}

if (fs.existsSync('.config')) {
  configBlock = JSON.parse(fs.readFileSync('.config', 'utf8'))
} else {
  configBlock.DISCORD_EMAIL = process.env.DISCORD_EMAIL
  configBlock.DISCORD_PASSWORD = process.env.DISCORD_PASSWORD
  if (!process.env.DISCORD_EMAIL || !process.env.DISCORD_PASSWORD) {
    LOG.error('Please define DISCORD_EMAIL and DISCORD_PASSWORD environment variables')
    process.exit(1)
  }
}
// Initialization
const bot = new DiscordClient({
  email: configBlock.DISCORD_EMAIL,
  password: configBlock.DISCORD_PASSWORD,
  autorun: true
})

Object.keys(configBlock).forEach(function (key) {
  process.env[key] = configBlock[key]
})

if (!process.env.DISCORD_GUILD) LOG.warn('DISCORD_GUILD unset - joining first guild')
if (!process.env.DISCORD_VOICE_CHANNEL) LOG.warn('DISCORD_VOICE_CHANNEL unset - joining "General"')
if (!process.env.DISCORD_TEXT_CHANNEL) LOG.warn('DISCORD_TEXT_CHANNEL unset - joining "general"')

bot.BOT_NAME = process.env.BOT_NAME
// This is the default text channel that
bot.DISCORD_GUILD = process.env.DISCORD_GUILD || undefined
bot.DISCORD_TEXT_CHANNEL = process.env.DISCORD_TEXT_CHANNEL || 'general'
bot.DISCORD_VOICE_CHANNEL = process.env.DISCORD_VOICE_CHANNEL || 'General'

// We will load all the code from the "modules" directory, and put the object each file exports into this array:
const modules = {}
// Used to sort thru modules
let sifter
// The bots name plus the identifier
let hailing_frequency = (process.env.HAILING || '!') + bot.BOT_NAME
if (process.env.NO_HAILING) hailing_frequency = bot.BOT_NAME

const initialize = function () {
  glob('_build/modules/*.js', function (err, files) {
    // Bail out if the modules directory doesn't exist. Your install is fucked?
    if (err) throw new Error(err)
    for (let i = 0; i < files.length; i++) {
      try {
        const module = require('./' + files[i].replace('_build/', ''))
        if (!module.names) {
          LOG.warn(`${files[i]} ignored as it has no .names propery`)
          continue
        }
        modules[module.names[0]] = module
        if (!module.help_text) module.help_text = 'Not documented'
        help_text.push([module.names[0], module.help_text])
      } catch (e) {
        LOG.error(`Failed to load ${module}:`, e.message)
      }
    }
    sifter = new Sifter(modules)
    setupBot()
  })
}

// Connect and determine default channels
const setupBot = function () {
  bot.on('ready', function (data) {
    // First wins if none is defined
    if (!bot.DISCORD_GUILD) bot.DISCORD_GUILD = data.d.guilds[0].id
    // Allow usage of text name or id
    bot.my_guild = data.d.guilds.filter(function (guild) {
      if (guild.name === bot.DISCORD_GUILD || guild.id === bot.DISCORD_GUILD) return true
    })
    if (bot.my_guild.length > 0) bot.my_guild = bot.my_guild[0]
    else {
      LOG.warn(`I was unable to join ${bot.DISCORD_GUILD}, falling back to first`)
      bot.my_guild = data.d.guilds[0]
    }
    LOG.debug(`Connected as "${hailing_frequency}" to "${bot.my_guild.name}"`)
    for (const i in bot.my_guild.channels) {
      const chan = bot.my_guild.channels[i]
      if (chan.name === 'general' && chan.type === 'text') bot.my_general_channel = chan
      if (chan.name === bot.DISCORD_TEXT_CHANNEL && chan.type === 'text') bot.my_text_channel = chan
      else if (chan.name === bot.DISCORD_VOICE_CHANNEL && chan.type === 'voice') bot.my_voice_channel = chan
    }
    onReady()
  })
  bot.on('disconnected', function () {
    if (process.env.NODE_ENV !== 'dev') {
      LOG.info('Disconnected from Discord! Reconnecting in 5 seconds.')
      const reconnecter = setInterval(function () {
        bot.connect()
        if (bot.connected) {
          clearInterval(reconnecter)
        }
      }, 5000)
    }
  })
}

bot._presences = {}
const onReady = function () {
// Cron Jobby bit
  schedule.scheduleJob('announceGuard', '*/5 * * * *', function () {
    announceGuard = 0
  })

  schedule.scheduleJob('xkcd', '0 0 0 * *', function () {
    modules['xkcd'].onMessage(bot, false, false, process.env.TEST_CHANNEL, false, false)
  })
// Bot tells us who it is in the test channel
  bot.sendMessage({
    to: process.env.TEST_CHANNEL,
    message: 'Hello My Name is ' + hailing_frequency
  })
// Set Bot UserName
  bot.editUserInfo({
    avatar: require('fs').readFileSync('discobot.png', 'base64'),
    password: configBlock.DISCORD_PASSWORD,
    username: process.env.BOT_NAME
  })
// Other actions
  bot.on('message', onMessage)
  bot.on('presence', function (name, id, status, game) {
    // Say nothing if the user is playing the same game but just went idle
    if (announceGuard === 0 && game && ((bot._presences[id] && bot._presences[id].game !== game) || !bot._presences[id])) {
      announceGuard = 1
      bot.sendMessage({
        to: bot.my_general_channel.id,
        message: `${name} has begun playing ${game}!`
      })
    }
    bot._presences[id] = { name, status, game }
  })
  bot.joinVoiceChannel(bot.my_voice_channel.id)
}

const onMessage = function (user, userID, channelID, message, rawEvent) {
  if (message.indexOf('!bots?') > -1 && message.indexOf('Force all bots to identify themselves') === -1) {
    return bot.sendMessage({
      to: channelID,
      message: 'Hello I am a robot and my name is ' + bot.BOT_NAME
    })
  }

  if (message.indexOf('(╯°□°）╯︵ ┻━┻') > -1) {
    const count = (message.match(/\(\╯\°\□\°\）\╯\︵ \┻\━\┻/g) || []).length
    const tableArray = []
    for (let i = 0; i < count; i++) {
      tableArray.push('┬─┬ ノ( ゜-゜ノ)')
    }
    return bot.sendMessage({
      to: channelID,
      message: tableArray.join(' ')
    })
  }

  // Anything after this point needs to be addressed to us (!bot action)
  if (message.substr(0, hailing_frequency.length) !== hailing_frequency) return undefined
  // Help text for "!bot"
  if (message === hailing_frequency || message === hailing_frequency + '--help' || message === hailing_frequency + 'help' || message === hailing_frequency + 'commands') {
    return bot.sendMessage({
      to: channelID,
      message: modules['help']['data'].join('\n')
    })
  }
  const trimmed = message.replace(hailing_frequency + ' ', '').trimLeft()
  const results = sifter.search(trimmed.split(' ')[0], {
    fields: [ 'names' ],
    sort: [{ field: 'score', direction: 'asc' }],
    limit: 1
  })
  if (results.total > 0) {
    const module = modules[results.items[0].id]
    LOG.debug(`${user} called ${module.names} with "${trimmed}" by a score of ${results.items[0].score}`)
    if (module.onMessage) module.onMessage(bot, user, userID, channelID, trimmed, rawEvent, dependencies)
  }
}

initialize()

// Log a message before exiting, for debugging purposes
process.stdin.resume()
process.on('exit', function () {
  LOG.info('Exiting!')
  process.exit()
})