-
Notifications
You must be signed in to change notification settings - Fork 0
/
server.js
69 lines (57 loc) · 1.67 KB
/
server.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
const express = require('express');
const app = express();
const server = require('http').Server(app);
const io = require('socket.io')(server,{cors:{origin:"*"}});
const PORT = 9999;
const rooms = new Map();
app.use(express.json())
app.get('/rooms/:id', (req, res) => {
const { id: roomId } = req.params;
const obj = rooms.has(roomId) ? {
users: [...rooms.get(roomId).get('users').values()],
messages: [...rooms.get(roomId).get('messages').values()]
} : { users: [], messages: []};
res.json(obj)
});
app.post('/rooms', (req, res) => {
const { roomId, userName } = req.body;
if (!rooms.has(roomId)) {
rooms.set(roomId, new Map([
['users', new Map()],
['messages', []],
]));
}
res.send();
});
io.on('connection', socket => {
console.log(`Connected ${socket.id}`);
socket.on('ROOM_JOIN', ({ roomId, userName }) => {
socket.join(roomId);
rooms.get(roomId).get('users').set(socket.id, userName);
const users = [...rooms.get(roomId).get('users').values()];
socket.to(roomId).emit('ROOM_SET_USERS', users);
});
socket.on('ROOM_NEW_MESSAGE', ({ roomId, userName, text }) => {
const obj = {
roomId,
userName,
text
};
rooms.get(roomId).get('messages').push(obj);
socket.to(roomId).emit('ROOM_NEW_MESSAGE', obj);
});
socket.on('disconnect', () => {
rooms.forEach((value, roomId) => {
if(value.get('users').delete(socket.id)) {
const users = [...value.get('users').values()];
socket.to(roomId).emit('ROOM_SET_USERS', users);
}
});
});
});
server.listen(PORT, (error) => {
if (error) {
throw Error(error);
}
console.log(`Server listen on port ${PORT}`);
});