You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

DiscordSocketClient.cs 102 kB

8 years ago
8 years ago
8 years ago
8 years ago
8 years ago
9 years ago
9 years ago
8 years ago
8 years ago
8 years ago
8 years ago
8 years ago
8 years ago
9 years ago
9 years ago
9 years ago
8 years ago
8 years ago
8 years ago
8 years ago
8 years ago
8 years ago
8 years ago
8 years ago
8 years ago
8 years ago
8 years ago
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826
  1. #pragma warning disable CS0618
  2. using Discord.API;
  3. using Discord.API.Gateway;
  4. using Discord.Logging;
  5. using Discord.Net.Converters;
  6. using Discord.Net.Udp;
  7. using Discord.Net.WebSockets;
  8. using Discord.Rest;
  9. using Newtonsoft.Json;
  10. using Newtonsoft.Json.Linq;
  11. using System;
  12. using System.Collections.Concurrent;
  13. using System.Collections.Generic;
  14. using System.Collections.Immutable;
  15. using System.IO;
  16. using System.Linq;
  17. using System.Threading;
  18. using System.Threading.Tasks;
  19. using GameModel = Discord.API.Game;
  20. namespace Discord.WebSocket
  21. {
  22. public partial class DiscordSocketClient : BaseSocketClient, IDiscordClient
  23. {
  24. private readonly ConcurrentQueue<ulong> _largeGuilds;
  25. private readonly JsonSerializer _serializer;
  26. private readonly SemaphoreSlim _connectionGroupLock;
  27. private readonly DiscordSocketClient _parentClient;
  28. private readonly ConcurrentQueue<long> _heartbeatTimes;
  29. private readonly ConnectionManager _connection;
  30. private readonly Logger _gatewayLogger;
  31. private readonly SemaphoreSlim _stateLock;
  32. private string _sessionId;
  33. private int _lastSeq;
  34. private ImmutableDictionary<string, RestVoiceRegion> _voiceRegions;
  35. private Task _heartbeatTask, _guildDownloadTask;
  36. private int _unavailableGuildCount;
  37. private long _lastGuildAvailableTime, _lastMessageTime;
  38. private int _nextAudioId;
  39. private DateTimeOffset? _statusSince;
  40. private RestApplication _applicationInfo;
  41. /// <summary> Gets the shard of of this client. </summary>
  42. public int ShardId { get; }
  43. /// <summary> Gets the current connection state of this client. </summary>
  44. public ConnectionState ConnectionState => _connection.State;
  45. /// <inheritdoc />
  46. public override int Latency { get; protected set; }
  47. public override UserStatus Status { get; protected set; } = UserStatus.Online;
  48. public override Game? Game { get; protected set; }
  49. //From DiscordSocketConfig
  50. internal int TotalShards { get; private set; }
  51. internal int MessageCacheSize { get; private set; }
  52. internal int LargeThreshold { get; private set; }
  53. internal ClientState State { get; private set; }
  54. internal UdpSocketProvider UdpSocketProvider { get; private set; }
  55. internal WebSocketProvider WebSocketProvider { get; private set; }
  56. internal bool AlwaysDownloadUsers { get; private set; }
  57. internal int? HandlerTimeout { get; private set; }
  58. internal new DiscordSocketApiClient ApiClient => base.ApiClient as DiscordSocketApiClient;
  59. public override IReadOnlyCollection<SocketGuild> Guilds => State.Guilds;
  60. public override IReadOnlyCollection<ISocketPrivateChannel> PrivateChannels => State.PrivateChannels;
  61. public IReadOnlyCollection<SocketDMChannel> DMChannels
  62. => State.PrivateChannels.Select(x => x as SocketDMChannel).Where(x => x != null).ToImmutableArray();
  63. public IReadOnlyCollection<SocketGroupChannel> GroupChannels
  64. => State.PrivateChannels.Select(x => x as SocketGroupChannel).Where(x => x != null).ToImmutableArray();
  65. public override IReadOnlyCollection<RestVoiceRegion> VoiceRegions => _voiceRegions.ToReadOnlyCollection();
  66. /// <summary> Creates a new REST/WebSocket discord client. </summary>
  67. public DiscordSocketClient() : this(new DiscordSocketConfig()) { }
  68. /// <summary> Creates a new REST/WebSocket discord client. </summary>
  69. public DiscordSocketClient(DiscordSocketConfig config) : this(config, CreateApiClient(config), null, null) { }
  70. internal DiscordSocketClient(DiscordSocketConfig config, SemaphoreSlim groupLock, DiscordSocketClient parentClient) : this(config, CreateApiClient(config), groupLock, parentClient) { }
  71. private DiscordSocketClient(DiscordSocketConfig config, API.DiscordSocketApiClient client, SemaphoreSlim groupLock, DiscordSocketClient parentClient)
  72. : base(config, client)
  73. {
  74. ShardId = config.ShardId ?? 0;
  75. TotalShards = config.TotalShards ?? 1;
  76. MessageCacheSize = config.MessageCacheSize;
  77. LargeThreshold = config.LargeThreshold;
  78. UdpSocketProvider = config.UdpSocketProvider;
  79. WebSocketProvider = config.WebSocketProvider;
  80. AlwaysDownloadUsers = config.AlwaysDownloadUsers;
  81. HandlerTimeout = config.HandlerTimeout;
  82. State = new ClientState(0, 0);
  83. _heartbeatTimes = new ConcurrentQueue<long>();
  84. _stateLock = new SemaphoreSlim(1, 1);
  85. _gatewayLogger = LogManager.CreateLogger(ShardId == 0 && TotalShards == 1 ? "Gateway" : $"Shard #{ShardId}");
  86. _connection = new ConnectionManager(_stateLock, _gatewayLogger, config.ConnectionTimeout,
  87. OnConnectingAsync, OnDisconnectingAsync, x => ApiClient.Disconnected += x);
  88. _connection.Connected += () => TimedInvokeAsync(_connectedEvent, nameof(Connected));
  89. _connection.Disconnected += (ex, recon) => TimedInvokeAsync(_disconnectedEvent, nameof(Disconnected), ex);
  90. _nextAudioId = 1;
  91. _connectionGroupLock = groupLock;
  92. _parentClient = parentClient;
  93. _serializer = new JsonSerializer { ContractResolver = new DiscordContractResolver() };
  94. _serializer.Error += (s, e) =>
  95. {
  96. _gatewayLogger.WarningAsync("Serializer Error", e.ErrorContext.Error).GetAwaiter().GetResult();
  97. e.ErrorContext.Handled = true;
  98. };
  99. ApiClient.SentGatewayMessage += async opCode => await _gatewayLogger.DebugAsync($"Sent {opCode}").ConfigureAwait(false);
  100. ApiClient.ReceivedGatewayEvent += ProcessMessageAsync;
  101. LeftGuild += async g => await _gatewayLogger.InfoAsync($"Left {g.Name}").ConfigureAwait(false);
  102. JoinedGuild += async g => await _gatewayLogger.InfoAsync($"Joined {g.Name}").ConfigureAwait(false);
  103. GuildAvailable += async g => await _gatewayLogger.VerboseAsync($"Connected to {g.Name}").ConfigureAwait(false);
  104. GuildUnavailable += async g => await _gatewayLogger.VerboseAsync($"Disconnected from {g.Name}").ConfigureAwait(false);
  105. LatencyUpdated += async (old, val) => await _gatewayLogger.DebugAsync($"Latency = {val} ms").ConfigureAwait(false);
  106. GuildAvailable += g =>
  107. {
  108. if (ConnectionState == ConnectionState.Connected && AlwaysDownloadUsers && !g.HasAllMembers)
  109. {
  110. var _ = g.DownloadUsersAsync();
  111. }
  112. return Task.Delay(0);
  113. };
  114. _voiceRegions = ImmutableDictionary.Create<string, RestVoiceRegion>();
  115. _largeGuilds = new ConcurrentQueue<ulong>();
  116. }
  117. private static API.DiscordSocketApiClient CreateApiClient(DiscordSocketConfig config)
  118. => new API.DiscordSocketApiClient(config.RestClientProvider, config.WebSocketProvider, DiscordRestConfig.UserAgent, config.GatewayHost);
  119. internal override void Dispose(bool disposing)
  120. {
  121. if (disposing)
  122. {
  123. StopAsync().GetAwaiter().GetResult();
  124. ApiClient.Dispose();
  125. }
  126. }
  127. internal override async Task OnLoginAsync(TokenType tokenType, string token)
  128. {
  129. if (_parentClient == null)
  130. {
  131. var voiceRegions = await ApiClient.GetVoiceRegionsAsync(new RequestOptions { IgnoreState = true, RetryMode = RetryMode.AlwaysRetry }).ConfigureAwait(false);
  132. _voiceRegions = voiceRegions.Select(x => RestVoiceRegion.Create(this, x)).ToImmutableDictionary(x => x.Id);
  133. }
  134. else
  135. _voiceRegions = _parentClient._voiceRegions;
  136. }
  137. internal override async Task OnLogoutAsync()
  138. {
  139. await StopAsync().ConfigureAwait(false);
  140. _applicationInfo = null;
  141. _voiceRegions = ImmutableDictionary.Create<string, RestVoiceRegion>();
  142. }
  143. public override async Task StartAsync()
  144. => await _connection.StartAsync().ConfigureAwait(false);
  145. public override async Task StopAsync()
  146. => await _connection.StopAsync().ConfigureAwait(false);
  147. private async Task OnConnectingAsync()
  148. {
  149. if (_connectionGroupLock != null)
  150. await _connectionGroupLock.WaitAsync(_connection.CancelToken).ConfigureAwait(false);
  151. try
  152. {
  153. await _gatewayLogger.DebugAsync("Connecting ApiClient").ConfigureAwait(false);
  154. await ApiClient.ConnectAsync().ConfigureAwait(false);
  155. if (_sessionId != null)
  156. {
  157. await _gatewayLogger.DebugAsync("Resuming").ConfigureAwait(false);
  158. await ApiClient.SendResumeAsync(_sessionId, _lastSeq).ConfigureAwait(false);
  159. }
  160. else
  161. {
  162. await _gatewayLogger.DebugAsync("Identifying").ConfigureAwait(false);
  163. await ApiClient.SendIdentifyAsync(shardID: ShardId, totalShards: TotalShards).ConfigureAwait(false);
  164. }
  165. //Wait for READY
  166. await _connection.WaitAsync().ConfigureAwait(false);
  167. await _gatewayLogger.DebugAsync("Sending Status").ConfigureAwait(false);
  168. await SendStatusAsync().ConfigureAwait(false);
  169. }
  170. finally
  171. {
  172. if (_connectionGroupLock != null)
  173. {
  174. await Task.Delay(5000).ConfigureAwait(false);
  175. _connectionGroupLock.Release();
  176. }
  177. }
  178. }
  179. private async Task OnDisconnectingAsync(Exception ex)
  180. {
  181. await _gatewayLogger.DebugAsync("Disconnecting ApiClient").ConfigureAwait(false);
  182. await ApiClient.DisconnectAsync().ConfigureAwait(false);
  183. //Wait for tasks to complete
  184. await _gatewayLogger.DebugAsync("Waiting for heartbeater").ConfigureAwait(false);
  185. var heartbeatTask = _heartbeatTask;
  186. if (heartbeatTask != null)
  187. await heartbeatTask.ConfigureAwait(false);
  188. _heartbeatTask = null;
  189. while (_heartbeatTimes.TryDequeue(out long time)) { }
  190. _lastMessageTime = 0;
  191. await _gatewayLogger.DebugAsync("Waiting for guild downloader").ConfigureAwait(false);
  192. var guildDownloadTask = _guildDownloadTask;
  193. if (guildDownloadTask != null)
  194. await guildDownloadTask.ConfigureAwait(false);
  195. _guildDownloadTask = null;
  196. //Clear large guild queue
  197. await _gatewayLogger.DebugAsync("Clearing large guild queue").ConfigureAwait(false);
  198. while (_largeGuilds.TryDequeue(out ulong guildId)) { }
  199. //Raise virtual GUILD_UNAVAILABLEs
  200. await _gatewayLogger.DebugAsync("Raising virtual GuildUnavailables").ConfigureAwait(false);
  201. foreach (var guild in State.Guilds)
  202. {
  203. if (guild.IsAvailable)
  204. await GuildUnavailableAsync(guild).ConfigureAwait(false);
  205. }
  206. }
  207. /// <inheritdoc />
  208. public override async Task<RestApplication> GetApplicationInfoAsync(RequestOptions options = null)
  209. => _applicationInfo ?? (_applicationInfo = await ClientHelper.GetApplicationInfoAsync(this, options ?? RequestOptions.Default).ConfigureAwait(false));
  210. /// <inheritdoc />
  211. public override SocketGuild GetGuild(ulong id)
  212. => State.GetGuild(id);
  213. /// <inheritdoc />
  214. public override SocketChannel GetChannel(ulong id)
  215. => State.GetChannel(id);
  216. /// <inheritdoc />
  217. public override SocketUser GetUser(ulong id)
  218. => State.GetUser(id);
  219. /// <inheritdoc />
  220. public override SocketUser GetUser(string username, string discriminator)
  221. => State.Users.FirstOrDefault(x => x.Discriminator == discriminator && x.Username == username);
  222. internal SocketGlobalUser GetOrCreateUser(ClientState state, Discord.API.User model)
  223. {
  224. return state.GetOrAddUser(model.Id, x =>
  225. {
  226. var user = SocketGlobalUser.Create(this, state, model);
  227. user.GlobalUser.AddRef();
  228. return user;
  229. });
  230. }
  231. internal SocketGlobalUser GetOrCreateSelfUser(ClientState state, Discord.API.User model)
  232. {
  233. return state.GetOrAddUser(model.Id, x =>
  234. {
  235. var user = SocketGlobalUser.Create(this, state, model);
  236. user.GlobalUser.AddRef();
  237. user.Presence = new SocketPresence(UserStatus.Online, null);
  238. return user;
  239. });
  240. }
  241. internal void RemoveUser(ulong id)
  242. => State.RemoveUser(id);
  243. /// <inheritdoc />
  244. public override RestVoiceRegion GetVoiceRegion(string id)
  245. {
  246. if (_voiceRegions.TryGetValue(id, out RestVoiceRegion region))
  247. return region;
  248. return null;
  249. }
  250. /// <summary> Downloads the users list for the provided guilds, if they don't have a complete list. </summary>
  251. public override async Task DownloadUsersAsync(IEnumerable<IGuild> guilds)
  252. {
  253. if (ConnectionState == ConnectionState.Connected)
  254. {
  255. //Race condition leads to guilds being requested twice, probably okay
  256. await ProcessUserDownloadsAsync(guilds.Select(x => GetGuild(x.Id)).Where(x => x != null)).ConfigureAwait(false);
  257. }
  258. }
  259. private async Task ProcessUserDownloadsAsync(IEnumerable<SocketGuild> guilds)
  260. {
  261. var cachedGuilds = guilds.ToImmutableArray();
  262. const short batchSize = 50;
  263. ulong[] batchIds = new ulong[Math.Min(batchSize, cachedGuilds.Length)];
  264. Task[] batchTasks = new Task[batchIds.Length];
  265. int batchCount = (cachedGuilds.Length + (batchSize - 1)) / batchSize;
  266. for (int i = 0, k = 0; i < batchCount; i++)
  267. {
  268. bool isLast = i == batchCount - 1;
  269. int count = isLast ? (batchIds.Length - (batchCount - 1) * batchSize) : batchSize;
  270. for (int j = 0; j < count; j++, k++)
  271. {
  272. var guild = cachedGuilds[k];
  273. batchIds[j] = guild.Id;
  274. batchTasks[j] = guild.DownloaderPromise;
  275. }
  276. await ApiClient.SendRequestMembersAsync(batchIds).ConfigureAwait(false);
  277. if (isLast && batchCount > 1)
  278. await Task.WhenAll(batchTasks.Take(count)).ConfigureAwait(false);
  279. else
  280. await Task.WhenAll(batchTasks).ConfigureAwait(false);
  281. }
  282. }
  283. public override async Task SetStatusAsync(UserStatus status)
  284. {
  285. Status = status;
  286. if (status == UserStatus.AFK)
  287. _statusSince = DateTimeOffset.UtcNow;
  288. else
  289. _statusSince = null;
  290. await SendStatusAsync().ConfigureAwait(false);
  291. }
  292. public override async Task SetGameAsync(string name, string streamUrl = null, StreamType streamType = StreamType.NotStreaming)
  293. {
  294. if (name != null)
  295. Game = new Game(name, streamUrl, streamType);
  296. else
  297. Game = null;
  298. await SendStatusAsync().ConfigureAwait(false);
  299. }
  300. private async Task SendStatusAsync()
  301. {
  302. if (CurrentUser == null)
  303. return;
  304. var game = Game;
  305. var status = Status;
  306. var statusSince = _statusSince;
  307. CurrentUser.Presence = new SocketPresence(status, game);
  308. GameModel gameModel;
  309. if (game != null)
  310. {
  311. gameModel = new API.Game
  312. {
  313. Name = game.Value.Name,
  314. StreamType = game.Value.StreamType,
  315. StreamUrl = game.Value.StreamUrl
  316. };
  317. }
  318. else
  319. gameModel = null;
  320. await ApiClient.SendStatusUpdateAsync(
  321. status,
  322. status == UserStatus.AFK,
  323. statusSince != null ? DateTimeUtils.ToUnixMilliseconds(_statusSince.Value) : (long?)null,
  324. gameModel).ConfigureAwait(false);
  325. }
  326. private async Task ProcessMessageAsync(GatewayOpCode opCode, int? seq, string type, object payload)
  327. {
  328. if (seq != null)
  329. _lastSeq = seq.Value;
  330. _lastMessageTime = Environment.TickCount;
  331. try
  332. {
  333. switch (opCode)
  334. {
  335. case GatewayOpCode.Hello:
  336. {
  337. await _gatewayLogger.DebugAsync("Received Hello").ConfigureAwait(false);
  338. var data = (payload as JToken).ToObject<HelloEvent>(_serializer);
  339. _heartbeatTask = RunHeartbeatAsync(data.HeartbeatInterval, _connection.CancelToken);
  340. }
  341. break;
  342. case GatewayOpCode.Heartbeat:
  343. {
  344. await _gatewayLogger.DebugAsync("Received Heartbeat").ConfigureAwait(false);
  345. await ApiClient.SendHeartbeatAsync(_lastSeq).ConfigureAwait(false);
  346. }
  347. break;
  348. case GatewayOpCode.HeartbeatAck:
  349. {
  350. await _gatewayLogger.DebugAsync("Received HeartbeatAck").ConfigureAwait(false);
  351. if (_heartbeatTimes.TryDequeue(out long time))
  352. {
  353. int latency = (int)(Environment.TickCount - time);
  354. int before = Latency;
  355. Latency = latency;
  356. await TimedInvokeAsync(_latencyUpdatedEvent, nameof(LatencyUpdated), before, latency).ConfigureAwait(false);
  357. }
  358. }
  359. break;
  360. case GatewayOpCode.InvalidSession:
  361. {
  362. await _gatewayLogger.DebugAsync("Received InvalidSession").ConfigureAwait(false);
  363. await _gatewayLogger.WarningAsync("Failed to resume previous session").ConfigureAwait(false);
  364. _sessionId = null;
  365. _lastSeq = 0;
  366. bool retry = (bool)payload;
  367. if (retry)
  368. _connection.Reconnect(); //TODO: Untested
  369. else
  370. await ApiClient.SendIdentifyAsync(shardID: ShardId, totalShards: TotalShards).ConfigureAwait(false);
  371. }
  372. break;
  373. case GatewayOpCode.Reconnect:
  374. {
  375. await _gatewayLogger.DebugAsync("Received Reconnect").ConfigureAwait(false);
  376. _connection.Error(new Exception("Server requested a reconnect"));
  377. }
  378. break;
  379. case GatewayOpCode.Dispatch:
  380. switch (type)
  381. {
  382. //Connection
  383. case "READY":
  384. {
  385. try
  386. {
  387. await _gatewayLogger.DebugAsync("Received Dispatch (READY)").ConfigureAwait(false);
  388. var data = (payload as JToken).ToObject<ReadyEvent>(_serializer);
  389. var state = new ClientState(data.Guilds.Length, data.PrivateChannels.Length);
  390. var currentUser = SocketSelfUser.Create(this, state, data.User);
  391. ApiClient.CurrentUserId = currentUser.Id;
  392. int unavailableGuilds = 0;
  393. for (int i = 0; i < data.Guilds.Length; i++)
  394. {
  395. var model = data.Guilds[i];
  396. var guild = AddGuild(model, state);
  397. if (!guild.IsAvailable || ApiClient.AuthTokenType == TokenType.User)
  398. unavailableGuilds++;
  399. else
  400. await GuildAvailableAsync(guild).ConfigureAwait(false);
  401. }
  402. for (int i = 0; i < data.PrivateChannels.Length; i++)
  403. AddPrivateChannel(data.PrivateChannels[i], state);
  404. _sessionId = data.SessionId;
  405. _unavailableGuildCount = unavailableGuilds;
  406. CurrentUser = currentUser;
  407. State = state;
  408. }
  409. catch (Exception ex)
  410. {
  411. _connection.CriticalError(new Exception("Processing READY failed", ex));
  412. return;
  413. }
  414. if (ApiClient.AuthTokenType == TokenType.User)
  415. await SyncGuildsAsync().ConfigureAwait(false);
  416. _lastGuildAvailableTime = Environment.TickCount;
  417. _guildDownloadTask = WaitForGuildsAsync(_connection.CancelToken, _gatewayLogger)
  418. .ContinueWith(async x =>
  419. {
  420. if (x.IsFaulted)
  421. {
  422. _connection.Error(x.Exception);
  423. return;
  424. }
  425. else if (_connection.CancelToken.IsCancellationRequested)
  426. return;
  427. await TimedInvokeAsync(_readyEvent, nameof(Ready)).ConfigureAwait(false);
  428. await _gatewayLogger.InfoAsync("Ready").ConfigureAwait(false);
  429. });
  430. var _ = _connection.CompleteAsync();
  431. }
  432. break;
  433. case "RESUMED":
  434. {
  435. await _gatewayLogger.DebugAsync("Received Dispatch (RESUMED)").ConfigureAwait(false);
  436. var _ = _connection.CompleteAsync();
  437. //Notify the client that these guilds are available again
  438. foreach (var guild in State.Guilds)
  439. {
  440. if (guild.IsAvailable)
  441. await GuildAvailableAsync(guild).ConfigureAwait(false);
  442. }
  443. await _gatewayLogger.InfoAsync("Resumed previous session").ConfigureAwait(false);
  444. }
  445. break;
  446. //Guilds
  447. case "GUILD_CREATE":
  448. {
  449. var data = (payload as JToken).ToObject<ExtendedGuild>(_serializer);
  450. if (data.Unavailable == false)
  451. {
  452. type = "GUILD_AVAILABLE";
  453. _lastGuildAvailableTime = Environment.TickCount;
  454. await _gatewayLogger.DebugAsync($"Received Dispatch (GUILD_AVAILABLE)").ConfigureAwait(false);
  455. var guild = State.GetGuild(data.Id);
  456. if (guild != null)
  457. {
  458. guild.Update(State, data);
  459. if (_unavailableGuildCount != 0)
  460. _unavailableGuildCount--;
  461. await GuildAvailableAsync(guild).ConfigureAwait(false);
  462. if (guild.DownloadedMemberCount >= guild.MemberCount && !guild.DownloaderPromise.IsCompleted)
  463. {
  464. guild.CompleteDownloadUsers();
  465. await TimedInvokeAsync(_guildMembersDownloadedEvent, nameof(GuildMembersDownloaded), guild).ConfigureAwait(false);
  466. }
  467. }
  468. else
  469. {
  470. await UnknownGuildAsync(type, data.Id).ConfigureAwait(false);
  471. return;
  472. }
  473. }
  474. else
  475. {
  476. await _gatewayLogger.DebugAsync($"Received Dispatch (GUILD_CREATE)").ConfigureAwait(false);
  477. var guild = AddGuild(data, State);
  478. if (guild != null)
  479. {
  480. if (ApiClient.AuthTokenType == TokenType.User)
  481. await SyncGuildsAsync().ConfigureAwait(false);
  482. await TimedInvokeAsync(_joinedGuildEvent, nameof(JoinedGuild), guild).ConfigureAwait(false);
  483. }
  484. else
  485. {
  486. await UnknownGuildAsync(type, data.Id).ConfigureAwait(false);
  487. return;
  488. }
  489. }
  490. }
  491. break;
  492. case "GUILD_UPDATE":
  493. {
  494. await _gatewayLogger.DebugAsync("Received Dispatch (GUILD_UPDATE)").ConfigureAwait(false);
  495. var data = (payload as JToken).ToObject<API.Guild>(_serializer);
  496. var guild = State.GetGuild(data.Id);
  497. if (guild != null)
  498. {
  499. var before = guild.Clone();
  500. guild.Update(State, data);
  501. await TimedInvokeAsync(_guildUpdatedEvent, nameof(GuildUpdated), before, guild).ConfigureAwait(false);
  502. }
  503. else
  504. {
  505. await UnknownGuildAsync(type, data.Id).ConfigureAwait(false);
  506. return;
  507. }
  508. }
  509. break;
  510. case "GUILD_EMOJIS_UPDATE":
  511. {
  512. await _gatewayLogger.DebugAsync("Received Dispatch (GUILD_EMOJIS_UPDATE)").ConfigureAwait(false);
  513. var data = (payload as JToken).ToObject<API.Gateway.GuildEmojiUpdateEvent>(_serializer);
  514. var guild = State.GetGuild(data.GuildId);
  515. if (guild != null)
  516. {
  517. var before = guild.Clone();
  518. guild.Update(State, data);
  519. await TimedInvokeAsync(_guildUpdatedEvent, nameof(GuildUpdated), before, guild).ConfigureAwait(false);
  520. }
  521. else
  522. {
  523. await UnknownGuildAsync(type, data.GuildId).ConfigureAwait(false);
  524. return;
  525. }
  526. }
  527. break;
  528. case "GUILD_SYNC":
  529. {
  530. await _gatewayLogger.DebugAsync("Received Dispatch (GUILD_SYNC)").ConfigureAwait(false);
  531. var data = (payload as JToken).ToObject<GuildSyncEvent>(_serializer);
  532. var guild = State.GetGuild(data.Id);
  533. if (guild != null)
  534. {
  535. var before = guild.Clone();
  536. guild.Update(State, data);
  537. //This is treated as an extension of GUILD_AVAILABLE
  538. _unavailableGuildCount--;
  539. _lastGuildAvailableTime = Environment.TickCount;
  540. await GuildAvailableAsync(guild).ConfigureAwait(false);
  541. await TimedInvokeAsync(_guildUpdatedEvent, nameof(GuildUpdated), before, guild).ConfigureAwait(false);
  542. }
  543. else
  544. {
  545. await UnknownGuildAsync(type, data.Id).ConfigureAwait(false);
  546. return;
  547. }
  548. }
  549. break;
  550. case "GUILD_DELETE":
  551. {
  552. var data = (payload as JToken).ToObject<ExtendedGuild>(_serializer);
  553. if (data.Unavailable == true)
  554. {
  555. type = "GUILD_UNAVAILABLE";
  556. await _gatewayLogger.DebugAsync($"Received Dispatch (GUILD_UNAVAILABLE)").ConfigureAwait(false);
  557. var guild = State.GetGuild(data.Id);
  558. if (guild != null)
  559. {
  560. await GuildUnavailableAsync(guild).ConfigureAwait(false);
  561. _unavailableGuildCount++;
  562. }
  563. else
  564. {
  565. await UnknownGuildAsync(type, data.Id).ConfigureAwait(false);
  566. return;
  567. }
  568. }
  569. else
  570. {
  571. await _gatewayLogger.DebugAsync($"Received Dispatch (GUILD_DELETE)").ConfigureAwait(false);
  572. var guild = RemoveGuild(data.Id);
  573. if (guild != null)
  574. {
  575. await GuildUnavailableAsync(guild).ConfigureAwait(false);
  576. await TimedInvokeAsync(_leftGuildEvent, nameof(LeftGuild), guild).ConfigureAwait(false);
  577. }
  578. else
  579. {
  580. await UnknownGuildAsync(type, data.Id).ConfigureAwait(false);
  581. return;
  582. }
  583. }
  584. }
  585. break;
  586. //Channels
  587. case "CHANNEL_CREATE":
  588. {
  589. await _gatewayLogger.DebugAsync("Received Dispatch (CHANNEL_CREATE)").ConfigureAwait(false);
  590. var data = (payload as JToken).ToObject<API.Channel>(_serializer);
  591. SocketChannel channel = null;
  592. if (data.GuildId.IsSpecified)
  593. {
  594. var guild = State.GetGuild(data.GuildId.Value);
  595. if (guild != null)
  596. {
  597. channel = guild.AddChannel(State, data);
  598. if (!guild.IsSynced)
  599. {
  600. await UnsyncedGuildAsync(type, guild.Id).ConfigureAwait(false);
  601. return;
  602. }
  603. }
  604. else
  605. {
  606. await UnknownGuildAsync(type, data.GuildId.Value).ConfigureAwait(false);
  607. return;
  608. }
  609. }
  610. else
  611. {
  612. channel = State.GetChannel(data.Id);
  613. if (channel != null)
  614. return; //Discord may send duplicate CHANNEL_CREATEs for DMs
  615. channel = AddPrivateChannel(data, State) as SocketChannel;
  616. }
  617. if (channel != null)
  618. await TimedInvokeAsync(_channelCreatedEvent, nameof(ChannelCreated), channel).ConfigureAwait(false);
  619. }
  620. break;
  621. case "CHANNEL_UPDATE":
  622. {
  623. await _gatewayLogger.DebugAsync("Received Dispatch (CHANNEL_UPDATE)").ConfigureAwait(false);
  624. var data = (payload as JToken).ToObject<API.Channel>(_serializer);
  625. var channel = State.GetChannel(data.Id);
  626. if (channel != null)
  627. {
  628. var before = channel.Clone();
  629. channel.Update(State, data);
  630. var guild = (channel as SocketGuildChannel)?.Guild;
  631. if (!(guild?.IsSynced ?? true))
  632. {
  633. await UnsyncedGuildAsync(type, guild.Id).ConfigureAwait(false);
  634. return;
  635. }
  636. await TimedInvokeAsync(_channelUpdatedEvent, nameof(ChannelUpdated), before, channel).ConfigureAwait(false);
  637. }
  638. else
  639. {
  640. await UnknownChannelAsync(type, data.Id).ConfigureAwait(false);
  641. return;
  642. }
  643. }
  644. break;
  645. case "CHANNEL_DELETE":
  646. {
  647. await _gatewayLogger.DebugAsync("Received Dispatch (CHANNEL_DELETE)").ConfigureAwait(false);
  648. SocketChannel channel = null;
  649. var data = (payload as JToken).ToObject<API.Channel>(_serializer);
  650. if (data.GuildId.IsSpecified)
  651. {
  652. var guild = State.GetGuild(data.GuildId.Value);
  653. if (guild != null)
  654. {
  655. channel = guild.RemoveChannel(State, data.Id);
  656. if (!guild.IsSynced)
  657. {
  658. await UnsyncedGuildAsync(type, guild.Id).ConfigureAwait(false);
  659. return;
  660. }
  661. }
  662. else
  663. {
  664. await UnknownGuildAsync(type, data.GuildId.Value).ConfigureAwait(false);
  665. return;
  666. }
  667. }
  668. else
  669. channel = RemovePrivateChannel(data.Id) as SocketChannel;
  670. if (channel != null)
  671. await TimedInvokeAsync(_channelDestroyedEvent, nameof(ChannelDestroyed), channel).ConfigureAwait(false);
  672. else
  673. {
  674. await UnknownChannelAsync(type, data.Id, data.GuildId.GetValueOrDefault(0)).ConfigureAwait(false);
  675. return;
  676. }
  677. }
  678. break;
  679. //Members
  680. case "GUILD_MEMBER_ADD":
  681. {
  682. await _gatewayLogger.DebugAsync("Received Dispatch (GUILD_MEMBER_ADD)").ConfigureAwait(false);
  683. var data = (payload as JToken).ToObject<GuildMemberAddEvent>(_serializer);
  684. var guild = State.GetGuild(data.GuildId);
  685. if (guild != null)
  686. {
  687. var user = guild.AddOrUpdateUser(data);
  688. guild.MemberCount++;
  689. if (!guild.IsSynced)
  690. {
  691. await UnsyncedGuildAsync(type, guild.Id).ConfigureAwait(false);
  692. return;
  693. }
  694. await TimedInvokeAsync(_userJoinedEvent, nameof(UserJoined), user).ConfigureAwait(false);
  695. }
  696. else
  697. {
  698. await UnknownGuildAsync(type, data.GuildId).ConfigureAwait(false);
  699. return;
  700. }
  701. }
  702. break;
  703. case "GUILD_MEMBER_UPDATE":
  704. {
  705. await _gatewayLogger.DebugAsync("Received Dispatch (GUILD_MEMBER_UPDATE)").ConfigureAwait(false);
  706. var data = (payload as JToken).ToObject<GuildMemberUpdateEvent>(_serializer);
  707. var guild = State.GetGuild(data.GuildId);
  708. if (guild != null)
  709. {
  710. var user = guild.GetUser(data.User.Id);
  711. if (!guild.IsSynced)
  712. {
  713. await UnsyncedGuildAsync(type, guild.Id).ConfigureAwait(false);
  714. return;
  715. }
  716. if (user != null)
  717. {
  718. var before = user.Clone();
  719. user.Update(State, data);
  720. await TimedInvokeAsync(_guildMemberUpdatedEvent, nameof(GuildMemberUpdated), before, user).ConfigureAwait(false);
  721. }
  722. else
  723. {
  724. if (!guild.HasAllMembers)
  725. await IncompleteGuildUserAsync(type, data.User.Id, data.GuildId).ConfigureAwait(false);
  726. else
  727. await UnknownGuildUserAsync(type, data.User.Id, data.GuildId).ConfigureAwait(false);
  728. return;
  729. }
  730. }
  731. else
  732. {
  733. await UnknownGuildAsync(type, data.GuildId).ConfigureAwait(false);
  734. return;
  735. }
  736. }
  737. break;
  738. case "GUILD_MEMBER_REMOVE":
  739. {
  740. await _gatewayLogger.DebugAsync("Received Dispatch (GUILD_MEMBER_REMOVE)").ConfigureAwait(false);
  741. var data = (payload as JToken).ToObject<GuildMemberRemoveEvent>(_serializer);
  742. var guild = State.GetGuild(data.GuildId);
  743. if (guild != null)
  744. {
  745. var user = guild.RemoveUser(data.User.Id);
  746. guild.MemberCount--;
  747. if (!guild.IsSynced)
  748. {
  749. await UnsyncedGuildAsync(type, guild.Id).ConfigureAwait(false);
  750. return;
  751. }
  752. if (user != null)
  753. await TimedInvokeAsync(_userLeftEvent, nameof(UserLeft), user).ConfigureAwait(false);
  754. else
  755. {
  756. if (!guild.HasAllMembers)
  757. await IncompleteGuildUserAsync(type, data.User.Id, data.GuildId).ConfigureAwait(false);
  758. else
  759. await UnknownGuildUserAsync(type, data.User.Id, data.GuildId).ConfigureAwait(false);
  760. return;
  761. }
  762. }
  763. else
  764. {
  765. await UnknownGuildAsync(type, data.GuildId).ConfigureAwait(false);
  766. return;
  767. }
  768. }
  769. break;
  770. case "GUILD_MEMBERS_CHUNK":
  771. {
  772. await _gatewayLogger.DebugAsync("Received Dispatch (GUILD_MEMBERS_CHUNK)").ConfigureAwait(false);
  773. var data = (payload as JToken).ToObject<GuildMembersChunkEvent>(_serializer);
  774. var guild = State.GetGuild(data.GuildId);
  775. if (guild != null)
  776. {
  777. foreach (var memberModel in data.Members)
  778. guild.AddOrUpdateUser(memberModel);
  779. if (guild.DownloadedMemberCount >= guild.MemberCount && !guild.DownloaderPromise.IsCompleted)
  780. {
  781. guild.CompleteDownloadUsers();
  782. await TimedInvokeAsync(_guildMembersDownloadedEvent, nameof(GuildMembersDownloaded), guild).ConfigureAwait(false);
  783. }
  784. }
  785. else
  786. {
  787. await UnknownGuildAsync(type, data.GuildId).ConfigureAwait(false);
  788. return;
  789. }
  790. }
  791. break;
  792. case "CHANNEL_RECIPIENT_ADD":
  793. {
  794. await _gatewayLogger.DebugAsync("Received Dispatch (CHANNEL_RECIPIENT_ADD)").ConfigureAwait(false);
  795. var data = (payload as JToken).ToObject<RecipientEvent>(_serializer);
  796. if (State.GetChannel(data.ChannelId) is SocketGroupChannel channel)
  797. {
  798. var user = channel.GetOrAddUser(data.User);
  799. await TimedInvokeAsync(_recipientAddedEvent, nameof(RecipientAdded), user).ConfigureAwait(false);
  800. }
  801. else
  802. {
  803. await UnknownChannelAsync(type, data.ChannelId).ConfigureAwait(false);
  804. return;
  805. }
  806. }
  807. break;
  808. case "CHANNEL_RECIPIENT_REMOVE":
  809. {
  810. await _gatewayLogger.DebugAsync("Received Dispatch (CHANNEL_RECIPIENT_REMOVE)").ConfigureAwait(false);
  811. var data = (payload as JToken).ToObject<RecipientEvent>(_serializer);
  812. if (State.GetChannel(data.ChannelId) is SocketGroupChannel channel)
  813. {
  814. var user = channel.RemoveUser(data.User.Id);
  815. if (user != null)
  816. await TimedInvokeAsync(_recipientRemovedEvent, nameof(RecipientRemoved), user).ConfigureAwait(false);
  817. else
  818. {
  819. await UnknownChannelUserAsync(type, data.User.Id, data.ChannelId).ConfigureAwait(false);
  820. return;
  821. }
  822. }
  823. else
  824. {
  825. await UnknownChannelAsync(type, data.ChannelId).ConfigureAwait(false);
  826. return;
  827. }
  828. }
  829. break;
  830. //Roles
  831. case "GUILD_ROLE_CREATE":
  832. {
  833. await _gatewayLogger.DebugAsync("Received Dispatch (GUILD_ROLE_CREATE)").ConfigureAwait(false);
  834. var data = (payload as JToken).ToObject<GuildRoleCreateEvent>(_serializer);
  835. var guild = State.GetGuild(data.GuildId);
  836. if (guild != null)
  837. {
  838. var role = guild.AddRole(data.Role);
  839. if (!guild.IsSynced)
  840. {
  841. await UnsyncedGuildAsync(type, guild.Id).ConfigureAwait(false);
  842. return;
  843. }
  844. await TimedInvokeAsync(_roleCreatedEvent, nameof(RoleCreated), role).ConfigureAwait(false);
  845. }
  846. else
  847. {
  848. await UnknownGuildAsync(type, data.GuildId).ConfigureAwait(false);
  849. return;
  850. }
  851. }
  852. break;
  853. case "GUILD_ROLE_UPDATE":
  854. {
  855. await _gatewayLogger.DebugAsync("Received Dispatch (GUILD_ROLE_UPDATE)").ConfigureAwait(false);
  856. var data = (payload as JToken).ToObject<GuildRoleUpdateEvent>(_serializer);
  857. var guild = State.GetGuild(data.GuildId);
  858. if (guild != null)
  859. {
  860. var role = guild.GetRole(data.Role.Id);
  861. if (role != null)
  862. {
  863. var before = role.Clone();
  864. role.Update(State, data.Role);
  865. if (!guild.IsSynced)
  866. {
  867. await UnsyncedGuildAsync(type, guild.Id).ConfigureAwait(false);
  868. return;
  869. }
  870. await TimedInvokeAsync(_roleUpdatedEvent, nameof(RoleUpdated), before, role).ConfigureAwait(false);
  871. }
  872. else
  873. {
  874. await UnknownRoleAsync(type, data.Role.Id, guild.Id).ConfigureAwait(false);
  875. return;
  876. }
  877. }
  878. else
  879. {
  880. await UnknownGuildAsync(type, data.GuildId).ConfigureAwait(false);
  881. return;
  882. }
  883. }
  884. break;
  885. case "GUILD_ROLE_DELETE":
  886. {
  887. await _gatewayLogger.DebugAsync("Received Dispatch (GUILD_ROLE_DELETE)").ConfigureAwait(false);
  888. var data = (payload as JToken).ToObject<GuildRoleDeleteEvent>(_serializer);
  889. var guild = State.GetGuild(data.GuildId);
  890. if (guild != null)
  891. {
  892. var role = guild.RemoveRole(data.RoleId);
  893. if (role != null)
  894. {
  895. if (!guild.IsSynced)
  896. {
  897. await UnsyncedGuildAsync(type, guild.Id).ConfigureAwait(false);
  898. return;
  899. }
  900. await TimedInvokeAsync(_roleDeletedEvent, nameof(RoleDeleted), role).ConfigureAwait(false);
  901. }
  902. else
  903. {
  904. await UnknownRoleAsync(type, data.RoleId, guild.Id).ConfigureAwait(false);
  905. return;
  906. }
  907. }
  908. else
  909. {
  910. await UnknownGuildAsync(type, data.GuildId).ConfigureAwait(false);
  911. return;
  912. }
  913. }
  914. break;
  915. //Bans
  916. case "GUILD_BAN_ADD":
  917. {
  918. await _gatewayLogger.DebugAsync("Received Dispatch (GUILD_BAN_ADD)").ConfigureAwait(false);
  919. var data = (payload as JToken).ToObject<GuildBanEvent>(_serializer);
  920. var guild = State.GetGuild(data.GuildId);
  921. if (guild != null)
  922. {
  923. if (!guild.IsSynced)
  924. {
  925. await UnsyncedGuildAsync(type, guild.Id).ConfigureAwait(false);
  926. return;
  927. }
  928. SocketUser user = guild.GetUser(data.User.Id);
  929. if (user == null)
  930. user = SocketUnknownUser.Create(this, State, data.User);
  931. await TimedInvokeAsync(_userBannedEvent, nameof(UserBanned), user, guild).ConfigureAwait(false);
  932. }
  933. else
  934. {
  935. await UnknownGuildAsync(type, data.GuildId).ConfigureAwait(false);
  936. return;
  937. }
  938. }
  939. break;
  940. case "GUILD_BAN_REMOVE":
  941. {
  942. await _gatewayLogger.DebugAsync("Received Dispatch (GUILD_BAN_REMOVE)").ConfigureAwait(false);
  943. var data = (payload as JToken).ToObject<GuildBanEvent>(_serializer);
  944. var guild = State.GetGuild(data.GuildId);
  945. if (guild != null)
  946. {
  947. if (!guild.IsSynced)
  948. {
  949. await UnsyncedGuildAsync(type, guild.Id).ConfigureAwait(false);
  950. return;
  951. }
  952. SocketUser user = State.GetUser(data.User.Id);
  953. if (user == null)
  954. user = SocketUnknownUser.Create(this, State, data.User);
  955. await TimedInvokeAsync(_userUnbannedEvent, nameof(UserUnbanned), user, guild).ConfigureAwait(false);
  956. }
  957. else
  958. {
  959. await UnknownGuildAsync(type, data.GuildId).ConfigureAwait(false);
  960. return;
  961. }
  962. }
  963. break;
  964. //Messages
  965. case "MESSAGE_CREATE":
  966. {
  967. await _gatewayLogger.DebugAsync("Received Dispatch (MESSAGE_CREATE)").ConfigureAwait(false);
  968. var data = (payload as JToken).ToObject<API.Message>(_serializer);
  969. if (State.GetChannel(data.ChannelId) is ISocketMessageChannel channel)
  970. {
  971. var guild = (channel as SocketGuildChannel)?.Guild;
  972. if (guild != null && !guild.IsSynced)
  973. {
  974. await UnsyncedGuildAsync(type, guild.Id).ConfigureAwait(false);
  975. return;
  976. }
  977. SocketUser author;
  978. if (guild != null)
  979. {
  980. if (data.WebhookId.IsSpecified)
  981. author = SocketWebhookUser.Create(guild, State, data.Author.Value, data.WebhookId.Value);
  982. else
  983. author = guild.GetUser(data.Author.Value.Id);
  984. }
  985. else
  986. author = (channel as SocketChannel).GetUser(data.Author.Value.Id);
  987. if (author == null)
  988. {
  989. if (guild != null)
  990. author = guild.AddOrUpdateUser(data.Author.Value); //User has no guild-specific data
  991. else if (channel is SocketGroupChannel)
  992. author = (channel as SocketGroupChannel).GetOrAddUser(data.Author.Value);
  993. else
  994. await UnknownChannelUserAsync(type, data.Author.Value.Id, channel.Id).ConfigureAwait(false);
  995. return;
  996. }
  997. var msg = SocketMessage.Create(this, State, author, channel, data);
  998. SocketChannelHelper.AddMessage(channel, this, msg);
  999. await TimedInvokeAsync(_messageReceivedEvent, nameof(MessageReceived), msg).ConfigureAwait(false);
  1000. }
  1001. else
  1002. {
  1003. await UnknownChannelAsync(type, data.ChannelId).ConfigureAwait(false);
  1004. return;
  1005. }
  1006. }
  1007. break;
  1008. case "MESSAGE_UPDATE":
  1009. {
  1010. await _gatewayLogger.DebugAsync("Received Dispatch (MESSAGE_UPDATE)").ConfigureAwait(false);
  1011. var data = (payload as JToken).ToObject<API.Message>(_serializer);
  1012. if (State.GetChannel(data.ChannelId) is ISocketMessageChannel channel)
  1013. {
  1014. var guild = (channel as SocketGuildChannel)?.Guild;
  1015. if (guild != null && !guild.IsSynced)
  1016. {
  1017. await UnsyncedGuildAsync(type, guild.Id).ConfigureAwait(false);
  1018. return;
  1019. }
  1020. SocketMessage before = null, after = null;
  1021. SocketMessage cachedMsg = channel.GetCachedMessage(data.Id);
  1022. bool isCached = cachedMsg != null;
  1023. if (isCached)
  1024. {
  1025. before = cachedMsg.Clone();
  1026. cachedMsg.Update(State, data);
  1027. after = cachedMsg;
  1028. }
  1029. else if (data.Author.IsSpecified)
  1030. {
  1031. //Edited message isnt in cache, create a detached one
  1032. SocketUser author;
  1033. if (guild != null)
  1034. author = guild.GetUser(data.Author.Value.Id);
  1035. else
  1036. author = (channel as SocketChannel).GetUser(data.Author.Value.Id);
  1037. if (author == null)
  1038. author = SocketUnknownUser.Create(this, State, data.Author.Value);
  1039. after = SocketMessage.Create(this, State, author, channel, data);
  1040. }
  1041. var cacheableBefore = new Cacheable<IMessage, ulong>(before, data.Id, isCached, async () => await channel.GetMessageAsync(data.Id));
  1042. await TimedInvokeAsync(_messageUpdatedEvent, nameof(MessageUpdated), cacheableBefore, after, channel).ConfigureAwait(false);
  1043. }
  1044. else
  1045. {
  1046. await UnknownChannelAsync(type, data.ChannelId).ConfigureAwait(false);
  1047. return;
  1048. }
  1049. }
  1050. break;
  1051. case "MESSAGE_DELETE":
  1052. {
  1053. await _gatewayLogger.DebugAsync("Received Dispatch (MESSAGE_DELETE)").ConfigureAwait(false);
  1054. var data = (payload as JToken).ToObject<API.Message>(_serializer);
  1055. if (State.GetChannel(data.ChannelId) is ISocketMessageChannel channel)
  1056. {
  1057. var guild = (channel as SocketGuildChannel)?.Guild;
  1058. if (!(guild?.IsSynced ?? true))
  1059. {
  1060. await UnsyncedGuildAsync(type, guild.Id).ConfigureAwait(false);
  1061. return;
  1062. }
  1063. var msg = SocketChannelHelper.RemoveMessage(channel, this, data.Id);
  1064. bool isCached = msg != null;
  1065. var cacheable = new Cacheable<IMessage, ulong>(msg, data.Id, isCached, async () => await channel.GetMessageAsync(data.Id));
  1066. await TimedInvokeAsync(_messageDeletedEvent, nameof(MessageDeleted), cacheable, channel).ConfigureAwait(false);
  1067. }
  1068. else
  1069. {
  1070. await UnknownChannelAsync(type, data.ChannelId).ConfigureAwait(false);
  1071. return;
  1072. }
  1073. }
  1074. break;
  1075. case "MESSAGE_REACTION_ADD":
  1076. {
  1077. await _gatewayLogger.DebugAsync("Received Dispatch (MESSAGE_REACTION_ADD)").ConfigureAwait(false);
  1078. var data = (payload as JToken).ToObject<API.Gateway.Reaction>(_serializer);
  1079. if (State.GetChannel(data.ChannelId) is ISocketMessageChannel channel)
  1080. {
  1081. var cachedMsg = channel.GetCachedMessage(data.MessageId) as SocketUserMessage;
  1082. bool isCached = cachedMsg != null;
  1083. var user = await channel.GetUserAsync(data.UserId, CacheMode.CacheOnly);
  1084. var reaction = SocketReaction.Create(data, channel, cachedMsg, Optional.Create(user));
  1085. var cacheable = new Cacheable<IUserMessage, ulong>(cachedMsg, data.MessageId, isCached, async () => await channel.GetMessageAsync(data.MessageId) as IUserMessage);
  1086. cachedMsg?.AddReaction(reaction);
  1087. await TimedInvokeAsync(_reactionAddedEvent, nameof(ReactionAdded), cacheable, channel, reaction).ConfigureAwait(false);
  1088. }
  1089. else
  1090. {
  1091. await UnknownChannelAsync(type, data.ChannelId).ConfigureAwait(false);
  1092. return;
  1093. }
  1094. }
  1095. break;
  1096. case "MESSAGE_REACTION_REMOVE":
  1097. {
  1098. await _gatewayLogger.DebugAsync("Received Dispatch (MESSAGE_REACTION_REMOVE)").ConfigureAwait(false);
  1099. var data = (payload as JToken).ToObject<API.Gateway.Reaction>(_serializer);
  1100. if (State.GetChannel(data.ChannelId) is ISocketMessageChannel channel)
  1101. {
  1102. var cachedMsg = channel.GetCachedMessage(data.MessageId) as SocketUserMessage;
  1103. bool isCached = cachedMsg != null;
  1104. var user = await channel.GetUserAsync(data.UserId, CacheMode.CacheOnly);
  1105. var reaction = SocketReaction.Create(data, channel, cachedMsg, Optional.Create(user));
  1106. var cacheable = new Cacheable<IUserMessage, ulong>(cachedMsg, data.MessageId, isCached, async () => await channel.GetMessageAsync(data.MessageId) as IUserMessage);
  1107. cachedMsg?.RemoveReaction(reaction);
  1108. await TimedInvokeAsync(_reactionRemovedEvent, nameof(ReactionRemoved), cacheable, channel, reaction).ConfigureAwait(false);
  1109. }
  1110. else
  1111. {
  1112. await UnknownChannelAsync(type, data.ChannelId).ConfigureAwait(false);
  1113. return;
  1114. }
  1115. }
  1116. break;
  1117. case "MESSAGE_REACTION_REMOVE_ALL":
  1118. {
  1119. await _gatewayLogger.DebugAsync("Received Dispatch (MESSAGE_REACTION_REMOVE_ALL)").ConfigureAwait(false);
  1120. var data = (payload as JToken).ToObject<RemoveAllReactionsEvent>(_serializer);
  1121. if (State.GetChannel(data.ChannelId) is ISocketMessageChannel channel)
  1122. {
  1123. var cachedMsg = channel.GetCachedMessage(data.MessageId) as SocketUserMessage;
  1124. bool isCached = cachedMsg != null;
  1125. var cacheable = new Cacheable<IUserMessage, ulong>(cachedMsg, data.MessageId, isCached, async () => await channel.GetMessageAsync(data.MessageId) as IUserMessage);
  1126. cachedMsg?.ClearReactions();
  1127. await TimedInvokeAsync(_reactionsClearedEvent, nameof(ReactionsCleared), cacheable, channel).ConfigureAwait(false);
  1128. }
  1129. else
  1130. {
  1131. await UnknownChannelAsync(type, data.ChannelId).ConfigureAwait(false);
  1132. return;
  1133. }
  1134. }
  1135. break;
  1136. case "MESSAGE_DELETE_BULK":
  1137. {
  1138. await _gatewayLogger.DebugAsync("Received Dispatch (MESSAGE_DELETE_BULK)").ConfigureAwait(false);
  1139. var data = (payload as JToken).ToObject<MessageDeleteBulkEvent>(_serializer);
  1140. if (State.GetChannel(data.ChannelId) is ISocketMessageChannel channel)
  1141. {
  1142. var guild = (channel as SocketGuildChannel)?.Guild;
  1143. if (!(guild?.IsSynced ?? true))
  1144. {
  1145. await UnsyncedGuildAsync(type, guild.Id).ConfigureAwait(false);
  1146. return;
  1147. }
  1148. foreach (ulong id in data.Ids)
  1149. {
  1150. var msg = SocketChannelHelper.RemoveMessage(channel, this, id);
  1151. bool isCached = msg != null;
  1152. var cacheable = new Cacheable<IMessage, ulong>(msg, id, isCached, async () => await channel.GetMessageAsync(id));
  1153. await TimedInvokeAsync(_messageDeletedEvent, nameof(MessageDeleted), cacheable, channel).ConfigureAwait(false);
  1154. }
  1155. }
  1156. else
  1157. {
  1158. await UnknownChannelAsync(type, data.ChannelId).ConfigureAwait(false);
  1159. return;
  1160. }
  1161. }
  1162. break;
  1163. //Statuses
  1164. case "PRESENCE_UPDATE":
  1165. {
  1166. await _gatewayLogger.DebugAsync("Received Dispatch (PRESENCE_UPDATE)").ConfigureAwait(false);
  1167. var data = (payload as JToken).ToObject<API.Presence>(_serializer);
  1168. if (data.GuildId.IsSpecified)
  1169. {
  1170. var guild = State.GetGuild(data.GuildId.Value);
  1171. if (guild == null)
  1172. {
  1173. await UnknownGuildAsync(type, data.GuildId.Value).ConfigureAwait(false);
  1174. return;
  1175. }
  1176. if (!guild.IsSynced)
  1177. {
  1178. await UnsyncedGuildAsync(type, guild.Id).ConfigureAwait(false);
  1179. return;
  1180. }
  1181. var user = guild.GetUser(data.User.Id);
  1182. if (user == null)
  1183. {
  1184. if (data.Status == UserStatus.Offline)
  1185. {
  1186. return;
  1187. }
  1188. user = guild.AddOrUpdateUser(data);
  1189. }
  1190. else
  1191. {
  1192. var globalBefore = user.GlobalUser.Clone();
  1193. if (user.GlobalUser.Update(State, data.User))
  1194. {
  1195. //Global data was updated, trigger UserUpdated
  1196. await TimedInvokeAsync(_userUpdatedEvent, nameof(UserUpdated), globalBefore, user).ConfigureAwait(false);
  1197. }
  1198. }
  1199. var before = user.Clone();
  1200. user.Update(State, data, true);
  1201. await TimedInvokeAsync(_guildMemberUpdatedEvent, nameof(GuildMemberUpdated), before, user).ConfigureAwait(false);
  1202. }
  1203. else
  1204. {
  1205. var globalUser = State.GetUser(data.User.Id);
  1206. if (globalUser == null)
  1207. {
  1208. await UnknownGlobalUserAsync(type, data.User.Id).ConfigureAwait(false);
  1209. return;
  1210. }
  1211. var before = globalUser.Clone();
  1212. globalUser.Update(State, data.User);
  1213. globalUser.Update(State, data);
  1214. await TimedInvokeAsync(_userUpdatedEvent, nameof(UserUpdated), before, globalUser).ConfigureAwait(false);
  1215. }
  1216. }
  1217. break;
  1218. case "TYPING_START":
  1219. {
  1220. await _gatewayLogger.DebugAsync("Received Dispatch (TYPING_START)").ConfigureAwait(false);
  1221. var data = (payload as JToken).ToObject<TypingStartEvent>(_serializer);
  1222. if (State.GetChannel(data.ChannelId) is ISocketMessageChannel channel)
  1223. {
  1224. var guild = (channel as SocketGuildChannel)?.Guild;
  1225. if (!(guild?.IsSynced ?? true))
  1226. {
  1227. await UnsyncedGuildAsync(type, guild.Id).ConfigureAwait(false);
  1228. return;
  1229. }
  1230. var user = (channel as SocketChannel).GetUser(data.UserId);
  1231. if (user != null)
  1232. await TimedInvokeAsync(_userIsTypingEvent, nameof(UserIsTyping), user, channel).ConfigureAwait(false);
  1233. }
  1234. }
  1235. break;
  1236. //Users
  1237. case "USER_UPDATE":
  1238. {
  1239. await _gatewayLogger.DebugAsync("Received Dispatch (USER_UPDATE)").ConfigureAwait(false);
  1240. var data = (payload as JToken).ToObject<API.User>(_serializer);
  1241. if (data.Id == CurrentUser.Id)
  1242. {
  1243. var before = CurrentUser.Clone();
  1244. CurrentUser.Update(State, data);
  1245. await TimedInvokeAsync(_selfUpdatedEvent, nameof(CurrentUserUpdated), before, CurrentUser).ConfigureAwait(false);
  1246. }
  1247. else
  1248. {
  1249. await _gatewayLogger.WarningAsync("Received USER_UPDATE for wrong user.").ConfigureAwait(false);
  1250. return;
  1251. }
  1252. }
  1253. break;
  1254. //Voice
  1255. case "VOICE_STATE_UPDATE":
  1256. {
  1257. await _gatewayLogger.DebugAsync("Received Dispatch (VOICE_STATE_UPDATE)").ConfigureAwait(false);
  1258. var data = (payload as JToken).ToObject<API.VoiceState>(_serializer);
  1259. SocketUser user;
  1260. SocketVoiceState before, after;
  1261. if (data.GuildId != null)
  1262. {
  1263. var guild = State.GetGuild(data.GuildId.Value);
  1264. if (guild == null)
  1265. {
  1266. await UnknownGuildAsync(type, data.GuildId.Value).ConfigureAwait(false);
  1267. return;
  1268. }
  1269. else if (!guild.IsSynced)
  1270. {
  1271. await UnsyncedGuildAsync(type, guild.Id).ConfigureAwait(false);
  1272. return;
  1273. }
  1274. if (data.ChannelId != null)
  1275. {
  1276. before = guild.GetVoiceState(data.UserId)?.Clone() ?? SocketVoiceState.Default;
  1277. after = await guild.AddOrUpdateVoiceStateAsync(State, data).ConfigureAwait(false);
  1278. /*if (data.UserId == CurrentUser.Id)
  1279. {
  1280. var _ = guild.FinishJoinAudioChannel().ConfigureAwait(false);
  1281. }*/
  1282. }
  1283. else
  1284. {
  1285. before = await guild.RemoveVoiceStateAsync(data.UserId).ConfigureAwait(false) ?? SocketVoiceState.Default;
  1286. after = SocketVoiceState.Create(null, data);
  1287. }
  1288. user = guild.GetUser(data.UserId);
  1289. if (user == null)
  1290. {
  1291. await UnknownGuildUserAsync(type, data.UserId, guild.Id).ConfigureAwait(false);
  1292. return;
  1293. }
  1294. }
  1295. else
  1296. {
  1297. var groupChannel = State.GetChannel(data.ChannelId.Value) as SocketGroupChannel;
  1298. if (groupChannel == null)
  1299. {
  1300. await UnknownChannelAsync(type, data.ChannelId.Value).ConfigureAwait(false);
  1301. return;
  1302. }
  1303. if (data.ChannelId != null)
  1304. {
  1305. before = groupChannel.GetVoiceState(data.UserId)?.Clone() ?? SocketVoiceState.Default;
  1306. after = groupChannel.AddOrUpdateVoiceState(State, data);
  1307. }
  1308. else
  1309. {
  1310. before = groupChannel.RemoveVoiceState(data.UserId) ?? SocketVoiceState.Default;
  1311. after = SocketVoiceState.Create(null, data);
  1312. }
  1313. user = groupChannel.GetUser(data.UserId);
  1314. if (user == null)
  1315. {
  1316. await UnknownChannelUserAsync(type, data.UserId, groupChannel.Id).ConfigureAwait(false);
  1317. return;
  1318. }
  1319. }
  1320. await TimedInvokeAsync(_userVoiceStateUpdatedEvent, nameof(UserVoiceStateUpdated), user, before, after).ConfigureAwait(false);
  1321. }
  1322. break;
  1323. case "VOICE_SERVER_UPDATE":
  1324. {
  1325. await _gatewayLogger.DebugAsync("Received Dispatch (VOICE_SERVER_UPDATE)").ConfigureAwait(false);
  1326. var data = (payload as JToken).ToObject<VoiceServerUpdateEvent>(_serializer);
  1327. var guild = State.GetGuild(data.GuildId);
  1328. if (guild != null)
  1329. {
  1330. string endpoint = data.Endpoint.Substring(0, data.Endpoint.LastIndexOf(':'));
  1331. var _ = guild.FinishConnectAudio(endpoint, data.Token).ConfigureAwait(false);
  1332. }
  1333. else
  1334. {
  1335. await UnknownGuildAsync(type, data.GuildId).ConfigureAwait(false);
  1336. return;
  1337. }
  1338. }
  1339. break;
  1340. //Ignored (User only)
  1341. case "CHANNEL_PINS_ACK":
  1342. await _gatewayLogger.DebugAsync("Ignored Dispatch (CHANNEL_PINS_ACK)").ConfigureAwait(false);
  1343. break;
  1344. case "CHANNEL_PINS_UPDATE":
  1345. await _gatewayLogger.DebugAsync("Ignored Dispatch (CHANNEL_PINS_UPDATE)").ConfigureAwait(false);
  1346. break;
  1347. case "GUILD_INTEGRATIONS_UPDATE":
  1348. await _gatewayLogger.DebugAsync("Ignored Dispatch (GUILD_INTEGRATIONS_UPDATE)").ConfigureAwait(false);
  1349. break;
  1350. case "MESSAGE_ACK":
  1351. await _gatewayLogger.DebugAsync("Ignored Dispatch (MESSAGE_ACK)").ConfigureAwait(false);
  1352. break;
  1353. case "USER_SETTINGS_UPDATE":
  1354. await _gatewayLogger.DebugAsync("Ignored Dispatch (USER_SETTINGS_UPDATE)").ConfigureAwait(false);
  1355. break;
  1356. case "WEBHOOKS_UPDATE":
  1357. await _gatewayLogger.DebugAsync("Ignored Dispatch (WEBHOOKS_UPDATE)").ConfigureAwait(false);
  1358. break;
  1359. //Others
  1360. default:
  1361. await _gatewayLogger.WarningAsync($"Unknown Dispatch ({type})").ConfigureAwait(false);
  1362. break;
  1363. }
  1364. break;
  1365. default:
  1366. await _gatewayLogger.WarningAsync($"Unknown OpCode ({opCode})").ConfigureAwait(false);
  1367. break;
  1368. }
  1369. }
  1370. catch (Exception ex)
  1371. {
  1372. await _gatewayLogger.ErrorAsync($"Error handling {opCode}{(type != null ? $" ({type})" : "")}", ex).ConfigureAwait(false);
  1373. }
  1374. }
  1375. private async Task RunHeartbeatAsync(int intervalMillis, CancellationToken cancelToken)
  1376. {
  1377. try
  1378. {
  1379. await _gatewayLogger.DebugAsync("Heartbeat Started").ConfigureAwait(false);
  1380. while (!cancelToken.IsCancellationRequested)
  1381. {
  1382. int now = Environment.TickCount;
  1383. //Did server respond to our last heartbeat, or are we still receiving messages (long load?)
  1384. if (_heartbeatTimes.Count != 0 && (now - _lastMessageTime) > intervalMillis)
  1385. {
  1386. if (ConnectionState == ConnectionState.Connected && (_guildDownloadTask?.IsCompleted ?? true))
  1387. {
  1388. _connection.Error(new Exception("Server missed last heartbeat"));
  1389. return;
  1390. }
  1391. }
  1392. _heartbeatTimes.Enqueue(now);
  1393. try
  1394. {
  1395. await ApiClient.SendHeartbeatAsync(_lastSeq).ConfigureAwait(false);
  1396. }
  1397. catch (Exception ex)
  1398. {
  1399. await _gatewayLogger.WarningAsync("Heartbeat Errored", ex).ConfigureAwait(false);
  1400. }
  1401. await Task.Delay(intervalMillis, cancelToken).ConfigureAwait(false);
  1402. }
  1403. await _gatewayLogger.DebugAsync("Heartbeat Stopped").ConfigureAwait(false);
  1404. }
  1405. catch (OperationCanceledException)
  1406. {
  1407. await _gatewayLogger.DebugAsync("Heartbeat Stopped").ConfigureAwait(false);
  1408. }
  1409. catch (Exception ex)
  1410. {
  1411. await _gatewayLogger.ErrorAsync("Heartbeat Errored", ex).ConfigureAwait(false);
  1412. }
  1413. }
  1414. /*public async Task WaitForGuildsAsync()
  1415. {
  1416. var downloadTask = _guildDownloadTask;
  1417. if (downloadTask != null)
  1418. await _guildDownloadTask.ConfigureAwait(false);
  1419. }*/
  1420. private async Task WaitForGuildsAsync(CancellationToken cancelToken, Logger logger)
  1421. {
  1422. //Wait for GUILD_AVAILABLEs
  1423. try
  1424. {
  1425. await logger.DebugAsync("GuildDownloader Started").ConfigureAwait(false);
  1426. while ((_unavailableGuildCount != 0) && (Environment.TickCount - _lastGuildAvailableTime < 2000))
  1427. await Task.Delay(500, cancelToken).ConfigureAwait(false);
  1428. await logger.DebugAsync("GuildDownloader Stopped").ConfigureAwait(false);
  1429. }
  1430. catch (OperationCanceledException)
  1431. {
  1432. await logger.DebugAsync("GuildDownloader Stopped").ConfigureAwait(false);
  1433. }
  1434. catch (Exception ex)
  1435. {
  1436. await logger.ErrorAsync("GuildDownloader Errored", ex).ConfigureAwait(false);
  1437. }
  1438. }
  1439. private async Task SyncGuildsAsync()
  1440. {
  1441. var guildIds = Guilds.Where(x => !x.IsSynced).Select(x => x.Id).ToImmutableArray();
  1442. if (guildIds.Length > 0)
  1443. await ApiClient.SendGuildSyncAsync(guildIds).ConfigureAwait(false);
  1444. }
  1445. internal SocketGuild AddGuild(ExtendedGuild model, ClientState state)
  1446. {
  1447. var guild = SocketGuild.Create(this, state, model);
  1448. state.AddGuild(guild);
  1449. if (model.Large)
  1450. _largeGuilds.Enqueue(model.Id);
  1451. return guild;
  1452. }
  1453. internal SocketGuild RemoveGuild(ulong id)
  1454. {
  1455. var guild = State.RemoveGuild(id);
  1456. if (guild != null)
  1457. {
  1458. foreach (var channel in guild.Channels)
  1459. State.RemoveChannel(id);
  1460. foreach (var user in guild.Users)
  1461. user.GlobalUser.RemoveRef(this);
  1462. }
  1463. return guild;
  1464. }
  1465. internal ISocketPrivateChannel AddPrivateChannel(API.Channel model, ClientState state)
  1466. {
  1467. var channel = SocketChannel.CreatePrivate(this, state, model);
  1468. state.AddChannel(channel as SocketChannel);
  1469. if (channel is SocketDMChannel dm)
  1470. dm.Recipient.GlobalUser.DMChannel = dm;
  1471. return channel;
  1472. }
  1473. internal ISocketPrivateChannel RemovePrivateChannel(ulong id)
  1474. {
  1475. var channel = State.RemoveChannel(id) as ISocketPrivateChannel;
  1476. if (channel != null)
  1477. {
  1478. if (channel is SocketDMChannel dmChannel)
  1479. dmChannel.Recipient.GlobalUser.DMChannel = null;
  1480. foreach (var recipient in channel.Recipients)
  1481. recipient.GlobalUser.RemoveRef(this);
  1482. }
  1483. return channel;
  1484. }
  1485. private async Task GuildAvailableAsync(SocketGuild guild)
  1486. {
  1487. if (!guild.IsConnected)
  1488. {
  1489. guild.IsConnected = true;
  1490. await TimedInvokeAsync(_guildAvailableEvent, nameof(GuildAvailable), guild).ConfigureAwait(false);
  1491. }
  1492. }
  1493. private async Task GuildUnavailableAsync(SocketGuild guild)
  1494. {
  1495. if (guild.IsConnected)
  1496. {
  1497. guild.IsConnected = false;
  1498. await TimedInvokeAsync(_guildUnavailableEvent, nameof(GuildUnavailable), guild).ConfigureAwait(false);
  1499. }
  1500. }
  1501. private async Task TimedInvokeAsync(AsyncEvent<Func<Task>> eventHandler, string name)
  1502. {
  1503. if (eventHandler.HasSubscribers)
  1504. {
  1505. if (HandlerTimeout.HasValue)
  1506. await TimeoutWrap(name, () => eventHandler.InvokeAsync()).ConfigureAwait(false);
  1507. else
  1508. await eventHandler.InvokeAsync().ConfigureAwait(false);
  1509. }
  1510. }
  1511. private async Task TimedInvokeAsync<T>(AsyncEvent<Func<T, Task>> eventHandler, string name, T arg)
  1512. {
  1513. if (eventHandler.HasSubscribers)
  1514. {
  1515. if (HandlerTimeout.HasValue)
  1516. await TimeoutWrap(name, () => eventHandler.InvokeAsync(arg)).ConfigureAwait(false);
  1517. else
  1518. await eventHandler.InvokeAsync(arg).ConfigureAwait(false);
  1519. }
  1520. }
  1521. private async Task TimedInvokeAsync<T1, T2>(AsyncEvent<Func<T1, T2, Task>> eventHandler, string name, T1 arg1, T2 arg2)
  1522. {
  1523. if (eventHandler.HasSubscribers)
  1524. {
  1525. if (HandlerTimeout.HasValue)
  1526. await TimeoutWrap(name, () => eventHandler.InvokeAsync(arg1, arg2)).ConfigureAwait(false);
  1527. else
  1528. await eventHandler.InvokeAsync(arg1, arg2).ConfigureAwait(false);
  1529. }
  1530. }
  1531. private async Task TimedInvokeAsync<T1, T2, T3>(AsyncEvent<Func<T1, T2, T3, Task>> eventHandler, string name, T1 arg1, T2 arg2, T3 arg3)
  1532. {
  1533. if (eventHandler.HasSubscribers)
  1534. {
  1535. if (HandlerTimeout.HasValue)
  1536. await TimeoutWrap(name, () => eventHandler.InvokeAsync(arg1, arg2, arg3)).ConfigureAwait(false);
  1537. else
  1538. await eventHandler.InvokeAsync(arg1, arg2, arg3).ConfigureAwait(false);
  1539. }
  1540. }
  1541. 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)
  1542. {
  1543. if (eventHandler.HasSubscribers)
  1544. {
  1545. if (HandlerTimeout.HasValue)
  1546. await TimeoutWrap(name, () => eventHandler.InvokeAsync(arg1, arg2, arg3, arg4)).ConfigureAwait(false);
  1547. else
  1548. await eventHandler.InvokeAsync(arg1, arg2, arg3, arg4).ConfigureAwait(false);
  1549. }
  1550. }
  1551. 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)
  1552. {
  1553. if (eventHandler.HasSubscribers)
  1554. {
  1555. if (HandlerTimeout.HasValue)
  1556. await TimeoutWrap(name, () => eventHandler.InvokeAsync(arg1, arg2, arg3, arg4, arg5)).ConfigureAwait(false);
  1557. else
  1558. await eventHandler.InvokeAsync(arg1, arg2, arg3, arg4, arg5).ConfigureAwait(false);
  1559. }
  1560. }
  1561. private async Task TimeoutWrap(string name, Func<Task> action)
  1562. {
  1563. try
  1564. {
  1565. var timeoutTask = Task.Delay(HandlerTimeout.Value);
  1566. var handlersTask = action();
  1567. if (await Task.WhenAny(timeoutTask, handlersTask).ConfigureAwait(false) == timeoutTask)
  1568. {
  1569. await _gatewayLogger.WarningAsync($"A {name} handler is blocking the gateway task.").ConfigureAwait(false);
  1570. await handlersTask.ConfigureAwait(false); //Ensure the handler completes
  1571. }
  1572. }
  1573. catch (Exception ex)
  1574. {
  1575. await _gatewayLogger.WarningAsync($"A {name} handler has thrown an unhandled exception.", ex).ConfigureAwait(false);
  1576. }
  1577. }
  1578. private async Task UnknownGlobalUserAsync(string evnt, ulong userId)
  1579. {
  1580. string details = $"{evnt} User={userId}";
  1581. await _gatewayLogger.WarningAsync($"Unknown User ({details}).").ConfigureAwait(false);
  1582. }
  1583. private async Task UnknownChannelUserAsync(string evnt, ulong userId, ulong channelId)
  1584. {
  1585. string details = $"{evnt} User={userId} Channel={channelId}";
  1586. await _gatewayLogger.WarningAsync($"Unknown User ({details}).").ConfigureAwait(false);
  1587. }
  1588. private async Task UnknownGuildUserAsync(string evnt, ulong userId, ulong guildId)
  1589. {
  1590. string details = $"{evnt} User={userId} Guild={guildId}";
  1591. await _gatewayLogger.WarningAsync($"Unknown User ({details}).").ConfigureAwait(false);
  1592. }
  1593. private async Task IncompleteGuildUserAsync(string evnt, ulong userId, ulong guildId)
  1594. {
  1595. string details = $"{evnt} User={userId} Guild={guildId}";
  1596. await _gatewayLogger.DebugAsync($"User has not been downloaded ({details}).").ConfigureAwait(false);
  1597. }
  1598. private async Task UnknownChannelAsync(string evnt, ulong channelId)
  1599. {
  1600. string details = $"{evnt} Channel={channelId}";
  1601. await _gatewayLogger.WarningAsync($"Unknown Channel ({details}).").ConfigureAwait(false);
  1602. }
  1603. private async Task UnknownChannelAsync(string evnt, ulong channelId, ulong guildId)
  1604. {
  1605. if (guildId == 0)
  1606. {
  1607. await UnknownChannelAsync(evnt, channelId).ConfigureAwait(false);
  1608. return;
  1609. }
  1610. string details = $"{evnt} Channel={channelId} Guild={guildId}";
  1611. await _gatewayLogger.WarningAsync($"Unknown Channel ({details}).").ConfigureAwait(false);
  1612. }
  1613. private async Task UnknownRoleAsync(string evnt, ulong roleId, ulong guildId)
  1614. {
  1615. string details = $"{evnt} Role={roleId} Guild={guildId}";
  1616. await _gatewayLogger.WarningAsync($"Unknown Role ({details}).").ConfigureAwait(false);
  1617. }
  1618. private async Task UnknownGuildAsync(string evnt, ulong guildId)
  1619. {
  1620. string details = $"{evnt} Guild={guildId}";
  1621. await _gatewayLogger.WarningAsync($"Unknown Guild ({details}).").ConfigureAwait(false);
  1622. }
  1623. private async Task UnsyncedGuildAsync(string evnt, ulong guildId)
  1624. {
  1625. string details = $"{evnt} Guild={guildId}";
  1626. await _gatewayLogger.DebugAsync($"Unsynced Guild ({details}).").ConfigureAwait(false);
  1627. }
  1628. internal int GetAudioId() => _nextAudioId++;
  1629. //IDiscordClient
  1630. async Task<IApplication> IDiscordClient.GetApplicationInfoAsync(RequestOptions options)
  1631. => await GetApplicationInfoAsync().ConfigureAwait(false);
  1632. Task<IChannel> IDiscordClient.GetChannelAsync(ulong id, CacheMode mode, RequestOptions options)
  1633. => Task.FromResult<IChannel>(GetChannel(id));
  1634. Task<IReadOnlyCollection<IPrivateChannel>> IDiscordClient.GetPrivateChannelsAsync(CacheMode mode, RequestOptions options)
  1635. => Task.FromResult<IReadOnlyCollection<IPrivateChannel>>(PrivateChannels);
  1636. Task<IReadOnlyCollection<IDMChannel>> IDiscordClient.GetDMChannelsAsync(CacheMode mode, RequestOptions options)
  1637. => Task.FromResult<IReadOnlyCollection<IDMChannel>>(DMChannels);
  1638. Task<IReadOnlyCollection<IGroupChannel>> IDiscordClient.GetGroupChannelsAsync(CacheMode mode, RequestOptions options)
  1639. => Task.FromResult<IReadOnlyCollection<IGroupChannel>>(GroupChannels);
  1640. async Task<IReadOnlyCollection<IConnection>> IDiscordClient.GetConnectionsAsync(RequestOptions options)
  1641. => await GetConnectionsAsync().ConfigureAwait(false);
  1642. async Task<IInvite> IDiscordClient.GetInviteAsync(string inviteId, RequestOptions options)
  1643. => await GetInviteAsync(inviteId).ConfigureAwait(false);
  1644. Task<IGuild> IDiscordClient.GetGuildAsync(ulong id, CacheMode mode, RequestOptions options)
  1645. => Task.FromResult<IGuild>(GetGuild(id));
  1646. Task<IReadOnlyCollection<IGuild>> IDiscordClient.GetGuildsAsync(CacheMode mode, RequestOptions options)
  1647. => Task.FromResult<IReadOnlyCollection<IGuild>>(Guilds);
  1648. async Task<IGuild> IDiscordClient.CreateGuildAsync(string name, IVoiceRegion region, Stream jpegIcon, RequestOptions options)
  1649. => await CreateGuildAsync(name, region, jpegIcon).ConfigureAwait(false);
  1650. Task<IUser> IDiscordClient.GetUserAsync(ulong id, CacheMode mode, RequestOptions options)
  1651. => Task.FromResult<IUser>(GetUser(id));
  1652. Task<IUser> IDiscordClient.GetUserAsync(string username, string discriminator, RequestOptions options)
  1653. => Task.FromResult<IUser>(GetUser(username, discriminator));
  1654. Task<IReadOnlyCollection<IVoiceRegion>> IDiscordClient.GetVoiceRegionsAsync(RequestOptions options)
  1655. => Task.FromResult<IReadOnlyCollection<IVoiceRegion>>(VoiceRegions);
  1656. Task<IVoiceRegion> IDiscordClient.GetVoiceRegionAsync(string id, RequestOptions options)
  1657. => Task.FromResult<IVoiceRegion>(GetVoiceRegion(id));
  1658. async Task IDiscordClient.StartAsync()
  1659. => await StartAsync().ConfigureAwait(false);
  1660. async Task IDiscordClient.StopAsync()
  1661. => await StopAsync().ConfigureAwait(false);
  1662. }
  1663. }