MagnusGreiff / server

src/socket.js

"use strict";

const WebSocket = require("ws");

const handleProtocols = (protocols /*, request */ ) => {
    console.info(`Incoming protocol requests '${protocols}'.`);
    for (var i = 0; i < protocols.length; i++) {
        if (protocols[i] === "text") {
            return "text";
        } else if (protocols[i] === "json") {
            return "json";
        }
    }
    return false;
};

let makeWss = (server) => {
    const wss = new WebSocket.Server({
        server: server,
        clientTracking: true, // keep track on connected clients
        handleProtocols: handleProtocols
    });

    return wss;
};

let broadcastExcept = (wss, ws, data) => {
    let clients = 0;

    wss.clients.forEach((client) => {
        if (client !== ws && client.readyState === WebSocket.OPEN) {
            clients++;
            if (ws.protocol === "json") {
                let msg = {
                    data: data
                };

                client.send(JSON.stringify(msg));
            } else {
                client.send(data);
            }
        }
    });
    console.info(`Broadcasted data to ${clients} (${wss.clients.size}) clients.`);
};

let socket = (server) => {
    const wss = makeWss(server);

    wss.on("connection", (ws /*, req*/ ) => {
        console.info("Connection received. Adding client.");

        ws.on("message", (message) => {
            console.log(message);

console.log looks like debug code. Are you sure you do not want to remove it?

            broadcastExcept(wss, ws, message);
        });

        ws.on("error", (error) => {
            console.error(`Server error: ${error}`);
        });

        ws.on("close", (code, reason) => {
            console.info(`Closing connection: ${code} ${reason}`);
        });
    });
}

module.exports = {
    socket: socket
};