You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
2008 lines
108 KiB
2008 lines
108 KiB
using Discord.API;
|
|
using Discord.API.Gateway;
|
|
using Discord.Logging;
|
|
using Discord.Net.Converters;
|
|
using Discord.Net.Udp;
|
|
using Discord.Net.WebSockets;
|
|
using Discord.Rest;
|
|
using Newtonsoft.Json;
|
|
using Newtonsoft.Json.Linq;
|
|
using System;
|
|
using System.Collections.Concurrent;
|
|
using System.Collections.Generic;
|
|
using System.Collections.Immutable;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
using GameModel = Discord.API.Game;
|
|
|
|
namespace Discord.WebSocket
|
|
{
|
|
/// <summary>
|
|
/// Represents a WebSocket-based Discord client.
|
|
/// </summary>
|
|
public partial class DiscordSocketClient : BaseSocketClient, IDiscordClient
|
|
{
|
|
private readonly ConcurrentQueue<ulong> _largeGuilds;
|
|
private readonly JsonSerializer _serializer;
|
|
private readonly SemaphoreSlim _connectionGroupLock;
|
|
private readonly DiscordSocketClient _parentClient;
|
|
private readonly ConcurrentQueue<long> _heartbeatTimes;
|
|
private readonly ConnectionManager _connection;
|
|
private readonly Logger _gatewayLogger;
|
|
private readonly SemaphoreSlim _stateLock;
|
|
|
|
private string _sessionId;
|
|
private int _lastSeq;
|
|
private ImmutableDictionary<string, RestVoiceRegion> _voiceRegions;
|
|
private Task _heartbeatTask, _guildDownloadTask;
|
|
private int _unavailableGuildCount;
|
|
private long _lastGuildAvailableTime, _lastMessageTime;
|
|
private int _nextAudioId;
|
|
private DateTimeOffset? _statusSince;
|
|
private RestApplication _applicationInfo;
|
|
private bool _isDisposed;
|
|
private bool _guildSubscriptions;
|
|
|
|
/// <summary>
|
|
/// Provides access to a REST-only client with a shared state from this client.
|
|
/// </summary>
|
|
public override DiscordSocketRestClient Rest { get; }
|
|
/// <summary> Gets the shard of of this client. </summary>
|
|
public int ShardId { get; }
|
|
/// <summary> Gets the current connection state of this client. </summary>
|
|
public ConnectionState ConnectionState => _connection.State;
|
|
/// <inheritdoc />
|
|
public override int Latency { get; protected set; }
|
|
/// <inheritdoc />
|
|
public override UserStatus Status { get; protected set; } = UserStatus.Online;
|
|
/// <inheritdoc />
|
|
public override IActivity Activity { get; protected set; }
|
|
|
|
//From DiscordSocketConfig
|
|
internal int TotalShards { get; private set; }
|
|
internal int MessageCacheSize { get; private set; }
|
|
internal int LargeThreshold { get; private set; }
|
|
internal ClientState State { get; private set; }
|
|
internal UdpSocketProvider UdpSocketProvider { get; private set; }
|
|
internal WebSocketProvider WebSocketProvider { get; private set; }
|
|
internal bool AlwaysDownloadUsers { get; private set; }
|
|
internal int? HandlerTimeout { get; private set; }
|
|
internal bool? ExclusiveBulkDelete { get; private set; }
|
|
|
|
internal new DiscordSocketApiClient ApiClient => base.ApiClient as DiscordSocketApiClient;
|
|
/// <inheritdoc />
|
|
public override IReadOnlyCollection<SocketGuild> Guilds => State.Guilds;
|
|
/// <inheritdoc />
|
|
public override IReadOnlyCollection<ISocketPrivateChannel> PrivateChannels => State.PrivateChannels;
|
|
/// <summary>
|
|
/// Gets a collection of direct message channels opened in this session.
|
|
/// </summary>
|
|
/// <remarks>
|
|
/// This method returns a collection of currently opened direct message channels.
|
|
/// <note type="warning">
|
|
/// This method will not return previously opened DM channels outside of the current session! If you
|
|
/// have just started the client, this may return an empty collection.
|
|
/// </note>
|
|
/// </remarks>
|
|
/// <returns>
|
|
/// A collection of DM channels that have been opened in this session.
|
|
/// </returns>
|
|
public IReadOnlyCollection<SocketDMChannel> DMChannels
|
|
=> State.PrivateChannels.OfType<SocketDMChannel>().ToImmutableArray();
|
|
/// <summary>
|
|
/// Gets a collection of group channels opened in this session.
|
|
/// </summary>
|
|
/// <remarks>
|
|
/// This method returns a collection of currently opened group channels.
|
|
/// <note type="warning">
|
|
/// This method will not return previously opened group channels outside of the current session! If you
|
|
/// have just started the client, this may return an empty collection.
|
|
/// </note>
|
|
/// </remarks>
|
|
/// <returns>
|
|
/// A collection of group channels that have been opened in this session.
|
|
/// </returns>
|
|
public IReadOnlyCollection<SocketGroupChannel> GroupChannels
|
|
=> State.PrivateChannels.OfType<SocketGroupChannel>().ToImmutableArray();
|
|
/// <inheritdoc />
|
|
public override IReadOnlyCollection<RestVoiceRegion> VoiceRegions => _voiceRegions.ToReadOnlyCollection();
|
|
|
|
/// <summary>
|
|
/// Initializes a new REST/WebSocket-based Discord client.
|
|
/// </summary>
|
|
public DiscordSocketClient() : this(new DiscordSocketConfig()) { }
|
|
/// <summary>
|
|
/// Initializes a new REST/WebSocket-based Discord client with the provided configuration.
|
|
/// </summary>
|
|
/// <param name="config">The configuration to be used with the client.</param>
|
|
#pragma warning disable IDISP004
|
|
public DiscordSocketClient(DiscordSocketConfig config) : this(config, CreateApiClient(config), null, null) { }
|
|
internal DiscordSocketClient(DiscordSocketConfig config, SemaphoreSlim groupLock, DiscordSocketClient parentClient) : this(config, CreateApiClient(config), groupLock, parentClient) { }
|
|
#pragma warning restore IDISP004
|
|
private DiscordSocketClient(DiscordSocketConfig config, API.DiscordSocketApiClient client, SemaphoreSlim groupLock, DiscordSocketClient parentClient)
|
|
: base(config, client)
|
|
{
|
|
ShardId = config.ShardId ?? 0;
|
|
TotalShards = config.TotalShards ?? 1;
|
|
MessageCacheSize = config.MessageCacheSize;
|
|
LargeThreshold = config.LargeThreshold;
|
|
UdpSocketProvider = config.UdpSocketProvider;
|
|
WebSocketProvider = config.WebSocketProvider;
|
|
AlwaysDownloadUsers = config.AlwaysDownloadUsers;
|
|
HandlerTimeout = config.HandlerTimeout;
|
|
ExclusiveBulkDelete = config.ExclusiveBulkDelete;
|
|
State = new ClientState(0, 0);
|
|
Rest = new DiscordSocketRestClient(config, ApiClient);
|
|
_heartbeatTimes = new ConcurrentQueue<long>();
|
|
_guildSubscriptions = config.GuildSubscriptions;
|
|
|
|
_stateLock = new SemaphoreSlim(1, 1);
|
|
_gatewayLogger = LogManager.CreateLogger(ShardId == 0 && TotalShards == 1 ? "Gateway" : $"Shard #{ShardId}");
|
|
_connection = new ConnectionManager(_stateLock, _gatewayLogger, config.ConnectionTimeout,
|
|
OnConnectingAsync, OnDisconnectingAsync, x => ApiClient.Disconnected += x);
|
|
_connection.Connected += () => TimedInvokeAsync(_connectedEvent, nameof(Connected));
|
|
_connection.Disconnected += (ex, recon) => TimedInvokeAsync(_disconnectedEvent, nameof(Disconnected), ex);
|
|
|
|
_nextAudioId = 1;
|
|
_connectionGroupLock = groupLock;
|
|
_parentClient = parentClient;
|
|
|
|
_serializer = new JsonSerializer { ContractResolver = new DiscordContractResolver() };
|
|
_serializer.Error += (s, e) =>
|
|
{
|
|
_gatewayLogger.WarningAsync("Serializer Error", e.ErrorContext.Error).GetAwaiter().GetResult();
|
|
e.ErrorContext.Handled = true;
|
|
};
|
|
|
|
ApiClient.SentGatewayMessage += async opCode => await _gatewayLogger.DebugAsync($"Sent {opCode}").ConfigureAwait(false);
|
|
ApiClient.ReceivedGatewayEvent += ProcessMessageAsync;
|
|
|
|
LeftGuild += async g => await _gatewayLogger.InfoAsync($"Left {g.Name}").ConfigureAwait(false);
|
|
JoinedGuild += async g => await _gatewayLogger.InfoAsync($"Joined {g.Name}").ConfigureAwait(false);
|
|
GuildAvailable += async g => await _gatewayLogger.VerboseAsync($"Connected to {g.Name}").ConfigureAwait(false);
|
|
GuildUnavailable += async g => await _gatewayLogger.VerboseAsync($"Disconnected from {g.Name}").ConfigureAwait(false);
|
|
LatencyUpdated += async (old, val) => await _gatewayLogger.DebugAsync($"Latency = {val} ms").ConfigureAwait(false);
|
|
|
|
GuildAvailable += g =>
|
|
{
|
|
if (ConnectionState == ConnectionState.Connected && AlwaysDownloadUsers && !g.HasAllMembers)
|
|
{
|
|
var _ = g.DownloadUsersAsync();
|
|
}
|
|
return Task.Delay(0);
|
|
};
|
|
|
|
_voiceRegions = ImmutableDictionary.Create<string, RestVoiceRegion>();
|
|
_largeGuilds = new ConcurrentQueue<ulong>();
|
|
}
|
|
private static API.DiscordSocketApiClient CreateApiClient(DiscordSocketConfig config)
|
|
=> new API.DiscordSocketApiClient(config.RestClientProvider, config.WebSocketProvider, DiscordRestConfig.UserAgent, config.GatewayHost,
|
|
rateLimitPrecision: config.RateLimitPrecision);
|
|
/// <inheritdoc />
|
|
internal override void Dispose(bool disposing)
|
|
{
|
|
if (!_isDisposed)
|
|
{
|
|
if (disposing)
|
|
{
|
|
StopAsync().GetAwaiter().GetResult();
|
|
ApiClient?.Dispose();
|
|
_stateLock?.Dispose();
|
|
}
|
|
_isDisposed = true;
|
|
}
|
|
|
|
base.Dispose(disposing);
|
|
}
|
|
|
|
/// <inheritdoc />
|
|
internal override async Task OnLoginAsync(TokenType tokenType, string token)
|
|
{
|
|
if (_parentClient == null)
|
|
{
|
|
var voiceRegions = await ApiClient.GetVoiceRegionsAsync(new RequestOptions { IgnoreState = true, RetryMode = RetryMode.AlwaysRetry }).ConfigureAwait(false);
|
|
_voiceRegions = voiceRegions.Select(x => RestVoiceRegion.Create(this, x)).ToImmutableDictionary(x => x.Id);
|
|
}
|
|
else
|
|
_voiceRegions = _parentClient._voiceRegions;
|
|
await Rest.OnLoginAsync(tokenType, token);
|
|
}
|
|
/// <inheritdoc />
|
|
internal override async Task OnLogoutAsync()
|
|
{
|
|
await StopAsync().ConfigureAwait(false);
|
|
_applicationInfo = null;
|
|
_voiceRegions = ImmutableDictionary.Create<string, RestVoiceRegion>();
|
|
await Rest.OnLogoutAsync();
|
|
}
|
|
|
|
/// <inheritdoc />
|
|
public override async Task StartAsync()
|
|
=> await _connection.StartAsync().ConfigureAwait(false);
|
|
/// <inheritdoc />
|
|
public override async Task StopAsync()
|
|
=> await _connection.StopAsync().ConfigureAwait(false);
|
|
|
|
private async Task OnConnectingAsync()
|
|
{
|
|
if (_connectionGroupLock != null)
|
|
await _connectionGroupLock.WaitAsync(_connection.CancelToken).ConfigureAwait(false);
|
|
try
|
|
{
|
|
await _gatewayLogger.DebugAsync("Connecting ApiClient").ConfigureAwait(false);
|
|
await ApiClient.ConnectAsync().ConfigureAwait(false);
|
|
|
|
if (_sessionId != null)
|
|
{
|
|
await _gatewayLogger.DebugAsync("Resuming").ConfigureAwait(false);
|
|
await ApiClient.SendResumeAsync(_sessionId, _lastSeq).ConfigureAwait(false);
|
|
}
|
|
else
|
|
{
|
|
await _gatewayLogger.DebugAsync("Identifying").ConfigureAwait(false);
|
|
await ApiClient.SendIdentifyAsync(shardID: ShardId, totalShards: TotalShards, guildSubscriptions: _guildSubscriptions).ConfigureAwait(false);
|
|
}
|
|
|
|
//Wait for READY
|
|
await _connection.WaitAsync().ConfigureAwait(false);
|
|
|
|
await _gatewayLogger.DebugAsync("Sending Status").ConfigureAwait(false);
|
|
await SendStatusAsync().ConfigureAwait(false);
|
|
}
|
|
finally
|
|
{
|
|
if (_connectionGroupLock != null)
|
|
{
|
|
await Task.Delay(5000).ConfigureAwait(false);
|
|
_connectionGroupLock.Release();
|
|
}
|
|
}
|
|
}
|
|
private async Task OnDisconnectingAsync(Exception ex)
|
|
{
|
|
|
|
await _gatewayLogger.DebugAsync("Disconnecting ApiClient").ConfigureAwait(false);
|
|
await ApiClient.DisconnectAsync().ConfigureAwait(false);
|
|
|
|
//Wait for tasks to complete
|
|
await _gatewayLogger.DebugAsync("Waiting for heartbeater").ConfigureAwait(false);
|
|
var heartbeatTask = _heartbeatTask;
|
|
if (heartbeatTask != null)
|
|
await heartbeatTask.ConfigureAwait(false);
|
|
_heartbeatTask = null;
|
|
|
|
while (_heartbeatTimes.TryDequeue(out _)) { }
|
|
_lastMessageTime = 0;
|
|
|
|
await _gatewayLogger.DebugAsync("Waiting for guild downloader").ConfigureAwait(false);
|
|
var guildDownloadTask = _guildDownloadTask;
|
|
if (guildDownloadTask != null)
|
|
await guildDownloadTask.ConfigureAwait(false);
|
|
_guildDownloadTask = null;
|
|
|
|
//Clear large guild queue
|
|
await _gatewayLogger.DebugAsync("Clearing large guild queue").ConfigureAwait(false);
|
|
while (_largeGuilds.TryDequeue(out _)) { }
|
|
|
|
//Raise virtual GUILD_UNAVAILABLEs
|
|
await _gatewayLogger.DebugAsync("Raising virtual GuildUnavailables").ConfigureAwait(false);
|
|
foreach (var guild in State.Guilds)
|
|
{
|
|
if (guild.IsAvailable)
|
|
await GuildUnavailableAsync(guild).ConfigureAwait(false);
|
|
}
|
|
}
|
|
|
|
/// <inheritdoc />
|
|
public override async Task<RestApplication> GetApplicationInfoAsync(RequestOptions options = null)
|
|
=> _applicationInfo ?? (_applicationInfo = await ClientHelper.GetApplicationInfoAsync(this, options ?? RequestOptions.Default).ConfigureAwait(false));
|
|
|
|
/// <inheritdoc />
|
|
public override SocketGuild GetGuild(ulong id)
|
|
=> State.GetGuild(id);
|
|
|
|
/// <inheritdoc />
|
|
public override SocketChannel GetChannel(ulong id)
|
|
=> State.GetChannel(id);
|
|
|
|
/// <inheritdoc />
|
|
public override SocketUser GetUser(ulong id)
|
|
=> State.GetUser(id);
|
|
/// <inheritdoc />
|
|
public override SocketUser GetUser(string username, string discriminator)
|
|
=> State.Users.FirstOrDefault(x => x.Discriminator == discriminator && x.Username == username);
|
|
internal SocketGlobalUser GetOrCreateUser(ClientState state, Discord.API.User model)
|
|
{
|
|
return state.GetOrAddUser(model.Id, x =>
|
|
{
|
|
var user = SocketGlobalUser.Create(this, state, model);
|
|
user.GlobalUser.AddRef();
|
|
return user;
|
|
});
|
|
}
|
|
internal SocketGlobalUser GetOrCreateSelfUser(ClientState state, Discord.API.User model)
|
|
{
|
|
return state.GetOrAddUser(model.Id, x =>
|
|
{
|
|
var user = SocketGlobalUser.Create(this, state, model);
|
|
user.GlobalUser.AddRef();
|
|
user.Presence = new SocketPresence(UserStatus.Online, null, null);
|
|
return user;
|
|
});
|
|
}
|
|
internal void RemoveUser(ulong id)
|
|
=> State.RemoveUser(id);
|
|
|
|
/// <inheritdoc />
|
|
public override RestVoiceRegion GetVoiceRegion(string id)
|
|
{
|
|
if (_voiceRegions.TryGetValue(id, out RestVoiceRegion region))
|
|
return region;
|
|
return null;
|
|
}
|
|
|
|
/// <inheritdoc />
|
|
public override async Task DownloadUsersAsync(IEnumerable<IGuild> guilds)
|
|
{
|
|
if (ConnectionState == ConnectionState.Connected)
|
|
{
|
|
//Race condition leads to guilds being requested twice, probably okay
|
|
await ProcessUserDownloadsAsync(guilds.Select(x => GetGuild(x.Id)).Where(x => x != null)).ConfigureAwait(false);
|
|
}
|
|
}
|
|
private async Task ProcessUserDownloadsAsync(IEnumerable<SocketGuild> guilds)
|
|
{
|
|
var cachedGuilds = guilds.ToImmutableArray();
|
|
|
|
const short batchSize = 50;
|
|
ulong[] batchIds = new ulong[Math.Min(batchSize, cachedGuilds.Length)];
|
|
Task[] batchTasks = new Task[batchIds.Length];
|
|
int batchCount = (cachedGuilds.Length + (batchSize - 1)) / batchSize;
|
|
|
|
for (int i = 0, k = 0; i < batchCount; i++)
|
|
{
|
|
bool isLast = i == batchCount - 1;
|
|
int count = isLast ? (batchIds.Length - (batchCount - 1) * batchSize) : batchSize;
|
|
|
|
for (int j = 0; j < count; j++, k++)
|
|
{
|
|
var guild = cachedGuilds[k];
|
|
batchIds[j] = guild.Id;
|
|
batchTasks[j] = guild.DownloaderPromise;
|
|
}
|
|
|
|
await ApiClient.SendRequestMembersAsync(batchIds).ConfigureAwait(false);
|
|
|
|
if (isLast && batchCount > 1)
|
|
await Task.WhenAll(batchTasks.Take(count)).ConfigureAwait(false);
|
|
else
|
|
await Task.WhenAll(batchTasks).ConfigureAwait(false);
|
|
}
|
|
}
|
|
|
|
/// <inheritdoc />
|
|
/// <example>
|
|
/// The following example sets the status of the current user to Do Not Disturb.
|
|
/// <code language="cs">
|
|
/// await client.SetStatusAsync(UserStatus.DoNotDisturb);
|
|
/// </code>
|
|
/// </example>
|
|
public override async Task SetStatusAsync(UserStatus status)
|
|
{
|
|
Status = status;
|
|
if (status == UserStatus.AFK)
|
|
_statusSince = DateTimeOffset.UtcNow;
|
|
else
|
|
_statusSince = null;
|
|
await SendStatusAsync().ConfigureAwait(false);
|
|
}
|
|
/// <inheritdoc />
|
|
/// <example>
|
|
/// <para>
|
|
/// The following example sets the activity of the current user to the specified game name.
|
|
/// <code language="cs">
|
|
/// await client.SetGameAsync("A Strange Game");
|
|
/// </code>
|
|
/// </para>
|
|
/// <para>
|
|
/// The following example sets the activity of the current user to a streaming status.
|
|
/// <code language="cs">
|
|
/// await client.SetGameAsync("Great Stream 10/10", "https://twitch.tv/MyAmazingStream1337", ActivityType.Streaming);
|
|
/// </code>
|
|
/// </para>
|
|
/// </example>
|
|
public override async Task SetGameAsync(string name, string streamUrl = null, ActivityType type = ActivityType.Playing)
|
|
{
|
|
if (!string.IsNullOrEmpty(streamUrl))
|
|
Activity = new StreamingGame(name, streamUrl);
|
|
else if (!string.IsNullOrEmpty(name))
|
|
Activity = new Game(name, type);
|
|
else
|
|
Activity = null;
|
|
await SendStatusAsync().ConfigureAwait(false);
|
|
}
|
|
/// <inheritdoc />
|
|
public override async Task SetActivityAsync(IActivity activity)
|
|
{
|
|
Activity = activity;
|
|
await SendStatusAsync().ConfigureAwait(false);
|
|
}
|
|
|
|
private async Task SendStatusAsync()
|
|
{
|
|
if (CurrentUser == null)
|
|
return;
|
|
var status = Status;
|
|
var statusSince = _statusSince;
|
|
CurrentUser.Presence = new SocketPresence(status, Activity, null);
|
|
|
|
var gameModel = new GameModel();
|
|
// Discord only accepts rich presence over RPC, don't even bother building a payload
|
|
if (Activity is RichGame)
|
|
throw new NotSupportedException("Outgoing Rich Presences are not supported via WebSocket.");
|
|
|
|
if (Activity != null)
|
|
{
|
|
gameModel.Name = Activity.Name;
|
|
gameModel.Type = Activity.Type;
|
|
if (Activity is StreamingGame streamGame)
|
|
gameModel.StreamUrl = streamGame.Url;
|
|
}
|
|
|
|
await ApiClient.SendStatusUpdateAsync(
|
|
status,
|
|
status == UserStatus.AFK,
|
|
statusSince != null ? _statusSince.Value.ToUnixTimeMilliseconds() : (long?)null,
|
|
gameModel).ConfigureAwait(false);
|
|
}
|
|
|
|
private async Task ProcessMessageAsync(GatewayOpCode opCode, int? seq, string type, object payload)
|
|
{
|
|
if (seq != null)
|
|
_lastSeq = seq.Value;
|
|
_lastMessageTime = Environment.TickCount;
|
|
|
|
try
|
|
{
|
|
switch (opCode)
|
|
{
|
|
case GatewayOpCode.Hello:
|
|
{
|
|
await _gatewayLogger.DebugAsync("Received Hello").ConfigureAwait(false);
|
|
var data = (payload as JToken).ToObject<HelloEvent>(_serializer);
|
|
|
|
_heartbeatTask = RunHeartbeatAsync(data.HeartbeatInterval, _connection.CancelToken);
|
|
}
|
|
break;
|
|
case GatewayOpCode.Heartbeat:
|
|
{
|
|
await _gatewayLogger.DebugAsync("Received Heartbeat").ConfigureAwait(false);
|
|
|
|
await ApiClient.SendHeartbeatAsync(_lastSeq).ConfigureAwait(false);
|
|
}
|
|
break;
|
|
case GatewayOpCode.HeartbeatAck:
|
|
{
|
|
await _gatewayLogger.DebugAsync("Received HeartbeatAck").ConfigureAwait(false);
|
|
|
|
if (_heartbeatTimes.TryDequeue(out long time))
|
|
{
|
|
int latency = (int)(Environment.TickCount - time);
|
|
int before = Latency;
|
|
Latency = latency;
|
|
|
|
await TimedInvokeAsync(_latencyUpdatedEvent, nameof(LatencyUpdated), before, latency).ConfigureAwait(false);
|
|
}
|
|
}
|
|
break;
|
|
case GatewayOpCode.InvalidSession:
|
|
{
|
|
await _gatewayLogger.DebugAsync("Received InvalidSession").ConfigureAwait(false);
|
|
await _gatewayLogger.WarningAsync("Failed to resume previous session").ConfigureAwait(false);
|
|
|
|
_sessionId = null;
|
|
_lastSeq = 0;
|
|
|
|
await ApiClient.SendIdentifyAsync(shardID: ShardId, totalShards: TotalShards).ConfigureAwait(false);
|
|
}
|
|
break;
|
|
case GatewayOpCode.Reconnect:
|
|
{
|
|
await _gatewayLogger.DebugAsync("Received Reconnect").ConfigureAwait(false);
|
|
_connection.Error(new Exception("Server requested a reconnect"));
|
|
}
|
|
break;
|
|
case GatewayOpCode.Dispatch:
|
|
switch (type)
|
|
{
|
|
//Connection
|
|
case "READY":
|
|
{
|
|
try
|
|
{
|
|
await _gatewayLogger.DebugAsync("Received Dispatch (READY)").ConfigureAwait(false);
|
|
|
|
var data = (payload as JToken).ToObject<ReadyEvent>(_serializer);
|
|
var state = new ClientState(data.Guilds.Length, data.PrivateChannels.Length);
|
|
|
|
var currentUser = SocketSelfUser.Create(this, state, data.User);
|
|
ApiClient.CurrentUserId = currentUser.Id;
|
|
int unavailableGuilds = 0;
|
|
for (int i = 0; i < data.Guilds.Length; i++)
|
|
{
|
|
var model = data.Guilds[i];
|
|
var guild = AddGuild(model, state);
|
|
if (!guild.IsAvailable || ApiClient.AuthTokenType == TokenType.User)
|
|
unavailableGuilds++;
|
|
else
|
|
await GuildAvailableAsync(guild).ConfigureAwait(false);
|
|
}
|
|
for (int i = 0; i < data.PrivateChannels.Length; i++)
|
|
AddPrivateChannel(data.PrivateChannels[i], state);
|
|
|
|
_sessionId = data.SessionId;
|
|
_unavailableGuildCount = unavailableGuilds;
|
|
CurrentUser = currentUser;
|
|
State = state;
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_connection.CriticalError(new Exception("Processing READY failed", ex));
|
|
return;
|
|
}
|
|
|
|
if (ApiClient.AuthTokenType == TokenType.User)
|
|
await SyncGuildsAsync().ConfigureAwait(false);
|
|
|
|
_lastGuildAvailableTime = Environment.TickCount;
|
|
_guildDownloadTask = WaitForGuildsAsync(_connection.CancelToken, _gatewayLogger)
|
|
.ContinueWith(async x =>
|
|
{
|
|
if (x.IsFaulted)
|
|
{
|
|
_connection.Error(x.Exception);
|
|
return;
|
|
}
|
|
else if (_connection.CancelToken.IsCancellationRequested)
|
|
return;
|
|
|
|
await TimedInvokeAsync(_readyEvent, nameof(Ready)).ConfigureAwait(false);
|
|
await _gatewayLogger.InfoAsync("Ready").ConfigureAwait(false);
|
|
});
|
|
_ = _connection.CompleteAsync();
|
|
}
|
|
break;
|
|
case "RESUMED":
|
|
{
|
|
await _gatewayLogger.DebugAsync("Received Dispatch (RESUMED)").ConfigureAwait(false);
|
|
|
|
_ = _connection.CompleteAsync();
|
|
|
|
//Notify the client that these guilds are available again
|
|
foreach (var guild in State.Guilds)
|
|
{
|
|
if (guild.IsAvailable)
|
|
await GuildAvailableAsync(guild).ConfigureAwait(false);
|
|
}
|
|
|
|
await _gatewayLogger.InfoAsync("Resumed previous session").ConfigureAwait(false);
|
|
}
|
|
break;
|
|
|
|
//Guilds
|
|
case "GUILD_CREATE":
|
|
{
|
|
var data = (payload as JToken).ToObject<ExtendedGuild>(_serializer);
|
|
|
|
if (data.Unavailable == false)
|
|
{
|
|
type = "GUILD_AVAILABLE";
|
|
_lastGuildAvailableTime = Environment.TickCount;
|
|
await _gatewayLogger.DebugAsync("Received Dispatch (GUILD_AVAILABLE)").ConfigureAwait(false);
|
|
|
|
var guild = State.GetGuild(data.Id);
|
|
if (guild != null)
|
|
{
|
|
guild.Update(State, data);
|
|
|
|
if (_unavailableGuildCount != 0)
|
|
_unavailableGuildCount--;
|
|
await GuildAvailableAsync(guild).ConfigureAwait(false);
|
|
|
|
if (guild.DownloadedMemberCount >= guild.MemberCount && !guild.DownloaderPromise.IsCompleted)
|
|
{
|
|
guild.CompleteDownloadUsers();
|
|
await TimedInvokeAsync(_guildMembersDownloadedEvent, nameof(GuildMembersDownloaded), guild).ConfigureAwait(false);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
await UnknownGuildAsync(type, data.Id).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
await _gatewayLogger.DebugAsync("Received Dispatch (GUILD_CREATE)").ConfigureAwait(false);
|
|
|
|
var guild = AddGuild(data, State);
|
|
if (guild != null)
|
|
{
|
|
if (ApiClient.AuthTokenType == TokenType.User)
|
|
await SyncGuildsAsync().ConfigureAwait(false);
|
|
await TimedInvokeAsync(_joinedGuildEvent, nameof(JoinedGuild), guild).ConfigureAwait(false);
|
|
}
|
|
else
|
|
{
|
|
await UnknownGuildAsync(type, data.Id).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
break;
|
|
case "GUILD_UPDATE":
|
|
{
|
|
await _gatewayLogger.DebugAsync("Received Dispatch (GUILD_UPDATE)").ConfigureAwait(false);
|
|
|
|
var data = (payload as JToken).ToObject<API.Guild>(_serializer);
|
|
var guild = State.GetGuild(data.Id);
|
|
if (guild != null)
|
|
{
|
|
var before = guild.Clone();
|
|
guild.Update(State, data);
|
|
await TimedInvokeAsync(_guildUpdatedEvent, nameof(GuildUpdated), before, guild).ConfigureAwait(false);
|
|
}
|
|
else
|
|
{
|
|
await UnknownGuildAsync(type, data.Id).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
}
|
|
break;
|
|
case "GUILD_EMOJIS_UPDATE":
|
|
{
|
|
await _gatewayLogger.DebugAsync("Received Dispatch (GUILD_EMOJIS_UPDATE)").ConfigureAwait(false);
|
|
|
|
var data = (payload as JToken).ToObject<API.Gateway.GuildEmojiUpdateEvent>(_serializer);
|
|
var guild = State.GetGuild(data.GuildId);
|
|
if (guild != null)
|
|
{
|
|
var before = guild.Clone();
|
|
guild.Update(State, data);
|
|
await TimedInvokeAsync(_guildUpdatedEvent, nameof(GuildUpdated), before, guild).ConfigureAwait(false);
|
|
}
|
|
else
|
|
{
|
|
await UnknownGuildAsync(type, data.GuildId).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
}
|
|
break;
|
|
case "GUILD_SYNC":
|
|
{
|
|
await _gatewayLogger.DebugAsync("Received Dispatch (GUILD_SYNC)").ConfigureAwait(false);
|
|
var data = (payload as JToken).ToObject<GuildSyncEvent>(_serializer);
|
|
var guild = State.GetGuild(data.Id);
|
|
if (guild != null)
|
|
{
|
|
var before = guild.Clone();
|
|
guild.Update(State, data);
|
|
//This is treated as an extension of GUILD_AVAILABLE
|
|
_unavailableGuildCount--;
|
|
_lastGuildAvailableTime = Environment.TickCount;
|
|
await GuildAvailableAsync(guild).ConfigureAwait(false);
|
|
await TimedInvokeAsync(_guildUpdatedEvent, nameof(GuildUpdated), before, guild).ConfigureAwait(false);
|
|
}
|
|
else
|
|
{
|
|
await UnknownGuildAsync(type, data.Id).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
}
|
|
break;
|
|
case "GUILD_DELETE":
|
|
{
|
|
var data = (payload as JToken).ToObject<ExtendedGuild>(_serializer);
|
|
if (data.Unavailable == true)
|
|
{
|
|
type = "GUILD_UNAVAILABLE";
|
|
await _gatewayLogger.DebugAsync("Received Dispatch (GUILD_UNAVAILABLE)").ConfigureAwait(false);
|
|
|
|
var guild = State.GetGuild(data.Id);
|
|
if (guild != null)
|
|
{
|
|
await GuildUnavailableAsync(guild).ConfigureAwait(false);
|
|
_unavailableGuildCount++;
|
|
}
|
|
else
|
|
{
|
|
await UnknownGuildAsync(type, data.Id).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
await _gatewayLogger.DebugAsync("Received Dispatch (GUILD_DELETE)").ConfigureAwait(false);
|
|
|
|
var guild = RemoveGuild(data.Id);
|
|
if (guild != null)
|
|
{
|
|
await GuildUnavailableAsync(guild).ConfigureAwait(false);
|
|
await TimedInvokeAsync(_leftGuildEvent, nameof(LeftGuild), guild).ConfigureAwait(false);
|
|
(guild as IDisposable).Dispose();
|
|
}
|
|
else
|
|
{
|
|
await UnknownGuildAsync(type, data.Id).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
break;
|
|
|
|
//Channels
|
|
case "CHANNEL_CREATE":
|
|
{
|
|
await _gatewayLogger.DebugAsync("Received Dispatch (CHANNEL_CREATE)").ConfigureAwait(false);
|
|
|
|
var data = (payload as JToken).ToObject<API.Channel>(_serializer);
|
|
SocketChannel channel = null;
|
|
if (data.GuildId.IsSpecified)
|
|
{
|
|
var guild = State.GetGuild(data.GuildId.Value);
|
|
if (guild != null)
|
|
{
|
|
channel = guild.AddChannel(State, data);
|
|
|
|
if (!guild.IsSynced)
|
|
{
|
|
await UnsyncedGuildAsync(type, guild.Id).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
await UnknownGuildAsync(type, data.GuildId.Value).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
channel = State.GetChannel(data.Id);
|
|
if (channel != null)
|
|
return; //Discord may send duplicate CHANNEL_CREATEs for DMs
|
|
channel = AddPrivateChannel(data, State) as SocketChannel;
|
|
}
|
|
|
|
if (channel != null)
|
|
await TimedInvokeAsync(_channelCreatedEvent, nameof(ChannelCreated), channel).ConfigureAwait(false);
|
|
}
|
|
break;
|
|
case "CHANNEL_UPDATE":
|
|
{
|
|
await _gatewayLogger.DebugAsync("Received Dispatch (CHANNEL_UPDATE)").ConfigureAwait(false);
|
|
|
|
var data = (payload as JToken).ToObject<API.Channel>(_serializer);
|
|
var channel = State.GetChannel(data.Id);
|
|
if (channel != null)
|
|
{
|
|
var before = channel.Clone();
|
|
channel.Update(State, data);
|
|
|
|
var guild = (channel as SocketGuildChannel)?.Guild;
|
|
if (!(guild?.IsSynced ?? true))
|
|
{
|
|
await UnsyncedGuildAsync(type, guild.Id).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
|
|
await TimedInvokeAsync(_channelUpdatedEvent, nameof(ChannelUpdated), before, channel).ConfigureAwait(false);
|
|
}
|
|
else
|
|
{
|
|
await UnknownChannelAsync(type, data.Id).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
}
|
|
break;
|
|
case "CHANNEL_DELETE":
|
|
{
|
|
await _gatewayLogger.DebugAsync("Received Dispatch (CHANNEL_DELETE)").ConfigureAwait(false);
|
|
|
|
SocketChannel channel = null;
|
|
var data = (payload as JToken).ToObject<API.Channel>(_serializer);
|
|
if (data.GuildId.IsSpecified)
|
|
{
|
|
var guild = State.GetGuild(data.GuildId.Value);
|
|
if (guild != null)
|
|
{
|
|
channel = guild.RemoveChannel(State, data.Id);
|
|
|
|
if (!guild.IsSynced)
|
|
{
|
|
await UnsyncedGuildAsync(type, guild.Id).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
await UnknownGuildAsync(type, data.GuildId.Value).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
}
|
|
else
|
|
channel = RemovePrivateChannel(data.Id) as SocketChannel;
|
|
|
|
if (channel != null)
|
|
await TimedInvokeAsync(_channelDestroyedEvent, nameof(ChannelDestroyed), channel).ConfigureAwait(false);
|
|
else
|
|
{
|
|
await UnknownChannelAsync(type, data.Id, data.GuildId.GetValueOrDefault(0)).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
}
|
|
break;
|
|
|
|
//Members
|
|
case "GUILD_MEMBER_ADD":
|
|
{
|
|
await _gatewayLogger.DebugAsync("Received Dispatch (GUILD_MEMBER_ADD)").ConfigureAwait(false);
|
|
|
|
var data = (payload as JToken).ToObject<GuildMemberAddEvent>(_serializer);
|
|
var guild = State.GetGuild(data.GuildId);
|
|
if (guild != null)
|
|
{
|
|
var user = guild.AddOrUpdateUser(data);
|
|
guild.MemberCount++;
|
|
|
|
if (!guild.IsSynced)
|
|
{
|
|
await UnsyncedGuildAsync(type, guild.Id).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
|
|
await TimedInvokeAsync(_userJoinedEvent, nameof(UserJoined), user).ConfigureAwait(false);
|
|
}
|
|
else
|
|
{
|
|
await UnknownGuildAsync(type, data.GuildId).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
}
|
|
break;
|
|
case "GUILD_MEMBER_UPDATE":
|
|
{
|
|
await _gatewayLogger.DebugAsync("Received Dispatch (GUILD_MEMBER_UPDATE)").ConfigureAwait(false);
|
|
|
|
var data = (payload as JToken).ToObject<GuildMemberUpdateEvent>(_serializer);
|
|
var guild = State.GetGuild(data.GuildId);
|
|
if (guild != null)
|
|
{
|
|
var user = guild.GetUser(data.User.Id);
|
|
|
|
if (!guild.IsSynced)
|
|
{
|
|
await UnsyncedGuildAsync(type, guild.Id).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
|
|
if (user != null)
|
|
{
|
|
var before = user.Clone();
|
|
user.Update(State, data);
|
|
await TimedInvokeAsync(_guildMemberUpdatedEvent, nameof(GuildMemberUpdated), before, user).ConfigureAwait(false);
|
|
}
|
|
else
|
|
{
|
|
if (!guild.HasAllMembers)
|
|
await IncompleteGuildUserAsync(type, data.User.Id, data.GuildId).ConfigureAwait(false);
|
|
else
|
|
await UnknownGuildUserAsync(type, data.User.Id, data.GuildId).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
await UnknownGuildAsync(type, data.GuildId).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
}
|
|
break;
|
|
case "GUILD_MEMBER_REMOVE":
|
|
{
|
|
await _gatewayLogger.DebugAsync("Received Dispatch (GUILD_MEMBER_REMOVE)").ConfigureAwait(false);
|
|
|
|
var data = (payload as JToken).ToObject<GuildMemberRemoveEvent>(_serializer);
|
|
var guild = State.GetGuild(data.GuildId);
|
|
if (guild != null)
|
|
{
|
|
var user = guild.RemoveUser(data.User.Id);
|
|
guild.MemberCount--;
|
|
|
|
if (!guild.IsSynced)
|
|
{
|
|
await UnsyncedGuildAsync(type, guild.Id).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
|
|
if (user != null)
|
|
await TimedInvokeAsync(_userLeftEvent, nameof(UserLeft), user).ConfigureAwait(false);
|
|
else
|
|
{
|
|
if (!guild.HasAllMembers)
|
|
await IncompleteGuildUserAsync(type, data.User.Id, data.GuildId).ConfigureAwait(false);
|
|
else
|
|
await UnknownGuildUserAsync(type, data.User.Id, data.GuildId).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
await UnknownGuildAsync(type, data.GuildId).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
}
|
|
break;
|
|
case "GUILD_MEMBERS_CHUNK":
|
|
{
|
|
await _gatewayLogger.DebugAsync("Received Dispatch (GUILD_MEMBERS_CHUNK)").ConfigureAwait(false);
|
|
|
|
var data = (payload as JToken).ToObject<GuildMembersChunkEvent>(_serializer);
|
|
var guild = State.GetGuild(data.GuildId);
|
|
if (guild != null)
|
|
{
|
|
foreach (var memberModel in data.Members)
|
|
guild.AddOrUpdateUser(memberModel);
|
|
|
|
if (guild.DownloadedMemberCount >= guild.MemberCount && !guild.DownloaderPromise.IsCompleted)
|
|
{
|
|
guild.CompleteDownloadUsers();
|
|
await TimedInvokeAsync(_guildMembersDownloadedEvent, nameof(GuildMembersDownloaded), guild).ConfigureAwait(false);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
await UnknownGuildAsync(type, data.GuildId).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
}
|
|
break;
|
|
case "CHANNEL_RECIPIENT_ADD":
|
|
{
|
|
await _gatewayLogger.DebugAsync("Received Dispatch (CHANNEL_RECIPIENT_ADD)").ConfigureAwait(false);
|
|
|
|
var data = (payload as JToken).ToObject<RecipientEvent>(_serializer);
|
|
if (State.GetChannel(data.ChannelId) is SocketGroupChannel channel)
|
|
{
|
|
var user = channel.GetOrAddUser(data.User);
|
|
await TimedInvokeAsync(_recipientAddedEvent, nameof(RecipientAdded), user).ConfigureAwait(false);
|
|
}
|
|
else
|
|
{
|
|
await UnknownChannelAsync(type, data.ChannelId).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
}
|
|
break;
|
|
case "CHANNEL_RECIPIENT_REMOVE":
|
|
{
|
|
await _gatewayLogger.DebugAsync("Received Dispatch (CHANNEL_RECIPIENT_REMOVE)").ConfigureAwait(false);
|
|
|
|
var data = (payload as JToken).ToObject<RecipientEvent>(_serializer);
|
|
if (State.GetChannel(data.ChannelId) is SocketGroupChannel channel)
|
|
{
|
|
var user = channel.RemoveUser(data.User.Id);
|
|
if (user != null)
|
|
await TimedInvokeAsync(_recipientRemovedEvent, nameof(RecipientRemoved), user).ConfigureAwait(false);
|
|
else
|
|
{
|
|
await UnknownChannelUserAsync(type, data.User.Id, data.ChannelId).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
await UnknownChannelAsync(type, data.ChannelId).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
}
|
|
break;
|
|
|
|
//Roles
|
|
case "GUILD_ROLE_CREATE":
|
|
{
|
|
await _gatewayLogger.DebugAsync("Received Dispatch (GUILD_ROLE_CREATE)").ConfigureAwait(false);
|
|
|
|
var data = (payload as JToken).ToObject<GuildRoleCreateEvent>(_serializer);
|
|
var guild = State.GetGuild(data.GuildId);
|
|
if (guild != null)
|
|
{
|
|
var role = guild.AddRole(data.Role);
|
|
|
|
if (!guild.IsSynced)
|
|
{
|
|
await UnsyncedGuildAsync(type, guild.Id).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
await TimedInvokeAsync(_roleCreatedEvent, nameof(RoleCreated), role).ConfigureAwait(false);
|
|
}
|
|
else
|
|
{
|
|
await UnknownGuildAsync(type, data.GuildId).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
}
|
|
break;
|
|
case "GUILD_ROLE_UPDATE":
|
|
{
|
|
await _gatewayLogger.DebugAsync("Received Dispatch (GUILD_ROLE_UPDATE)").ConfigureAwait(false);
|
|
|
|
var data = (payload as JToken).ToObject<GuildRoleUpdateEvent>(_serializer);
|
|
var guild = State.GetGuild(data.GuildId);
|
|
if (guild != null)
|
|
{
|
|
var role = guild.GetRole(data.Role.Id);
|
|
if (role != null)
|
|
{
|
|
var before = role.Clone();
|
|
role.Update(State, data.Role);
|
|
|
|
if (!guild.IsSynced)
|
|
{
|
|
await UnsyncedGuildAsync(type, guild.Id).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
|
|
await TimedInvokeAsync(_roleUpdatedEvent, nameof(RoleUpdated), before, role).ConfigureAwait(false);
|
|
}
|
|
else
|
|
{
|
|
await UnknownRoleAsync(type, data.Role.Id, guild.Id).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
await UnknownGuildAsync(type, data.GuildId).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
}
|
|
break;
|
|
case "GUILD_ROLE_DELETE":
|
|
{
|
|
await _gatewayLogger.DebugAsync("Received Dispatch (GUILD_ROLE_DELETE)").ConfigureAwait(false);
|
|
|
|
var data = (payload as JToken).ToObject<GuildRoleDeleteEvent>(_serializer);
|
|
var guild = State.GetGuild(data.GuildId);
|
|
if (guild != null)
|
|
{
|
|
var role = guild.RemoveRole(data.RoleId);
|
|
if (role != null)
|
|
{
|
|
if (!guild.IsSynced)
|
|
{
|
|
await UnsyncedGuildAsync(type, guild.Id).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
|
|
await TimedInvokeAsync(_roleDeletedEvent, nameof(RoleDeleted), role).ConfigureAwait(false);
|
|
}
|
|
else
|
|
{
|
|
await UnknownRoleAsync(type, data.RoleId, guild.Id).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
await UnknownGuildAsync(type, data.GuildId).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
}
|
|
break;
|
|
|
|
//Bans
|
|
case "GUILD_BAN_ADD":
|
|
{
|
|
await _gatewayLogger.DebugAsync("Received Dispatch (GUILD_BAN_ADD)").ConfigureAwait(false);
|
|
|
|
var data = (payload as JToken).ToObject<GuildBanEvent>(_serializer);
|
|
var guild = State.GetGuild(data.GuildId);
|
|
if (guild != null)
|
|
{
|
|
if (!guild.IsSynced)
|
|
{
|
|
await UnsyncedGuildAsync(type, guild.Id).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
|
|
SocketUser user = guild.GetUser(data.User.Id);
|
|
if (user == null)
|
|
user = SocketUnknownUser.Create(this, State, data.User);
|
|
await TimedInvokeAsync(_userBannedEvent, nameof(UserBanned), user, guild).ConfigureAwait(false);
|
|
}
|
|
else
|
|
{
|
|
await UnknownGuildAsync(type, data.GuildId).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
}
|
|
break;
|
|
case "GUILD_BAN_REMOVE":
|
|
{
|
|
await _gatewayLogger.DebugAsync("Received Dispatch (GUILD_BAN_REMOVE)").ConfigureAwait(false);
|
|
|
|
var data = (payload as JToken).ToObject<GuildBanEvent>(_serializer);
|
|
var guild = State.GetGuild(data.GuildId);
|
|
if (guild != null)
|
|
{
|
|
if (!guild.IsSynced)
|
|
{
|
|
await UnsyncedGuildAsync(type, guild.Id).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
|
|
SocketUser user = State.GetUser(data.User.Id);
|
|
if (user == null)
|
|
user = SocketUnknownUser.Create(this, State, data.User);
|
|
await TimedInvokeAsync(_userUnbannedEvent, nameof(UserUnbanned), user, guild).ConfigureAwait(false);
|
|
}
|
|
else
|
|
{
|
|
await UnknownGuildAsync(type, data.GuildId).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
}
|
|
break;
|
|
|
|
//Messages
|
|
case "MESSAGE_CREATE":
|
|
{
|
|
await _gatewayLogger.DebugAsync("Received Dispatch (MESSAGE_CREATE)").ConfigureAwait(false);
|
|
|
|
var data = (payload as JToken).ToObject<API.Message>(_serializer);
|
|
if (State.GetChannel(data.ChannelId) is ISocketMessageChannel channel)
|
|
{
|
|
var guild = (channel as SocketGuildChannel)?.Guild;
|
|
if (guild != null && !guild.IsSynced)
|
|
{
|
|
await UnsyncedGuildAsync(type, guild.Id).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
|
|
SocketUser author;
|
|
if (guild != null)
|
|
{
|
|
if (data.WebhookId.IsSpecified)
|
|
author = SocketWebhookUser.Create(guild, State, data.Author.Value, data.WebhookId.Value);
|
|
else
|
|
author = guild.GetUser(data.Author.Value.Id);
|
|
}
|
|
else
|
|
author = (channel as SocketChannel).GetUser(data.Author.Value.Id);
|
|
|
|
if (author == null)
|
|
{
|
|
if (guild != null)
|
|
{
|
|
if (data.Member.IsSpecified) // member isn't always included, but use it when we can
|
|
{
|
|
data.Member.Value.User = data.Author.Value;
|
|
author = guild.AddOrUpdateUser(data.Member.Value);
|
|
}
|
|
else
|
|
author = guild.AddOrUpdateUser(data.Author.Value); // user has no guild-specific data
|
|
}
|
|
else if (channel is SocketGroupChannel)
|
|
author = (channel as SocketGroupChannel).GetOrAddUser(data.Author.Value);
|
|
else
|
|
{
|
|
await UnknownChannelUserAsync(type, data.Author.Value.Id, channel.Id).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
}
|
|
|
|
var msg = SocketMessage.Create(this, State, author, channel, data);
|
|
SocketChannelHelper.AddMessage(channel, this, msg);
|
|
await TimedInvokeAsync(_messageReceivedEvent, nameof(MessageReceived), msg).ConfigureAwait(false);
|
|
}
|
|
else
|
|
{
|
|
await UnknownChannelAsync(type, data.ChannelId).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
}
|
|
break;
|
|
case "MESSAGE_UPDATE":
|
|
{
|
|
await _gatewayLogger.DebugAsync("Received Dispatch (MESSAGE_UPDATE)").ConfigureAwait(false);
|
|
|
|
var data = (payload as JToken).ToObject<API.Message>(_serializer);
|
|
if (State.GetChannel(data.ChannelId) is ISocketMessageChannel channel)
|
|
{
|
|
var guild = (channel as SocketGuildChannel)?.Guild;
|
|
if (guild != null && !guild.IsSynced)
|
|
{
|
|
await UnsyncedGuildAsync(type, guild.Id).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
|
|
SocketMessage before = null, after = null;
|
|
SocketMessage cachedMsg = channel.GetCachedMessage(data.Id);
|
|
bool isCached = cachedMsg != null;
|
|
if (isCached)
|
|
{
|
|
before = cachedMsg.Clone();
|
|
cachedMsg.Update(State, data);
|
|
after = cachedMsg;
|
|
}
|
|
else
|
|
{
|
|
//Edited message isnt in cache, create a detached one
|
|
SocketUser author;
|
|
if (data.Author.IsSpecified)
|
|
{
|
|
if (guild != null)
|
|
author = guild.GetUser(data.Author.Value.Id);
|
|
else
|
|
author = (channel as SocketChannel).GetUser(data.Author.Value.Id);
|
|
if (author == null)
|
|
author = SocketUnknownUser.Create(this, State, data.Author.Value);
|
|
}
|
|
else
|
|
// Message author wasn't specified in the payload, so create a completely anonymous unknown user
|
|
author = new SocketUnknownUser(this, id: 0);
|
|
|
|
after = SocketMessage.Create(this, State, author, channel, data);
|
|
}
|
|
var cacheableBefore = new Cacheable<IMessage, ulong>(before, data.Id, isCached, async () => await channel.GetMessageAsync(data.Id).ConfigureAwait(false));
|
|
|
|
await TimedInvokeAsync(_messageUpdatedEvent, nameof(MessageUpdated), cacheableBefore, after, channel).ConfigureAwait(false);
|
|
}
|
|
else
|
|
{
|
|
await UnknownChannelAsync(type, data.ChannelId).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
}
|
|
break;
|
|
case "MESSAGE_DELETE":
|
|
{
|
|
await _gatewayLogger.DebugAsync("Received Dispatch (MESSAGE_DELETE)").ConfigureAwait(false);
|
|
|
|
var data = (payload as JToken).ToObject<API.Message>(_serializer);
|
|
if (State.GetChannel(data.ChannelId) is ISocketMessageChannel channel)
|
|
{
|
|
var guild = (channel as SocketGuildChannel)?.Guild;
|
|
if (!(guild?.IsSynced ?? true))
|
|
{
|
|
await UnsyncedGuildAsync(type, guild.Id).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
|
|
var msg = SocketChannelHelper.RemoveMessage(channel, this, data.Id);
|
|
bool isCached = msg != null;
|
|
var cacheable = new Cacheable<IMessage, ulong>(msg, data.Id, isCached, async () => await channel.GetMessageAsync(data.Id).ConfigureAwait(false));
|
|
|
|
await TimedInvokeAsync(_messageDeletedEvent, nameof(MessageDeleted), cacheable, channel).ConfigureAwait(false);
|
|
}
|
|
else
|
|
{
|
|
await UnknownChannelAsync(type, data.ChannelId).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
}
|
|
break;
|
|
case "MESSAGE_REACTION_ADD":
|
|
{
|
|
await _gatewayLogger.DebugAsync("Received Dispatch (MESSAGE_REACTION_ADD)").ConfigureAwait(false);
|
|
|
|
var data = (payload as JToken).ToObject<API.Gateway.Reaction>(_serializer);
|
|
if (State.GetChannel(data.ChannelId) is ISocketMessageChannel channel)
|
|
{
|
|
var cachedMsg = channel.GetCachedMessage(data.MessageId) as SocketUserMessage;
|
|
bool isCached = cachedMsg != null;
|
|
var user = await channel.GetUserAsync(data.UserId, CacheMode.CacheOnly).ConfigureAwait(false);
|
|
|
|
var optionalMsg = !isCached
|
|
? Optional.Create<SocketUserMessage>()
|
|
: Optional.Create(cachedMsg);
|
|
|
|
var optionalUser = user is null
|
|
? Optional.Create<IUser>()
|
|
: Optional.Create(user);
|
|
|
|
var reaction = SocketReaction.Create(data, channel, optionalMsg, optionalUser);
|
|
var cacheable = new Cacheable<IUserMessage, ulong>(cachedMsg, data.MessageId, isCached, async () => await channel.GetMessageAsync(data.MessageId).ConfigureAwait(false) as IUserMessage);
|
|
|
|
cachedMsg?.AddReaction(reaction);
|
|
|
|
await TimedInvokeAsync(_reactionAddedEvent, nameof(ReactionAdded), cacheable, channel, reaction).ConfigureAwait(false);
|
|
}
|
|
else
|
|
{
|
|
await UnknownChannelAsync(type, data.ChannelId).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
}
|
|
break;
|
|
case "MESSAGE_REACTION_REMOVE":
|
|
{
|
|
await _gatewayLogger.DebugAsync("Received Dispatch (MESSAGE_REACTION_REMOVE)").ConfigureAwait(false);
|
|
|
|
var data = (payload as JToken).ToObject<API.Gateway.Reaction>(_serializer);
|
|
if (State.GetChannel(data.ChannelId) is ISocketMessageChannel channel)
|
|
{
|
|
var cachedMsg = channel.GetCachedMessage(data.MessageId) as SocketUserMessage;
|
|
bool isCached = cachedMsg != null;
|
|
var user = await channel.GetUserAsync(data.UserId, CacheMode.CacheOnly).ConfigureAwait(false);
|
|
|
|
var optionalMsg = !isCached
|
|
? Optional.Create<SocketUserMessage>()
|
|
: Optional.Create(cachedMsg);
|
|
|
|
var optionalUser = user is null
|
|
? Optional.Create<IUser>()
|
|
: Optional.Create(user);
|
|
|
|
var reaction = SocketReaction.Create(data, channel, optionalMsg, optionalUser);
|
|
var cacheable = new Cacheable<IUserMessage, ulong>(cachedMsg, data.MessageId, isCached, async () => await channel.GetMessageAsync(data.MessageId).ConfigureAwait(false) as IUserMessage);
|
|
|
|
cachedMsg?.RemoveReaction(reaction);
|
|
|
|
await TimedInvokeAsync(_reactionRemovedEvent, nameof(ReactionRemoved), cacheable, channel, reaction).ConfigureAwait(false);
|
|
}
|
|
else
|
|
{
|
|
await UnknownChannelAsync(type, data.ChannelId).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
}
|
|
break;
|
|
case "MESSAGE_REACTION_REMOVE_ALL":
|
|
{
|
|
await _gatewayLogger.DebugAsync("Received Dispatch (MESSAGE_REACTION_REMOVE_ALL)").ConfigureAwait(false);
|
|
|
|
var data = (payload as JToken).ToObject<RemoveAllReactionsEvent>(_serializer);
|
|
if (State.GetChannel(data.ChannelId) is ISocketMessageChannel channel)
|
|
{
|
|
var cachedMsg = channel.GetCachedMessage(data.MessageId) as SocketUserMessage;
|
|
bool isCached = cachedMsg != null;
|
|
var cacheable = new Cacheable<IUserMessage, ulong>(cachedMsg, data.MessageId, isCached, async () => (await channel.GetMessageAsync(data.MessageId).ConfigureAwait(false)) as IUserMessage);
|
|
|
|
cachedMsg?.ClearReactions();
|
|
|
|
await TimedInvokeAsync(_reactionsClearedEvent, nameof(ReactionsCleared), cacheable, channel).ConfigureAwait(false);
|
|
}
|
|
else
|
|
{
|
|
await UnknownChannelAsync(type, data.ChannelId).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
}
|
|
break;
|
|
case "MESSAGE_DELETE_BULK":
|
|
{
|
|
await _gatewayLogger.DebugAsync("Received Dispatch (MESSAGE_DELETE_BULK)").ConfigureAwait(false);
|
|
|
|
if (!ExclusiveBulkDelete.HasValue)
|
|
{
|
|
await _gatewayLogger.WarningAsync("A bulk delete event has been received, but the event handling behavior has not been set. " +
|
|
"To suppress this message, set the ExclusiveBulkDelete configuration property. " +
|
|
"This message will appear only once.");
|
|
ExclusiveBulkDelete = false;
|
|
}
|
|
|
|
var data = (payload as JToken).ToObject<MessageDeleteBulkEvent>(_serializer);
|
|
if (State.GetChannel(data.ChannelId) is ISocketMessageChannel channel)
|
|
{
|
|
var guild = (channel as SocketGuildChannel)?.Guild;
|
|
if (!(guild?.IsSynced ?? true))
|
|
{
|
|
await UnsyncedGuildAsync(type, guild.Id).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
|
|
var cacheableList = new List<Cacheable<IMessage, ulong>>(data.Ids.Length);
|
|
foreach (ulong id in data.Ids)
|
|
{
|
|
var msg = SocketChannelHelper.RemoveMessage(channel, this, id);
|
|
bool isCached = msg != null;
|
|
var cacheable = new Cacheable<IMessage, ulong>(msg, id, isCached, async () => await channel.GetMessageAsync(id).ConfigureAwait(false));
|
|
cacheableList.Add(cacheable);
|
|
|
|
if (!ExclusiveBulkDelete ?? false) // this shouldn't happen, but we'll play it safe anyways
|
|
await TimedInvokeAsync(_messageDeletedEvent, nameof(MessageDeleted), cacheable, channel).ConfigureAwait(false);
|
|
}
|
|
|
|
await TimedInvokeAsync(_messagesBulkDeletedEvent, nameof(MessagesBulkDeleted), cacheableList, channel).ConfigureAwait(false);
|
|
}
|
|
else
|
|
{
|
|
await UnknownChannelAsync(type, data.ChannelId).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
}
|
|
break;
|
|
|
|
//Statuses
|
|
case "PRESENCE_UPDATE":
|
|
{
|
|
await _gatewayLogger.DebugAsync("Received Dispatch (PRESENCE_UPDATE)").ConfigureAwait(false);
|
|
|
|
var data = (payload as JToken).ToObject<API.Presence>(_serializer);
|
|
|
|
if (data.GuildId.IsSpecified)
|
|
{
|
|
var guild = State.GetGuild(data.GuildId.Value);
|
|
if (guild == null)
|
|
{
|
|
await UnknownGuildAsync(type, data.GuildId.Value).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
if (!guild.IsSynced)
|
|
{
|
|
await UnsyncedGuildAsync(type, guild.Id).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
|
|
var user = guild.GetUser(data.User.Id);
|
|
if (user == null)
|
|
{
|
|
if (data.Status == UserStatus.Offline)
|
|
{
|
|
return;
|
|
}
|
|
user = guild.AddOrUpdateUser(data);
|
|
}
|
|
else
|
|
{
|
|
var globalBefore = user.GlobalUser.Clone();
|
|
if (user.GlobalUser.Update(State, data.User))
|
|
{
|
|
//Global data was updated, trigger UserUpdated
|
|
await TimedInvokeAsync(_userUpdatedEvent, nameof(UserUpdated), globalBefore, user).ConfigureAwait(false);
|
|
}
|
|
}
|
|
|
|
var before = user.Clone();
|
|
user.Update(State, data, true);
|
|
await TimedInvokeAsync(_guildMemberUpdatedEvent, nameof(GuildMemberUpdated), before, user).ConfigureAwait(false);
|
|
}
|
|
else
|
|
{
|
|
var globalUser = State.GetUser(data.User.Id);
|
|
if (globalUser == null)
|
|
{
|
|
await UnknownGlobalUserAsync(type, data.User.Id).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
|
|
var before = globalUser.Clone();
|
|
globalUser.Update(State, data.User);
|
|
globalUser.Update(State, data);
|
|
await TimedInvokeAsync(_userUpdatedEvent, nameof(UserUpdated), before, globalUser).ConfigureAwait(false);
|
|
}
|
|
}
|
|
break;
|
|
case "TYPING_START":
|
|
{
|
|
await _gatewayLogger.DebugAsync("Received Dispatch (TYPING_START)").ConfigureAwait(false);
|
|
|
|
var data = (payload as JToken).ToObject<TypingStartEvent>(_serializer);
|
|
if (State.GetChannel(data.ChannelId) is ISocketMessageChannel channel)
|
|
{
|
|
var guild = (channel as SocketGuildChannel)?.Guild;
|
|
if (!(guild?.IsSynced ?? true))
|
|
{
|
|
await UnsyncedGuildAsync(type, guild.Id).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
|
|
var user = (channel as SocketChannel).GetUser(data.UserId);
|
|
if (user == null)
|
|
{
|
|
if (guild != null)
|
|
user = guild.AddOrUpdateUser(data.Member);
|
|
}
|
|
if (user != null)
|
|
await TimedInvokeAsync(_userIsTypingEvent, nameof(UserIsTyping), user, channel).ConfigureAwait(false);
|
|
}
|
|
}
|
|
break;
|
|
|
|
//Users
|
|
case "USER_UPDATE":
|
|
{
|
|
await _gatewayLogger.DebugAsync("Received Dispatch (USER_UPDATE)").ConfigureAwait(false);
|
|
|
|
var data = (payload as JToken).ToObject<API.User>(_serializer);
|
|
if (data.Id == CurrentUser.Id)
|
|
{
|
|
var before = CurrentUser.Clone();
|
|
CurrentUser.Update(State, data);
|
|
await TimedInvokeAsync(_selfUpdatedEvent, nameof(CurrentUserUpdated), before, CurrentUser).ConfigureAwait(false);
|
|
}
|
|
else
|
|
{
|
|
await _gatewayLogger.WarningAsync("Received USER_UPDATE for wrong user.").ConfigureAwait(false);
|
|
return;
|
|
}
|
|
}
|
|
break;
|
|
|
|
//Voice
|
|
case "VOICE_STATE_UPDATE":
|
|
{
|
|
await _gatewayLogger.DebugAsync("Received Dispatch (VOICE_STATE_UPDATE)").ConfigureAwait(false);
|
|
|
|
var data = (payload as JToken).ToObject<API.VoiceState>(_serializer);
|
|
SocketUser user;
|
|
SocketVoiceState before, after;
|
|
if (data.GuildId != null)
|
|
{
|
|
var guild = State.GetGuild(data.GuildId.Value);
|
|
if (guild == null)
|
|
{
|
|
await UnknownGuildAsync(type, data.GuildId.Value).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
else if (!guild.IsSynced)
|
|
{
|
|
await UnsyncedGuildAsync(type, guild.Id).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
|
|
if (data.ChannelId != null)
|
|
{
|
|
before = guild.GetVoiceState(data.UserId)?.Clone() ?? SocketVoiceState.Default;
|
|
after = await guild.AddOrUpdateVoiceStateAsync(State, data).ConfigureAwait(false);
|
|
/*if (data.UserId == CurrentUser.Id)
|
|
{
|
|
var _ = guild.FinishJoinAudioChannel().ConfigureAwait(false);
|
|
}*/
|
|
}
|
|
else
|
|
{
|
|
before = await guild.RemoveVoiceStateAsync(data.UserId).ConfigureAwait(false) ?? SocketVoiceState.Default;
|
|
after = SocketVoiceState.Create(null, data);
|
|
}
|
|
|
|
// per g250k, this should always be sent, but apparently not always
|
|
user = guild.GetUser(data.UserId)
|
|
?? (data.Member.IsSpecified ? guild.AddOrUpdateUser(data.Member.Value) : null);
|
|
if (user == null)
|
|
{
|
|
await UnknownGuildUserAsync(type, data.UserId, guild.Id).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
var groupChannel = State.GetChannel(data.ChannelId.Value) as SocketGroupChannel;
|
|
if (groupChannel == null)
|
|
{
|
|
await UnknownChannelAsync(type, data.ChannelId.Value).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
if (data.ChannelId != null)
|
|
{
|
|
before = groupChannel.GetVoiceState(data.UserId)?.Clone() ?? SocketVoiceState.Default;
|
|
after = groupChannel.AddOrUpdateVoiceState(State, data);
|
|
}
|
|
else
|
|
{
|
|
before = groupChannel.RemoveVoiceState(data.UserId) ?? SocketVoiceState.Default;
|
|
after = SocketVoiceState.Create(null, data);
|
|
}
|
|
user = groupChannel.GetUser(data.UserId);
|
|
if (user == null)
|
|
{
|
|
await UnknownChannelUserAsync(type, data.UserId, groupChannel.Id).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
}
|
|
|
|
await TimedInvokeAsync(_userVoiceStateUpdatedEvent, nameof(UserVoiceStateUpdated), user, before, after).ConfigureAwait(false);
|
|
}
|
|
break;
|
|
case "VOICE_SERVER_UPDATE":
|
|
{
|
|
await _gatewayLogger.DebugAsync("Received Dispatch (VOICE_SERVER_UPDATE)").ConfigureAwait(false);
|
|
|
|
var data = (payload as JToken).ToObject<VoiceServerUpdateEvent>(_serializer);
|
|
var guild = State.GetGuild(data.GuildId);
|
|
var isCached = guild != null;
|
|
var cachedGuild = new Cacheable<IGuild, ulong>(guild, data.GuildId, isCached,
|
|
() => Task.FromResult(State.GetGuild(data.GuildId) as IGuild));
|
|
|
|
var voiceServer = new SocketVoiceServer(cachedGuild, data.Endpoint, data.Token);
|
|
await TimedInvokeAsync(_voiceServerUpdatedEvent, nameof(UserVoiceStateUpdated), voiceServer).ConfigureAwait(false);
|
|
|
|
if (isCached)
|
|
{
|
|
var endpoint = data.Endpoint;
|
|
|
|
//Only strip out the port if the endpoint contains it
|
|
var portBegin = endpoint.LastIndexOf(':');
|
|
if (portBegin > 0)
|
|
endpoint = endpoint.Substring(0, portBegin);
|
|
|
|
var _ = guild.FinishConnectAudio(endpoint, data.Token).ConfigureAwait(false);
|
|
}
|
|
else
|
|
{
|
|
await UnknownGuildAsync(type, data.GuildId).ConfigureAwait(false);
|
|
}
|
|
|
|
}
|
|
break;
|
|
|
|
//Ignored (User only)
|
|
case "CHANNEL_PINS_ACK":
|
|
await _gatewayLogger.DebugAsync("Ignored Dispatch (CHANNEL_PINS_ACK)").ConfigureAwait(false);
|
|
break;
|
|
case "CHANNEL_PINS_UPDATE":
|
|
await _gatewayLogger.DebugAsync("Ignored Dispatch (CHANNEL_PINS_UPDATE)").ConfigureAwait(false);
|
|
break;
|
|
case "GUILD_INTEGRATIONS_UPDATE":
|
|
await _gatewayLogger.DebugAsync("Ignored Dispatch (GUILD_INTEGRATIONS_UPDATE)").ConfigureAwait(false);
|
|
break;
|
|
case "MESSAGE_ACK":
|
|
await _gatewayLogger.DebugAsync("Ignored Dispatch (MESSAGE_ACK)").ConfigureAwait(false);
|
|
break;
|
|
case "PRESENCES_REPLACE":
|
|
await _gatewayLogger.DebugAsync("Ignored Dispatch (PRESENCES_REPLACE)").ConfigureAwait(false);
|
|
break;
|
|
case "USER_SETTINGS_UPDATE":
|
|
await _gatewayLogger.DebugAsync("Ignored Dispatch (USER_SETTINGS_UPDATE)").ConfigureAwait(false);
|
|
break;
|
|
case "WEBHOOKS_UPDATE":
|
|
await _gatewayLogger.DebugAsync("Ignored Dispatch (WEBHOOKS_UPDATE)").ConfigureAwait(false);
|
|
break;
|
|
|
|
//Others
|
|
default:
|
|
await _gatewayLogger.WarningAsync($"Unknown Dispatch ({type})").ConfigureAwait(false);
|
|
break;
|
|
}
|
|
break;
|
|
default:
|
|
await _gatewayLogger.WarningAsync($"Unknown OpCode ({opCode})").ConfigureAwait(false);
|
|
break;
|
|
}
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
await _gatewayLogger.ErrorAsync($"Error handling {opCode}{(type != null ? $" ({type})" : "")}", ex).ConfigureAwait(false);
|
|
}
|
|
}
|
|
|
|
private async Task RunHeartbeatAsync(int intervalMillis, CancellationToken cancelToken)
|
|
{
|
|
try
|
|
{
|
|
await _gatewayLogger.DebugAsync("Heartbeat Started").ConfigureAwait(false);
|
|
while (!cancelToken.IsCancellationRequested)
|
|
{
|
|
int now = Environment.TickCount;
|
|
|
|
//Did server respond to our last heartbeat, or are we still receiving messages (long load?)
|
|
if (_heartbeatTimes.Count != 0 && (now - _lastMessageTime) > intervalMillis)
|
|
{
|
|
if (ConnectionState == ConnectionState.Connected && (_guildDownloadTask?.IsCompleted ?? true))
|
|
{
|
|
_connection.Error(new Exception("Server missed last heartbeat"));
|
|
return;
|
|
}
|
|
}
|
|
|
|
_heartbeatTimes.Enqueue(now);
|
|
try
|
|
{
|
|
await ApiClient.SendHeartbeatAsync(_lastSeq).ConfigureAwait(false);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
await _gatewayLogger.WarningAsync("Heartbeat Errored", ex).ConfigureAwait(false);
|
|
}
|
|
|
|
await Task.Delay(intervalMillis, cancelToken).ConfigureAwait(false);
|
|
}
|
|
await _gatewayLogger.DebugAsync("Heartbeat Stopped").ConfigureAwait(false);
|
|
}
|
|
catch (OperationCanceledException)
|
|
{
|
|
await _gatewayLogger.DebugAsync("Heartbeat Stopped").ConfigureAwait(false);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
await _gatewayLogger.ErrorAsync("Heartbeat Errored", ex).ConfigureAwait(false);
|
|
}
|
|
}
|
|
/*public async Task WaitForGuildsAsync()
|
|
{
|
|
var downloadTask = _guildDownloadTask;
|
|
if (downloadTask != null)
|
|
await _guildDownloadTask.ConfigureAwait(false);
|
|
}*/
|
|
private async Task WaitForGuildsAsync(CancellationToken cancelToken, Logger logger)
|
|
{
|
|
//Wait for GUILD_AVAILABLEs
|
|
try
|
|
{
|
|
await logger.DebugAsync("GuildDownloader Started").ConfigureAwait(false);
|
|
while ((_unavailableGuildCount != 0) && (Environment.TickCount - _lastGuildAvailableTime < 2000))
|
|
await Task.Delay(500, cancelToken).ConfigureAwait(false);
|
|
await logger.DebugAsync("GuildDownloader Stopped").ConfigureAwait(false);
|
|
}
|
|
catch (OperationCanceledException)
|
|
{
|
|
await logger.DebugAsync("GuildDownloader Stopped").ConfigureAwait(false);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
await logger.ErrorAsync("GuildDownloader Errored", ex).ConfigureAwait(false);
|
|
}
|
|
}
|
|
private async Task SyncGuildsAsync()
|
|
{
|
|
var guildIds = Guilds.Where(x => !x.IsSynced).Select(x => x.Id).ToImmutableArray();
|
|
if (guildIds.Length > 0)
|
|
await ApiClient.SendGuildSyncAsync(guildIds).ConfigureAwait(false);
|
|
}
|
|
|
|
internal SocketGuild AddGuild(ExtendedGuild model, ClientState state)
|
|
{
|
|
var guild = SocketGuild.Create(this, state, model);
|
|
state.AddGuild(guild);
|
|
if (model.Large)
|
|
_largeGuilds.Enqueue(model.Id);
|
|
return guild;
|
|
}
|
|
internal SocketGuild RemoveGuild(ulong id)
|
|
{
|
|
var guild = State.RemoveGuild(id);
|
|
if (guild != null)
|
|
{
|
|
foreach (var _ in guild.Channels)
|
|
State.RemoveChannel(id);
|
|
foreach (var user in guild.Users)
|
|
user.GlobalUser.RemoveRef(this);
|
|
}
|
|
return guild;
|
|
}
|
|
|
|
/// <exception cref="InvalidOperationException">Unexpected channel type is created.</exception>
|
|
internal ISocketPrivateChannel AddPrivateChannel(API.Channel model, ClientState state)
|
|
{
|
|
var channel = SocketChannel.CreatePrivate(this, state, model);
|
|
state.AddChannel(channel as SocketChannel);
|
|
if (channel is SocketDMChannel dm)
|
|
dm.Recipient.GlobalUser.DMChannel = dm;
|
|
|
|
return channel;
|
|
}
|
|
internal ISocketPrivateChannel RemovePrivateChannel(ulong id)
|
|
{
|
|
var channel = State.RemoveChannel(id) as ISocketPrivateChannel;
|
|
if (channel != null)
|
|
{
|
|
if (channel is SocketDMChannel dmChannel)
|
|
dmChannel.Recipient.GlobalUser.DMChannel = null;
|
|
|
|
foreach (var recipient in channel.Recipients)
|
|
recipient.GlobalUser.RemoveRef(this);
|
|
}
|
|
return channel;
|
|
}
|
|
|
|
private async Task GuildAvailableAsync(SocketGuild guild)
|
|
{
|
|
if (!guild.IsConnected)
|
|
{
|
|
guild.IsConnected = true;
|
|
await TimedInvokeAsync(_guildAvailableEvent, nameof(GuildAvailable), guild).ConfigureAwait(false);
|
|
}
|
|
}
|
|
private async Task GuildUnavailableAsync(SocketGuild guild)
|
|
{
|
|
if (guild.IsConnected)
|
|
{
|
|
guild.IsConnected = false;
|
|
await TimedInvokeAsync(_guildUnavailableEvent, nameof(GuildUnavailable), guild).ConfigureAwait(false);
|
|
}
|
|
}
|
|
|
|
private async Task TimedInvokeAsync(AsyncEvent<Func<Task>> eventHandler, string name)
|
|
{
|
|
if (eventHandler.HasSubscribers)
|
|
{
|
|
if (HandlerTimeout.HasValue)
|
|
await TimeoutWrap(name, eventHandler.InvokeAsync).ConfigureAwait(false);
|
|
else
|
|
await eventHandler.InvokeAsync().ConfigureAwait(false);
|
|
}
|
|
}
|
|
private async Task TimedInvokeAsync<T>(AsyncEvent<Func<T, Task>> eventHandler, string name, T arg)
|
|
{
|
|
if (eventHandler.HasSubscribers)
|
|
{
|
|
if (HandlerTimeout.HasValue)
|
|
await TimeoutWrap(name, () => eventHandler.InvokeAsync(arg)).ConfigureAwait(false);
|
|
else
|
|
await eventHandler.InvokeAsync(arg).ConfigureAwait(false);
|
|
}
|
|
}
|
|
private async Task TimedInvokeAsync<T1, T2>(AsyncEvent<Func<T1, T2, Task>> eventHandler, string name, T1 arg1, T2 arg2)
|
|
{
|
|
if (eventHandler.HasSubscribers)
|
|
{
|
|
if (HandlerTimeout.HasValue)
|
|
await TimeoutWrap(name, () => eventHandler.InvokeAsync(arg1, arg2)).ConfigureAwait(false);
|
|
else
|
|
await eventHandler.InvokeAsync(arg1, arg2).ConfigureAwait(false);
|
|
}
|
|
}
|
|
private async Task TimedInvokeAsync<T1, T2, T3>(AsyncEvent<Func<T1, T2, T3, Task>> eventHandler, string name, T1 arg1, T2 arg2, T3 arg3)
|
|
{
|
|
if (eventHandler.HasSubscribers)
|
|
{
|
|
if (HandlerTimeout.HasValue)
|
|
await TimeoutWrap(name, () => eventHandler.InvokeAsync(arg1, arg2, arg3)).ConfigureAwait(false);
|
|
else
|
|
await eventHandler.InvokeAsync(arg1, arg2, arg3).ConfigureAwait(false);
|
|
}
|
|
}
|
|
private async Task TimedInvokeAsync<T1, T2, T3, T4>(AsyncEvent<Func<T1, T2, T3, T4, Task>> eventHandler, string name, T1 arg1, T2 arg2, T3 arg3, T4 arg4)
|
|
{
|
|
if (eventHandler.HasSubscribers)
|
|
{
|
|
if (HandlerTimeout.HasValue)
|
|
await TimeoutWrap(name, () => eventHandler.InvokeAsync(arg1, arg2, arg3, arg4)).ConfigureAwait(false);
|
|
else
|
|
await eventHandler.InvokeAsync(arg1, arg2, arg3, arg4).ConfigureAwait(false);
|
|
}
|
|
}
|
|
private async Task TimedInvokeAsync<T1, T2, T3, T4, T5>(AsyncEvent<Func<T1, T2, T3, T4, T5, Task>> eventHandler, string name, T1 arg1, T2 arg2, T3 arg3, T4 arg4, T5 arg5)
|
|
{
|
|
if (eventHandler.HasSubscribers)
|
|
{
|
|
if (HandlerTimeout.HasValue)
|
|
await TimeoutWrap(name, () => eventHandler.InvokeAsync(arg1, arg2, arg3, arg4, arg5)).ConfigureAwait(false);
|
|
else
|
|
await eventHandler.InvokeAsync(arg1, arg2, arg3, arg4, arg5).ConfigureAwait(false);
|
|
}
|
|
}
|
|
private async Task TimeoutWrap(string name, Func<Task> action)
|
|
{
|
|
try
|
|
{
|
|
var timeoutTask = Task.Delay(HandlerTimeout.Value);
|
|
var handlersTask = action();
|
|
if (await Task.WhenAny(timeoutTask, handlersTask).ConfigureAwait(false) == timeoutTask)
|
|
{
|
|
await _gatewayLogger.WarningAsync($"A {name} handler is blocking the gateway task.").ConfigureAwait(false);
|
|
}
|
|
await handlersTask.ConfigureAwait(false); //Ensure the handler completes
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
await _gatewayLogger.WarningAsync($"A {name} handler has thrown an unhandled exception.", ex).ConfigureAwait(false);
|
|
}
|
|
}
|
|
|
|
private async Task UnknownGlobalUserAsync(string evnt, ulong userId)
|
|
{
|
|
string details = $"{evnt} User={userId}";
|
|
await _gatewayLogger.WarningAsync($"Unknown User ({details}).").ConfigureAwait(false);
|
|
}
|
|
private async Task UnknownChannelUserAsync(string evnt, ulong userId, ulong channelId)
|
|
{
|
|
string details = $"{evnt} User={userId} Channel={channelId}";
|
|
await _gatewayLogger.WarningAsync($"Unknown User ({details}).").ConfigureAwait(false);
|
|
}
|
|
private async Task UnknownGuildUserAsync(string evnt, ulong userId, ulong guildId)
|
|
{
|
|
string details = $"{evnt} User={userId} Guild={guildId}";
|
|
await _gatewayLogger.WarningAsync($"Unknown User ({details}).").ConfigureAwait(false);
|
|
}
|
|
private async Task IncompleteGuildUserAsync(string evnt, ulong userId, ulong guildId)
|
|
{
|
|
string details = $"{evnt} User={userId} Guild={guildId}";
|
|
await _gatewayLogger.DebugAsync($"User has not been downloaded ({details}).").ConfigureAwait(false);
|
|
}
|
|
private async Task UnknownChannelAsync(string evnt, ulong channelId)
|
|
{
|
|
string details = $"{evnt} Channel={channelId}";
|
|
await _gatewayLogger.WarningAsync($"Unknown Channel ({details}).").ConfigureAwait(false);
|
|
}
|
|
private async Task UnknownChannelAsync(string evnt, ulong channelId, ulong guildId)
|
|
{
|
|
if (guildId == 0)
|
|
{
|
|
await UnknownChannelAsync(evnt, channelId).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
string details = $"{evnt} Channel={channelId} Guild={guildId}";
|
|
await _gatewayLogger.WarningAsync($"Unknown Channel ({details}).").ConfigureAwait(false);
|
|
}
|
|
private async Task UnknownRoleAsync(string evnt, ulong roleId, ulong guildId)
|
|
{
|
|
string details = $"{evnt} Role={roleId} Guild={guildId}";
|
|
await _gatewayLogger.WarningAsync($"Unknown Role ({details}).").ConfigureAwait(false);
|
|
}
|
|
private async Task UnknownGuildAsync(string evnt, ulong guildId)
|
|
{
|
|
string details = $"{evnt} Guild={guildId}";
|
|
await _gatewayLogger.WarningAsync($"Unknown Guild ({details}).").ConfigureAwait(false);
|
|
}
|
|
private async Task UnsyncedGuildAsync(string evnt, ulong guildId)
|
|
{
|
|
string details = $"{evnt} Guild={guildId}";
|
|
await _gatewayLogger.DebugAsync($"Unsynced Guild ({details}).").ConfigureAwait(false);
|
|
}
|
|
|
|
internal int GetAudioId() => _nextAudioId++;
|
|
|
|
//IDiscordClient
|
|
/// <inheritdoc />
|
|
async Task<IApplication> IDiscordClient.GetApplicationInfoAsync(RequestOptions options)
|
|
=> await GetApplicationInfoAsync().ConfigureAwait(false);
|
|
|
|
/// <inheritdoc />
|
|
Task<IChannel> IDiscordClient.GetChannelAsync(ulong id, CacheMode mode, RequestOptions options)
|
|
=> Task.FromResult<IChannel>(GetChannel(id));
|
|
/// <inheritdoc />
|
|
Task<IReadOnlyCollection<IPrivateChannel>> IDiscordClient.GetPrivateChannelsAsync(CacheMode mode, RequestOptions options)
|
|
=> Task.FromResult<IReadOnlyCollection<IPrivateChannel>>(PrivateChannels);
|
|
/// <inheritdoc />
|
|
Task<IReadOnlyCollection<IDMChannel>> IDiscordClient.GetDMChannelsAsync(CacheMode mode, RequestOptions options)
|
|
=> Task.FromResult<IReadOnlyCollection<IDMChannel>>(DMChannels);
|
|
/// <inheritdoc />
|
|
Task<IReadOnlyCollection<IGroupChannel>> IDiscordClient.GetGroupChannelsAsync(CacheMode mode, RequestOptions options)
|
|
=> Task.FromResult<IReadOnlyCollection<IGroupChannel>>(GroupChannels);
|
|
|
|
/// <inheritdoc />
|
|
async Task<IReadOnlyCollection<IConnection>> IDiscordClient.GetConnectionsAsync(RequestOptions options)
|
|
=> await GetConnectionsAsync().ConfigureAwait(false);
|
|
|
|
/// <inheritdoc />
|
|
async Task<IInvite> IDiscordClient.GetInviteAsync(string inviteId, RequestOptions options)
|
|
=> await GetInviteAsync(inviteId, options).ConfigureAwait(false);
|
|
|
|
/// <inheritdoc />
|
|
Task<IGuild> IDiscordClient.GetGuildAsync(ulong id, CacheMode mode, RequestOptions options)
|
|
=> Task.FromResult<IGuild>(GetGuild(id));
|
|
/// <inheritdoc />
|
|
Task<IReadOnlyCollection<IGuild>> IDiscordClient.GetGuildsAsync(CacheMode mode, RequestOptions options)
|
|
=> Task.FromResult<IReadOnlyCollection<IGuild>>(Guilds);
|
|
/// <inheritdoc />
|
|
async Task<IGuild> IDiscordClient.CreateGuildAsync(string name, IVoiceRegion region, Stream jpegIcon, RequestOptions options)
|
|
=> await CreateGuildAsync(name, region, jpegIcon).ConfigureAwait(false);
|
|
|
|
/// <inheritdoc />
|
|
Task<IUser> IDiscordClient.GetUserAsync(ulong id, CacheMode mode, RequestOptions options)
|
|
=> Task.FromResult<IUser>(GetUser(id));
|
|
/// <inheritdoc />
|
|
Task<IUser> IDiscordClient.GetUserAsync(string username, string discriminator, RequestOptions options)
|
|
=> Task.FromResult<IUser>(GetUser(username, discriminator));
|
|
|
|
/// <inheritdoc />
|
|
Task<IReadOnlyCollection<IVoiceRegion>> IDiscordClient.GetVoiceRegionsAsync(RequestOptions options)
|
|
=> Task.FromResult<IReadOnlyCollection<IVoiceRegion>>(VoiceRegions);
|
|
/// <inheritdoc />
|
|
Task<IVoiceRegion> IDiscordClient.GetVoiceRegionAsync(string id, RequestOptions options)
|
|
=> Task.FromResult<IVoiceRegion>(GetVoiceRegion(id));
|
|
|
|
/// <inheritdoc />
|
|
async Task IDiscordClient.StartAsync()
|
|
=> await StartAsync().ConfigureAwait(false);
|
|
/// <inheritdoc />
|
|
async Task IDiscordClient.StopAsync()
|
|
=> await StopAsync().ConfigureAwait(false);
|
|
}
|
|
}
|