using System.ComponentModel; using System.ComponentModel.DataAnnotations; using System.Text; using JetBrains.Annotations; using Remora.Commands.Attributes; using Remora.Commands.Groups; using Remora.Discord.API.Abstractions.Objects; using Remora.Discord.API.Abstractions.Rest; using Remora.Discord.Commands.Attributes; using Remora.Discord.Commands.Conditions; using Remora.Discord.Commands.Contexts; using Remora.Discord.Commands.Feedback.Services; using Remora.Discord.Extensions.Embeds; using Remora.Discord.Extensions.Formatting; using Remora.Rest.Core; using Remora.Results; using TeamOctolings.Octobot.Data; using TeamOctolings.Octobot.Extensions; using TeamOctolings.Octobot.Parsers; using TeamOctolings.Octobot.Services; namespace TeamOctolings.Octobot.Commands; /// /// Handles commands to manage reminders: /remind, /listremind, /delremind /// [UsedImplicitly] public sealed class RemindCommandGroup : CommandGroup { private readonly IInteractionCommandContext _context; private readonly IFeedbackService _feedback; private readonly GuildDataService _guildData; private readonly IDiscordRestInteractionAPI _interactionApi; private readonly IDiscordRestUserAPI _userApi; public RemindCommandGroup( IInteractionCommandContext context, GuildDataService guildData, IFeedbackService feedback, IDiscordRestUserAPI userApi, IDiscordRestInteractionAPI interactionApi) { _context = context; _guildData = guildData; _feedback = feedback; _userApi = userApi; _interactionApi = interactionApi; } /// /// A slash command that lists reminders of the user that called it. /// /// A feedback sending result which may or may not have succeeded. [Command("listremind")] [Description("List your reminders")] [DiscordDefaultDMPermission(false)] [RequireContext(ChannelContext.Guild)] [UsedImplicitly] public async Task ExecuteListReminderAsync() { if (!_context.TryGetContextIDs(out var guildId, out _, out var executorId)) { return new ArgumentInvalidError(nameof(_context), "Unable to retrieve necessary IDs from command context"); } var botResult = await _userApi.GetCurrentUserAsync(CancellationToken); if (!botResult.IsDefined(out var bot)) { return ResultExtensions.FromError(botResult); } var executorResult = await _userApi.GetUserAsync(executorId, CancellationToken); if (!executorResult.IsDefined(out var executor)) { return ResultExtensions.FromError(executorResult); } var data = await _guildData.GetData(guildId, CancellationToken); Messages.Culture = GuildSettings.Language.Get(data.Settings); return await ListRemindersAsync(data.GetOrCreateMemberData(executorId), guildId, executor, bot, CancellationToken); } private Task ListRemindersAsync(MemberData data, Snowflake guildId, IUser executor, IUser bot, CancellationToken ct) { if (data.Reminders.Count == 0) { var failedEmbed = new EmbedBuilder().WithSmallTitle(Messages.NoRemindersFound, bot) .WithColour(ColorsList.Red) .Build(); return _feedback.SendContextualEmbedResultAsync(failedEmbed, ct: ct); } var builder = new StringBuilder(); for (var i = 0; i < data.Reminders.Count; i++) { var reminder = data.Reminders[i]; builder.AppendBulletPointLine(string.Format(Messages.ReminderPosition, Markdown.InlineCode((i + 1).ToString()))) .AppendSubBulletPointLine(string.Format(Messages.ReminderText, Markdown.InlineCode(reminder.Text))) .AppendSubBulletPointLine(string.Format(Messages.ReminderTime, Markdown.Timestamp(reminder.At))) .AppendSubBulletPointLine(string.Format(Messages.DescriptionActionJumpToMessage, $"https://discord.com/channels/{guildId.Value}/{reminder.ChannelId}/{reminder.MessageId}")); } var embed = new EmbedBuilder().WithSmallTitle( string.Format(Messages.ReminderList, executor.GetTag()), executor) .WithDescription(builder.ToString()) .WithColour(ColorsList.Cyan) .Build(); return _feedback.SendContextualEmbedResultAsync(embed, ct: ct); } /// /// A slash command that schedules a reminder with the specified text. /// /// The period of time which must pass before the reminder will be sent. /// The text of the reminder. /// A feedback sending result which may or may not have succeeded. [Command("remind")] [Description("Create a reminder")] [DiscordDefaultDMPermission(false)] [RequireContext(ChannelContext.Guild)] [UsedImplicitly] public async Task ExecuteReminderAsync( [Description("After what period of time mention the reminder (e.g. 1h30m)")] [Option("in")] string timeSpanString, [Description("Reminder text")] [MaxLength(512)] string text) { if (!_context.TryGetContextIDs(out var guildId, out var channelId, out var executorId)) { return new ArgumentInvalidError(nameof(_context), "Unable to retrieve necessary IDs from command context"); } var botResult = await _userApi.GetCurrentUserAsync(CancellationToken); if (!botResult.IsDefined(out var bot)) { return ResultExtensions.FromError(botResult); } var executorResult = await _userApi.GetUserAsync(executorId, CancellationToken); if (!executorResult.IsDefined(out var executor)) { return ResultExtensions.FromError(executorResult); } var data = await _guildData.GetData(guildId, CancellationToken); Messages.Culture = GuildSettings.Language.Get(data.Settings); var parseResult = TimeSpanParser.TryParse(timeSpanString); if (!parseResult.IsDefined(out var timeSpan)) { var failedEmbed = new EmbedBuilder() .WithSmallTitle(Messages.InvalidTimeSpan, bot) .WithDescription(Messages.TimeSpanExample) .WithColour(ColorsList.Red) .Build(); return await _feedback.SendContextualEmbedResultAsync(failedEmbed, ct: CancellationToken); } return await AddReminderAsync(timeSpan, text, data, channelId, executor, CancellationToken); } private async Task AddReminderAsync(TimeSpan timeSpan, string text, GuildData data, Snowflake channelId, IUser executor, CancellationToken ct = default) { var memberData = data.GetOrCreateMemberData(executor.ID); var remindAt = DateTimeOffset.UtcNow.Add(timeSpan); var responseResult = await _interactionApi.GetOriginalInteractionResponseAsync(_context.Interaction.ApplicationID, _context.Interaction.Token, ct); if (!responseResult.IsDefined(out var response)) { return (Result)responseResult; } memberData.Reminders.Add( new Reminder { At = remindAt, ChannelId = channelId.Value, Text = text, MessageId = response.ID.Value }); var builder = new StringBuilder() .AppendBulletPointLine(string.Format(Messages.ReminderText, Markdown.InlineCode(text))) .AppendBulletPoint(string.Format(Messages.ReminderTime, Markdown.Timestamp(remindAt))); var embed = new EmbedBuilder().WithSmallTitle( string.Format(Messages.ReminderCreated, executor.GetTag()), executor) .WithDescription(builder.ToString()) .WithColour(ColorsList.Green) .WithFooter(string.Format(Messages.ReminderPosition, memberData.Reminders.Count)) .Build(); return await _feedback.SendContextualEmbedResultAsync(embed, ct: ct); } public enum Parameters { [UsedImplicitly] Time, [UsedImplicitly] Text } /// /// A slash command that edits a scheduled reminder using the specified text or time. /// /// The list position of the reminder to edit. /// The reminder's parameter to edit. /// The new value for the reminder as a text or time. /// A feedback sending result which may or may not have succeeded. [Command("editremind")] [Description("Edit a reminder")] [DiscordDefaultDMPermission(false)] [RequireContext(ChannelContext.Guild)] [UsedImplicitly] public async Task ExecuteEditReminderAsync( [Description("Position in list")] [MinValue(1)] int position, [Description("Parameter to edit")] Parameters parameter, [Description("Parameter's new value")] string value) { if (!_context.TryGetContextIDs(out var guildId, out _, out var executorId)) { return new ArgumentInvalidError(nameof(_context), "Unable to retrieve necessary IDs from command context"); } var botResult = await _userApi.GetCurrentUserAsync(CancellationToken); if (!botResult.IsDefined(out var bot)) { return ResultExtensions.FromError(botResult); } var executorResult = await _userApi.GetUserAsync(executorId, CancellationToken); if (!executorResult.IsDefined(out var executor)) { return ResultExtensions.FromError(executorResult); } var data = await _guildData.GetData(guildId, CancellationToken); Messages.Culture = GuildSettings.Language.Get(data.Settings); var memberData = data.GetOrCreateMemberData(executor.ID); if (parameter is Parameters.Time) { return await EditReminderTimeAsync(position - 1, value, memberData, bot, executor, CancellationToken); } return await EditReminderTextAsync(position - 1, value, memberData, bot, executor, CancellationToken); } private async Task EditReminderTimeAsync(int index, string value, MemberData data, IUser bot, IUser executor, CancellationToken ct = default) { if (index >= data.Reminders.Count) { var failedEmbed = new EmbedBuilder().WithSmallTitle(Messages.InvalidReminderPosition, bot) .WithColour(ColorsList.Red) .Build(); return await _feedback.SendContextualEmbedResultAsync(failedEmbed, ct: ct); } var parseResult = TimeSpanParser.TryParse(value); if (!parseResult.IsDefined(out var timeSpan)) { var failedEmbed = new EmbedBuilder() .WithSmallTitle(Messages.InvalidTimeSpan, bot) .WithDescription(Messages.TimeSpanExample) .WithColour(ColorsList.Red) .Build(); return await _feedback.SendContextualEmbedResultAsync(failedEmbed, ct: ct); } var oldReminder = data.Reminders[index]; var remindAt = DateTimeOffset.UtcNow.Add(timeSpan); data.Reminders.Add(oldReminder with { At = remindAt }); data.Reminders.RemoveAt(index); var builder = new StringBuilder() .AppendBulletPointLine(string.Format(Messages.ReminderText, Markdown.InlineCode(oldReminder.Text))) .AppendBulletPoint(string.Format(Messages.ReminderTime, Markdown.Timestamp(remindAt))); var embed = new EmbedBuilder().WithSmallTitle( string.Format(Messages.ReminderEdited, executor.GetTag()), executor) .WithDescription(builder.ToString()) .WithColour(ColorsList.Cyan) .WithFooter(string.Format(Messages.ReminderPosition, data.Reminders.Count)) .Build(); return await _feedback.SendContextualEmbedResultAsync(embed, ct: ct); } private async Task EditReminderTextAsync(int index, string value, MemberData data, IUser bot, IUser executor, CancellationToken ct = default) { if (index >= data.Reminders.Count) { var failedEmbed = new EmbedBuilder().WithSmallTitle(Messages.InvalidReminderPosition, bot) .WithColour(ColorsList.Red) .Build(); return await _feedback.SendContextualEmbedResultAsync(failedEmbed, ct: ct); } var oldReminder = data.Reminders[index]; data.Reminders.Add(oldReminder with { Text = value }); data.Reminders.RemoveAt(index); var builder = new StringBuilder() .AppendBulletPointLine(string.Format(Messages.ReminderText, Markdown.InlineCode(value))) .AppendBulletPoint(string.Format(Messages.ReminderTime, Markdown.Timestamp(oldReminder.At))); var embed = new EmbedBuilder().WithSmallTitle( string.Format(Messages.ReminderEdited, executor.GetTag()), executor) .WithDescription(builder.ToString()) .WithColour(ColorsList.Cyan) .WithFooter(string.Format(Messages.ReminderPosition, data.Reminders.Count)) .Build(); return await _feedback.SendContextualEmbedResultAsync(embed, ct: ct); } /// /// A slash command that deletes a reminder using its list position. /// /// The list position of the reminder to delete. /// A feedback sending result which may or may not have succeeded. [Command("delremind")] [Description("Delete one of your reminders")] [DiscordDefaultDMPermission(false)] [RequireContext(ChannelContext.Guild)] [UsedImplicitly] public async Task ExecuteDeleteReminderAsync( [Description("Position in list")] [MinValue(1)] int position) { if (!_context.TryGetContextIDs(out var guildId, out _, out var executorId)) { return new ArgumentInvalidError(nameof(_context), "Unable to retrieve necessary IDs from command context"); } var botResult = await _userApi.GetCurrentUserAsync(CancellationToken); if (!botResult.IsDefined(out var bot)) { return ResultExtensions.FromError(botResult); } var data = await _guildData.GetData(guildId, CancellationToken); Messages.Culture = GuildSettings.Language.Get(data.Settings); return await DeleteReminderAsync(data.GetOrCreateMemberData(executorId), position - 1, bot, CancellationToken); } private Task DeleteReminderAsync(MemberData data, int index, IUser bot, CancellationToken ct) { if (index >= data.Reminders.Count) { var failedEmbed = new EmbedBuilder().WithSmallTitle(Messages.InvalidReminderPosition, bot) .WithColour(ColorsList.Red) .Build(); return _feedback.SendContextualEmbedResultAsync(failedEmbed, ct: ct); } var reminder = data.Reminders[index]; var description = new StringBuilder() .AppendBulletPointLine(string.Format(Messages.ReminderText, Markdown.InlineCode(reminder.Text))) .AppendBulletPointLine(string.Format(Messages.ReminderTime, Markdown.Timestamp(reminder.At))); data.Reminders.RemoveAt(index); var embed = new EmbedBuilder().WithSmallTitle(Messages.ReminderDeleted, bot) .WithDescription(description.ToString()) .WithColour(ColorsList.Green) .Build(); return _feedback.SendContextualEmbedResultAsync(embed, ct: ct); } }