SupportChild/Commands/RandomAssignCommand.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
using DSharpPlus.Entities;
using DSharpPlus.Exceptions;
using DSharpPlus.SlashCommands;
using DSharpPlus.SlashCommands.Attributes;
using Microsoft.Extensions.Logging;

namespace SupportChild.Commands;

public class RandomAssignCommand : ApplicationCommandModule
{
	[SlashRequireGuild]
	[SlashCommand("rassign", "Randomly assigns a staff member to a ticket.")]
	public async Task OnExecute(InteractionContext command, [Option("Role", "(Optional) Limit the random assignment to a specific role.")] DiscordRole role = null)
	{
		// Check if ticket exists in the database
		if (!Database.TryGetOpenTicket(command.Channel.Id, out Database.Ticket ticket))
		{
			await command.CreateResponseAsync(new DiscordEmbedBuilder
			{
				Color = DiscordColor.Red,
				Description = "Error: This channel is not a ticket."
			}, true);
			return;
		}

		// Get a random staff member who is verified to have the correct role if applicable
		DiscordMember staffMember = await GetRandomVerifiedStaffMember(command, role, ticket);
		if (staffMember == null)
		{
			return;
		}

		// Attempt to assign the staff member to the ticket
		if (!Database.AssignStaff(ticket, staffMember.Id))
		{
			await command.CreateResponseAsync(new DiscordEmbedBuilder
			{
				Color = DiscordColor.Red,
				Description = "Error: Failed to assign " + staffMember.Mention + " to ticket."
			}, true);
			return;
		}

		// Respond that the command was successful
		await command.CreateResponseAsync(new DiscordEmbedBuilder
		{
			Color = DiscordColor.Green,
			Description = "Randomly assigned " + staffMember.Mention + " to ticket."
		});

		// Send a notification to the staff member if applicable
		if (Config.assignmentNotifications)
		{
			try
			{
				await staffMember.SendMessageAsync(new DiscordEmbedBuilder
				{
					Color = DiscordColor.Green,
					Description = "You have been randomly assigned to a support ticket: " + command.Channel.Mention
				});
			}
			catch (UnauthorizedException) { }
		}

		// 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 randomly assigned to " + command.Channel.Mention + " by " + command.Member.Mention + "."
			});
		}
	}

	private static async Task<DiscordMember> GetRandomVerifiedStaffMember(InteractionContext command, DiscordRole targetRole, Database.Ticket ticket)
	{
		if (targetRole != null) // A role was provided
		{
			// Check if role rassign should override staff's active status
			List<Database.StaffMember> staffMembers = Config.randomAssignRoleOverride
				? Database.GetAllStaff(ticket.assignedStaffID, ticket.creatorID)
				: Database.GetActiveStaff(ticket.assignedStaffID, ticket.creatorID);

			// Randomize the list before checking for roles in order to reduce number of API calls
			staffMembers.Shuffle();

			// Get the first staff member that has the role
			foreach (Database.StaffMember sm in staffMembers)
			{
				try
				{
					DiscordMember verifiedMember = await command.Guild.GetMemberAsync(sm.userID);
					if (verifiedMember?.Roles?.Any(role => role.Id == targetRole.Id) ?? false)
					{
						return verifiedMember;
					}
				}
				catch (Exception e)
				{
					command.Client.Logger.Log(LogLevel.Information, e, "Error occured trying to find a staff member in the rassign command.");
				}
			}
		}
		else // No role was specified, any active staff will be picked
		{
			Database.StaffMember staffEntry = Database.GetRandomActiveStaff(ticket.assignedStaffID, ticket.creatorID);
			if (staffEntry == null)
			{
				await command.CreateResponseAsync(new DiscordEmbedBuilder
				{
					Color = DiscordColor.Red,
					Description = "Error: There are no other staff members to choose from."
				}, true);
				return null;
			}

			// Get the staff member from discord
			try
			{
				return await command.Guild.GetMemberAsync(staffEntry.userID);
			}
			catch (NotFoundException) { }
		}

		// Send a more generic error if we get to this point and still haven't found the staff member
		await command.CreateResponseAsync(new DiscordEmbedBuilder
		{
			Color = DiscordColor.Red,
			Description = "Error: Could not find an applicable staff member."
		}, true);
		return null;
	}
}