void HandleGuildMemberUpdateEvent(DiscordApiData data)
{
Snowflake guildId = data.GetSnowflake("guild_id").Value;
DiscordApiData userData = data.Get("user");
Snowflake userId = userData.GetSnowflake("id").Value;
// Get user
MutableUser mutableUser;
if (!cache.Users.TryGetValue(userId, out mutableUser))
{
mutableUser = new MutableUser(userId, false, http);
cache.Users[userId] = mutableUser;
}
mutableUser.Update(userData);
// Get member
if (cache.GuildMembers.TryGetValue(guildId, userId, out MutableGuildMember mutableMember))
{
// Update member
mutableMember.PartialUpdate(data);
// Fire event
OnGuildMemberUpdated?.Invoke(this, new GuildMemberEventArgs(shard, guildId, mutableMember.ImmutableEntity));
}
// It is technically valid for the member to not exist here, especially if the guild is considered large.
}
void HandleGuildMemberAddEvent(DiscordApiData data)
{
Snowflake guildId = data.GetSnowflake("guild_id").Value;
DiscordApiData userData = data.Get("user");
Snowflake userId = userData.GetSnowflake("id").Value;
// Get user
MutableUser mutableUser;
if (!cache.Users.TryGetValue(userId, out mutableUser))
{
mutableUser = new MutableUser(userId, false, http);
cache.Users[userId] = mutableUser;
}
// Update user
mutableUser.Update(userData);
// Get or create member
MutableGuildMember mutableMember;
if (!cache.GuildMembers.TryGetValue(guildId, userId, out mutableMember))
{
mutableMember = new MutableGuildMember(mutableUser, guildId, http);
cache.GuildMembers[guildId, userId] = mutableMember;
}
// Update member
mutableMember.Update(data);
// Fire event
OnGuildMemberAdded?.Invoke(this, new GuildMemberEventArgs(shard, guildId, mutableMember.ImmutableEntity));
}
void HandleGuildMemberRemoveEvent(DiscordApiData data)
{
Snowflake guildId = data.GetSnowflake("guild_id").Value;
DiscordApiData userData = data.Get("user");
Snowflake userId = userData.GetSnowflake("id").Value;
// Get user
MutableUser mutableUser;
if (!cache.Users.TryGetValue(userId, out mutableUser))
{
mutableUser = new MutableUser(userId, false, http);
cache.Users[userId] = mutableUser;
}
mutableUser.Update(userData);
// Get and remove member
if (cache.GuildMembers.TryRemove(guildId, userId, out MutableGuildMember mutableMember))
{
// Ensure all references are removed
mutableMember.ClearReferences();
// Fire event
OnGuildMemberRemoved?.Invoke(this, new GuildMemberEventArgs(shard, guildId, mutableMember.ImmutableEntity));
}
}
void HandleChannelCreateEvent(DiscordApiData data)
{
Snowflake id = data.GetSnowflake("id").Value;
DiscordChannelType type = (DiscordChannelType)data.GetInteger("type").Value;
if (type == DiscordChannelType.DirectMessage)
{
// DM channel
DiscordApiData recipientData = data.GetArray("recipients").First();
Snowflake recipientId = recipientData.GetSnowflake("id").Value;
MutableUser recipient;
if (!cache.Users.TryGetValue(recipientId, out recipient))
{
recipient = new MutableUser(recipientId, false, http);
cache.Users[recipientId] = recipient;
}
recipient.Update(recipientData);
MutableDMChannel mutableDMChannel;
if (!cache.DMChannels.TryGetValue(id, out mutableDMChannel))
{
mutableDMChannel = new MutableDMChannel(id, recipient, http);
cache.DMChannels[id] = mutableDMChannel;
}
OnDMChannelCreated?.Invoke(this, new DMChannelEventArgs(shard, mutableDMChannel.ImmutableEntity));
}
else if (type == DiscordChannelType.GuildText || type == DiscordChannelType.GuildVoice)
{
// Guild channel
Snowflake guildId = data.GetSnowflake("guild_id").Value;
DiscordGuildChannel channel;
if (type == DiscordChannelType.GuildText)
{
channel = new DiscordGuildTextChannel(http, data, guildId);
}
else if (type == DiscordChannelType.GuildVoice)
{
channel = new DiscordGuildVoiceChannel(http, data, guildId);
}
else if (type == DiscordChannelType.GuildCategory)
{
channel = new DiscordGuildCategoryChannel(http, data, guildId);
}
else
{
throw new NotImplementedException($"Guild channel type \"{type}\" has no implementation!");
}
cache.GuildChannels[id] = channel;
OnGuildChannelCreated?.Invoke(this, new GuildChannelEventArgs(shard, guildId, channel));
}
}
void HandleUserUpdateEvent(DiscordApiData data)
{
Snowflake userId = data.GetSnowflake("id").Value;
MutableUser mutableUser;
if (!cache.Users.TryGetValue(userId, out mutableUser))
{
mutableUser = new MutableUser(userId, false, http);
cache.Users[userId] = mutableUser;
}
mutableUser.Update(data);
OnUserUpdated?.Invoke(this, new UserEventArgs(shard, mutableUser.ImmutableEntity));
}
void HandleMessageUpdateEvent(DiscordApiData data)
{
// Get author
DiscordApiData authorData = data.Get("author");
if (authorData != null)
{
Snowflake authorId = authorData.GetSnowflake("id").Value;
bool isWebhookUser = !string.IsNullOrWhiteSpace(data.GetString("webhook_id"));
MutableUser mutableAuthor;
if (!cache.Users.TryGetValue(authorId, out mutableAuthor))
{
mutableAuthor = new MutableUser(authorId, isWebhookUser, http);
cache.Users[authorId] = mutableAuthor;
}
mutableAuthor.Update(authorData);
}
// Get mentioned users
IList <DiscordApiData> mentionsArray = data.GetArray("mentions");
if (mentionsArray != null)
{
for (int i = 0; i < mentionsArray.Count; i++)
{
DiscordApiData userData = mentionsArray[i];
Snowflake userId = userData.GetSnowflake("id").Value;
MutableUser mutableUser;
if (!cache.Users.TryGetValue(userId, out mutableUser))
{
mutableUser = new MutableUser(userId, false, http);
cache.Users[userId] = mutableUser;
}
mutableUser.Update(userData);
}
}
// Create message
DiscordMessage message = new DiscordMessage(http, data);
OnMessageUpdated?.Invoke(this, new MessageUpdateEventArgs(shard, message));
}
internal DiscordUser(MutableUser user)
{
Id = user.Id;
IsWebhookUser = user.IsWebhookUser;
Username = user.Username;
Discriminator = user.Discriminator;
IsBot = user.IsBot;
HasTwoFactorAuth = user.HasTwoFactorAuth;
IsVerified = user.IsVerified;
Email = user.Email;
if (user.Avatar != null)
{
Avatar = DiscordCdnUrl.ForUserAvatar(user.Id, user.Avatar);
}
}
void HandleGuildBanRemoveEvent(DiscordApiData data)
{
Snowflake guildId = data.GetSnowflake("guild_id").Value;
DiscordApiData userData = data.Get("user");
Snowflake userId = userData.GetSnowflake("id").Value;
MutableUser mutableUser;
if (!cache.Users.TryGetValue(userId, out mutableUser))
{
mutableUser = new MutableUser(userId, false, http);
cache.Users[userId] = mutableUser;
}
mutableUser.Update(userData);
OnGuildBanRemoved?.Invoke(this, new GuildUserEventArgs(shard, guildId, mutableUser.ImmutableEntity));
}
void HandleGuildMembersChunkEvent(DiscordApiData data)
{
Snowflake guildId = data.GetSnowflake("guild_id").Value;
// Get every member and ensure they are cached
IList <DiscordApiData> membersData = data.GetArray("members");
DiscordGuildMember[] members = new DiscordGuildMember[membersData.Count];
for (int i = 0; i < membersData.Count; i++)
{
DiscordApiData memberData = membersData[i];
DiscordApiData userData = memberData.Get("user");
Snowflake userId = userData.GetSnowflake("id").Value;
// Get user
MutableUser mutableUser;
if (!cache.Users.TryGetValue(userId, out mutableUser))
{
mutableUser = new MutableUser(userId, false, http);
cache.Users[userId] = mutableUser;
}
mutableUser.Update(userData);
// Get or create member
MutableGuildMember mutableMember;
if (!cache.GuildMembers.TryGetValue(guildId, userId, out mutableMember))
{
mutableMember = new MutableGuildMember(mutableUser, guildId, http);
mutableMember.Update(memberData);
cache.GuildMembers[guildId, userId] = mutableMember;
}
members[i] = mutableMember.ImmutableEntity;
}
// Fire event
OnGuildMembersChunk?.Invoke(this, new GuildMemberChunkEventArgs(shard, guildId, members));
}
void HandleReadyEvent(DiscordApiData data)
{
// Check gateway protocol
int protocolVersion = data.GetInteger("v").Value;
if (protocolVersion != GATEWAY_VERSION)
{
log.LogError($"[Ready] Gateway protocol mismatch! Expected v{GATEWAY_VERSION}, got {protocolVersion}.");
}
// Clear the cache
cache.Clear();
// Get the current bot's user object
DiscordApiData userData = data.Get("user");
Snowflake userId = userData.GetSnowflake("id").Value;
MutableUser user;
if (!cache.Users.TryGetValue(userId, out user))
{
user = new MutableUser(userId, false, http);
cache.Users[userId] = user;
}
user.Update(userData);
shard.UserId = userId;
log.LogInfo($"[Ready] user = {user.Username}#{user.Discriminator}");
// Get session ID
sessionId = data.GetString("session_id");
// Get unavailable guilds
foreach (DiscordApiData unavailableGuildData in data.GetArray("guilds"))
{
Snowflake guildId = unavailableGuildData.GetSnowflake("id").Value;
cache.AddGuildId(guildId);
cache.SetGuildAvailability(guildId, false);
}
// Get DM channels
foreach (DiscordApiData dmChannelData in data.GetArray("private_channels"))
{
DiscordChannelType type = (DiscordChannelType)dmChannelData.GetInteger("type").Value;
if (type != DiscordChannelType.DirectMessage)
{
// TODO: Support group DMs
continue;
}
Snowflake channelId = dmChannelData.GetSnowflake("id").Value;
DiscordApiData recipientData = dmChannelData.GetArray("recipients").First();
Snowflake recipientId = recipientData.GetSnowflake("id").Value;
MutableUser recipient;
if (!cache.Users.TryGetValue(recipientId, out recipient))
{
recipient = new MutableUser(recipientId, false, http);
cache.Users[recipientId] = recipient;
}
recipient.Update(recipientData);
MutableDMChannel mutableDMChannel;
if (!cache.DMChannels.TryGetValue(channelId, out mutableDMChannel))
{
mutableDMChannel = new MutableDMChannel(channelId, recipient, http);
cache.DMChannels[channelId] = mutableDMChannel;
}
mutableDMChannel.Update(dmChannelData);
}
LogServerTrace("Ready", data);
// Signal that the connection is ready
handshakeCompleteEvent.Set();
}
void HandleGuildCreateEvent(DiscordApiData data)
{
Snowflake guildId = data.GetSnowflake("id").Value;
bool wasUnavailable = !cache.IsGuildAvailable(guildId);
// Update guild
MutableGuild mutableGuild;
if (!cache.Guilds.TryGetValue(guildId, out mutableGuild))
{
mutableGuild = new MutableGuild(guildId, http);
cache.Guilds[guildId] = mutableGuild;
}
mutableGuild.Update(data);
// Ensure the cache guildId list contains this guild (it uses a hashset so don't worry about duplicates).
cache.AddGuildId(guildId);
// GUILD_CREATE specifics
// Update metadata
cache.GuildMetadata[guildId] = new DiscordGuildMetadata(data);
// Deserialize members
cache.GuildMembers.Clear(guildId);
IList <DiscordApiData> membersArray = data.GetArray("members");
for (int i = 0; i < membersArray.Count; i++)
{
DiscordApiData memberData = membersArray[i];
DiscordApiData userData = memberData.Get("user");
Snowflake userId = userData.GetSnowflake("id").Value;
MutableUser user;
if (!cache.Users.TryGetValue(userId, out user))
{
user = new MutableUser(userId, false, http);
cache.Users[userId] = user;
}
user.Update(userData);
MutableGuildMember member;
if (!cache.GuildMembers.TryGetValue(guildId, userId, out member))
{
member = new MutableGuildMember(user, guildId, http);
cache.GuildMembers[guildId, userId] = member;
}
member.Update(memberData);
}
// Deserialize channels
cache.ClearGuildChannels(guildId);
IList <DiscordApiData> channelsArray = data.GetArray("channels");
for (int i = 0; i < channelsArray.Count; i++)
{
DiscordApiData channelData = channelsArray[i];
DiscordChannelType channelType = (DiscordChannelType)channelData.GetInteger("type");
DiscordGuildChannel channel = null;
if (channelType == DiscordChannelType.GuildText)
{
channel = new DiscordGuildTextChannel(http, channelData, guildId);
}
else if (channelType == DiscordChannelType.GuildVoice)
{
channel = new DiscordGuildVoiceChannel(http, channelData, guildId);
}
else if (channelType == DiscordChannelType.GuildCategory)
{
channel = new DiscordGuildCategoryChannel(http, channelData, guildId);
}
if (channel != null)
{
cache.AddGuildChannel(channel);
}
}
// Deserialize voice states
cache.GuildVoiceStates.Clear(guildId);
IList <DiscordApiData> voiceStatesArray = data.GetArray("voice_states");
for (int i = 0; i < voiceStatesArray.Count; i++)
{
DiscordVoiceState state = new DiscordVoiceState(guildId, voiceStatesArray[i]);
UpdateMemberVoiceState(state);
}
// Deserialize presences
cache.GuildPresences.Clear(guildId);
IList <DiscordApiData> presencesArray = data.GetArray("presences");
for (int i = 0; i < presencesArray.Count; i++)
{
// Presence's in GUILD_CREATE do not contain full user objects,
// so don't attempt to update them here.
DiscordApiData presenceData = presencesArray[i];
Snowflake userId = presenceData.LocateSnowflake("user.id").Value;
cache.GuildPresences[guildId, userId] = new DiscordUserPresence(userId, presenceData);
}
// Mark the guild as available
cache.SetGuildAvailability(guildId, true);
// Fire event
if (wasUnavailable)
{
OnGuildAvailable?.Invoke(this, new GuildEventArgs(shard, mutableGuild.ImmutableEntity));
}
else
{
OnGuildCreated?.Invoke(this, new GuildEventArgs(shard, mutableGuild.ImmutableEntity));
}
}
void HandleReadyEvent(DiscordApiData data)
{
// Check gateway protocol
int protocolVersion = data.GetInteger("v").Value;
if (protocolVersion != GATEWAY_VERSION)
{
log.LogError($"[Ready] Gateway protocol mismatch! Expected v{GATEWAY_VERSION}, got {protocolVersion}.");
}
// Check shard
if (shard.Id != 0 || totalShards > 1)
{
IList <DiscordApiData> shardData = data.GetArray("shard");
if (shardData != null)
{
if (shardData.Count > 0 && shardData[0].ToInteger() != shard.Id)
{
log.LogError($"[Ready] Shard ID mismatch! Expected {shard.Id}, got {shardData[0].ToInteger()}");
}
if (shardData.Count > 1 && shardData[1].ToInteger() != totalShards)
{
log.LogError($"[Ready] Total shards mismatch! Expected {totalShards}, got {shardData[1].ToInteger()}");
}
}
}
// Clear the cache
cache.Clear();
// Get the current bot's user object
DiscordApiData userData = data.Get("user");
Snowflake userId = userData.GetSnowflake("id").Value;
MutableUser user;
if (!cache.Users.TryGetValue(userId, out user))
{
user = new MutableUser(userId, false, http);
cache.Users[userId] = user;
}
user.Update(userData);
shard.UserId = userId;
log.LogInfo($"[Ready] user = {user.Username}#{user.Discriminator}");
// Get session ID
sessionId = data.GetString("session_id");
// Get unavailable guilds
foreach (DiscordApiData unavailableGuildData in data.GetArray("guilds"))
{
Snowflake guildId = unavailableGuildData.GetSnowflake("id").Value;
cache.AddGuildId(guildId);
cache.SetGuildAvailability(guildId, false);
}
LogServerTrace("Ready", data);
// Signal that the connection is ready
handshakeCompleteEvent.Set();
}
