const Discord = require("discord.js"); const got = require("got"); module.exports = (client) => { client.permlevel = message => { 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)) { permlvl = currentLevel.level; break; } } return permlvl; }; const defaultSettings = { "prefix": "b&", "modLogChannel": "mod-log", "modRole": "modinator", "adminRole": "adminator", "systemNotice": "true", "welcomeChannel": "welcome", "welcomeMessage": "Say hello to {{user}}, everyone! We all need a warm welcome sometimes :D", "welcomeEnabled": "false" }; client.getSettings = (guild) => { client.settings.ensure("default", defaultSettings); if(!guild) return client.settings.get("default"); const guildConf = client.settings.get(guild.id) || {}; return ({...client.settings.get("default"), ...guildConf}); }; 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; } }; 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; }; client.loadCommand = (commandName) => { try { client.logger.log(`Loading Command: ${commandName}`); const props = require(`../${commandName}`); if (props.init) { props.init(client); } client.commands.set(props.help.name, props); props.conf.aliases.forEach(alias => { client.aliases.set(alias, props.help.name); }); return false; } catch (e) { return `Unable to load command ${commandName}: ${e}`; } }; 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.category}/${command.help.name}`)]; delete require.cache[require.resolve(`../commands/${command.help.category}/${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; }; Object.defineProperty(String.prototype, "toProperCase", { value: function() { return this.replace(/([^\W_]+[^\s-]*) */g, (txt) => txt.charAt(0).toUpperCase() + txt.substr(1).toLowerCase()); } }); Object.defineProperty(Array.prototype, "random", { value: function() { return this[Math.floor(Math.random() * this.length)]; } }); client.wait = require("util").promisify(setTimeout); client.randomSelection = choices => choices[Math.floor(Math.random() * choices.length)]; process.on("uncaughtException", (err) => { const errorMsg = err.stack.replace(new RegExp(`${__dirname}/`, "g"), "./"); client.logger.error(`Uncaught Exception: ${errorMsg}`); console.error(err); process.exit(1); }); process.on("unhandledRejection", err => { client.logger.error(`Unhandled rejection: ${err}`); console.error(err); }); client.parseArgs = (args, options) => { if (!options) return args; if (typeof options === 'string') options = [options]; let optionValues = {}; let i; for (i = 0; i < args.length; i++) { let arg = args[i]; if (!arg.startsWith('-')) { break; } let label = arg.substr(1); if (options.indexOf(label + ':') > -1) { let leftover = args.slice(i + 1).join(' '); let matches = leftover.match(/^"(.+?)"/); if (matches) { optionValues[label] = matches[1]; i += matches[0].split(' ').length; } else { i++; optionValues[label] = args[i]; } } else if (options.indexOf(label) > -1) { optionValues[label] = true; } else { break; } } return { options: optionValues, leftover: args.slice(i) }; }; client.embed = (title, description = '', fields = [], options = {}) => { let url = options.url || ''; let color = options.color || randomColor(); if (options.inline) { if (fields.length % 3 === 2) { fields.push({ name: '\u200b', value: '\u200b' }); } fields.forEach(obj => { obj.inline = true; }); } return new Discord.RichEmbed({ fields, video: options.video || url }) .setTitle(title) .setColor(color) .setDescription(description) .setURL(url) .setImage(options.image) .setTimestamp(options.timestamp ? timestampToDate(options.timestamp) : null) .setFooter(options.footer === true ? randomFooter() : (options.footer ? options.footer : ''), options.footer ? client.user.avatarURL : undefined) .setAuthor(options.author === undefined ? '' : options.author) .setThumbnail(options.thumbnail); }; const randomColor = () => [ Math.floor(Math.random() * 256), Math.floor(Math.random() * 256), Math.floor(Math.random() * 256) ]; const randomFooter = () => { return randomSelection([ 'just add water!', 'Powered by squirrels!', 'codeisluvcodeislife', 'Where did you get that?', 'WHAT DID YOU BREAK!?', 'D-D-D-DROP THE BASS', 'Eat, Sleep, JavaScript', '#BlameRhearmas', 'SharpBot was the best, right?', 'I like turtles', 'Hi mom!' ]); }; const timestampToDate = timestamp => { if (timestamp === true) { return new Date(); } if (typeof timestamp === 'number') { return new Date(timestamp); } return timestamp; }; const parseArgs = (args, options) => { if (!options) return args; if (typeof options === 'string') options = [options]; let optionValues = {}; let i; for (i = 0; i < args.length; i++) { let arg = args[i]; if (!arg.startsWith('-')) { break; } let label = arg.substr(1); if (options.indexOf(label + ':') > -1) { let leftover = args.slice(i + 1).join(' '); let matches = leftover.match(/^"(.+?)"/); if (matches) { optionValues[label] = matches[1]; i += matches[0].split(' ').length; } else { i++; optionValues[label] = args[i]; } } else if (options.indexOf(label) > -1) { optionValues[label] = true; } else { break; } } return { options: optionValues, leftover: args.slice(i) }; }; const multiSend = (channel, messages, delay) => { delay = delay || 100; messages.forEach((m, i) => { setTimeout(() => { channel.send(m); }, delay * i); }); }; client.textUpload = (text, options) => { options = options || {}; let method = (options.method || uploadMethods.ix).toLowerCase(); if (method === uploadMethods.ix) { return ixUpload(text); } else if (method === uploadMethods.hastebin) { return hastebinUpload(text); } }; const uploadMethods = { hastebin: 'hastebin', ix: 'ix.io' }; const hastebinUpload = text => { return got('https://hastebin.com/documents', { body: { 'contents': text }, form: true }) .then(res => { if (res && res.body && res.body.key) { const key = res.body.key; return { key: key, success: true, url: `https://hastebin.com/${key}`, rawUrl: `https://hastebin.com/raw/${key}` }; } else { return { success: false }; } }); }; const ixUpload = text => { return got('http://ix.io', { body: { 'f:1': text }, form: true }) .then(res => { if (res && res.body) { return { success: true, url: res.body, rawUrl: res.body }; } else { return { success: false }; } }); }; client.quoteRegex = input => `${input}`.replace(/[.?*+^$[\]\\(){}|-]/g, '\\$&'); };