This commit is contained in:
Skillz4Killz
2021-10-24 19:13:23 +00:00
committed by GitHub
parent ebdbfcf721
commit 2427a5d2c8
17 changed files with 293 additions and 211 deletions
+15 -62
View File
@@ -1,67 +1,20 @@
import { eventHandlers, setApplicationId, setBotId } from "../../bot.ts";
import { cache } from "../../cache.ts";
import { Bot } from "../../bot.ts";
import type { DiscordGatewayPayload } from "../../types/gateway/gateway_payload.ts";
import type { Ready } from "../../types/gateway/ready.ts";
import { snowflakeToBigint } from "../../util/bigint.ts";
import { DiscordenoShard, ws } from "../../ws/ws.ts";
import type { DiscordReady } from "../../types/gateway/ready.ts";
export function handleReady(data: DiscordGatewayPayload, shardId: number) {
export function handleReady(bot: Bot, data: DiscordGatewayPayload, shardId: number) {
const payload = data.d as DiscordReady;
// Triggered on each shard
eventHandlers.shardReady?.(shardId);
bot.events.ready(bot, {
shardId,
v: payload.v,
user: bot.transformers.user(bot, payload.user),
guilds: payload.guilds.map((p) => bot.transformers.snowflake(p.id)),
sessionId: payload.session_id,
shard: payload.shard,
applicationId: bot.transformers.snowflake(payload.application.id),
}, payload);
// The bot has already started, the last shard is resumed, however.
if (cache.isReady) return;
const shard = ws.shards.get(shardId);
if (!shard) return;
const payload = data.d as Ready;
setBotId(payload.user.id);
setApplicationId(payload.application.id);
// Set ready to false just to go sure
shard.ready = false;
// All guilds are unavailable at first
shard.unavailableGuildIds = new Set(payload.guilds.map((g) => snowflakeToBigint(g.id)));
// Falied to load check
shard.failedToLoadTimeoutId = setTimeout(() => {
eventHandlers.shardFailedToLoad?.(shard.id, shard.unavailableGuildIds);
// Force execute the loaded function to prevent infinite loop
return loaded(shard);
}, 5000);
}
export function guildAvailable(shard: DiscordenoShard, guildId: bigint) {
if (!shard.failedToLoadTimeoutId) return;
clearTimeout(shard.failedToLoadTimeoutId);
shard.unavailableGuildIds.delete(guildId);
if (!shard.unavailableGuildIds.size) return loaded(shard);
shard.failedToLoadTimeoutId = setTimeout(() => {
eventHandlers.shardFailedToLoad?.(shard.id, shard.unavailableGuildIds);
// Force execute the loaded function to prevent infinite loop
return loaded(shard);
}, 5000);
}
function loaded(shard: DiscordenoShard) {
shard.ready = true;
// If it is not the last shard we can't go full ready
if (shard.id !== ws.lastShardId) return;
// Still some shards are loading so wait another 2 seconds for them
if (ws.shards.some((shard) => !shard.ready)) {
setTimeout(() => {
eventHandlers.debug?.("loop", `3. Running setTimeout in READY file.`);
loaded(shard);
}, 2000);
return;
}
cache.isReady = true;
eventHandlers.ready?.();
if (!bot.id) bot.id = bot.transformers.snowflake(payload.user.id);
if (!bot.applicationId) bot.applicationId = bot.transformers.snowflake(payload.application.id);
}