server.js

const express = require('express');
const app = express();
const http = require('http');
const path = require('path');
const { Server } = require('socket.io');
const ACTIONS = require('./src/Actions');

const server = http.createServer(app);
const io = new Server(server);

app.use(express.static('build'));
app.use((req, res, next) => {
    res.sendFile(path.join(__dirname, 'build', 'index.html'));
});

const userSocketMap = {};
const { spawn } = require('child_process');

function getAllConnectedClients(roomId) {
    // Map
    return Array.from(io.sockets.adapter.rooms.get(roomId) || []).map(
        (socketId) => {
            return {
                socketId,
                username: userSocketMap[socketId],
            };
        }
    );
}

io.on('connection', (socket) => {
    console.log('socket connected', socket.id);

    socket.on(ACTIONS.JOIN, ({ roomId, username }) => {
        userSocketMap[socket.id] = username;
        socket.join(roomId);
        const clients = getAllConnectedClients(roomId);
        clients.forEach(({ socketId }) => {
            io.to(socketId).emit(ACTIONS.JOINED, {
                clients,
                username,
                socketId: socket.id,
            });
        });
    });

    socket.on(ACTIONS.CODE_CHANGE, ({ roomId, code }) => {
        socket.in(roomId).emit(ACTIONS.CODE_CHANGE, { code });
    });

    socket.on(ACTIONS.SYNC_CODE, ({ socketId, code }) => {
        io.to(socketId).emit(ACTIONS.CODE_CHANGE, { code });
    });

    socket.on(ACTIONS.RUN_PYTHON, ({ code }) => {
        let stdout = '';
        let stderr = '';

        const proc = spawn('python', ['-c', code], {
            windowsHide: true,
        });

        proc.stdout.on('data', (d) => {
            stdout += d.toString();
        });

        proc.stderr.on('data', (d) => {
            stderr += d.toString();
        });

        proc.on('close', () => {
            if (stderr.trim().length > 0) {
                socket.emit(ACTIONS.PYTHON_OUTPUT, {
                    output: stdout,
                    error: stderr,
                });
            } else {
                socket.emit(ACTIONS.PYTHON_OUTPUT, {
                    output: stdout,
                    error: null,
                });
            }
        });

        proc.on('error', (err) => {
            socket.emit(ACTIONS.PYTHON_OUTPUT, {
                output: stdout,
                error: String(err?.message ?? err),
            });
        });
    });

    socket.on(ACTIONS.RUN_JAVASCRIPT, ({ code }) => {
        let stdout = '';
        let stderr = '';

        // Executes JS safely only for trusted users.
        const proc = spawn('node', ['-e', code], {
            windowsHide: true,
        });

        proc.stdout.on('data', (d) => {
            stdout += d.toString();
        });

        proc.stderr.on('data', (d) => {
            stderr += d.toString();
        });

        proc.on('close', () => {
            if (stderr.trim().length > 0) {
                socket.emit(ACTIONS.JAVASCRIPT_OUTPUT, {
                    output: stdout,
                    error: stderr,
                });
            } else {
                socket.emit(ACTIONS.JAVASCRIPT_OUTPUT, {
                    output: stdout,
                    error: null,
                });
            }
        });

        proc.on('error', (err) => {
            socket.emit(ACTIONS.JAVASCRIPT_OUTPUT, {
                output: stdout,
                error: String(err?.message ?? err),
            });
        });
    });


    socket.on('disconnecting', () => {
        const rooms = [...socket.rooms];
        rooms.forEach((roomId) => {
            socket.in(roomId).emit(ACTIONS.DISCONNECTED, {
                socketId: socket.id,
                username: userSocketMap[socket.id],
            });
        });
        delete userSocketMap[socket.id];
        socket.leave();
    });
});

const PORT = process.env.PORT || 5000;
server.listen(PORT, () => console.log(`Listening on port ${PORT}`));