feat(bot): store command message info in redis

PluralKit.Bot/CommandSystem/Context/Context.cs

@@ -94,12 +94,12 @@ public class Context
             AllowedMentions = mentions ?? new AllowedMentions()
         });
 
-        if (embed != null)
-        {
+        // if (embed != null)
+        // {
             // Sensitive information that might want to be deleted by :x: reaction is typically in an embed format (member cards, for example)
-            // This may need to be changed at some point but works well enough for now
+            // but since we can, we just store all sent messages for possible deletion
             await _commandMessageService.RegisterMessage(msg.Id, msg.ChannelId, Author.Id);
-        }
+        // }
 
         return msg;
     }

PluralKit.Bot/Commands/Message.cs

@@ -2,6 +2,8 @@
 using System.Text;
 using System.Text.RegularExpressions;
 
+using Autofac;
+
 using Myriad.Builders;
 using Myriad.Cache;
 using Myriad.Extensions;
@@ -304,14 +306,14 @@ public class ProxiedMessage
 
     private async Task DeleteCommandMessage(Context ctx, ulong messageId)
     {
-        var message = await ctx.Repository.GetCommandMessage(messageId);
-        if (message == null)
+        var (authorId, channelId) = await ctx.Services.Resolve<CommandMessageService>().GetCommandMessage(messageId);
+        if (authorId == null)
             throw Errors.MessageNotFound(messageId);
 
-        if (message.AuthorId != ctx.Author.Id)
+        if (authorId != ctx.Author.Id)
             throw new PKError("You can only delete command messages queried by this account.");
 
-        await ctx.Rest.DeleteMessage(message.ChannelId, message.MessageId);
+        await ctx.Rest.DeleteMessage(channelId!.Value, messageId);
 
         if (ctx.Guild != null)
             await ctx.Rest.DeleteMessage(ctx.Message);

PluralKit.Bot/Handlers/ReactionAdded.cs

@@ -73,10 +73,10 @@ public class ReactionAdded: IEventHandler<MessageReactionAddEvent>
                 return;
             }
 
-            var commandMsg = await _commandMessageService.GetCommandMessage(evt.MessageId);
-            if (commandMsg != null)
+            var (authorId, _) = await _commandMessageService.GetCommandMessage(evt.MessageId);
+            if (authorId != null)
             {
-                await HandleCommandDeleteReaction(evt, commandMsg);
+                await HandleCommandDeleteReaction(evt, authorId.Value);
                 return;
             }
         }
@@ -141,11 +141,11 @@ public class ReactionAdded: IEventHandler<MessageReactionAddEvent>
         await _repo.DeleteMessage(evt.MessageId);
     }
 
-    private async ValueTask HandleCommandDeleteReaction(MessageReactionAddEvent evt, CommandMessage? msg)
+    private async ValueTask HandleCommandDeleteReaction(MessageReactionAddEvent evt, ulong? authorId)
     {
         // Can only delete your own message
         // (except in DMs, where msg will be null)
-        if (msg != null && msg.AuthorId != evt.UserId)
+        if (authorId != null && authorId != evt.UserId)
             return;
 
         // todo: don't try to delete the user's own messages in DMs

PluralKit.Bot/Services/CommandMessageService.cs

@@ -8,16 +8,13 @@ namespace PluralKit.Bot;
 
 public class CommandMessageService
 {
-    private readonly IClock _clock;
-    private readonly IDatabase _db;
+    private readonly RedisService _redis;
     private readonly ILogger _logger;
-    private readonly ModelRepository _repo;
+    private static readonly TimeSpan CommandMessageRetention = TimeSpan.FromHours(24);
 
-    public CommandMessageService(IDatabase db, ModelRepository repo, IClock clock, ILogger logger)
+    public CommandMessageService(RedisService redis, IClock clock, ILogger logger)
     {
-        _db = db;
-        _repo = repo;
-        _clock = clock;
+        _redis = redis;
         _logger = logger.ForContext<CommandMessageService>();
     }
 
@@ -27,9 +24,18 @@ public class CommandMessageService
             "Registering command response {MessageId} from author {AuthorId} in {ChannelId}",
             messageId, authorId, channelId
         );
-        await _repo.SaveCommandMessage(messageId, channelId, authorId);
+
+        await _redis.Connection.GetDatabase().StringSetAsync(messageId.ToString(), $"{authorId}-{channelId}", expiry: CommandMessageRetention);
     }
 
-    public async Task<CommandMessage?> GetCommandMessage(ulong messageId) =>
-        await _repo.GetCommandMessage(messageId);
+    public async Task<(ulong?, ulong?)> GetCommandMessage(ulong messageId)
+    {
+        var str = await _redis.Connection.GetDatabase().StringGetAsync(messageId.ToString());
+        if (str.HasValue)
+        {
+            var split = ((string)str).Split("-");
+            return (ulong.Parse(split[0]), ulong.Parse(split[1]));
+        }
+        return (null, null);
+    }
 }