find or create, then save guild on each interaction

commands/admin/custom_vc/slash.js

@@ -4,7 +4,7 @@ import {
 	SlashCommandBuilder,
 	ChatInputCommandInteraction
 } from 'discord.js';
-import { VoiceChannel } from '../../../database.js';
+import { Guild, VoiceChannel } from '../../../database.js';
 
 export const data = new SlashCommandBuilder()
 	.setName('custom_vc')
@@ -52,6 +52,7 @@ export async function execute(interaction) {
 
 	/** @type {string} */
 	let step;
+	const guildData = { id: guild.id };
 	try {
 		switch (options.getSubcommand()) {
 			case 'create': {
@@ -65,10 +66,16 @@ export async function execute(interaction) {
 					type: ChannelType.GuildVoice
 				});
 
-				// Save channel data
 				step = 'save';
+				// Create guild if not exists
+				await Guild.findOrCreate({
+					where: guildData,
+					defaults: guildData
+				});
+				// Save channel data
 				await VoiceChannel.create({
 					id: channel.id,
+					guild: guild.id,
 					create: true
 				});
 
@@ -85,9 +92,18 @@ export async function execute(interaction) {
 				// Get channel id from user input
 				const { id } = options.getChannel('channel');
 
-				// Save channel data
 				step = 'save';
-				await VoiceChannel.create({ id, create: true });
+				// Create guild if not exists
+				await Guild.findOrCreate({
+					where: guildData,
+					defaults: guildData
+				});
+				// Save channel data
+				await VoiceChannel.create({
+					id,
+					guild: guild.id,
+					create: true
+				});
 
 				// Reply success to acknowledge command
 				await interaction.reply({

commands/admin/member_roles/slash.js

@@ -1,5 +1,25 @@
 import { SlashCommandBuilder, PermissionFlagsBits } from 'discord.js';
-import { Role } from '../../../database.js';
+import { Role, Guild } from '../../../database.js';
+
+/**
+ * @param {Guild} guild
+ * @param {Role} role
+ */
+const registerRole = async (guild, role) => {
+	// Check if guild exists in database, otherwise create it
+	const guildData = { id: guild.id };
+	await Guild.findOrCreate({
+		where: guildData,
+		defaults: guildData
+	});
+
+	// Register role in database
+	await Role.create({
+		guild: guild.id,
+		id: role.id,
+		assign: true
+	});
+};
 
 export const data = new SlashCommandBuilder()
 	.setName('member_roles')
@@ -49,12 +69,7 @@ export async function execute(interaction) {
 				found.assign = true;
 				await found.save();
 				// Otherwise create new database entry
-			} else
+			} else await registerRole(interaction.guild, role);
-				await Role.create({
-					id: role.id,
-					assign: true
-				});
-
 			// Reply successfully to acknowledge command
 			await interaction.reply({
 				content: 'Successfully registered role.',

commands/admin/self_roles/slash.js

@@ -1,6 +1,6 @@
 import { PermissionFlagsBits, SlashCommandBuilder, ChatInputCommandInteraction } from 'discord.js';
 import { addSelfRoles, removeSelfRoles } from '../../../shared.js';
-import { Message } from '../../../database.js';
+import { Guild, Message } from '../../../database.js';
 
 /**
  * Sends a `Message` in the current channel and registers for self roles.
@@ -46,6 +46,13 @@ const registerSelfRoles = async (interaction) => {
 		// Get message by id
 		await channel.messages.fetch(id);
 
+		// Check if message is already registered
+		const found = await Message.findOne({
+			where: { id }
+		});
+
+		if (found) throw new Error('Message already registered!');
+
 		// Reply successfully to acknowledge command
 		await interaction.reply({
 			content: 'Successfully fetched message!',
@@ -186,8 +193,18 @@ export async function execute(interaction) {
 
 	if (createNew) {
 		try {
+			// Create guild if not exists
+			const guildData = { id: interaction.guild.id };
+			await Guild.findOrCreate({
+				where: guildData,
+				defaults: guildData
+			});
+
 			// Create database entry
-			await Message.create({ id });
+			await Message.create({
+				id,
+				guild: interaction.guild.id
+			});
 		} catch (error) {
 			console.error(error);
 

events/members/guildMemberAdd.js

@@ -15,8 +15,13 @@ export async function execute(member) {
 	// Ignore if no none found
 	if (roles.length === 0) return;
 
-	// Add roles to member
+	try {
-	await member.roles.add(roles.map((role) => role.id));
+		// Add roles to member
-
+		await member.roles.add(roles.map((role) => role.id));
-	console.info(`[INFO] Added ${roles.length} roles to user with ID '${member.user.id}'.`);
+	} catch (error) {
+		// Missing permissions
+		console.error(error);
+		await member.user.send('Could not assign roles. Please contact server staff.');
+	}
+	console.info(`[INFO] Added ${roles.length} roles to new member with ID '${member.user.id}'.`);
 }

models/guilds.js

@@ -6,6 +6,7 @@ import { DataTypes, Sequelize } from 'sequelize';
  * @property {(model: Object) => void} hasMany Defines an One-To-Many relationship.
  * @property {(conditions: Object) => Promise<Guild>} findOne Finds one instance in the database matching the provided condition(-s).
  * @property {(conditions: Object) => Promise<Array<Guild>>} findAll Finds all instances in the database matching the provided condition(-s).
+ * @property {(conditions: Object) => Promise<Guild>} findOrCreate Finds or creates an instance in the database matching the provided condition(-s) or default values.
  */
 
 /**

models/messages.js

@@ -7,6 +7,7 @@ import { DataTypes, Deferrable, Sequelize } from 'sequelize';
  * @property {(model: Object) => void} hasMany Defines an One-To-Many relationship.
  * @property {(conditions: Object) => Promise<Message>} findOne Finds one instance in the database matching the provided condition(-s).
  * @property {(conditions: Object) => Promise<Array<Message>>} findAll Finds all instances in the database matching the provided condition(-s).
+ * @property {(conditions: Object) => Promise<Message>} findOrCreate Finds or creates an instance in the database matching the provided condition(-s) or default values.
  */
 
 /**

models/roleEmojiPairs.js

@@ -9,6 +9,7 @@ import { DataTypes, Deferrable, Sequelize } from 'sequelize';
  * @property {(model: Object) => void} hasMany Defines an One-To-Many relationship.
  * @property {(conditions: Object) => Promise<RoleEmojiPair>} findOne Finds one instance in the database matching the provided condition(-s).
  * @property {(conditions: Object) => Promise<Array<RoleEmojiPair>>} findAll Finds all instances in the database matching the provided condition(-s).
+ * @property {(conditions: Object) => Promise<RoleEmojiPair>} findOrCreate Finds or creates an instance in the database matching the provided condition(-s) or default values.
  */
 
 /**

models/roles.js

@@ -8,6 +8,7 @@ import { DataTypes, Deferrable, Sequelize } from 'sequelize';
  * @property {(model: Object) => void} hasMany Defines an One-To-Many relationship.
  * @property {(conditions: Object) => Promise<Role>} findOne Finds one instance in the database matching the provided condition(-s).
  * @property {(conditions: Object) => Promise<Array<Role>>} findAll Finds all instances in the database matching the provided condition(-s).
+ * @property {(conditions: Object) => Promise<Role>} findOrCreate Finds or creates an instance in the database matching the provided condition(-s) or default values.
  */
 
 /**

models/voiceChannels.js

@@ -9,6 +9,7 @@ import { DataTypes, Deferrable, Sequelize } from 'sequelize';
  * @property {(model: Object) => void} hasMany Defines an One-To-Many relationship.
  * @property {(conditions: Object) => Promise<VoiceChannel>} findOne Finds one instance in the database matching the provided condition(-s).
  * @property {(conditions: Object) => Promise<Array<VoiceChannel>>} findAll Finds all instances in the database matching the provided condition(-s).
+ * @property {(conditions: Object) => Promise<VoiceChannel>} findOrCreate Finds or creates an instance in the database matching the provided condition(-s) or default values.
  */
 
 /**

shared.js

@@ -65,10 +65,18 @@ const saveMessageData = async (id, role, emoji) => {
 			`Existing RoleEmojiPair entry with (partial) data {message:${id},role:${role.id},emoji:${emoji}}!`
 		);
 
+	// Create guild if not exists
+	const guildData = { id: role.guild.id };
+	await Guild.findOrCreate({
+		where: guildData,
+		defaults: guildData
+	});
+
 	// Create database entry for pair
 	await RoleEmojiPair.create({
 		message: id,
 		role: role.id,
+		guild: guildData.id,
 		emoji: emoji.replace(/:(\s*[^:]*\s*):/, ':_:')
 	});
 };