const express = require('express'), app = express(), server = require('http').createServer(app), io = require('socket.io').listen(server), //用于保存用户信息的数组 users = []; let kit = { //判断用户是否存在 isHaveUser(user) { var flag = false; users.forEach(function (item) { if (item.name == user.name) { flag = true; } }) return flag; }, //删除某一用户 delUser(id) { users.forEach(function (item, index) { if (item.id == id) { users.splice(index, 1); } }) } } //设置静态资源 app.use('/static', express.static(__dirname + '/static')); //用户访问网站页面会根据浏览器userAgent返回不同的页面 app.get("/", (req, res) => { let userAgent = req.headers['user-agent'].toLowerCase(); let bIsIpad = userAgent.match(/ipad/i) == "ipad"; let bIsIphoneOs = userAgent.match(/iphone os/i) == "iphone os"; let bIsMidp = userAgent.match(/midp/i) == "midp"; let bIsUc7 = userAgent.match(/rv:1.2.3.4/i) == "rv:1.2.3.4"; let bIsUc = userAgent.match(/ucweb/i) == "ucweb"; let bIsAndroid = userAgent.match(/android/i) == "android"; let bIsCE = userAgent.match(/windows ce/i) == "windows ce"; let bIsWM = userAgent.match(/windows mobile/i) == "windows mobile"; if (bIsIpad || bIsIphoneOs || bIsMidp || bIsUc7 || bIsUc || bIsAndroid || bIsCE || bIsWM) { let path = __dirname + '/static/iChat.html'; res.sendFile(path); } else { let path = __dirname + '/static/index.html'; res.sendFile(path); } }) io.sockets.on('connection',(socket)=>{ //创建用户链接 socket.on('login', (user)=> { if (kit.isHaveUser(user)) { console.log("登录失败!", user) socket.emit('loginFail', "登录失败,昵称已存在!"); } else { socket.user = user; user.id = socket.id; user.roomId=socket.id; user.address = socket.handshake.address.replace(/::ffff:/,""); console.log("登录成功!", user) socket.emit('loginSuccess', user, users); users.push(user) socket.broadcast.emit('system', user, 'join'); } }); //用户注销链接 socket.on('disconnect',()=> { if (socket.user != null) { kit.delUser(socket.user.id); console.log("用户退出!", socket.user) socket.broadcast.emit('system', socket.user, 'logout'); } }); //群发消息 socket.on('groupMessage',(from, to,message,type)=>{ //用户登录状态掉线,重置用户登录状态 if (!socket.user) { from.roomId = socket.id; socket.user = from; users.push(from); socket.broadcast.emit('system', from, 'join'); socket.emit('loginSuccess', from, []); } socket.broadcast.emit('groupMessage', socket.user, to,message,type); }); //发送私信 socket.on('message',(from, to,message,type)=> { //用户登录状态掉线,重置用户登录状态 if (!socket.user) { from.roomId = socket.id; socket.user = from; users.push(from); socket.broadcast.emit('system', from, 'join'); socket.emit('loginSuccess', from, []); } socket.broadcast.to(to.roomId).emit('message', socket.user, to,message,type); }); //判断用户重新连接 if(socket.handshake.query.User){ let user=JSON.parse(socket.handshake.query.User); socket.user = user; user.roomId = socket.id; user.address = socket.handshake.address.replace(/::ffff:/,""); console.log("重新连接成功!", user) socket.emit('loginSuccess', user, users); users.push(user) socket.broadcast.emit('system', user, 'join'); } }); //启动服务器 server.listen(3000, function () { console.log("服务器已启动在:3000端口", "http://localhost:3000") });