diff --git a/webrtc/src/Server.ts b/webrtc/src/Server.ts
index 06a36df9..cdda10ec 100644
--- a/webrtc/src/Server.ts
+++ b/webrtc/src/Server.ts
@@ -1,15 +1,19 @@
import { Server as WebSocketServer } from "ws";
-import { WebSocket, CLOSECODES, Payload, OPCODES } from "@fosscord/gateway";
+import { WebSocket, Payload, } from "@fosscord/gateway";
import { Config, initDatabase } from "@fosscord/util";
import OPCodeHandlers from "./opcodes";
-import { setHeartbeat } from "./util"
-import mediasoup from "mediasoup";
+import { setHeartbeat } from "./util";
+import * as mediasoup from "mediasoup";
+import { types as MediasoupTypes } from "mediasoup";
var port = Number(process.env.PORT);
if (isNaN(port)) port = 3004;
export class Server {
public ws: WebSocketServer;
+ public mediasoupWorkers: MediasoupTypes.Worker[] = [];
+ public mediasoupRouters: MediasoupTypes.Router[] = [];
+ public mediasoupTransports: MediasoupTypes.Transport[] = [];
constructor() {
this.ws = new WebSocketServer({
@@ -23,9 +27,9 @@ export class Server {
const payload: Payload = JSON.parse(message);
if (OPCodeHandlers[payload.op])
- await OPCodeHandlers[payload.op](socket, payload);
+ await OPCodeHandlers[payload.op].call(this, socket, payload);
else
- console.error(`Unimplemented`, payload)
+ console.error(`Unimplemented`, payload);
});
});
}
@@ -34,7 +38,63 @@ export class Server {
// @ts-ignore
await initDatabase();
await Config.init();
+ await this.createWorkers();
console.log("[DB] connected");
console.log(`[WebRTC] online on 0.0.0.0:${port}`);
}
+
+ async createWorkers(): Promise<void> {
+ const numWorkers = 1;
+ for (let i = 0; i < numWorkers; i++) {
+ const worker = await mediasoup.createWorker();
+ if (!worker) return;
+
+ worker.on("died", () => {
+ console.error("mediasoup worker died");
+ });
+
+ worker.observer.on("newrouter", async (router: MediasoupTypes.Router) => {
+ console.log("new router");
+
+ this.mediasoupRouters.push(router);
+
+ router.observer.on("newtransport", (transport: MediasoupTypes.Transport) => {
+ console.log("new transport");
+
+ this.mediasoupTransports.push(transport);
+ })
+
+ await router.createWebRtcTransport({
+ listenIps: [{ ip: "127.0.0.1" }],
+ enableUdp: true,
+ enableTcp: true,
+ preferUdp: true
+ });
+ });
+
+ await worker.createRouter({
+ mediaCodecs: [
+ {
+ kind: "audio",
+ mimeType: "audio/opus",
+ clockRate: 48000,
+ channels: 2
+ },
+ {
+ kind: "video",
+ mimeType: "video/H264",
+ clockRate: 90000,
+ parameters:
+ {
+ "packetization-mode": 1,
+ "profile-level-id": "42e01f",
+ "level-asymmetry-allowed": 1
+ }
+ }
+ ]
+ });
+
+ this.mediasoupWorkers.push(worker);
+ }
+ }
}
|
