chat-server/server/chat/index.js

const jwt = require('jsonwebtoken');
const { chatlog } = require('../database/models');
const { Op } = require('sequelize');

const chat = io => {
	io.on('connection', socket => {
		//middleware
		socket.use((request, next) => {
			//verify request format
			if (!['open chat', 'message'].includes(request[0])) {
				return next(`Invalid request to the chat server ${request[0]}`);
			}
			return next();
		});

		socket.use((request, next) => {
			//authenticate the jwt
			return jwt.verify(request[1].accessToken, process.env.SECRET_ACCESS, (err, user) => {
				if (err) {
					return next(err);
				}

				const room = socket.user?.room; //save room, if any
				socket.user = user;
				socket.user.room = room;

				return next();
			});
		});

		//handle errors
		socket.on('error', err => {
			console.log('socket error:', err);
			socket.emit(err);
		});

		//from here, handles all normal messages
		socket.on('open chat', message => {
			//handle rooms - only in a room if you've opened chat
			const newlyOpened = !socket.user.room;
			socket.user.room = socket.user.room || 'general'; //default to general

			if (!newlyOpened) {
				return;
			}

			socket.join(socket.user.room);

			//broadcast to this room
			socket.broadcast.to(socket.user.room).emit('message', { emphasis: true, text: `${socket.user.username} entered chat` });

			//log
			chatlog.create({
				username: socket.user.username,
				text: `${socket.user.username} entered chat`,
				room: socket.user.room,
				emphasis: true
			});

			//send backlog to the user
			chatlog.findAll({
				where: {
					room: {
						[Op.eq]: socket.user.room
					}
				},
				order: [
					['id', 'ASC']
				]
			})
				.then(rows => rows.map(row => row.dataValues))
				.then(rows => rows.filter(row => {
					//emphasis and strong don't use usernames
					return !(row.emphasis || row.strong);
				}))
				.then(rows => socket.emit('backlog', rows))
				.then(() => {
					//send a # to the user
					const count = io.sockets.size;
					socket.emit('message', { emphasis: true, text: count == 1 ? `${count} person in the chat` : `${count} people in the chat` });
				})
			;
		});

		socket.on('message', message => {
			//server commands begin with a '/'
			if (message.text.startsWith('/')) {
				return executeCommand(socket, message.text);
			}

			//broadcast to this room
			socket.broadcast.to(socket.user.room).emit('message', { username: socket.user.username, text: message.text });

			//log
			chatlog.create({
				username: socket.user.username,
				text: message.text,
				room: socket.user.room
			});
		});

		socket.on('disconnect', reason => {
			//broadcast to this room
			if (!socket.user) {
				return;
			}

			socket.broadcast.to(socket.user.room || '.error').emit('message', { emphasis: true, text: `${socket.user.username} left chat` });

			//log
			chatlog.create({
				username: socket.user.username,
				text: `${socket.user.username} left chat`,
				room: socket.user.room,
				emphasis: true
			});
		});
	});
};

//handle commands
const executeCommand = (socket, command) => {
	switch(command.split(' ')[0]) {
		case '/room':
			const room = command.split(' ')[1];

			if (!room) {
				socket.emit('no room argument');
				break;
			}

			//broadcast to the old room
			socket.broadcast.to(socket.user.room).emit('message', { emphasis: true, text: `${socket.user.username} left the room (going to ${room})` });

			//log
			chatlog.create({
				username: socket.user.username,
				text: `${socket.user.username} left the room`,
				room: socket.user.room,
				emphasis: true
			});

			//move
			socket.leave(socket.user.room);
			socket.user.room = room;
			socket.join(socket.user.room);

			//broadcast to the new room
			socket.broadcast.to(socket.user.room).emit('message', { emphasis: true, text: `${socket.user.username} entered the room` });

			//log
			chatlog.create({
				username: socket.user.username,
				text: `${socket.user.username} entered the room`,
				room: socket.user.room,
				emphasis: true
			});

			//update the user
			socket.emit('message', { emphasis: true, text: `Entered room ${socket.user.room}` });
			break;

		default:
			socket.emit('message', { emphasis: true, text: 'Unknown command' });
	}
};

module.exports = chat;

//TODO: add banning and muting controls