-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.js
253 lines (224 loc) · 7.94 KB
/
main.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
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
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
/*
启动服务端程序
*/
const app = require('express')()
const server = require('http').Server(app)
const io = require('socket.io')(server)
const path = require('path');
const db = require('./db/db.js')
const chinaTime = require('china-time')
const svgCaptcha = require('svg-captcha');
// //支持截图功能
// const webshot = require('webshot');
//记录所有已经登陆过的用户
var users = []
var id_now = 1
var captcha = svgCaptcha.create();
//启动了服务器
server.listen(3000, () => {
console.log('服务器启动成功了')
})
//express处理静态资源
//把public目录设置为静态资源
app.use(require('express').static(path.join(__dirname, 'public')))
app.get('/', function (req, res) {
res.redirect('/index.html')
})
db.selectAll('select count(*) as sum from message', (e, r) => {
//id 按照消息发送的先后顺序递增
console.log('数据库共有' + r[0].sum + '条历史消息记录')
id_now = r[0].sum + 1
})
//一进入聊天室就加载信息 传个socket对象
function initMessage(socket) {
db.selectAll('select * from message order by id asc', (e, res) => {
// console.log(r)
// console.log(r[0])
for (var i = 0; i < res.length; i++) {
// console.log(res[i])
if (res[i].type === 'image') {
//广播给当前进入聊天室的用户
socket.emit('receiveImage', res[i])
console.log('历史消息:')
console.log(res[i])
} else {
//广播给当前进入聊天室的用户
socket.emit('receiveMessage', res[i])
console.log('历史消息:')
console.log(res[i])
}
}
})
console.log("当前用户:");
console.log(users);
}
/*
此处有个bug,如果用户没通过验证,就会显示一名undefined用户离开了聊天室,因为它connection了
已修复 解决方案:disconnect的时候不当他是位用户
另一种不好的思路:在客户端连接数据库判断,这样暴露安全隐患
*/
io.on('connection', function (socket) {
//验证码
socket.on('getCaptcha', () => {
captcha = svgCaptcha.create({
size: 4,
fontSize: 50,
width: 100,
height: 40,
background: '#cc9966'
})
socket.emit('captcha', captcha.data)
})
//验证验证码
socket.on('checkoutFirst', (data, callback) => {
console.log(data);
if (data.verify.toLowerCase() === captcha.text.toLowerCase()) {
callback({
status: "ok"
});
} else {
callback({
status: "error"
});
}
});
socket.on('checkoutLogin', data => {
// 连接数据库验证
let msg = '',
resultData = '';
db.selectAll("select * from usersInformation where username ='" + data.username + "' ", (e, r) => {
let tt = r.length;
if (tt == 0) {
msg = "用户名不存在";
} else if (data.password != r[0].password) {
msg = "用户密码错误";
} else {
resultData = r[0];
msg = "用户密码正确"
}
socket.emit('checkoutAnswer', {
msg: msg,
avatar: resultData.avatar,
online: users.find(item => item.username === data.username) ? true : false
})
console.log(msg, resultData)
})
})
socket.on('login', data => {
//判断,如果在data在users中存在,说明该用户登陆过了,不允许登录
//如果data在user中不存在,说明用户没有登陆,允许登录
let user = users.find(item => item.username === data.username)
if (user) {
//表示用户在线
socket.emit('loginError', {
msg: '登陆失败'
})
// console.log('登陆失败')
} else {
// // //连接数据库获取用户头像
// db.selectAll("select * from usersInformation where username ='" + data.username + "' ", (e, r) => {
// data.avatar = r[0].avatar
// console.log(r[0].avatar)
// })
//把登陆成功的用户信息存储起来
//socket.username ? avatar 内置对象? 不太像
socket.username = data.username
socket.avatar = data.avatar
//表示用户不在线,把用户存入user数组
users.push(data)
//告诉用户,登陆成功
socket.emit('loginSuccess', data)
// console.log('登陆成功')
//告诉所有人,有新用户加入到了聊天室,广播消息
//socket.emit 给当前用户发消息 io.emit 给所有用户发消息
io.emit('addUser', data)
//告诉所有用户,当前聊天室用户列表以及数量
io.emit('userList', users)
//一进入聊天室就加载信息
initMessage(socket)
}
})
//用户断开连接功能
//监听用户断开连接
socket.on('disconnect', () => {
if (socket.username === 'undefined') return;
//把当前用户信息从user中删除
let idx = users.findIndex(item => item.username === socket.username)
//删除掉断开连接的人
users.splice(idx, 1)
// 1.告诉所有人,有人离开了聊天室
io.emit('deleteUser', {
username: socket.username,
avatar: socket.avatar
})
// 2.告诉所有人,userList发生更新
io.emit('userList', users)
})
//监听聊天的消息
socket.on('sendMessage', data => {
//存入数据库
var time = chinaTime('YY/MM/DD HH:mm')
let saveData = {
id: id_now,
username: data.username,
content: data.content,
time: time,
avatar: data.avatar,
type: data.type
}
db.insertData('message', saveData, (e, r) => {
console.log('消息存入成功')
id_now++
})
console.log('聊天消息')
console.log(saveData)
//广播给所有用户
io.emit('receiveMessage', saveData)
})
//接受图片的信息
// 由于图片所占内存过大,为节省数据库空间,可以不做处理
socket.on('sendImage', data => {
var time = chinaTime('YY/MM/DD HH:mm')
let saveData = {
id: id_now,
username: data.username,
content: data.img,
time: time,
avatar: data.avatar,
type: data.type
}
db.insertData('message', saveData, (e, r) => {
console.log('消息存入成功')
id_now++
})
console.log('聊天消息')
console.log(saveData)
//广播给所有用户
io.emit('receiveImage', saveData)
})
// //实现截图功能
// socket.on('webshot', url => {
// webshot(url, 'hello_world.png', {siteType:'html'}, function(err) {
// // screenshot now saved to hello_world.png
// })
// })
//注册用户
socket.on('registerUser', data => {
// console.log(data)
//插入之前要查询一遍,确保用户名未被注册
db.selectAll("select * from usersInformation where username = '" + data.username + "' ", (e, r) => {
let tt = r.length;
if (tt == 1) {
console.log("账号已经被注册")
socket.emit('registerError')
} else {
let saveData = data
//注册
db.insertData('usersInformation', saveData, (e, r) => {
console.log('注册成功')
socket.emit('registerSuccess')
})
}
})
})
})