commands/admin/ctx_add_self_roles.js

@@ -1,65 +1,30 @@
-import { TextInputBuilder, TextInputStyle } from 'discord.js';
-import {
-	ModalBuilder,
-	ActionRowBuilder,
-	ApplicationCommandType,
-	ContextMenuCommandBuilder
-} from 'discord.js';
-import { addSelfRoles } from '../../shared.js';
+import { Message } from '../../database.js';
+import { ApplicationCommandType, ContextMenuCommandBuilder } from 'discord.js';
 
 export const data = new ContextMenuCommandBuilder()
 	.setDMPermission(false)
-	.setName('Add role emoji pair')
+	.setName('Register self roles')
 	.setType(ApplicationCommandType.Message);
-export async function modalSubmit(interaction) {
-	const { fields, guild } = interaction;
-	// Get text inputs from modal
-	const message = fields.getTextInputValue('message');
-	const roleID = fields.getTextInputValue('role');
-	const emoji = fields.getTextInputValue('emoji');
+export async function execute(interaction) {
+	const id = interaction.targetMessage.id;
 
-	// Fetch role from guild
-	const role = await guild.roles.fetch(roleID);
-	// Role not found
-	if (role === null) {
+	try {
+		// Create database entry
+		await Message.create({ id });
+
+		// Reply successfully to acknowledge command
 		await interaction.reply({
-			content: 'Could not fetch role! Please contact server staff.',
+			content: `Successfully saved data from message! Add roles to it with reference ID '${id}'.`,
+			ephemeral: true,
+		});
+
+		console.info(`[INFO] New self roles on message with ID: '${id}'.`);
+	} catch (error) {
+		console.error(error);
+		// Reply failed to acknowledge command
+		await interaction.reply({
+			content: 'Failed to save data from message!',
 			ephemeral: true,
 		});
-		return;
 	}
-
-	await addSelfRoles(interaction, message, role, emoji);
-}
-export async function execute(interaction) {
-	const modal = new ModalBuilder()
-		.setCustomId('Add role emoji pair-pair')
-		.setTitle('Role Emoji Pair');
-
-	const id = interaction.targetMessage.id;
-	const message = new ActionRowBuilder().addComponents(
-		new TextInputBuilder()
-			.setLabel('The message ID this command is run on.')
-			.setStyle(TextInputStyle.Short)
-			.setCustomId('message')
-			.setRequired(true)
-			.setValue(id));
-
-	const role = new ActionRowBuilder().addComponents(
-		new TextInputBuilder()
-			.setLabel('Enter exactly one role ID.')
-			.setStyle(TextInputStyle.Short)
-			.setCustomId('role')
-			.setRequired(true));
-
-	const emoji = new ActionRowBuilder().addComponents(
-		new TextInputBuilder()
-			.setLabel('Enter exactly one emoji.')
-			.setStyle(TextInputStyle.Short)
-			.setCustomId('emoji')
-			.setRequired(true));
-
-	modal.addComponents(message, role, emoji);
-
-	await interaction.showModal(modal);
 }

commands/admin/slash_self_roles.js

@@ -1,7 +1,6 @@
 import { Op } from 'sequelize';
 import { SlashCommandBuilder } from 'discord.js';
 import { Message, RoleEmojiPair } from './../../database.js';
-import { addSelfRoles } from '../../shared.js';
 
 const createSelfRoles = async (interaction) => {
 	const { options, channel } = interaction;
@@ -51,6 +50,67 @@ const registerSelfRoles = async (interaction) => {
 	return response;
 };
 
+const saveMessageData = async (id, role, emoji) => {
+	// Try finding existing entry
+	const rep = await RoleEmojiPair.findOne({
+		where: {
+			[Op.or]: [
+				{
+					message: id,
+					role: role.id
+				}, {
+					message: id,
+					emoji
+				}
+			]
+		}
+	});
+	if (rep !== null) throw new Error(`Failed to fetch RoleEmojiPair entry with data {message:${id},role:${role.id},emoji:${emoji}}!`);
+
+	// Create database entry for pair
+	await RoleEmojiPair.create({ message: id, role: role.id, emoji });
+};
+
+const addSelfRoles = async (interaction) => {
+	const { options, channel } = interaction;
+	const id = options.getString('id');
+
+	let step = 'fetch';
+	try {
+		// Get message by id
+		const message = await channel.messages.fetch(id);
+
+		// Get user arguments
+		const role = options.getRole('role');
+		const emoji = options
+			.getString('emoji')
+			.replace(/:.*?:/, ':_:');
+
+		step = 'save data from';
+		await saveMessageData(id, role, emoji);
+
+		step = 'react to';
+		// React with emoji to message
+		await message.react(emoji);
+
+		// Reply successfully to acknowledge command
+		await interaction.reply({
+			content: 'Added new entry for self roles!',
+			ephemeral: true,
+		});
+
+		console.info(`[INFO] Added new entry to get role with ID '${role.id}' using '${emoji}'.`);
+	} catch (error) {
+		console.error(error);
+
+		// Reply failed to acknowledge command
+		await interaction.reply({
+			content: `Failed to ${step} message!`,
+			ephemeral: true,
+		});
+	}
+}
+
 export const data = new SlashCommandBuilder()
 	.setName('self_roles')
 	.setDMPermission(false)
@@ -103,18 +163,13 @@ export async function execute(interaction) {
 			createNew = true;
 			break;
 		case 'register':
-			const response = await registerSelfRoles(interaction);
-			id = response.msgID ?? id;
+			const { success, msgID } = await registerSelfRoles(interaction);
+			id = msgID ?? id;
 			// Flag to create new database entry
-			createNew = response.success;
+			createNew = success;
 			break;
 		case 'add':
-			// Get command options
-			const msgID = options.getString('id');
-			const role = options.getRole('role');
-			const emoji = options.getString('emoji');
-			// Try adding self role pair
-			await addSelfRoles(interaction, msgID, role, emoji);
+			await addSelfRoles(interaction);
 			break;
 	}
 

events/interactionCreate.js

@@ -21,9 +21,9 @@ const executeCommand = async (interaction, command) => {
 	}
 };
 
-const genericExecute = async (interaction, command, name, description, cmdName) => {
+const genericExecute = async (interaction, command, name, description) => {
 	try {
-		console.info(`[INFO] Command ${(cmdName ?? interaction.commandName) ?? 'anonymous'} ${description ?? `used "${name}"`}.`);
+		console.info(`[INFO] Command ${interaction.commandName} ${description ?? `used "${name}"`}.`);
 		await command[name](interaction);
 	} catch (error) {
 		console.error(error);
@@ -49,7 +49,7 @@ export async function execute(interaction) {
 		const name = interaction.customId.split('-')[0];
 		command = interaction.client.commands.get(name);
 
-		await genericExecute(interaction, command, 'modalSubmit', 'submitted a modal', name);
+		await genericExecute(interaction, command, 'modalSubmit', 'submitted a modal');
 		return;
 	}
 

shared.js

@@ -1,60 +0,0 @@
-import { Op } from "sequelize";
-import { Message, RoleEmojiPair } from "./database.js";
-
-const saveMessageData = async (id, role, emoji) => {
-	// Try finding message
-	const msg = await Message.findOne({ where: { id } });
-	if (msg === null) throw new Error(`No message with ID '${id}' could be found!`);
-
-	// Try finding existing entry
-	const rep = await RoleEmojiPair.findOne({
-		where: {
-			[Op.or]: [
-				{
-					message: id,
-					role: role.id
-				}, {
-					message: id,
-					emoji
-				}
-			]
-		}
-	});
-	if (rep !== null) throw new Error(`Existing RoleEmojiPair entry with (partial) data {message:${id},role:${role.id},emoji:${emoji}}!`);
-
-	// Create database entry for pair
-	await RoleEmojiPair.create({ message: id, role: role.id, emoji });
-};
-
-export const addSelfRoles = async (interaction, msgID, role, emoji) => {
-	const { channel } = interaction;
-
-	let step = 'fetch';
-	try {
-		// Get message by id
-		const message = await channel.messages.fetch(msgID);
-
-		step = 'save data from';
-		await saveMessageData(msgID, role, emoji);
-
-		step = 'react to';
-		// React with emoji to message
-		await message.react(emoji);
-
-		// Reply successfully to acknowledge command
-		await interaction.reply({
-			content: 'Added new entry for self roles!',
-			ephemeral: true,
-		});
-
-		console.info(`[INFO] Added new entry to get role with ID '${role.id}' using '${emoji}'.`);
-	} catch (error) {
-		console.error(error);
-
-		// Reply failed to acknowledge command
-		await interaction.reply({
-			content: `Failed to ${step} message!`,
-			ephemeral: true,
-		});
-	}
-}