SupportChild/Commands/AddStaffCommand.cs

using System;
using System.Threading.Tasks;
using DSharpPlus.Entities;
using DSharpPlus.SlashCommands;
using DSharpPlus.SlashCommands.Attributes;
using MySqlConnector;

namespace SupportChild.Commands;

public class AddStaffCommand : ApplicationCommandModule
{
	[SlashRequireGuild]
	[SlashCommand("addstaff", "Adds a new staff member.")]
	public async Task OnExecute(InteractionContext command, [Option("User", "User to add to staff.")] DiscordUser user)
	{
		DiscordMember staffMember = null;
		try
		{
			staffMember = user == null ? command.Member : await command.Guild.GetMemberAsync(user.Id);

			if (staffMember == null)
			{
				await command.CreateResponseAsync(new DiscordEmbedBuilder
				{
					Color = DiscordColor.Red,
					Description = "Could not find that user in this server."
				}, true);
				return;
			}
		}
		catch (Exception)
		{
			await command.CreateResponseAsync(new DiscordEmbedBuilder
			{
				Color = DiscordColor.Red,
				Description = "Could not find that user in this server."
			}, true);
			return;
		}

		await using MySqlConnection c = Database.GetConnection();
		MySqlCommand cmd = Database.IsStaff(staffMember.Id) ? new MySqlCommand(@"UPDATE staff SET name = @name WHERE user_id = @user_id", c) : new MySqlCommand(@"INSERT INTO staff (user_id, name) VALUES (@user_id, @name);", c);

		c.Open();
		cmd.Parameters.AddWithValue("@user_id", staffMember.Id);
		cmd.Parameters.AddWithValue("@name", staffMember.DisplayName);
		cmd.ExecuteNonQuery();
		cmd.Dispose();

		await command.CreateResponseAsync(new DiscordEmbedBuilder
		{
			Color = DiscordColor.Green,
			Description = staffMember.Mention + " was added to staff."
		});

		// Log it if the log channel exists
		DiscordChannel logChannel = command.Guild.GetChannel(Config.logChannel);
		if (logChannel != null)
		{
			await logChannel.SendMessageAsync(new DiscordEmbedBuilder
			{
				Color = DiscordColor.Green,
				Description = staffMember.Mention + " was added to staff.\n"
			});
		}
	}
}