mirror of
https://github.com/discordjs/discord.js.git
synced 2026-03-10 16:43:31 +01:00
816aed478e
* feat(core): implement some ws send events * fix: check guild id and add a timeout * fix: only check for nonce * chore: update readme * chore: make requested changes * chore: make methods consistent * chore: fix readme example * chore: move shard ID calculation to util Co-authored-by: kodiakhq[bot] <49736102+kodiakhq[bot]@users.noreply.github.com>
272 lines
12 KiB
TypeScript
272 lines
12 KiB
TypeScript
import { setTimeout } from 'node:timers';
|
|
import type { REST } from '@discordjs/rest';
|
|
import { calculateShardId } from '@discordjs/util';
|
|
import { WebSocketShardEvents, type WebSocketManager } from '@discordjs/ws';
|
|
import { DiscordSnowflake } from '@sapphire/snowflake';
|
|
import { AsyncEventEmitter } from '@vladfrangu/async_event_emitter';
|
|
import {
|
|
GatewayDispatchEvents,
|
|
GatewayOpcodes,
|
|
type GatewayVoiceStateUpdateData,
|
|
type APIGuildMember,
|
|
type GatewayAutoModerationActionExecutionDispatchData,
|
|
type GatewayAutoModerationRuleCreateDispatchData,
|
|
type GatewayAutoModerationRuleDeleteDispatchData,
|
|
type GatewayAutoModerationRuleUpdateDispatchData,
|
|
type GatewayChannelCreateDispatchData,
|
|
type GatewayChannelDeleteDispatchData,
|
|
type GatewayChannelPinsUpdateDispatchData,
|
|
type GatewayChannelUpdateDispatchData,
|
|
type GatewayGuildBanAddDispatchData,
|
|
type GatewayGuildBanRemoveDispatchData,
|
|
type GatewayGuildCreateDispatchData,
|
|
type GatewayGuildDeleteDispatchData,
|
|
type GatewayGuildEmojisUpdateDispatchData,
|
|
type GatewayGuildIntegrationsUpdateDispatchData,
|
|
type GatewayGuildMemberAddDispatchData,
|
|
type GatewayGuildMemberRemoveDispatchData,
|
|
type GatewayGuildMembersChunkDispatchData,
|
|
type GatewayGuildMemberUpdateDispatchData,
|
|
type GatewayGuildRoleCreateDispatchData,
|
|
type GatewayGuildRoleDeleteDispatchData,
|
|
type GatewayGuildRoleUpdateDispatchData,
|
|
type GatewayGuildScheduledEventCreateDispatchData,
|
|
type GatewayGuildScheduledEventDeleteDispatchData,
|
|
type GatewayGuildScheduledEventUpdateDispatchData,
|
|
type GatewayGuildScheduledEventUserAddDispatchData,
|
|
type GatewayGuildScheduledEventUserRemoveDispatchData,
|
|
type GatewayGuildStickersUpdateDispatchData,
|
|
type GatewayGuildUpdateDispatchData,
|
|
type GatewayIntegrationCreateDispatchData,
|
|
type GatewayIntegrationDeleteDispatchData,
|
|
type GatewayIntegrationUpdateDispatchData,
|
|
type GatewayInteractionCreateDispatchData,
|
|
type GatewayInviteCreateDispatchData,
|
|
type GatewayInviteDeleteDispatchData,
|
|
type GatewayMessageCreateDispatchData,
|
|
type GatewayMessageDeleteBulkDispatchData,
|
|
type GatewayMessageDeleteDispatchData,
|
|
type GatewayMessageReactionAddDispatchData,
|
|
type GatewayMessageReactionRemoveAllDispatchData,
|
|
type GatewayMessageReactionRemoveDispatchData,
|
|
type GatewayMessageReactionRemoveEmojiDispatchData,
|
|
type GatewayMessageUpdateDispatchData,
|
|
type GatewayPresenceUpdateDispatchData,
|
|
type GatewayReadyDispatchData,
|
|
type GatewayRequestGuildMembersData,
|
|
type GatewayStageInstanceCreateDispatchData,
|
|
type GatewayStageInstanceDeleteDispatchData,
|
|
type GatewayStageInstanceUpdateDispatchData,
|
|
type GatewayThreadCreateDispatchData,
|
|
type GatewayThreadDeleteDispatchData,
|
|
type GatewayThreadListSyncDispatchData,
|
|
type GatewayThreadMembersUpdateDispatchData,
|
|
type GatewayThreadMemberUpdateDispatchData,
|
|
type GatewayThreadUpdateDispatchData,
|
|
type GatewayTypingStartDispatchData,
|
|
type GatewayUserUpdateDispatchData,
|
|
type GatewayVoiceServerUpdateDispatchData,
|
|
type GatewayVoiceStateUpdateDispatchData,
|
|
type GatewayWebhooksUpdateDispatchData,
|
|
type GatewayPresenceUpdateData,
|
|
} from 'discord-api-types/v10';
|
|
import { API } from './api/index.js';
|
|
|
|
export interface IntrinsicProps {
|
|
/**
|
|
* The REST API
|
|
*/
|
|
api: API;
|
|
/**
|
|
* The id of the shard that emitted the event
|
|
*/
|
|
shardId: number;
|
|
}
|
|
|
|
export interface WithIntrinsicProps<T> extends IntrinsicProps {
|
|
data: T;
|
|
}
|
|
|
|
export interface MappedEvents {
|
|
[GatewayDispatchEvents.ChannelCreate]: [WithIntrinsicProps<GatewayChannelCreateDispatchData>];
|
|
[GatewayDispatchEvents.ChannelDelete]: [WithIntrinsicProps<GatewayChannelDeleteDispatchData>];
|
|
[GatewayDispatchEvents.ChannelPinsUpdate]: [WithIntrinsicProps<GatewayChannelPinsUpdateDispatchData>];
|
|
[GatewayDispatchEvents.ChannelUpdate]: [WithIntrinsicProps<GatewayChannelUpdateDispatchData>];
|
|
[GatewayDispatchEvents.GuildBanAdd]: [WithIntrinsicProps<GatewayGuildBanAddDispatchData>];
|
|
[GatewayDispatchEvents.GuildBanRemove]: [WithIntrinsicProps<GatewayGuildBanRemoveDispatchData>];
|
|
[GatewayDispatchEvents.GuildCreate]: [WithIntrinsicProps<GatewayGuildCreateDispatchData>];
|
|
[GatewayDispatchEvents.GuildDelete]: [WithIntrinsicProps<GatewayGuildDeleteDispatchData>];
|
|
[GatewayDispatchEvents.GuildEmojisUpdate]: [WithIntrinsicProps<GatewayGuildEmojisUpdateDispatchData>];
|
|
[GatewayDispatchEvents.GuildIntegrationsUpdate]: [WithIntrinsicProps<GatewayGuildIntegrationsUpdateDispatchData>];
|
|
[GatewayDispatchEvents.GuildMemberAdd]: [WithIntrinsicProps<GatewayGuildMemberAddDispatchData>];
|
|
[GatewayDispatchEvents.GuildMemberRemove]: [WithIntrinsicProps<GatewayGuildMemberRemoveDispatchData>];
|
|
[GatewayDispatchEvents.GuildMemberUpdate]: [WithIntrinsicProps<GatewayGuildMemberUpdateDispatchData>];
|
|
[GatewayDispatchEvents.GuildMembersChunk]: [WithIntrinsicProps<GatewayGuildMembersChunkDispatchData>];
|
|
[GatewayDispatchEvents.GuildRoleCreate]: [WithIntrinsicProps<GatewayGuildRoleCreateDispatchData>];
|
|
[GatewayDispatchEvents.GuildRoleDelete]: [WithIntrinsicProps<GatewayGuildRoleDeleteDispatchData>];
|
|
[GatewayDispatchEvents.GuildRoleUpdate]: [WithIntrinsicProps<GatewayGuildRoleUpdateDispatchData>];
|
|
[GatewayDispatchEvents.GuildScheduledEventCreate]: [WithIntrinsicProps<GatewayGuildScheduledEventCreateDispatchData>];
|
|
[GatewayDispatchEvents.GuildScheduledEventDelete]: [WithIntrinsicProps<GatewayGuildScheduledEventDeleteDispatchData>];
|
|
[GatewayDispatchEvents.GuildScheduledEventUpdate]: [WithIntrinsicProps<GatewayGuildScheduledEventUpdateDispatchData>];
|
|
[GatewayDispatchEvents.GuildScheduledEventUserAdd]: [
|
|
WithIntrinsicProps<GatewayGuildScheduledEventUserAddDispatchData>,
|
|
];
|
|
[GatewayDispatchEvents.GuildScheduledEventUserRemove]: [
|
|
WithIntrinsicProps<GatewayGuildScheduledEventUserRemoveDispatchData>,
|
|
];
|
|
[GatewayDispatchEvents.GuildStickersUpdate]: [WithIntrinsicProps<GatewayGuildStickersUpdateDispatchData>];
|
|
[GatewayDispatchEvents.GuildUpdate]: [WithIntrinsicProps<GatewayGuildUpdateDispatchData>];
|
|
[GatewayDispatchEvents.IntegrationCreate]: [WithIntrinsicProps<GatewayIntegrationCreateDispatchData>];
|
|
[GatewayDispatchEvents.IntegrationDelete]: [WithIntrinsicProps<GatewayIntegrationDeleteDispatchData>];
|
|
[GatewayDispatchEvents.IntegrationUpdate]: [WithIntrinsicProps<GatewayIntegrationUpdateDispatchData>];
|
|
[GatewayDispatchEvents.InteractionCreate]: [WithIntrinsicProps<GatewayInteractionCreateDispatchData>];
|
|
[GatewayDispatchEvents.InviteCreate]: [WithIntrinsicProps<GatewayInviteCreateDispatchData>];
|
|
[GatewayDispatchEvents.InviteDelete]: [WithIntrinsicProps<GatewayInviteDeleteDispatchData>];
|
|
[GatewayDispatchEvents.MessageCreate]: [WithIntrinsicProps<GatewayMessageCreateDispatchData>];
|
|
[GatewayDispatchEvents.MessageDelete]: [WithIntrinsicProps<GatewayMessageDeleteDispatchData>];
|
|
[GatewayDispatchEvents.MessageDeleteBulk]: [WithIntrinsicProps<GatewayMessageDeleteBulkDispatchData>];
|
|
[GatewayDispatchEvents.MessageReactionAdd]: [WithIntrinsicProps<GatewayMessageReactionAddDispatchData>];
|
|
[GatewayDispatchEvents.MessageReactionRemove]: [WithIntrinsicProps<GatewayMessageReactionRemoveDispatchData>];
|
|
[GatewayDispatchEvents.MessageReactionRemoveAll]: [WithIntrinsicProps<GatewayMessageReactionRemoveAllDispatchData>];
|
|
[GatewayDispatchEvents.MessageReactionRemoveEmoji]: [
|
|
WithIntrinsicProps<GatewayMessageReactionRemoveEmojiDispatchData>,
|
|
];
|
|
[GatewayDispatchEvents.MessageUpdate]: [WithIntrinsicProps<GatewayMessageUpdateDispatchData>];
|
|
[GatewayDispatchEvents.PresenceUpdate]: [WithIntrinsicProps<GatewayPresenceUpdateDispatchData>];
|
|
[GatewayDispatchEvents.Ready]: [WithIntrinsicProps<GatewayReadyDispatchData>];
|
|
[GatewayDispatchEvents.StageInstanceCreate]: [WithIntrinsicProps<GatewayStageInstanceCreateDispatchData>];
|
|
[GatewayDispatchEvents.StageInstanceDelete]: [WithIntrinsicProps<GatewayStageInstanceDeleteDispatchData>];
|
|
[GatewayDispatchEvents.StageInstanceUpdate]: [WithIntrinsicProps<GatewayStageInstanceUpdateDispatchData>];
|
|
[GatewayDispatchEvents.ThreadCreate]: [WithIntrinsicProps<GatewayThreadCreateDispatchData>];
|
|
[GatewayDispatchEvents.ThreadDelete]: [WithIntrinsicProps<GatewayThreadDeleteDispatchData>];
|
|
[GatewayDispatchEvents.ThreadListSync]: [WithIntrinsicProps<GatewayThreadListSyncDispatchData>];
|
|
[GatewayDispatchEvents.ThreadMemberUpdate]: [WithIntrinsicProps<GatewayThreadMemberUpdateDispatchData>];
|
|
[GatewayDispatchEvents.ThreadMembersUpdate]: [WithIntrinsicProps<GatewayThreadMembersUpdateDispatchData>];
|
|
[GatewayDispatchEvents.ThreadUpdate]: [WithIntrinsicProps<GatewayThreadUpdateDispatchData>];
|
|
[GatewayDispatchEvents.UserUpdate]: [WithIntrinsicProps<GatewayUserUpdateDispatchData>];
|
|
[GatewayDispatchEvents.VoiceServerUpdate]: [WithIntrinsicProps<GatewayVoiceServerUpdateDispatchData>];
|
|
[GatewayDispatchEvents.VoiceStateUpdate]: [WithIntrinsicProps<GatewayVoiceStateUpdateDispatchData>];
|
|
[GatewayDispatchEvents.WebhooksUpdate]: [WithIntrinsicProps<GatewayWebhooksUpdateDispatchData>];
|
|
[GatewayDispatchEvents.Resumed]: [WithIntrinsicProps<never>];
|
|
[GatewayDispatchEvents.TypingStart]: [WithIntrinsicProps<GatewayTypingStartDispatchData>];
|
|
[GatewayDispatchEvents.AutoModerationActionExecution]: [
|
|
WithIntrinsicProps<GatewayAutoModerationActionExecutionDispatchData>,
|
|
];
|
|
[GatewayDispatchEvents.AutoModerationRuleCreate]: [WithIntrinsicProps<GatewayAutoModerationRuleCreateDispatchData>];
|
|
[GatewayDispatchEvents.AutoModerationRuleDelete]: [WithIntrinsicProps<GatewayAutoModerationRuleDeleteDispatchData>];
|
|
[GatewayDispatchEvents.AutoModerationRuleUpdate]: [WithIntrinsicProps<GatewayAutoModerationRuleUpdateDispatchData>];
|
|
}
|
|
|
|
export type ManagerShardEventsMap = {
|
|
[K in keyof MappedEvents]: MappedEvents[K];
|
|
};
|
|
|
|
export interface ClientOptions {
|
|
rest: REST;
|
|
ws: WebSocketManager;
|
|
}
|
|
|
|
export class Client extends AsyncEventEmitter<ManagerShardEventsMap> {
|
|
public readonly rest: REST;
|
|
|
|
public readonly ws: WebSocketManager;
|
|
|
|
public readonly api: API;
|
|
|
|
public constructor({ rest, ws }: ClientOptions) {
|
|
super();
|
|
this.rest = rest;
|
|
this.ws = ws;
|
|
this.api = new API(rest);
|
|
|
|
this.ws.on(WebSocketShardEvents.Dispatch, ({ data: dispatch, shardId }) => {
|
|
// @ts-expect-error event props can't be resolved properly, but they are correct
|
|
this.emit(dispatch.t, this.wrapIntrinsicProps(dispatch.d, shardId));
|
|
});
|
|
}
|
|
|
|
/**
|
|
* Requests guild members from the gateway.
|
|
*
|
|
* @see {@link https://discord.com/developers/docs/topics/gateway-events#request-guild-members}
|
|
* @param options - The options for the request
|
|
* @param timeout - The timeout for waiting for each guild members chunk event
|
|
*/
|
|
public async requestGuildMembers(options: GatewayRequestGuildMembersData, timeout = 10_000) {
|
|
const shardId = calculateShardId(options.guild_id, await this.ws.getShardCount());
|
|
const nonce = options.nonce ?? DiscordSnowflake.generate().toString();
|
|
|
|
const promise = new Promise<APIGuildMember[]>((resolve, reject) => {
|
|
const guildMembers: APIGuildMember[] = [];
|
|
|
|
const timer = setTimeout(() => {
|
|
reject(new Error('Request timed out'));
|
|
}, timeout);
|
|
|
|
const handler = ({ data }: MappedEvents[GatewayDispatchEvents.GuildMembersChunk][0]) => {
|
|
timer.refresh();
|
|
|
|
if (data.nonce !== nonce) return;
|
|
|
|
guildMembers.push(...data.members);
|
|
|
|
if (data.chunk_index >= data.chunk_count - 1) {
|
|
this.off(GatewayDispatchEvents.GuildMembersChunk, handler);
|
|
resolve(guildMembers);
|
|
}
|
|
};
|
|
|
|
this.on(GatewayDispatchEvents.GuildMembersChunk, handler);
|
|
});
|
|
|
|
await this.ws.send(shardId, {
|
|
op: GatewayOpcodes.RequestGuildMembers,
|
|
// eslint-disable-next-line id-length
|
|
d: {
|
|
...options,
|
|
nonce,
|
|
},
|
|
});
|
|
|
|
return promise;
|
|
}
|
|
|
|
/**
|
|
* Updates the voice state of the bot user
|
|
*
|
|
* @see {@link https://discord.com/developers/docs/topics/gateway-events#update-voice-state}
|
|
* @param options - The options for updating the voice state
|
|
*/
|
|
public async updateVoiceState(options: GatewayVoiceStateUpdateData) {
|
|
const shardId = calculateShardId(options.guild_id, await this.ws.getShardCount());
|
|
|
|
await this.ws.send(shardId, {
|
|
op: GatewayOpcodes.VoiceStateUpdate,
|
|
// eslint-disable-next-line id-length
|
|
d: options,
|
|
});
|
|
}
|
|
|
|
/**
|
|
* Updates the presence of the bot user
|
|
*
|
|
* @param shardId - The id of the shard to update the presence in
|
|
* @param options - The options for updating the presence
|
|
*/
|
|
public async updatePresence(shardId: number, options: GatewayPresenceUpdateData) {
|
|
await this.ws.send(shardId, {
|
|
op: GatewayOpcodes.PresenceUpdate,
|
|
// eslint-disable-next-line id-length
|
|
d: options,
|
|
});
|
|
}
|
|
|
|
private wrapIntrinsicProps<T>(obj: T, shardId: number): WithIntrinsicProps<T> {
|
|
return {
|
|
api: this.api,
|
|
shardId,
|
|
data: obj,
|
|
};
|
|
}
|
|
}
|