woomy/helpers/_functions.js

const mongoose = require('mongoose')
const Guild = require('../models/guild')
const User = require('../models/user')

module.exports = client => {
  // Permission level function
  client.permlevel = (message, settings) => {
    let permlvl = 0

    const permOrder = client.config.permLevels.slice(0).sort((p, c) => p.level < c.level ? 1 : -1)

    while (permOrder.length) {
      const currentLevel = permOrder.shift()
      if (message.guild && currentLevel.guildOnly) continue
      if (currentLevel.check(message, settings)) {
        permlvl = currentLevel.level
        break
      }
    }
    return permlvl
  }

  // Get guild settings
  client.findOrCreateGuild = async (guild) => {
    const data = await Guild.findOne({ guildID: guild.id })
    if (data) {
      return data
    } else {
      const merged = Object.assign({ _id: mongoose.Types.ObjectId() }, { guildID: guild.id })
      const newGuild = await new Guild(merged)
      return newGuild.save()
    }
  }

  // Update guild settings
  client.updateGuild = async (guild, settings) => {
    let data = await client.getGuild(guild)

    if (typeof data !== 'object') data = {}
    for (const key in settings) {
      if (data[key] !== settings[key]) {
        data[key] = settings[key]
      } else return
    }

    return data.updateOne(settings)
  }

  // Delete guild settings
  client.deleteGuild = async (guild) => {
    const data = await client.getGuild(guild)
    if (data) {
      data.deleteOne({ guildID: guild.id })
    }
  }

  // Get user settings
  client.findOrCreateUser = async (user) => {
    const data = await User.findOne({ userID: user.id })
    if (data) {
      return data
    } else {
      const merged = Object.assign({ _id: mongoose.Types.ObjectId() }, { userID: user.id })
      const newUser = await new User(merged)
      return newUser.save()
    }
  }

  // Update user settings
  client.updateUser = async (user, settings) => {
    let data = await client.getUser(user)

    if (typeof data !== 'object') data = {}
    for (const key in settings) {
      if (data[key] !== settings[key]) {
        data[key] = settings[key]
      } else return
    }

    return data.updateOne(settings)
  }

  // Delete user settings
  client.deleteUser = async (user) => {
    const data = await client.getUser(user)
    if (data) {
      data.deleteOne({ userID: user.id })
    }
  }

  // Loads commands
  client.loadCommand = (commandName) => {
    try {
      const props = require(`../commands/${commandName}`)
      if (props.init) {
        props.init(client)
      }
      client.commands.set(props.help.name, props)
      // So commands can each have their own cooldown time
      client.cooldown.set(props.help.name, new Map())
      props.conf.aliases.forEach(alias => {
        client.aliases.set(alias, props.help.name)
      })
      return false
    } catch (e) {
      return `Failed to load ${commandName}: ${e}`
    }
  }

  // Command unloader
  client.unloadCommand = async (commandName) => {
    let command
    if (client.commands.has(commandName)) {
      command = client.commands.get(commandName)
    } else if (client.aliases.has(commandName)) {
      command = client.commands.get(client.aliases.get(commandName))
    }
    if (!command) return `The command \`${commandName}\` doesn't seem to exist, nor is it an alias. Try again!`

    if (command.shutdown) {
      await command.shutdown(client)
    }
    const mod = require.cache[require.resolve(`../commands/${command.help.name}`)]
    delete require.cache[require.resolve(`../commands/${command.help.name}.js`)]
    for (let i = 0; i < mod.parent.children.length; i++) {
      if (mod.parent.children[i] === mod) {
        mod.parent.children.splice(i, 1)
        break
      }
    }
    return false
  }

  // Clean up input to remove @everyone, token, etc
  client.clean = async (client, text) => {
    if (text && text.constructor.name === 'Promise') {
      text = await text
    }
    if (typeof text !== 'string') {
      text = require('util').inspect(text, { depth: 1 })
    }
    text = text
      .replace(/`/g, '`' + String.fromCharCode(8203))
      .replace(/@/g, '@' + String.fromCharCode(8203))
      .replace(client.token, 'mfa.VkO_2G4Qv3T--NO--lWetW_tjND--TOKEN--QFTm6YGtzq9PH--4U--tG0')

    return text
  }

  // Single line await messages
  client.awaitReply = async (msg, question, limit = 60000) => {
    const filter = m => m.author.id === msg.author.id
    await msg.channel.send(question)
    try {
      const collected = await msg.channel.awaitMessages(filter, {
        max: 1,
        time: limit,
        errors: ['time']
      })
      return collected.first().content
    } catch (e) {
      return false
    }
  }

  // Embed colour function
  client.embedColour = function (guild) {
    if (!guild || guild.member(client.user).displayHexColor === '#000000') {
      return ['#ff9d68', '#ff97cb', '#d789ff', '#74FFFF'].random()
    } else {
      return guild.member(client.user).displayHexColor
    }
  }

  // Returns a random int between min and max
  client.intBetween = function (min, max) {
    return Math.floor((Math.random() * (max - min)) + min)
  }

  // Get random array object
  // eslint-disable-next-line no-extend-native
  Object.defineProperty(Array.prototype, 'random', {
    value: function () {
      return this[Math.floor(Math.random() * this.length)]
    }
  })

  // `await client.wait(1000);` to 'pause' for 1 second.
  client.wait = require('util').promisify(setTimeout)

  // Find guild members
  client.findMembers = function (guild, search) {
    if (!search || typeof search !== 'string') return

    const members = []
    let member

    // Try ID search
    if (!isNaN(search) === true) {
      members.push(guild.members.cache.get(search))
      if (members[0]) {
        return members[0]
      }
    }

    // Try username search
    try {
      member = guild.members.cache.find(m => m.displayName.toLowerCase() === search)
      if (!member) {
        guild.members.cache.find(m => m.user.tag.toLowerCase() === search)
      }
    } catch (err) {}
    if (member) {
      members.push(member)
    }
    guild.members.cache.forEach(m => {
      if (m.displayName.toLowerCase().startsWith(search) || m.user.tag.toLowerCase().startsWith(search)) {
        members.push(m)
      }
    })

    return members
  }

  // USER OBJECT FROM MENTION
  client.getUserFromMention = mention => {
    if (!mention) return

    if (mention.startsWith('<@') && mention.endsWith('>')) {
      mention = mention.slice(2, -1)

      if (mention.startsWith('!')) {
        mention = mention.slice(1)
      }

      return client.users.cache.get(mention)
    }
  }

  client.findRole = function (guild, search) {
    var role = guild.roles.cache.find(r => r.name.toLowerCase() === search.toLowerCase())
    if (!role) {
      role = guild.roles.cache.get(search.toLowerCase())
    }

    if (!role) {
      return
    }

    return role
  }

  // Both of these functions catch errors and log them (maybe we could use sentry?)
  process.on('uncaughtException', (err) => {
    const errorMsg = err.stack.replace(new RegExp(`${__dirname}/`, 'g'), './')
    client.logger.fatal(`Uncaught Exception: ${errorMsg}`)
    process.exit(1)
  })

  process.on('unhandledRejection', err => {
    client.logger.error(`Unhandled rejection: ${err.stack}`)
  })
}