Initial Support for Spacebar WebRTC

Signed-off-by: Christopher Lentocha <[email protected]>

Author: CE1CECL

package-lock.json

@@ -116,16 +116,19 @@
 		"@spacebar/api": "dist/api",
 		"@spacebar/cdn": "dist/cdn",
 		"@spacebar/gateway": "dist/gateway",
-		"@spacebar/util": "dist/util"
+		"@spacebar/util": "dist/util",
+		"@spacebar/webrtc": "dist/webrtc"
 	},
 	"optionalDependencies": {
 		"@yukikaze-bot/erlpack": "^1.0.1",
 		"jimp": "^1.6.0",
 		"mysql": "^2.18.1",
+		"medooze-media-server": "0.129.9",
 		"nodemailer-mailgun-transport": "^2.1.5",
 		"nodemailer-mailjet-transport": "github:n0script22/nodemailer-mailjet-transport",
 		"nodemailer-sendgrid-transport": "github:Maria-Golomb/nodemailer-sendgrid-transport",
 		"pg": "^8.13.1",
+		"semantic-sdp": "3.26.0",
 		"sqlite3": "^5.1.7"
 	}
 }

package.json

@@ -23,6 +23,7 @@ import http from "http";
 import * as Api from "@spacebar/api";
 import * as Gateway from "@spacebar/gateway";
 import { CDNServer } from "@spacebar/cdn";
+import * as Webrtc from "@spacebar/webrtc";
 import express from "express";
 import { green, bold } from "picocolors";
 import { Config, initDatabase, Sentry } from "@spacebar/util";
@@ -36,12 +37,14 @@ server.on("request", app);
 const api = new Api.SpacebarServer({ server, port, production, app });
 const cdn = new CDNServer({ server, port, production, app });
 const gateway = new Gateway.Server({ server, port, production });
+const webrtc = new Webrtc.Server({ server, port, production });
 
 process.on("SIGTERM", async () => {
 	console.log("Shutting down due to SIGTERM");
 	await gateway.stop();
 	await cdn.stop();
 	await api.stop();
+	await webrtc.stop();
 	server.close();
 	Sentry.close();
 });
@@ -54,7 +57,12 @@ async function main() {
 	await new Promise((resolve) =>
 		server.listen({ port }, () => resolve(undefined)),
 	);
-	await Promise.all([api.start(), cdn.start(), gateway.start()]);
+	await Promise.all([
+		api.start(),
+		cdn.start(),
+		gateway.start(),
+		webrtc.start(),
+	]);
 
 	Sentry.errorHandler(app);
 

src/bundle/Server.ts

@@ -56,6 +56,7 @@ export class Server {
 		}
 
 		this.server.on("upgrade", (request, socket, head) => {
+			if (request.url?.includes("voice")) return;
 			this.ws.handleUpgrade(request, socket, head, (socket) => {
 				this.ws.emit("connection", socket, request);
 			});

src/gateway/Server.ts

@@ -39,6 +39,8 @@ import {
 export async function onVoiceStateUpdate(this: WebSocket, data: Payload) {
 	check.call(this, VoiceStateUpdateSchema, data.d);
 	const body = data.d as VoiceStateUpdateSchema;
+	const isNew = body.channel_id === null && body.guild_id === null;
+	let isChanged = false;
 
 	let voiceState: VoiceState;
 	try {
@@ -54,6 +56,8 @@ export async function onVoiceStateUpdate(this: WebSocket, data: Payload) {
 			return;
 		}
 
+		if (voiceState.channel_id !== body.channel_id) isChanged = true;
+
 		//If a user change voice channel between guild we should send a left event first
 		if (
 			voiceState.guild_id !== body.guild_id &&
@@ -111,7 +115,7 @@ export async function onVoiceStateUpdate(this: WebSocket, data: Payload) {
 	]);
 
 	//If it's null it means that we are leaving the channel and this event is not needed
-	if (voiceState.channel_id !== null) {
+	if ((isNew || isChanged) && voiceState.channel_id !== null) {
 		const guild = await Guild.findOne({
 			where: { id: voiceState.guild_id },
 		});
@@ -134,7 +138,7 @@ export async function onVoiceStateUpdate(this: WebSocket, data: Payload) {
 				guild_id: voiceState.guild_id,
 				endpoint: guildRegion.endpoint,
 			},
-			guild_id: voiceState.guild_id,
+			user_id: this.user_id,
 		} as VoiceServerUpdateEvent);
 	}
 }

src/gateway/opcodes/VoiceStateUpdate.ts

@@ -16,7 +16,7 @@
 	along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */
 
-// import { VoiceOPCodes } from "@spacebar/webrtc";
+import { VoiceOPCodes } from "@spacebar/webrtc";
 
 export enum OPCODES {
 	Dispatch = 0,
@@ -63,7 +63,7 @@ export enum CLOSECODES {
 }
 
 export interface Payload {
-	op: OPCODES /*  | VoiceOPCodes */;
+	op: OPCODES | VoiceOPCodes;
 	// eslint-disable-next-line @typescript-eslint/no-explicit-any
 	d?: any;
 	s?: number;