public override ValueTask <CheckResult> CheckAsync(DiscordGuildCommandContext context)
{
if (context.Channel is null)
{
throw new InvalidOperationException($"{nameof(RequireNsfwAttribute)} requires the context channel.");
}
var isNsfw = context.Channel switch
{
CachedTextChannel textChannel => textChannel.IsNsfw,
CachedThreadChannel threadChannel => threadChannel.GetChannel()?.IsNsfw ?? false,
_ => false
};
if (isNsfw)
{
return(Success());
}
return(Failure("This can only be executed in a NSFW channel."));
}
}
// TODO: possible cache inconsistencies on unavailable guilds? No idea what Discord sends nor does while it's unavailable.
private IGatewayGuild UpdateCache(GatewayGuildJsonModel model, bool isPending)
{
IGatewayGuild guild = null;
if (CacheProvider.TryGetGuilds(out var guildCache))
{
if (isPending)
{
guild = new CachedGuild(Client, model);
guildCache.Add(model.Id, guild as CachedGuild);
}
else
{
guild = guildCache.GetValueOrDefault(model.Id);
guild?.Update(model);
}
}
guild ??= new TransientGatewayGuild(Client, model);
if (CacheProvider.TryGetUsers(out var userCache) && CacheProvider.TryGetMembers(model.Id, out var memberCache))
{
foreach (var memberModel in model.Members)
{
Dispatcher.GetOrAddMember(userCache, memberCache, model.Id, memberModel);
}
}
if (CacheProvider.TryGetChannels(model.Id, out var channelCache))
{
foreach (var channelModel in model.Channels)
{
if (isPending)
{
channelModel.GuildId = model.Id;
var channel = CachedGuildChannel.Create(Client, channelModel);
channelCache.Add(channel.Id, channel);
}
else
{
var channel = channelCache.GetValueOrDefault(channelModel.Id);
channel?.Update(channelModel);
}
}
foreach (var threadModel in model.Threads)
{
if (isPending)
{
threadModel.GuildId = model.Id;
if (threadModel.Member.HasValue)
{
threadModel.Member.Value.Id = threadModel.Id;
threadModel.Member.Value.UserId = Client.CurrentUser.Id;
}
var channel = new CachedThreadChannel(Client, threadModel);
channelCache.Add(channel.Id, channel);
}
else
{
var channel = channelCache.GetValueOrDefault(threadModel.Id);
channel?.Update(threadModel);
}
}
}
if (CacheProvider.TryGetRoles(model.Id, out var roleCache))
{
foreach (var roleModel in model.Roles)
{
if (isPending)
{
var role = new CachedRole(Client, model.Id, roleModel);
roleCache.Add(role.Id, role);
}
else
{
var role = roleCache.GetValueOrDefault(roleModel.Id);
role?.Update(roleModel);
}
}
}
if (CacheProvider.TryGetVoiceStates(model.Id, out var voiceStateCache))
{
foreach (var voiceStateModel in model.VoiceStates)
{
if (isPending)
{
var voiceState = new CachedVoiceState(Client, model.Id, voiceStateModel);
voiceStateCache.Add(voiceState.MemberId, voiceState);
}
else
{
var voiceState = voiceStateCache.GetValueOrDefault(voiceStateModel.UserId);
voiceState?.Update(voiceStateModel);
}
}
}
if (CacheProvider.TryGetPresences(model.Id, out var presenceCache))
{
foreach (var presenceModel in model.Presences)
{
if (isPending)
{
var presence = new CachedPresence(Client, presenceModel);
presenceCache.Add(presence.MemberId, presence);
}
else
{
var presence = presenceCache.GetValueOrDefault(presenceModel.User.Id);
presence?.Update(presenceModel);
}
}
}
if (CacheProvider.TryGetStages(model.Id, out var stageCache))
{
foreach (var stageModel in model.StageInstances)
{
if (isPending)
{
var stage = new CachedStage(Client, stageModel);
stageCache.Add(stage.Id, stage);
}
else
{
var stage = stageCache.GetValueOrDefault(stageModel.Id);
stage?.Update(stageModel);
}
}
}
return(guild);
}
static void UncacheThread(ISynchronizedDictionary <Snowflake, CachedGuildChannel> channelCache, Dictionary <Snowflake, IReadOnlyList <CachedThreadChannel> > uncachedThreads, KeyValuePair <Snowflake, List <ChannelJsonModel> > kvp, CachedThreadChannel cachedThreadChannel)
{
channelCache.Remove(cachedThreadChannel.Id);
List <CachedThreadChannel> list;
if (uncachedThreads.TryGetValue(kvp.Key, out var boxedList))
{
list = boxedList as List <CachedThreadChannel>;
}
else
{
uncachedThreads.Add(kvp.Key, list = new List <CachedThreadChannel>());
}
list.Add(cachedThreadChannel);
}
