ericN93 / module

index.js

'use strict';

const webSocket = require('ws');

module.exports = function(server) {
	const wss = new webSocket.Server({
		server: server,
		clientTracking: true,
		handleProtocols: (protocol) => {return 'json'}

The parameter protocol is not used and could be removed.

	});

	let sendToAll = (ws, data) => {
		wss.clients.forEach((client) => {
	      	if (data.type==='connection'){
				client.send(JSON.stringify(data));
			} else {
				if(client !== ws && client.readyState === webSocket.OPEN){
					client.send(JSON.stringify(data));
				}
			}
	    });
	}

	wss.on('connection', (ws, data) => {

The parameter data is not used and could be removed.

			let sendData = {
				'message': {author: 'Server', message:'test'},
				'data': {clients: wss.clients.size},
				'type': 'connection'
			}
			sendToAll(ws, sendData)

	  ws.on('message', (data) => {

		  let dataFromSend = JSON.parse(data)
		  let sendData = {
			  'message': {author: dataFromSend.message.author, message: dataFromSend.message.message},
			  'data': {clients: wss.clients.size},
			  'type': 'message',
		  }
	  		sendToAll(ws, sendData);
	  });

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

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

		  let sendData = {
			  'message': {author: 'Server', message:'A user has left the channel'},
			  'data': {clients: wss.clients.size},
			  'type': 'close'
		  }
		  sendToAll(ws, sendData)
      });
	});

	return wss;
};