StyAndCo/zeppelin
3
0
Fork 0
mirror of https://github.com/ZeppelinBot/Zeppelin.git synced 2025-03-15 05:41:51 +00:00
Code Releases Activity

Move action handlers to util file

Browse source
This commit is contained in:
Dark 2021-06-08 04:47:36 +02:00
parent 5c7c3c8cba
commit b0329dca05
No known key found for this signature in database
GPG key ID: 384C4B4F5B1E25A8
4 changed files with 89 additions and 60 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

64
backend/src/plugins/ReactionRoles/events/ButtonInteractionEvt.ts
View file

@ -1,4 +1,4 @@
import { MessageActionRow, MessageButton, MessageComponentInteraction } from "discord.js";
import { MessageComponentInteraction } from "discord.js";
import moment from "moment";
import { LogType } from "src/data/LogType";
import { logger } from "src/logger";
@ -6,13 +6,10 @@ import { LogsPlugin } from "src/plugins/Logs/LogsPlugin";
import { MINUTES } from "src/utils";
import { idToTimestamp } from "src/utils/idToTimestamp";
import { reactionRolesEvt } from "../types";
import {
generateStatelessCustomId,
resolveStatefulCustomId,
BUTTON_CONTEXT_SEPARATOR,
} from "../util/buttonCustomIdFunctions";
import { resolveStatefulCustomId, BUTTON_CONTEXT_SEPARATOR } from "../util/buttonCustomIdFunctions";
import { ButtonMenuActions } from "../util/buttonMenuActions";
import humanizeDuration from "humanize-duration";
import { handleModifyRole, handleOpenMenu } from "../util/buttonActionHandlers";
const BUTTON_INVALIDATION_TIME = 15 * MINUTES;
@ -70,62 +67,13 @@ export const ButtonInteractionEvt = reactionRolesEvt({
return;
}
if (context.action === ButtonMenuActions.GRANT_ROLE) {
const role = await meta.pluginData.guild.roles.fetch(context.roleOrMenu);
if (!role) {
await sendEphemeralReply(int, `A configuration error was encountered, please contact the Administrators!`);
meta.pluginData
.getPlugin(LogsPlugin)
.log(
LogType.BOT_ALERT,
`**A configuration error occured** on buttons for message ${int.message.id}, group **${context.groupName}** not found in config`,
);
return;
}
const member = await meta.pluginData.guild.members.fetch(int.user.id);
if (member.roles.cache.has(role.id)) {
await member.roles.remove(role, `Button Roles on message ${int.message.id}`);
await sendEphemeralReply(int, `Role **${role.name}** removed`);
} else {
await member.roles.add(role, `Button Roles on message ${int.message.id}`);
await sendEphemeralReply(int, `Role **${role.name}** added`);
}
if (context.action === ButtonMenuActions.MODIFY_ROLE) {
await handleModifyRole(meta.pluginData, int, group, context);
return;
}
if (context.action === ButtonMenuActions.OPEN_MENU) {
const menuButtons: MessageButton[] = [];
for (const menuButton of Object.values(group.button_menus[context.roleOrMenu])) {
const customId = await generateStatelessCustomId(meta.pluginData, context.groupName, menuButton.role_or_menu);
const btn = new MessageButton()
.setLabel(menuButton.label)
.setStyle("PRIMARY")
.setType("BUTTON")
.setCustomID(customId);
if (menuButton.emoji) {
const emo = meta.pluginData.client.emojis.resolve(menuButton.emoji) ?? menuButton.emoji;
btn.setEmoji(emo);
}
menuButtons.push(btn);
}
if (menuButtons.length === 0) {
await sendEphemeralReply(int, `A configuration error was encountered, please contact the Administrators!`);
meta.pluginData
.getPlugin(LogsPlugin)
.log(
LogType.BOT_ALERT,
`**A configuration error occured** on buttons for message ${int.message.id}, menu **${context.roleOrMenu}** not found in config`,
);
return;
}
const row = new MessageActionRow().addComponents(menuButtons);
int.reply({ content: `Click to add/remove a role`, components: [row], ephemeral: true, split: false });
await handleOpenMenu(meta.pluginData, int, group, context);
return;
}

81
backend/src/plugins/ReactionRoles/util/buttonActionHandlers.ts Normal file
View file

@ -0,0 +1,81 @@
import { MessageButton, MessageActionRow, MessageComponentInteraction } from "discord.js";
import { GuildPluginData } from "knub";
import { LogType } from "../../../data/LogType";
import { LogsPlugin } from "../../../plugins/Logs/LogsPlugin";
import { ReactionRolesPluginType, TButtonPairOpts } from "../types";
import { generateStatelessCustomId } from "./buttonCustomIdFunctions";
export async function handleOpenMenu(
pluginData: GuildPluginData<ReactionRolesPluginType>,
int: MessageComponentInteraction,
group: TButtonPairOpts,
context,
) {
const menuButtons: MessageButton[] = [];
for (const menuButton of Object.values(group.button_menus[context.roleOrMenu])) {
const customId = await generateStatelessCustomId(pluginData, context.groupName, menuButton.role_or_menu);
const btn = new MessageButton()
.setLabel(menuButton.label)
.setStyle("PRIMARY")
.setType("BUTTON")
.setCustomID(customId);
if (menuButton.emoji) {
const emo = pluginData.client.emojis.resolve(menuButton.emoji) ?? menuButton.emoji;
btn.setEmoji(emo);
}
menuButtons.push(btn);
}
if (menuButtons.length === 0) {
await int.reply({
content: `A configuration error was encountered, please contact the Administrators!`,
ephemeral: true,
});
pluginData
.getPlugin(LogsPlugin)
.log(
LogType.BOT_ALERT,
`**A configuration error occured** on buttons for message ${int.message.id}, menu **${context.roleOrMenu}** not found in config`,
);
return;
}
const row = new MessageActionRow().addComponents(menuButtons);
int.reply({ content: `Click to add/remove a role`, components: [row], ephemeral: true });
return;
}
export async function handleModifyRole(
pluginData: GuildPluginData<ReactionRolesPluginType>,
int: MessageComponentInteraction,
group: TButtonPairOpts,
context,
) {
const role = await pluginData.guild.roles.fetch(context.roleOrMenu);
if (!role) {
await int.reply({
content: `A configuration error was encountered, please contact the Administrators!`,
ephemeral: true,
});
pluginData
.getPlugin(LogsPlugin)
.log(
LogType.BOT_ALERT,
`**A configuration error occured** on buttons for message ${int.message.id}, role **${context.roleOrMenu}** not found on server`,
);
return;
}
const member = await pluginData.guild.members.fetch(int.user.id);
if (member.roles.cache.has(role.id)) {
await member.roles.remove(role, `Button Roles on message ${int.message.id}`);
await int.reply({ content: `Role **${role.name}** removed`, ephemeral: true });
} else {
await member.roles.add(role, `Button Roles on message ${int.message.id}`);
await int.reply({ content: `Role **${role.name}** added`, ephemeral: true });
}
return;
}

2
backend/src/plugins/ReactionRoles/util/buttonCustomIdFunctions.ts
View file

@ -6,7 +6,7 @@ export const BUTTON_CONTEXT_SEPARATOR = "::";
export async function getButtonAction(pluginData: GuildPluginData<ReactionRolesPluginType>, roleOrMenu: string) {
if (await pluginData.guild.roles.fetch(roleOrMenu)) {
return ButtonMenuActions.GRANT_ROLE;
return ButtonMenuActions.MODIFY_ROLE;
} else {
return ButtonMenuActions.OPEN_MENU;
}

2
backend/src/plugins/ReactionRoles/util/buttonMenuActions.ts
View file

@ -1,4 +1,4 @@
export enum ButtonMenuActions {
OPEN_MENU = "goto",
GRANT_ROLE = "grant",
MODIFY_ROLE = "grant",
}