87 lines
2.7 KiB
C#
87 lines
2.7 KiB
C#
// Copyright (C) 2021-present sam (starshines.gay)
|
|
//
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Affero General Public License as published
|
|
// by the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU Affero General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Affero General Public License
|
|
// along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
using System.Collections.Concurrent;
|
|
using System.Diagnostics.CodeAnalysis;
|
|
using Remora.Discord.API.Abstractions.Objects;
|
|
using Remora.Rest.Core;
|
|
|
|
namespace Catalogger.Backend.Cache.InMemoryCache;
|
|
|
|
public class RoleCache
|
|
{
|
|
private readonly ConcurrentDictionary<Snowflake, IRole> _roles = new();
|
|
private readonly ConcurrentDictionary<Snowflake, HashSet<Snowflake>> _guildRoles = new();
|
|
|
|
public int Size => _roles.Count;
|
|
|
|
public void Set(IRole role, Snowflake guildId)
|
|
{
|
|
_roles[role.ID] = role;
|
|
// Add to set of guild channels
|
|
_guildRoles.AddOrUpdate(
|
|
guildId,
|
|
_ => [role.ID],
|
|
(_, l) =>
|
|
{
|
|
l.Add(role.ID);
|
|
return l;
|
|
}
|
|
);
|
|
}
|
|
|
|
public bool TryGet(Snowflake id, [NotNullWhen(true)] out IRole? role) =>
|
|
_roles.TryGetValue(id, out role);
|
|
|
|
public void Remove(Snowflake guildId, Snowflake id, out IRole? role)
|
|
{
|
|
_roles.Remove(id, out role);
|
|
// Remove from set of guild channels
|
|
_guildRoles.AddOrUpdate(
|
|
guildId,
|
|
_ => [],
|
|
(_, s) =>
|
|
{
|
|
s.Remove(id);
|
|
return s;
|
|
}
|
|
);
|
|
}
|
|
|
|
public void RemoveGuild(Snowflake guildId)
|
|
{
|
|
if (!_guildRoles.TryGetValue(guildId, out var roleIds))
|
|
return;
|
|
foreach (var id in roleIds)
|
|
{
|
|
_roles.Remove(id, out _);
|
|
}
|
|
|
|
_guildRoles.Remove(guildId, out _);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Gets all of a guild's cached roles.
|
|
/// </summary>
|
|
/// <param name="guildId">The guild to get the roles of</param>
|
|
/// <returns>A list of cached roles</returns>
|
|
public IEnumerable<IRole> GuildRoles(Snowflake guildId) =>
|
|
!_guildRoles.TryGetValue(guildId, out var roleIds)
|
|
? []
|
|
: roleIds
|
|
.Select(id => _roles.GetValueOrDefault(id))
|
|
.Where(r => r != null)
|
|
.Select(r => r!);
|
|
}
|