wip: guild member remove

This commit is contained in:
sam 2024-09-02 17:00:33 +02:00
parent db01f879bd
commit 516ce3a6e9
Signed by: sam
GPG key ID: 5F3C3C1B3166639D
6 changed files with 95 additions and 1 deletions

View file

@ -8,6 +8,7 @@ public interface IMemberCache
public Task<IGuildMember?> TryGetAsync(Snowflake guildId, Snowflake userId);
public Task SetAsync(Snowflake guildId, IGuildMember member);
public Task SetManyAsync(Snowflake guildId, IReadOnlyList<IGuildMember> members);
public Task RemoveAsync(Snowflake guildId, Snowflake userId);
public Task<bool> IsGuildCachedAsync(Snowflake guildId);
public Task MarkAsCachedAsync(Snowflake guildId);
public Task MarkAsUncachedAsync(Snowflake guildId);

View file

@ -0,0 +1,21 @@
using System.Collections.Concurrent;
using Remora.Discord.API;
using Remora.Rest.Core;
namespace Catalogger.Backend.Cache.InMemoryCache;
public class AuditLogCache
{
private readonly ConcurrentDictionary<(Snowflake, Snowflake), ModactionData> _kicks = new();
private readonly ConcurrentDictionary<(Snowflake, Snowflake), ModactionData> _bans = new();
public void SetKick(Snowflake guildId, string targetId, Snowflake moderatorId, Optional<string> reason)
{
if (!DiscordSnowflake.TryParse(targetId, out var targetUser))
throw new CataloggerError("Target ID was not a valid snowflake");
_kicks[(guildId, targetUser.Value)] = new ModactionData(moderatorId, reason.OrDefault());
}
public record struct ModactionData(Snowflake ModeratorId, string? Reason);
}

View file

@ -19,7 +19,7 @@ public class InMemoryMemberCache : IMemberCache
public Task SetAsync(Snowflake guildId, IGuildMember member)
{
if (!member.User.IsDefined())
throw new CataloggerError("Member with undefined User passed to RedisMemberCache.SetAsync");
throw new CataloggerError("Member with undefined User passed to InMemoryMemberCache.SetAsync");
_members[(guildId, member.User.Value.ID)] = member;
return Task.CompletedTask;
}
@ -30,6 +30,12 @@ public class InMemoryMemberCache : IMemberCache
await SetAsync(guildId, member);
}
public Task RemoveAsync(Snowflake guildId, Snowflake userId)
{
_members.Remove((guildId, userId), out _);
return Task.CompletedTask;
}
public Task<bool> IsGuildCachedAsync(Snowflake guildId) => Task.FromResult(_guilds.ContainsKey(guildId));
public Task MarkAsCachedAsync(Snowflake guildId)

View file

@ -31,6 +31,9 @@ public class RedisMemberCache(RedisService redisService) : IMemberCache
await redisService.SetHashAsync(GuildMembersKey(guildId), redisMembers, m => m.User.Id.ToString());
}
public async Task RemoveAsync(Snowflake guildId, Snowflake userId) =>
await redisService.GetDatabase().HashDeleteAsync(GuildMembersKey(guildId), userId.ToString());
public async Task<bool> IsGuildCachedAsync(Snowflake guildId) =>
await redisService.GetDatabase().SetContainsAsync(GuildCacheKey, guildId.ToString());