1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
|
import { Server as WebSocketServer } from "ws";
import { WebSocket, CLOSECODES } from "@fosscord/gateway";
import { Config, initDatabase } from "@fosscord/util";
import OPCodeHandlers, { Payload } from "./opcodes";
import { setHeartbeat } from "./util";
import * as mediasoup from "mediasoup";
import { types as MediasoupTypes } from "mediasoup";
import udp from "dgram";
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.WebRtcTransport[] = [];
public mediasoupProducers: MediasoupTypes.Producer[] = [];
public mediasoupConsumers: MediasoupTypes.Consumer[] = [];
public testUdp = udp.createSocket("udp6");
constructor() {
this.ws = new WebSocketServer({
port,
maxPayload: 4096,
});
this.ws.on("connection", async (socket: WebSocket) => {
await setHeartbeat(socket);
socket.on("message", async (message: string) => {
const payload: Payload = JSON.parse(message);
if (OPCodeHandlers[payload.op])
try {
await OPCodeHandlers[payload.op].call(this, socket, payload);
}
catch (e) {
console.error(e);
socket.close(CLOSECODES.Unknown_error);
}
else {
console.error(`Unimplemented`, payload);
socket.close(CLOSECODES.Unknown_opcode);
}
});
});
// this.testUdp.bind(50001);
// this.testUdp.on("message", (msg, rinfo) => {
// if (msg[0] === 0 && msg[1] === 1 && msg[2] === 0) { //idk stun?
// }
// })
}
async listen(): Promise<void> {
// @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({ logLevel: "debug", logTags: ["dtls", "ice", "info", "message", "bwe"] });
if (!worker) return;
worker.on("died", () => {
console.error("mediasoup worker died");
});
worker.observer.on("newrouter", async (router: MediasoupTypes.Router) => {
console.log("new router created [id:%s]", router.id);
this.mediasoupRouters.push(router);
router.observer.on("newtransport", async (transport: MediasoupTypes.WebRtcTransport) => {
console.log("new transport created [id:%s]", transport.id);
await transport.enableTraceEvent();
transport.on('dtlsstatechange', (dtlsstate) => {
console.log(dtlsstate);
})
transport.on("sctpstatechange", (sctpstate) => {
console.log(sctpstate)
})
router.observer.on("newrtpobserver", (rtpObserver: MediasoupTypes.RtpObserver) => {
console.log("new RTP observer created [id:%s]", rtpObserver.id);
// rtpObserver.observer.on("")
})
transport.on("connect", () => {
console.log("transport connect");
});
transport.observer.on("newproducer", (producer: MediasoupTypes.Producer) => {
console.log("new producer created [id:%s]", producer.id);
this.mediasoupProducers.push(producer);
});
transport.observer.on("newconsumer", (consumer: MediasoupTypes.Consumer) => {
console.log("new consumer created [id:%s]", consumer.id);
this.mediasoupConsumers.push(consumer);
consumer.on("rtp", (rtpPacket) => {
console.log(rtpPacket);
});
});
transport.observer.on("newdataproducer", (dataProducer) => {
console.log("new data producer created [id:%s]", dataProducer.id);
});
transport.on("trace", (trace) => {
console.log(trace);
});
this.mediasoupTransports.push(transport);
});
});
await worker.createRouter({
mediaCodecs: [
{
kind: "audio",
mimeType: "audio/opus",
clockRate: 48000,
channels: 2,
preferredPayloadType: 111,
},
],
});
this.mediasoupWorkers.push(worker);
}
}
}
|
