eightball/PluralKit
1
0
Fork 0
mirror of https://github.com/PluralKit/PluralKit.git synced 2026-02-07 06:17:55 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge 21a63a43cc into da4c05d4ed

Browse source
This commit is contained in:
Petal Ladenson 2024-12-06 01:17:19 +01:00 • committed by GitHub
commit 2bdcc1d3c0
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
6 changed files with 65 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

31
PluralKit.Bot/Handlers/MessageDeleted.cs
View file

@ -32,6 +32,16 @@ public class MessageDeleted: IEventHandler<MessageDeleteEvent>, IEventHandler<Me
async Task Inner()
{
await Task.Delay(MessageDeleteDelay);
var message = await _repo.GetMessage(evt.Id);
if (message == null) return;
if (message.Member.HasValue)
{
var memberFull = await _repo.GetMember(message.Member.Value);
await _repo.UpdateMemberForDeletedMessage(message.Member.Value, memberFull.MessageCount);
}
await _repo.DeleteMessage(evt.Id);
}
@ -53,6 +63,27 @@ public class MessageDeleted: IEventHandler<MessageDeleteEvent>, IEventHandler<Me
_logger.Information("Bulk deleting {Count} messages in channel {Channel}",
evt.Ids.Length, evt.ChannelId);
await _repo.DeleteMessagesBulk(evt.Ids);
// get all the messages from the database
var messages = await _repo.GetMessagesBulk(evt.Ids);
// make a dictionary of every member id associated with a message and how many messages it's associated with
var memberMessageCounts = new Dictionary<MemberId, int>();
foreach (PKMessage msg in messages)
{
if (!msg.Member.HasValue) continue;
if (memberMessageCounts.ContainsKey(msg.Member.Value))
memberMessageCounts[msg.Member.Value] = memberMessageCounts[msg.Member.Value] + 1;
else
memberMessageCounts[msg.Member.Value] = 1;
}
// go through each member id in dictionary and decrement that member's message count by appropriate amount
foreach (MemberId member in memberMessageCounts.Keys)
{
await _repo.UpdateMemberForDeletedMessage(member, (await _repo.GetMember(member)).MessageCount, memberMessageCounts[member]);
}
}
_lastMessage.HandleMessageDeletion(evt.ChannelId, evt.Ids.ToList());

2
PluralKit.Bot/Handlers/ReactionAdded.cs
View file

@ -141,8 +141,6 @@ public class ReactionAdded: IEventHandler<MessageReactionAddEvent>
{
// Message was deleted by something/someone else before we got to it
}
await _repo.DeleteMessage(evt.MessageId);
}
private async ValueTask HandleCommandDeleteReaction(MessageReactionAddEvent evt, ulong? authorId, bool isDM)