public override ValueTask <ChannelCreatedEventArgs> HandleDispatchAsync(IGatewayApiClient shard, ChannelJsonModel model)
{
if (!model.GuildId.HasValue)
{
return(new(result : null));
}
IGuildChannel channel;
if (CacheProvider.TryGetChannels(model.GuildId.Value, out var cache))
{
channel = CachedGuildChannel.Create(Client, model);
cache.Add(channel.Id, channel as CachedGuildChannel);
}
else
{
channel = TransientGuildChannel.Create(Client, model);
}
var e = new ChannelCreatedEventArgs(channel);
return(new(e));
}
// 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);
}
public override async ValueTask <EventArgs> HandleDispatchAsync(IGatewayApiClient shard, GatewayGuildJsonModel model)
{
IGatewayGuild guild = null;
// Check if the event is guild availability or we joined a new guild.
if (model.Unavailable.HasValue)
{
// A guild became available.
var isPending = _readyHandler.IsPendingGuild(shard.Id, model.Id);
try
{
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);
}
}
if (guild == null)
{
guild = new TransientGatewayGuild(Client, model);
}
// TODO: optimise member cache retrieval
if (CacheProvider.TryGetMembers(model.Id, out var memberCache))
{
foreach (var memberModel in model.Members)
{
Dispatcher.GetOrAddMember(model.Id, memberModel);
}
}
if (CacheProvider.TryGetChannels(model.Id, out var channelCache))
{
foreach (var channelModel in model.Channels)
{
if (isPending)
{
var channel = CachedGuildChannel.Create(Client, model.Id, channelModel);
channelCache.Add(channel.Id, channel);
}
else
{
var channel = channelCache.GetValueOrDefault(channelModel.Id);
channel?.Update(channelModel);
}
}
}
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.Id, voiceState);
}
else
{
var voiceState = voiceStateCache.GetValueOrDefault(voiceStateModel.UserId);
voiceState?.Update(voiceStateModel);
}
}
}
var logLevel = isPending
? LogLevel.Debug
: LogLevel.Information;
var message = isPending
? "Pending guild {0} ({1}) is available."
: "Guild {0} ({1}) became available.";
shard.Logger.Log(logLevel, message, guild.Name, guild.Id.RawValue);
// Invoke the event and possibly invoke ready afterwards.
await InvokeEventAsync(new GuildAvailableEventArgs(guild)).ConfigureAwait(false);
}
finally
{
if (isPending)
{
_readyHandler.PopPendingGuild(shard.Id, model.Id);
}
}
return(null);
}
else
{
// We joined a new guild.
if (Client.CacheProvider.TryGetGuilds(out var cache))
{
guild = new CachedGuild(Client, model);
cache.Add(model.Id, guild as CachedGuild);
}
else
{
guild = new TransientGatewayGuild(Client, model);
}
// TODO: optimise member cache retrieval
if (CacheProvider.TryGetMembers(model.Id, out var memberCache))
{
foreach (var memberModel in model.Members)
{
Dispatcher.GetOrAddMember(model.Id, memberModel);
}
}
if (CacheProvider.TryGetChannels(model.Id, out var channelCache))
{
foreach (var channelModel in model.Channels)
{
var channel = CachedGuildChannel.Create(Client, model.Id, channelModel);
channelCache.Add(channel.Id, channel);
}
}
if (CacheProvider.TryGetRoles(model.Id, out var roleCache))
{
foreach (var roleModel in model.Roles)
{
var role = new CachedRole(Client, model.Id, roleModel);
roleCache.Add(role.Id, role);
}
}
shard.Logger.LogInformation("Joined guild {0} ({1}).", guild.Name, guild.Id.RawValue);
return(new JoinedGuildEventArgs(guild));
}
}