94 lines
3.2 KiB
JavaScript
94 lines
3.2 KiB
JavaScript
const CoreUtil = require("./Util.js");
|
|
const Camo = require("camo");
|
|
const CronJob = require("cron").CronJob;
|
|
const Discord = require("discord.js");
|
|
const HandleGuildMessage = require("./HandleGuildMessage");
|
|
const InternalConfig = require("./internal-config.json");
|
|
const RequireAll = require("require-all");
|
|
const Util = require("./Util.js");
|
|
|
|
let neDB;
|
|
|
|
module.exports = class Client extends Discord.Client {
|
|
/**
|
|
* Construct a new Discord.Client with some added functionality
|
|
* @param {string} token bot token
|
|
* @param {string} dataFile location for json data file
|
|
* @param {string} commandsDir location of dir containing commands .js files
|
|
* @param {*} guildDataModel GuildData model to be used for app; must extend BaseGuildData
|
|
*/
|
|
constructor(token, dataFile, commandsDir, guildDataModel) {
|
|
super();
|
|
|
|
this._token = token;
|
|
this.dataFile = dataFile;
|
|
this.commandsDir = commandsDir;
|
|
this.guildDataModel = guildDataModel;
|
|
|
|
this.commands = RequireAll(this.commandsDir);
|
|
|
|
this.on("ready", this._onReady);
|
|
this.on("message", this._onMessage);
|
|
this.on("debug", this._onDebug);
|
|
this.on("guildCreate", this._onGuildCreate);
|
|
this.on("guildDelete", this._onGuildDelete);
|
|
process.on("uncaughtException", err => this._onUnhandledException(this, err));
|
|
}
|
|
|
|
bootstrap() {
|
|
Camo.connect("nedb://guilds-data").then(db => {
|
|
neDB = db;
|
|
|
|
this.emit("beforeLogin");
|
|
this.login(this._token);
|
|
});
|
|
}
|
|
|
|
_onReady() {
|
|
this.user.setGame(InternalConfig.website.replace(/^https?:\/\//, ""));
|
|
CoreUtil.dateLog(`Registered bot ${this.user.username}`);
|
|
|
|
new CronJob(InternalConfig.dbCompactionSchedule, () => compactCollections(), null, true);
|
|
}
|
|
|
|
_onMessage(message) {
|
|
if (message.channel.type === "text" && message.member)
|
|
HandleGuildMessage(this, message, this.commands);
|
|
}
|
|
|
|
_onDebug(info) {
|
|
info = info.replace(/Authenticated using token [^ ]+/, "Authenticated using token [redacted]");
|
|
if (!InternalConfig.debugIgnores.some(x => info.startsWith(x)))
|
|
CoreUtil.dateDebug(info);
|
|
}
|
|
|
|
_onGuildCreate(guild) {
|
|
CoreUtil.dateLog(`Added to guild ${guild.name}`);
|
|
}
|
|
|
|
_onGuildDelete(guild) {
|
|
this.guildDataModel.findOneAndDelete({ guildID: guild.id });
|
|
|
|
CoreUtil.dateLog(`Removed from guild ${guild.name}, removing data for this guild`);
|
|
}
|
|
|
|
_onUnhandledException(client, err) {
|
|
CoreUtil.dateError("Unhandled exception!\n", err);
|
|
CoreUtil.dateLog("Destroying existing client...");
|
|
client.destroy().then(() => {
|
|
CoreUtil.dateLog("Client destroyed, recreating...");
|
|
setTimeout(() => client.login(client._token), InternalConfig.reconnectTimeout);
|
|
});
|
|
}
|
|
};
|
|
|
|
function compactCollections() {
|
|
/*I realise it is a bit of a cheat to just access _collections in this manner, but in the absence of
|
|
camo actually having any kind of solution for this it's the easiest method I could come up with.
|
|
Maybe at some point in future I should fork camo and add this feature. The compaction function is NeDB only
|
|
and camo is designed to work with both NeDB and MongoDB, which is presumably why it doesn't alraedy exist */
|
|
for (let collectionName of Object.keys(neDB._collections))
|
|
neDB._collections[collectionName].persistence.compactDatafile();
|
|
|
|
Util.dateLog("Executed compaction on loaded NeDB collections");
|
|
} |