TravBot-v3/src/commands/utility/lsemotes.ts

109 lines
4.7 KiB
TypeScript
Raw Normal View History

import {GuildEmoji, MessageEmbed, TextChannel, DMChannel, NewsChannel, User} from "discord.js";
import {Command, NamedCommand, paginate} from "../../core";
import {split} from "../../lib";
2021-03-28 17:49:45 +00:00
import vm from "vm";
const REGEX_TIMEOUT_MS = 1000;
export default new NamedCommand({
2020-12-15 01:44:28 +00:00
description: "Lists all emotes the bot has in it's registry,",
usage: "<regex pattern> (-flags)",
async run({message, channel, guild, author, member, client, args}) {
displayEmoteList(client.emojis.cache.array(), channel, author);
},
any: new Command({
description:
"Filters emotes by via a regular expression. Flags can be added by adding a dash at the end. For example, to do a case-insensitive search, do %prefix%lsemotes somepattern -i",
async run({message, channel, guild, author, member, client, args}) {
// If a guild ID is provided, filter all emotes by that guild (but only if there aren't any arguments afterward)
if (args.length === 1 && /^\d{17,19}$/.test(args[0])) {
const guildID: string = args[0];
displayEmoteList(
client.emojis.cache.filter((emote) => emote.guild.id === guildID).array(),
channel,
author
);
} else {
2021-03-28 17:49:45 +00:00
// Otherwise, search via a regex pattern
let flags: string | undefined = undefined;
if (/^-[dgimsuy]{1,7}$/.test(args[args.length - 1])) {
flags = args.pop().substring(1);
2021-03-28 17:49:45 +00:00
}
let emoteCollection = client.emojis.cache.array();
2021-03-28 17:49:45 +00:00
// Creates a sandbox to stop a regular expression if it takes too much time to search.
// To avoid passing in a giant data structure, I'll just pass in the structure {[id: string]: [name: string]}.
//let emotes: {[id: string]: string} = {};
let emotes = new Map<string, string>();
for (const emote of emoteCollection) {
emotes.set(emote.id, emote.name);
}
// The result will be sandbox.emotes because it'll be modified in-place.
const sandbox = {
regex: new RegExp(args.join(" "), flags),
2021-03-28 17:49:45 +00:00
emotes
};
const context = vm.createContext(sandbox);
2021-03-28 17:49:45 +00:00
if (vm.isContext(sandbox)) {
// Restrict an entire query to the timeout specified.
try {
const script = new vm.Script(
"for(const [id, name] of emotes.entries()) if(!regex.test(name)) emotes.delete(id);"
);
script.runInContext(context, {timeout: REGEX_TIMEOUT_MS});
emotes = sandbox.emotes;
2021-03-28 18:58:10 +00:00
emoteCollection = emoteCollection.filter((emote) => emotes.has(emote.id)); // Only allow emotes that haven't been deleted.
displayEmoteList(emoteCollection, channel, author);
2021-03-28 17:49:45 +00:00
} catch (error) {
if (error.code === "ERR_SCRIPT_EXECUTION_TIMEOUT") {
channel.send(
2021-03-28 17:49:45 +00:00
`The regular expression you entered exceeded the time limit of ${REGEX_TIMEOUT_MS} milliseconds.`
);
} else {
throw new Error(error);
}
}
} else {
channel.send("Failed to initialize sandbox.");
2021-03-28 17:49:45 +00:00
}
}
}
})
});
async function displayEmoteList(emotes: GuildEmoji[], channel: TextChannel | DMChannel | NewsChannel, author: User) {
emotes.sort((a, b) => {
const first = a.name.toLowerCase();
const second = b.name.toLowerCase();
if (first > second) return 1;
else if (first < second) return -1;
else return 0;
});
const sections = split(emotes, 20);
const pages = sections.length;
2021-03-30 12:16:31 +00:00
const embed = new MessageEmbed().setColor("AQUA");
// Gather the first page (if it even exists, which it might not if there no valid emotes appear)
if (pages > 0) {
2021-03-30 12:16:31 +00:00
paginate(channel, author.id, pages, (page, hasMultiplePages) => {
embed.setTitle(hasMultiplePages ? `**Emotes** (Page ${page + 1} of ${pages})` : "**Emotes**");
2020-12-15 01:44:28 +00:00
2021-03-30 12:16:31 +00:00
let desc = "";
for (const emote of sections[page]) {
desc += `${emote} ${emote.name} (**${emote.guild.name}**)\n`;
}
embed.setDescription(desc);
2021-03-30 12:16:31 +00:00
return embed;
});
} else {
channel.send("No valid emotes found by that query.");
2020-12-15 01:44:28 +00:00
}
}