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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
| const roomMgr = require("./roomManager6");
const userMgr = require("./userManager6");
const HEARTBEAT_TIMEOUT = 5000;
function broadcastRoom(roomId, data){
const room = roomMgr.getRoom(roomId);
const json = JSON.stringify(data);
for(const [ws] of room.users.entries()){
if(ws.readyState === ws.OPEN){
ws.send(json);
}
}
}
function broadcastUserList(roomId){
const list = roomMgr.getUserList(roomId);
broadcastRoom(roomId, {
type:"userList",
list
});
}
function handleMessage(ws, msg, heartbeatMap){
const type = msg.type;
switch(type){
case "ping":{
ws.send(JSON.stringify({type:"pong"}));
clearTimeout(heartbeatMap.timer);
heartbeatMap.timer = setTimeout(()=> ws.close(), HEARTBEAT_TIMEOUT);
break;
}
case "login":{
const {username, roomId} = msg;
const inputName = username.trim();
const rid = roomId || "default";
const room = roomMgr.getRoom(rid);
// 当前ws已经登录
if(heartbeatMap.username){
ws.send(JSON.stringify({
type:"init",
score: room.score,
msg:`你已经登录【${heartbeatMap.username}】房间${rid}`
}));
break;
}
// 房间用户名查重
if(userMgr.isNameExist(rid, inputName)){
ws.send(JSON.stringify({
type:"loginError",
msg:"❌ 用户名已存在，请换一个名字"
}));
break;
}
// 登录成功
heartbeatMap.username = inputName;
heartbeatMap.roomId = rid;
room.users.set(ws, inputName);
userMgr.addUserName(rid, inputName);
console.log(`用户【${inputName}】进入房间【${rid}】`);
clearTimeout(heartbeatMap.timer);
heartbeatMap.timer = setTimeout(() => ws.close(), HEARTBEAT_TIMEOUT);
ws.send(JSON.stringify({
type:"init",
score: room.score,
msg:`欢迎你【${inputName}】进入房间【${rid}】`,
roomId: rid
}));
broadcastRoom(rid, {
type:"notice",
content:`🎉 用户【${inputName}】加入房间【${rid}】！`,
color: userMgr.getColorByName(inputName)
});
broadcastUserList(rid);
break;
}
case "click":{
const username = heartbeatMap.username;
const roomId = heartbeatMap.roomId;
if(!username || !roomId) return;
if(!userMgr.checkClickLimit(username)){
ws.send(JSON.stringify({
type:"notice",
content:"⚠️ 点击太快，请稍后再试"
}));
return;
}
const room = roomMgr.getRoom(roomId);
room.score++;
roomMgr.saveRooms();
console.log(`【${username}】房间【${roomId}】加分:${room.score}`);
broadcastRoom(roomId,{
type:"update",
score: room.score,
user: username,
color: userMgr.getColorByName(username)
});
break;
}
case "chat":{
const username = heartbeatMap.username;
const roomId = heartbeatMap.roomId;
if(!username || !roomId) return;
const rawContent = msg.content.trim();
if(!rawContent) return;
if(!userMgr.checkChatLimit(username)){
ws.send(JSON.stringify({
type:"notice",
content:"⚠️ 聊天频率太快，3秒后再发"
}));
return;
}
const safeText = userMgr.filterBadWord(rawContent);
broadcastRoom(roomId,{
type:"chat",
user: username,
color: userMgr.getColorByName(username),
content: safeText
});
break;
}
}
}
// 用户下线清理
function handleClose(ws, heartbeatMap){
const username = heartbeatMap.username;
const roomId = heartbeatMap.roomId;
if(!username || !roomId) return;
const room = roomMgr.getRoom(roomId);
room.users.delete(ws);
userMgr.removeUserName(roomId, username);
userMgr.cleanUserCache(username);
console.log(`用户【${username}】离开房间【${roomId}】`);
broadcastRoom(roomId, {
type:"notice",
content:`👋 用户【${username}】离开房间【${roomId}】`,
color: userMgr.getColorByName(username)
});
broadcastUserList(roomId);
}
// 踢人
function kickUser(wss, targetName){
for(const [ws, username] of wss.clients.entries()){
if(username === targetName){
ws.send(JSON.stringify({
type:"kick",
msg:`你被管理员踢出房间`
}));
ws.close(1008, "被踢下线");
console.log(`✅ 已踢用户【${targetName}】`);
return true;
}
}
console.log(`❌ 用户【${targetName}】不存在`);
return false;
}
module.exports = {
handleMessage,
handleClose,
kickUser
};
|