PluralKit/PluralKit.Bot/Handlers/MessageDeleted.cs

using System;
using System.Threading.Tasks;

using Myriad.Gateway;

using PluralKit.Core;

using Serilog;

namespace PluralKit.Bot
{
    // Double duty :)
    public class MessageDeleted: IEventHandler<MessageDeleteEvent>, IEventHandler<MessageDeleteBulkEvent>
    {
        private static readonly TimeSpan MessageDeleteDelay = TimeSpan.FromSeconds(15);
        
        private readonly IDatabase _db;
        private readonly ModelRepository _repo;
        private readonly ILogger _logger;

        public MessageDeleted(ILogger logger, IDatabase db, ModelRepository repo)
        {
            _db = db;
            _repo = repo;
            _logger = logger.ForContext<MessageDeleted>();
        }
        
        public Task Handle(Shard shard, MessageDeleteEvent evt)
        {
            // Delete deleted webhook messages from the data store
            // Most of the data in the given message is wrong/missing, so always delete just to be sure.

            async Task Inner()
            {
                await Task.Delay(MessageDeleteDelay);
                // TODO
                // await _db.Execute(c => _repo.DeleteMessage(c, evt.Message.Id));
            }

            // Fork a task to delete the message after a short delay
            // to allow for lookups to happen for a little while after deletion
            _ = Inner();
            return Task.CompletedTask;
        }

        public Task Handle(Shard shard, MessageDeleteBulkEvent evt)
        {
            // Same as above, but bulk
            async Task Inner()
            {
                await Task.Delay(MessageDeleteDelay);
                // TODO
                // _logger.Information("Bulk deleting {Count} messages in channel {Channel}", evt.Messages.Count, evt.Channel.Id);
                // await _db.Execute(c => _repo.DeleteMessagesBulk(c, evt.Messages.Select(m => m.Id).ToList()));
            }
            
            _ = Inner();
            return Task.CompletedTask;
        }
    }
}