forked from ronaldburns/guidebot-class
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathindex.js
243 lines (206 loc) · 9 KB
/
index.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
// This will check if the node version you are running is the required
// Node version, if it isn't it will throw the following error to inform
// you.
if (Number(process.version.slice(1).split(".")[0]) < 8) throw new Error("Node 8.0.0 or higher is required. Update Node on your system.");
// Load up the discord.js library
const { Client, Collection } = require("discord.js");
// We also load the rest of the things we need in this file:
const { promisify } = require("util");
const readdir = promisify(require("fs").readdir);
const Enmap = require("enmap");
const klaw = require("klaw");
const path = require("path");
class GuideBot extends Client {
constructor (options) {
super(options);
// Here we load the config.js file that contains our token and our prefix values.
this.config = require("./config.js");
// client.config.token contains the bot's token
// client.config.prefix contains the message prefix
// Aliases and commands are put in collections where they can be read from,
// catalogued, listed, etc.
this.commands = new Collection();
this.aliases = new Collection();
// Now we integrate the use of Evie's awesome Enhanced Map module, which
// essentially saves a collection to disk. This is great for per-server configs,
// and makes things extremely easy for this purpose.
this.settings = new Enmap({ name: "settings", cloneLevel: "deep", fetchAll: false, autoFetch: true });
//requiring the Logger class for easy console logging
this.logger = require("./modules/Logger");
// Basically just an async shortcut to using a setTimeout. Nothing fancy!
this.wait = require("util").promisify(setTimeout);
}
/*
PERMISSION LEVEL FUNCTION
This is a very basic permission system for commands which uses "levels"
"spaces" are intentionally left black so you can add them if you want.
NEVER GIVE ANYONE BUT OWNER THE LEVEL 10! By default this can run any
command including the VERY DANGEROUS `eval` command!
*/
permlevel (message) {
let permlvl = 0;
const permOrder = this.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;
}
/*
COMMAND LOAD AND UNLOAD
To simplify the loading and unloading of commands from multiple locations
including the index.js load loop, and the reload function, these 2 ensure
that unloading happens in a consistent manner across the board.
*/
loadCommand (commandPath, commandName) {
try {
const props = new (require(`${commandPath}${path.sep}${commandName}`))(this);
this.logger.log(`Loading Command: ${props.help.name}. 👌`, "log");
props.conf.location = commandPath;
if (props.init) {
props.init(this);
}
this.commands.set(props.help.name, props);
props.conf.aliases.forEach(alias => {
this.aliases.set(alias, props.help.name);
});
return false;
} catch (e) {
return `Unable to load command ${commandName}: ${e}`;
}
}
async unloadCommand (commandPath, commandName) {
let command;
if (this.commands.has(commandName)) {
command = this.commands.get(commandName);
} else if (this.aliases.has(commandName)) {
command = this.commands.get(this.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(this);
}
delete require.cache[require.resolve(`${commandPath}${path.sep}${commandName}.js`)];
return false;
}
/* SETTINGS FUNCTIONS
These functions are used by any and all location in the bot that wants to either
read the current *complete* guild settings (default + overrides, merged) or that
wants to change settings for a specific guild.
*/
// getSettings merges the client defaults with the guild settings. guild settings in
// enmap should only have *unique* overrides that are different from defaults.
getSettings (guild) {
const defaults = this.config.defaultSettings || {};
const guildData = this.settings.get(guild.id) || {};
const returnObject = {};
Object.keys(defaults).forEach((key) => {
returnObject[key] = guildData[key] ? guildData[key] : defaults[key];
});
return returnObject;
}
// writeSettings overrides, or adds, any configuration item that is different
// than the defaults. This ensures less storage wasted and to detect overrides.
writeSettings (id, newSettings) {
const defaults = this.settings.get("default");
let settings = this.settings.get(id);
if (typeof settings != "object") settings = {};
for (const key in newSettings) {
if (defaults[key] !== newSettings[key]) {
settings[key] = newSettings[key];
} else {
delete settings[key];
}
}
this.settings.set(id, settings);
}
/*
SINGLE-LINE AWAITMESSAGE
A simple way to grab a single reply, from the user that initiated
the command. Useful to get "precisions" on certain things...
USAGE
const response = await client.awaitReply(msg, "Favourite Color?");
msg.reply(`Oh, I really love ${response} too!`);
*/
async awaitReply (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;
}
}
}
// This is your client. Some people call it `bot`, some people call it `self`,
// some might call it `cootchie`. Either way, when you see `client.something`,
// or `bot.something`, this is what we're refering to. Your client.
const client = new GuideBot();
console.log(client.config.permLevels.map(p => `${p.level} : ${p.name}`));
// We're doing real fancy node 8 async/await stuff here, and to do that
// we need to wrap stuff in an anonymous function. It's annoying but it works.
const init = async () => {
// Here we load **commands** into memory, as a collection, so they're accessible
// here and everywhere else.
klaw("./commands").on("data", (item) => {
const cmdFile = path.parse(item.path);
if (!cmdFile.ext || cmdFile.ext !== ".js") return;
const response = client.loadCommand(cmdFile.dir, `${cmdFile.name}${cmdFile.ext}`);
if (response) client.logger.error(response);
});
// Then we load events, which will include our message and ready event.
const evtFiles = await readdir("./events/");
client.logger.log(`Loading a total of ${evtFiles.length} events.`, "log");
evtFiles.forEach(file => {
const eventName = file.split(".")[0];
client.logger.log(`Loading Event: ${eventName}`);
const event = new (require(`./events/${file}`))(client);
// This line is awesome by the way. Just sayin'.
client.on(eventName, (...args) => event.run(...args));
delete require.cache[require.resolve(`./events/${file}`)];
});
client.levelCache = {};
for (let i = 0; i < client.config.permLevels.length; i++) {
const thisLevel = client.config.permLevels[i];
client.levelCache[thisLevel.name] = thisLevel.level;
}
// Here we login the client.
client.login(client.config.token);
// End top-level async/await function.
};
init();
client.on("disconnect", () => client.logger.warn("Bot is disconnecting..."))
.on("reconnecting", () => client.logger.log("Bot reconnecting...", "log"))
.on("error", e => client.logger.error(e))
.on("warn", info => client.logger.warn(info));
/* MISCELANEOUS NON-CRITICAL FUNCTIONS */
// EXTENDING NATIVE TYPES IS BAD PRACTICE. Why? Because if JavaScript adds this
// later, this conflicts with native code. Also, if some other lib you use does
// this, a conflict also occurs. KNOWING THIS however, the following methods
// are, we feel, very useful in code.
// <String>.toPropercase() returns a proper-cased string such as:
// "Mary had a little lamb".toProperCase() returns "Mary Had A Little Lamb"
String.prototype.toProperCase = function () {
return this.replace(/([^\W_]+[^\s-]*) */g, function (txt) {return txt.charAt(0).toUpperCase() + txt.substr(1).toLowerCase();});
};
// <Array>.random() returns a single random element from an array
// [1, 2, 3, 4, 5].random() can return 1, 2, 3, 4 or 5.
Array.prototype.random = function () {
return this[Math.floor(Math.random() * this.length)];
};
// These 2 process methods will catch exceptions and give *more details* about the error and stack trace.
process.on("uncaughtException", (err) => {
const errorMsg = err.stack.replace(new RegExp(`${__dirname}/`, "g"), "./");
console.error("Uncaught Exception: ", errorMsg);
// Always best practice to let the code crash on uncaught exceptions.
// Because you should be catching them anyway.
process.exit(1);
});
process.on("unhandledRejection", err => {
console.error("Uncaught Promise Error: ", err);
});