summary refs log tree commit diff
path: root/gateway/src/opcodes/VoiceStateUpdate.ts
diff options
context:
space:
mode:
Diffstat (limited to 'gateway/src/opcodes/VoiceStateUpdate.ts')
-rw-r--r--gateway/src/opcodes/VoiceStateUpdate.ts67
1 files changed, 50 insertions, 17 deletions
diff --git a/gateway/src/opcodes/VoiceStateUpdate.ts b/gateway/src/opcodes/VoiceStateUpdate.ts
index 0d51513d..4719949c 100644
--- a/gateway/src/opcodes/VoiceStateUpdate.ts
+++ b/gateway/src/opcodes/VoiceStateUpdate.ts
@@ -1,26 +1,59 @@
-import { VoiceStateUpdateSchema } from "../schema/VoiceStateUpdate.ts";
-import { CLOSECODES, Payload } from "../util/Constants";
-import { Send } from "../util/Send";
-
+import { VoiceStateUpdateSchema } from "../schema/VoiceStateUpdateSchema";
+import { Payload } from "../util/Constants";
 import WebSocket from "../util/WebSocket";
 import { check } from "./instanceOf";
-// TODO: implementation
+import { Config, emitEvent, VoiceServerUpdateEvent, VoiceState, VoiceStateUpdateEvent } from "@fosscord/util";
+import { genVoiceToken } from "../util/SessionUtils";
 // TODO: check if a voice server is setup
-// TODO: save voice servers in database and retrieve them
 // Notice: Bot users respect the voice channel's user limit, if set. When the voice channel is full, you will not receive the Voice State Update or Voice Server Update events in response to your own Voice State Update. Having MANAGE_CHANNELS permission bypasses this limit and allows you to join regardless of the channel being full or not.
 
 export async function onVoiceStateUpdate(this: WebSocket, data: Payload) {
 	check.call(this, VoiceStateUpdateSchema, data.d);
 	const body = data.d as VoiceStateUpdateSchema;
 
-	await Send(this, {
-		op: 0,
-		s: this.sequence++,
-		t: "VOICE_SERVER_UPDATE",
-		d: {
-			token: ``,
-			guild_id: body.guild_id,
-			endpoint: `localhost:3004`,
-		},
-	});
-}
+	let voiceState
+	try {
+		voiceState = await VoiceState.findOneOrFail({where:{ user_id: this.user_id },relations: ["member", "member.user", "member.roles"]});
+		if(voiceState.session_id !== this.session_id && body.channel_id === null) { //Should we also check guild_id === null?
+			//changing deaf or mute on a client that's not the one with the same session of the voicestate in the database should be ignored
+			return
+		}
+
+		//The event send by Discord's client on channel leave has both guild_id and channel_id as null
+		if(body.guild_id === null) body.guild_id = voiceState.guild_id;
+		voiceState.assign(body);
+	} catch (error) {
+		voiceState = new VoiceState({
+			...body,
+			user_id: this.user_id,
+			deaf: false,
+			mute: false,
+			suppress: false
+		})
+	}
+
+	//If the session changed we generate a new token
+	if(voiceState.session_id !== this.session_id)
+		voiceState.token = genVoiceToken()
+	voiceState.session_id = this.session_id
+
+	//TODO the member should only have these properties: hoisted_role, deaf, joined_at, mute, roles, user
+	//TODO the member.user should only have these properties: avatar, discriminator, id, username
+	const {id, ...newObj} = voiceState;
+
+	await Promise.all([
+		voiceState.save(),
+		emitEvent({ event: "VOICE_STATE_UPDATE", data: newObj, guild_id: voiceState.guild_id} as VoiceStateUpdateEvent),
+	]);
+
+	//If it's null it means that we are leaving the channel and this event is not needed
+	if(voiceState.channel_id !== null) {
+		const regions = Config.get().regions;
+
+		await emitEvent({ event: "VOICE_SERVER_UPDATE", data: {
+				token: voiceState.token,
+				guild_id: voiceState.guild_id,
+				endpoint: regions.available[0].endpoint, //TODO return best endpoint or default
+			}, guild_id: voiceState.guild_id } as VoiceServerUpdateEvent)
+	}
+}
\ No newline at end of file