internal CachedMember(CachedSharedUser user, CachedGuild guild, MemberModel model) : base(user)
{
SharedUser = user;
Guild = guild;
JoinedAt = model.JoinedAt;
IsMuted = model.Mute;
IsDeafened = model.Deaf;
Update(model);
}
internal CachedMember(CachedSharedUser user, CachedGuild guild, MemberModel model) : base(user)
{
_roles = Extensions.CreateConcurrentDictionary<Snowflake, CachedRole>(model.Roles.Value.Length);
Roles = new ReadOnlyDictionary<Snowflake, CachedRole>(_roles);
SharedUser = user;
Guild = guild;
JoinedAt = model.JoinedAt;
IsMuted = model.Mute;
IsDeafened = model.Deaf;
Update(model);
}
internal CachedCurrentUser(CachedSharedUser user, UserModel model, int relationshipCount, int noteCount) : base(user)
{
SharedUser = user;
if (!Client.IsBot)
{
_relationships = new LockedDictionary <Snowflake, CachedRelationship>(relationshipCount);
_notes = new LockedDictionary <Snowflake, string>(noteCount);
}
Update(model);
}
internal CachedCurrentUser(CachedSharedUser user, UserModel model, int relationshipCount, int noteCount) : base(user)
{
SharedUser = user;
if (Client.TokenType != TokenType.Bot)
{
_relationships = Extensions.CreateConcurrentDictionary <Snowflake, CachedRelationship>(relationshipCount);
_relationshipsWrapper = new ReadOnlyDictionary <Snowflake, CachedRelationship>(_relationships);
_notes = Extensions.CreateConcurrentDictionary <Snowflake, string>(noteCount);
_notesWrapper = new ReadOnlyDictionary <Snowflake, string>(_notes);
}
Update(model);
}
public Task HandleReadyAsync(ReadyModel model)
{
if (_currentUser == null)
{
_client.RestClient.CurrentUser.Value = new RestCurrentUser(_client.RestClient, model.User);
var sharedUser = new CachedSharedUser(_client, model.User);
_currentUser = new CachedCurrentUser(sharedUser, model.User);
sharedUser.References++;
_users.TryAdd(model.User.Id, _currentUser.SharedUser);
}
else
{
_client.RestClient.CurrentUser.Value.Update(model.User);
_currentUser.Update(model.User);
}
// TODO: this won't work for the sharder
//// TODO: more, more, more stale checking
//// I can't remember what I was supposed to be checking though
//foreach (var guild in _guilds.Values)
//{
// if (_client.IsBot)
// {
// if (guild.IsLarge)
// {
// guild.ChunksExpected = (int) Math.Ceiling(guild.MemberCount / 1000.0);
// guild.ChunkTcs = new TaskCompletionSource<bool>();
// }
// }
// else
// {
// guild.SyncTcs = new TaskCompletionSource<bool>();
// }
// var found = false;
// for (var i = 0; i < model.Guilds.Length; i++)
// {
// if (guild.Id == model.Guilds[i].Id)
// {
// found = true;
// break;
// }
// }
// if (!found)
// _guilds.TryRemove(guild.Id, out _);
//}
return(Task.CompletedTask);
}
internal CachedUser(CachedSharedUser sharedUser) : base(sharedUser.Client, sharedUser.Id)
{
IsBot = sharedUser.IsBot;
}
internal CachedCurrentUser(CachedSharedUser user, UserModel model) : base(user)
{
SharedUser = user;
Update(model);
}
public Task HandleReadyAsync(ReadyModel model)
{
if (_currentUser == null)
{
_client.RestClient.CurrentUser.Value = new RestCurrentUser(_client.RestClient, model.User);
var sharedUser = new CachedSharedUser(_client, model.User);
_currentUser = new CachedCurrentUser(sharedUser, model.User, model.Relationships?.Length ?? 0, model.Notes?.Count ?? 0);
sharedUser.References++;
_users.TryAdd(model.User.Id, _currentUser.SharedUser);
}
else
{
_client.RestClient.CurrentUser.Value.Update(model.User);
_currentUser.Update(model.User);
}
// TODO: this won't work for the sharder
//// TODO: more, more, more stale checking
//// I can't remember what I was supposed to be checking though
//foreach (var guild in _guilds.Values)
//{
// if (_client.IsBot)
// {
// if (guild.IsLarge)
// {
// guild.ChunksExpected = (int) Math.Ceiling(guild.MemberCount / 1000.0);
// guild.ChunkTcs = new TaskCompletionSource<bool>();
// }
// }
// else
// {
// guild.SyncTcs = new TaskCompletionSource<bool>();
// }
// var found = false;
// for (var i = 0; i < model.Guilds.Length; i++)
// {
// if (guild.Id == model.Guilds[i].Id)
// {
// found = true;
// break;
// }
// }
// if (!found)
// _guilds.TryRemove(guild.Id, out _);
//}
if (!_client.IsBot)
{
for (var i = 0; i < model.Guilds.Length; i++)
{
var guildModel = model.Guilds[i];
if (guildModel.Unavailable.HasValue && guildModel.Unavailable.Value)
{
Log(LogMessageSeverity.Information, $"Guild {guildModel.Id} is unavailable.");
continue;
}
_guilds.AddOrUpdate(guildModel.Id, _ => new CachedGuild(_client, guildModel), (_, old) =>
{
old.Update(guildModel);
return(old);
});
}
if (model.Guilds.Length != _guilds.Count)
{
foreach (var key in _guilds.Keys)
{
var found = false;
for (var i = 0; i < model.Guilds.Length; i++)
{
if (key == model.Guilds[i].Id)
{
found = true;
break;
}
}
if (!found)
{
_guilds.TryRemove(key, out _);
}
}
}
_currentUser.Update(model.Relationships);
_currentUser.Update(model.Notes);
for (var i = 0; i < model.PrivateChannels.Length; i++)
{
var channelModel = model.PrivateChannels[i];
_privateChannels.AddOrUpdate(channelModel.Id, _ => CachedPrivateChannel.Create(_client, channelModel), (_, old) =>
{
old.Update(channelModel);
return(old);
});
}
if (model.PrivateChannels.Length != _privateChannels.Count)
{
foreach (var key in _privateChannels.Keys)
{
var found = false;
for (var i = 0; i < model.PrivateChannels.Length; i++)
{
if (key == model.PrivateChannels[i].Id)
{
found = true;
break;
}
}
if (!found)
{
_privateChannels.TryRemove(key, out _);
}
}
}
return(_client.GetGateway(0).SendGuildSyncAsync(_guilds.Keys.Select(x => x.RawValue)));
}
return(Task.CompletedTask);
}
