adapter creator for djs v12
This commit is contained in:
parent
18aa70531e
commit
2b82fe2b70
7 changed files with 81 additions and 17 deletions
62
src/VoiceInterface/AdapterCreator.ts
Normal file
62
src/VoiceInterface/AdapterCreator.ts
Normal file
|
@ -0,0 +1,62 @@
|
||||||
|
import { DiscordGatewayAdapterCreator, DiscordGatewayAdapterLibraryMethods } from "@discordjs/voice";
|
||||||
|
import { VoiceChannel, Snowflake, Client, Constants, WebSocketShard, Guild, StageChannel } from "discord.js";
|
||||||
|
import { GatewayVoiceServerUpdateDispatchData, GatewayVoiceStateUpdateDispatchData } from "discord-api-types/v8";
|
||||||
|
|
||||||
|
const adapters = new Map<Snowflake, DiscordGatewayAdapterLibraryMethods>();
|
||||||
|
const trackedClients = new Set<Client>();
|
||||||
|
|
||||||
|
function trackClient(client: Client) {
|
||||||
|
if (trackedClients.has(client)) return;
|
||||||
|
trackedClients.add(client);
|
||||||
|
client.ws.on(Constants.WSEvents.VOICE_SERVER_UPDATE, (payload: GatewayVoiceServerUpdateDispatchData) => {
|
||||||
|
adapters.get(payload.guild_id)?.onVoiceServerUpdate(payload);
|
||||||
|
});
|
||||||
|
client.ws.on(Constants.WSEvents.VOICE_STATE_UPDATE, (payload: GatewayVoiceStateUpdateDispatchData) => {
|
||||||
|
if (payload.guild_id && payload.session_id && payload.user_id === client.user?.id) {
|
||||||
|
adapters.get(payload.guild_id)?.onVoiceStateUpdate(payload);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
const trackedGuilds = new Map<WebSocketShard, Set<Snowflake>>();
|
||||||
|
|
||||||
|
function cleanupGuilds(shard: WebSocketShard) {
|
||||||
|
const guilds = trackedGuilds.get(shard);
|
||||||
|
if (guilds) {
|
||||||
|
for (const guildID of guilds.values()) {
|
||||||
|
adapters.get(guildID)?.destroy();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function trackGuild(guild: Guild) {
|
||||||
|
let guilds = trackedGuilds.get(guild.shard);
|
||||||
|
if (!guilds) {
|
||||||
|
const cleanup = () => cleanupGuilds(guild.shard);
|
||||||
|
guild.shard.on("close", cleanup);
|
||||||
|
guild.shard.on("destroyed", cleanup);
|
||||||
|
guilds = new Set();
|
||||||
|
trackedGuilds.set(guild.shard, guilds);
|
||||||
|
}
|
||||||
|
guilds.add(guild.id);
|
||||||
|
}
|
||||||
|
|
||||||
|
export function VoiceAdapterCreator(channel: VoiceChannel | StageChannel): DiscordGatewayAdapterCreator {
|
||||||
|
return (methods) => {
|
||||||
|
adapters.set(channel.guild.id, methods);
|
||||||
|
trackClient(channel.client);
|
||||||
|
trackGuild(channel.guild);
|
||||||
|
return {
|
||||||
|
sendPayload(data) {
|
||||||
|
if (channel.guild.shard.status === Constants.Status.READY) {
|
||||||
|
channel.guild.shard.send(data);
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
},
|
||||||
|
destroy() {
|
||||||
|
return adapters.delete(channel.guild.id);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
};
|
||||||
|
}
|
|
@ -1,5 +1,5 @@
|
||||||
import { VoiceChannel, StageChannel, Collection, Snowflake } from "discord.js";
|
import { VoiceChannel, StageChannel, Collection, Snowflake } from "discord.js";
|
||||||
import { entersState, joinVoiceChannel, VoiceConnection, VoiceConnectionStatus } from "@discordjs/voice";
|
import { DiscordGatewayAdapterCreator, entersState, joinVoiceChannel, VoiceConnection, VoiceConnectionStatus } from "@discordjs/voice";
|
||||||
import { StreamDispatcher } from "./BasicStreamDispatcher";
|
import { StreamDispatcher } from "./BasicStreamDispatcher";
|
||||||
|
|
||||||
class VoiceUtils {
|
class VoiceUtils {
|
||||||
|
@ -47,12 +47,13 @@ class VoiceUtils {
|
||||||
options?: {
|
options?: {
|
||||||
deaf?: boolean;
|
deaf?: boolean;
|
||||||
maxTime?: number;
|
maxTime?: number;
|
||||||
|
adapter?: DiscordGatewayAdapterCreator;
|
||||||
}
|
}
|
||||||
) {
|
) {
|
||||||
let conn = joinVoiceChannel({
|
let conn = joinVoiceChannel({
|
||||||
guildId: channel.guild.id,
|
guildId: channel.guild.id,
|
||||||
channelId: channel.id,
|
channelId: channel.id,
|
||||||
adapterCreator: (channel.guild as any).voiceAdapterCreator, // eslint-disable-line @typescript-eslint/no-explicit-any
|
adapterCreator: options.adapter ?? (channel.guild as any).voiceAdapterCreator, // eslint-disable-line @typescript-eslint/no-explicit-any
|
||||||
selfDeaf: Boolean(options.deaf)
|
selfDeaf: Boolean(options.deaf)
|
||||||
});
|
});
|
||||||
|
|
||||||
|
|
|
@ -6,4 +6,5 @@ export { Queue } from "./Structures/Queue";
|
||||||
export { Track } from "./Structures/Track";
|
export { Track } from "./Structures/Track";
|
||||||
export { VoiceUtils } from "./VoiceInterface/VoiceUtils";
|
export { VoiceUtils } from "./VoiceInterface/VoiceUtils";
|
||||||
export { VoiceEvents, StreamDispatcher } from "./VoiceInterface/BasicStreamDispatcher";
|
export { VoiceEvents, StreamDispatcher } from "./VoiceInterface/BasicStreamDispatcher";
|
||||||
|
export { VoiceAdapterCreator } from "./VoiceInterface/AdapterCreator";
|
||||||
export * from "./types/types";
|
export * from "./types/types";
|
||||||
|
|
Loading…
Reference in a new issue