SpecialRoleCommands.cs

using System;
using System.Collections.Generic;
using System.Text;
using System.Linq;
using Discord;
using Discord.WebSocket;
using DiscordBotBase;
using DiscordBotBase.CommandHandlers;
using ModBot.Database;
using ModBot.WarningHandlers;
using ModBot.Core;
using FreneticUtilities.FreneticExtensions;
using FreneticUtilities.FreneticToolkit;
using System.Threading.Tasks;

namespace ModBot.CommandHandlers
{
    /// <summary>Command handler for special role commands.</summary>
    public class SpecialRoleCommands : UserCommands
    {
        /// <summary>User command to add a special role to a user.</summary>
        public void CMD_AddSpecialRole(GuildConfig.SpecialRole role, CommandData command)
        {
            SocketGuild guild = (command.Message.Channel as SocketGuildChannel).Guild;
            GuildConfig config = DiscordModBot.GetConfig(guild.Id);
            if (!DiscordModBot.IsModerator(command.Message.Author as SocketGuildUser))
            {
                SendErrorMessageReply(command.Message, "Not Authorized", "You're not allowed to do that.");
                return;
            }
            if (!DiscordModBot.WarningCommandHandler.GetTargetUser(command, true, true, out ulong userID))
            {
                return;
            }
            SocketGuildUser guildUser = guild.GetUser(userID);
            WarnableUser warnable = WarningUtilities.GetWarnableUser(guild.Id, userID);
            if (warnable.SeenNames.IsEmpty() && !config.AllowWarningUnknownUsers)
            {
                SendErrorMessageReply(command.Message, "Invalid Input", $"Cannot alter that user: user <@{userID}> has never been seen before. Did you reference a user that hasn't joined this guild yet, or accidentally copy a message ID instead of user ID?");
                return;
            }
            if (guildUser is not null)
            {
                SocketRole socketRole = guild.GetRole(role.RoleID);
                if (socketRole is null)
                {
                    SendErrorMessageReply(command.Message, "Failed To Apply", "Cannot apply special role: no matching role found.");
                    return;
                }
                guildUser.AddRoleAsync(socketRole).Wait();
            }
            bool alreadyHadRole = warnable.SpecialRoles.Contains(role.Name);
            if (alreadyHadRole)
            {
                SendGenericNegativeMessageReply(command.Message, "Cannot Apply Special Role", $"User <@{userID}> is already marked with special role `{role.Name}`.");
                return;
            }
            warnable.SpecialRoles.Add(role.Name);
            warnable.Save();
            string hadSameRoleBefore = "";
            if (!string.IsNullOrWhiteSpace(role.AddWarnText) && warnable.Warnings.Any(w => w.Reason.StartsWith(role.AddWarnText)))
            {
                hadSameRoleBefore = $"\n\nUser has previously had the special role `{role.Name}` applied.";
            }
            string addedRef = string.Join(" ", command.RawArguments.Skip(1)).Trim();
            string refLink = null;
            if (addedRef.StartsWith("https://discord.com/channels/") && REF_LINK_AFTER_BASE_MATCHER.IsOnlyMatches(addedRef["https://discord.com/channels/".Length..]))
            {
                refLink = addedRef;
                addedRef = "";
            }
            string addedText = string.IsNullOrWhiteSpace(addedRef) ? "." : $" with reference input: {addedRef}\n";
            string contextRef = "";
            IMessageChannel targetChannel = command.Message.Channel;
            if (role.ChannelNoticeType > 0 && role.PutNoticeInChannel != 0 && role.PutNoticeInChannel != targetChannel.Id && (targetChannel is not SocketThreadChannel threaded || threaded.ParentChannel.Id != role.PutNoticeInChannel))
            {
                string name = NameUtilities.AcceptableSymbolMatcher.TrimToMatches(warnable.LastKnownUsername?.BeforeLast('#') ?? "Unknown");
                if (name.Length > 20)
                {
                    name = name[..20];
                }
                SocketGuildChannel baseTargetChannel = guild.GetChannel(role.PutNoticeInChannel);
                if (role.ChannelNoticeType == 1)
                {
                    targetChannel = baseTargetChannel as IMessageChannel;
                }
                else if (role.ChannelNoticeType == 2)
                {
                    targetChannel = (baseTargetChannel as SocketTextChannel).CreateThreadAsync($"[Auto] {name}", ThreadType.PublicThread).Result;
                }
                else if (role.ChannelNoticeType == 3)
                {
                    targetChannel = (baseTargetChannel as SocketTextChannel).CreateThreadAsync($"[Auto] {name}", ThreadType.PrivateThread).Result;
                }
                if (targetChannel is null)
                {
                    targetChannel = command.Message.Channel;
                    Console.WriteLine($"Failed to channel thread type {role.ChannelNoticeType} in {role.PutNoticeInChannel}");
                }
                else
                {
                    contextRef = $"[Original Command Message Reference]({command.Message.GetJumpUrl()})\n\n";
                }
            }
            IUserMessage sentMessage = targetChannel.SendMessageAsync(text: $"<@{userID}>", embed: new EmbedBuilder().WithTitle("Special Role Applied").WithDescription($"{contextRef}<@{command.Message.Author.Id}> has given <@{userID}> special role `{role.Name}`{addedText}\n\n{role.AddExplanation}\n{warnable.GetPastWarningsText()}{hadSameRoleBefore}").Build()).Result;
            if (!string.IsNullOrWhiteSpace(role.AddWarnText))
            {
                Warning warning = new() { GivenTo = userID, GivenBy = command.Message.Author.Id, TimeGiven = DateTimeOffset.UtcNow, Level = role.AddLevel, Reason = role.AddWarnText + addedText, RefLink = refLink, Link = LinkToMessage(sentMessage) };
                warnable.AddWarning(warning);
            }
            if (targetChannel.Id != command.Message.Channel.Id)
            {
                if (targetChannel is IThreadChannel)
                {
                    SendGenericPositiveMessageReply(command.Message, "Applied", $"Role applied.\n\nThread created: <#{targetChannel.Id}>");
                }
                else
                {
                    SendGenericPositiveMessageReply(command.Message, "Applied", "Role applied.");
                }
            }
        }

        /// <summary>Matcher for characters after the base of a Discord message link.</summary>
        public static AsciiMatcher REF_LINK_AFTER_BASE_MATCHER = new("0123456789/");

        /// <summary>Util to remove a single special role from a user.</summary>
        public static void RemoveSpecialRole(GuildConfig.SpecialRole role, SocketGuildUser guildUser, WarnableUser warnable, CommandData command, GuildConfig config)
        {
            bool hasRole = warnable.SpecialRoles.Contains(role.Name);
            if (!hasRole)
            {
                SendErrorMessageReply(command.Message, "Failed To Remove", $"Cannot remove special role `{EscapeUserInput(role.Name)}` from user <@{warnable.UserID()}>: user does not have that special role.");
                return;
            }
            if (guildUser is not null && !warnable.SpecialRoles.Any(r => r != role.Name && config.SpecialRoles.TryGetValue(r, out GuildConfig.SpecialRole roleObj) && roleObj.RoleID == role.RoleID))
            {
                SocketRole socketRole = guildUser.Guild.GetRole(role.RoleID);
                if (socketRole is null)
                {
                    SendErrorMessageReply(command.Message, "Failed To Remove", $"Cannot remove special role `{EscapeUserInput(role.Name)}` with Discord role ID `{role.RoleID}` / <@&{role.RoleID}>: no matching role found. Was it deleted, or is the ID wrong?");
                }
                else
                {
                    guildUser.RemoveRoleAsync(socketRole).Wait();
                }
            }
            warnable.SpecialRoles.Remove(role.Name);
            warnable.Save();
            IUserMessage sentMessage = command.Message.Channel.SendMessageAsync(embed: new EmbedBuilder().WithTitle("Special Role Removed").WithDescription($"<@{command.Message.Author.Id}> has removed the special role `{role.Name}` from <@{warnable.UserID()}>.\n{role.RemoveExplanation}").Build()).Result;
            if (!string.IsNullOrWhiteSpace(role.RemoveWarnText))
            {
                Warning warning = new() { GivenTo = warnable.UserID(), GivenBy = command.Message.Author.Id, TimeGiven = DateTimeOffset.UtcNow, Level = role.RemoveLevel, Reason = role.RemoveWarnText, Link = LinkToMessage(sentMessage) };
                warnable.AddWarning(warning);
            }
        }

        /// <summary>User command to remove a special role from a user.</summary>
        public void CMD_RemoveSpecialRole(GuildConfig.SpecialRole role, CommandData command)
        {
            SocketGuild guild = (command.Message.Channel as SocketGuildChannel).Guild;
            GuildConfig config = DiscordModBot.GetConfig(guild.Id);
            if (!DiscordModBot.IsModerator(command.Message.Author as SocketGuildUser))
            {
                SendErrorMessageReply(command.Message, "Not Authorized", "You're not allowed to do that.");
                return;
            }
            if (!DiscordModBot.WarningCommandHandler.GetTargetUser(command, true, true, out ulong userID))
            {
                return;
            }
            SocketGuildUser guildUser = guild.GetUser(userID);
            WarnableUser warnable = WarningUtilities.GetWarnableUser(guild.Id, userID);
            if (warnable.SeenNames.IsEmpty() && warnable.SpecialRoles.IsEmpty())
            {
                SendErrorMessageReply(command.Message, "Invalid Input", $"Cannot alter that user: user <@{userID}> has never been seen before. Did you reference a user that hasn't joined this guild yet, or accidentally copy a message ID instead of user ID?");
                return;
            }
            RemoveSpecialRole(role, guildUser, warnable, command, config);
        }

        /// <summary>User command to clear ALL special roles from a user.</summary>
        public void CMD_ClearSpecialRoles(CommandData command)
        {
            SocketGuild guild = (command.Message.Channel as SocketGuildChannel).Guild;
            GuildConfig config = DiscordModBot.GetConfig(guild.Id);
            if (!DiscordModBot.IsModerator(command.Message.Author as SocketGuildUser))
            {
                SendErrorMessageReply(command.Message, "Not Authorized", "You're not allowed to do that.");
                return;
            }
            if (!DiscordModBot.WarningCommandHandler.GetTargetUser(command, true, true, out ulong userID))
            {
                return;
            }
            SocketGuildUser guildUser = guild.GetUser(userID);
            WarnableUser warnable = WarningUtilities.GetWarnableUser(guild.Id, userID);
            if (warnable.SeenNames.IsEmpty() && warnable.SpecialRoles.IsEmpty())
            {
                SendErrorMessageReply(command.Message, "Invalid Input", $"Cannot alter that user: user <@{userID}> has never been seen before. Did you reference a user that hasn't joined this guild yet, or accidentally copy a message ID instead of user ID?");
                return;
            }
            string[] roles = [.. warnable.SpecialRoles];
            if (roles.IsEmpty())
            {
                SendErrorMessageReply(command.Message, "Invalid Input", $"User <@{userID}> already does not have any special roles.");
                return;
            }
            foreach (string roleName in roles)
            {
                if (config.SpecialRoles.TryGetValue(roleName, out GuildConfig.SpecialRole role))
                {
                    RemoveSpecialRole(role, guildUser, warnable, command, config);
                }
            }
            warnable.SpecialRoles.Clear();
            warnable.Save();
        }
    }
}