server.js

const path = require('path');
const http = require('http');
const express = require('express');
const socketio = require('socket.io');
const formatMessage = require('./utils/messages');
const { userJoin, getCurrentUser, userLeave, getRoomUsers } = require('./utils/users');

const app = express();
const server = http.createServer(app);
const io = socketio(server);

// Set static folder
app.use(express.static(path.join(__dirname, 'public')));

const botName = 'Chatte-R bot';
// Run when user connects
io.on('connection', socket => {
    // console.log('new connection');

    socket.on('joinRoom', ({ username, room }) => {

        const user = userJoin(socket.id, username, room)

        socket.join(user.room);

        // welcome for current user
        socket.emit('message', formatMessage(botName, 'Welcome to Chatte-R'));

        // broadcast when new user connects
        socket.broadcast.to(user.room)
            .emit('message', formatMessage(botName, `${user.username} has joined the chat`));

    });

    // Listen for chatMessage
    socket.on('chatMessage', (msg) => {
        // console.log(msg);
        const user = getCurrentUser(socket.id);
        io.to(user.room).emit('message', formatMessage(user.username, msg));
    });

    // when user disconnects
    socket.on('disconnect', () => {
        const user = userLeave(socket.id);

        if (user) {
            io.to(user.room).emit('message', formatMessage(botName, `${user.username} has left the chat`));
        }
    });
});

const PORT = process.env.PORT || 3000;

server.listen(PORT, () => console.log(`Server is up and running on port ${PORT}`));