galactic-bot/structure/extensions/Message.js

135 lines
4.3 KiB
JavaScript

const { Structures } = require('discord.js');
const escapeRegex = require('escape-string-regexp');
const emojis = require('../../util/emojis.json');
const { Util } = require('../../util/');
const Message = Structures.extend('Message', (Message) => {
class ExtendedMessage extends Message {
constructor(...args) {
super(...args);
this.command = null; //Will set to command if the message induces a command.
this.args = null;
this.parameters = null;
this._pending = null;
}
format(index, parameters = { }, code = false) {
let language = this.author._settings.locale || 'en_us';
if(this.guild && this.guild._settings.locale) language = this.guild._settings.locale;
parameters.prefix = this.guild ? this.guild.prefix : this.client._config.bot.prefix;
let template = this.client.localeLoader.template(language, index); //.languages[language][index];
if(!template) {
return `**Missing language index \`${language} [${index}]\` in languages. Contact a bot developer about this.**`;
}
for (const [param, val] of Object.entries(parameters)) {
template = template.replace(new RegExp(`{${escapeRegex(param.toLowerCase())}}`, 'gi'), val);
}
if(code) {
try {
template = eval(template);
} catch(error) {
this.command.client.logger.error(`Error in locale ${language}:${index} while executing code.\n${error.stack || error}`);
}
}
return template;
}
async resolve() {
if (this.command.showUsage && !this.parameters.length) {
return this._showUsage();
}
try {
const resolved = this.command.execute(this, {
params: this.parameters,
args: this.args
});
if(resolved instanceof Promise) await resolved;
return { error: false };
} catch(error) {
return { error: true, message: error.stack || error };
}
}
async embed(embed, opts = {}) {
if (!embed.color) embed.color = 619452;
let send = { content: opts.reply ? `<@${this.author.id}>` : '', embed };
this._pending = await this.channel.send(send);
return this._pending;
}
async respond(str, opts = {}) {
if(typeof str === 'string') {
if(opts.emoji) {
const emoji = emojis[opts.emoji];
if(!emoji) this.command.client.logger.warn(`Invalid emoji provided to command ${this.command.resolveable}: "${opts.emoji}".`);
str = `${emoji} ${str}`;
}
if(opts.reply) str = `<@!${this.author.id}> ${str}`;
}
//console.log(str)
this._pending = await this.channel.send(str);
return this._pending;
}
async edit(str, opts) {
if(!this.editable) return null;
if(typeof str === 'string') {
if(opts.emoji) str = `${emojis[opts.emoji]} ${str}`;
if(opts.reply) str = `<@!${this.author.id}> ${str}`;
}
return super.edit(str);
}
async prompt(str, opts) {
if(typeof str === 'string') {
if(opts.emoji) str = `${emojis[opts.emoji]} ${str}`;
if(opts.reply) str = `<@!${this.author.id}> ${str}`;
}
const message = await this.channel.send(str);
return await this.channel.awaitMessages(m => m.author.id === this.author.id, { max: 1, time: 30000, errors: ['time'] })
.then((collected) => {
return collected.first();
})
.catch((error) => {
return null;
});
}
async _showUsage() {
//TODO: format this
return await this.embed({
title: `**${this.command.name.toUpperCase()} USAGE**`,
description: this.format(`C_${this.command.name.toUpperCase()}_USAGE`)
});
}
}
return ExtendedMessage;
});
module.exports = Message;