mirror of https://github.com/bra1n/townsquare.git
59 lines
1.3 KiB
JavaScript
59 lines
1.3 KiB
JavaScript
const fs = require("fs");
|
|
const https = require("https");
|
|
const WebSocket = require("ws");
|
|
|
|
const server = https.createServer({
|
|
cert: fs.readFileSync("cert.pem"),
|
|
key: fs.readFileSync("key.pem")
|
|
});
|
|
const wss = new WebSocket.Server({
|
|
...(process.env.NODE_ENV === "development" ? { port: 8081 } : { server }),
|
|
verifyClient: info =>
|
|
!!info.origin.match(/^https?:\/\/(bra1n\.github\.io|localhost)/i)
|
|
});
|
|
|
|
function noop() {}
|
|
|
|
function heartbeat() {
|
|
this.isAlive = true;
|
|
}
|
|
|
|
wss.on("connection", function connection(ws, req) {
|
|
ws.channel = req.url
|
|
.split("/")
|
|
.pop()
|
|
.toLocaleLowerCase();
|
|
ws.isAlive = true;
|
|
ws.on("pong", heartbeat);
|
|
ws.on("message", function incoming(data) {
|
|
if (!data.match(/^\["ping/i)) {
|
|
console.log(ws.channel, wss.clients.size, data);
|
|
}
|
|
wss.clients.forEach(function each(client) {
|
|
if (
|
|
client !== ws &&
|
|
client.readyState === WebSocket.OPEN &&
|
|
client.channel === ws.channel
|
|
) {
|
|
client.send(data);
|
|
}
|
|
});
|
|
});
|
|
});
|
|
|
|
const interval = setInterval(function ping() {
|
|
wss.clients.forEach(function each(ws) {
|
|
if (ws.isAlive === false) return ws.terminate();
|
|
ws.isAlive = false;
|
|
ws.ping(noop);
|
|
});
|
|
}, 30000);
|
|
|
|
wss.on("close", function close() {
|
|
clearInterval(interval);
|
|
});
|
|
|
|
if (process.env.NODE_ENV !== "development") {
|
|
server.listen(8080);
|
|
}
|