NodeChatRoom/server.js at master · noobakong/NodeChatRoom · GitHub

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
var express = require('express');
var app = express();
var http = require('http').Server(app);

var io = require("socket.io")(http);

var users = []; // 储存登录用户
var usersInfo = [];  // 存储用户姓名和头像

// 路由为/默认www静态文件夹
app.use('/', express.static(__dirname + '/www'));

// 每个连接的用户都有专有的socket
/*
   io.emit(foo); //会触发所有用户的foo事件
   socket.emit(foo); //只触发当前用户的foo事件
   socket.broadcast.emit(foo); //触发除了当前用户的其他用户的foo事件
*/
io.on('connection', (socket)=> {
    // 渲染在线人员
    io.emit('disUser', usersInfo);

    // 登录，检测用户名
    socket.on('login', (user)=> {
        // 检查昵称是否重复
        if(users.indexOf(user.name) > -1) {
            socket.emit('loginError');
        } else {
            users.push(user.name);
            usersInfo.push(user);

            socket.emit('loginSuc');
            socket.nickname = user.name;
            // 给每一个用户发消息
            io.emit('system', {
                name: user.name,
                status: '进入'
            });

            // 重新渲染用户列表
            io.emit('disUser', usersInfo);
            console.log(users.length + ' user connect.');
            // console.log(io.sockets.sockets)
        }
    });

    // 发送窗口抖动
    socket.on('shake', ()=> {
        socket.emit('shake', {
            name: '您'
        });
        socket.broadcast.emit('shake', {
            name: socket.nickname
        });
    });

    // 发送消息事件
    socket.on('sendMsg', (data)=> {
        var img = '';
        // 循环找到发消息者
        for(var i = 0; i < usersInfo.length; i++) {
            if(usersInfo[i].name == socket.nickname) {
                img = usersInfo[i].img;
            }
        }

        if (data.target === '所有人') {
            //  其他用户
            socket.broadcast.emit('receiveMsg', {
                name: socket.nickname,
                img: img,
                msg: data.msg,
                color: data.color,
                type: data.type,
                side: 'left',
                Privacy: false
            });

            // 当前用户
            socket.emit('receiveMsg', {
                name: socket.nickname,
                img: img,
                msg: data.msg,
                color: data.color,
                type: data.type,
                side: 'right',
                Privacy: false
            });
        } else {
            var to = ''
            var from = io.sockets.sockets[data.id]

            // 当前用户
            for (k in io.sockets.sockets) {
                if (io.sockets.sockets[k].nickname === data.target) {
                    to = io.sockets.sockets[k]
                }
            }
            if (!to) {
                socket.emit('sendErr', {
                    noname: data.target
                })
            }
            socket.emit('receiveMsg', {
                name: socket.nickname,
                img: img,
                msg: data.msg,
                color: data.color,
                type: data.type,
                side: 'right',
                Privacy: true,
                from: from.nickname,
                to: to.nickname
            });
            to.emit('receiveMsg', {
                name: socket.nickname,
                img: img,
                msg: data.msg,
                color: data.color,
                type: data.type,
                side: 'left',
                Privacy: true,
                from: from.nickname,
                to: to.nickname
            });
        }

    });

    // 断开连接时
    socket.on('disconnect', ()=> {
        var index = users.indexOf(socket.nickname);
        if(index > -1 ) {  // 避免是undefined
            users.splice(index, 1);  // 删除用户信息
            usersInfo.splice(index, 1);  // 删除用户信息

            io.emit('system', {  // 系统通知
                name: socket.nickname,
                status: '离开'
            });

            io.emit('disUser', usersInfo);  // 重新渲染
            console.log('a user left.');
        }
    });
});

http.listen(3000, function() {
    console.log('listen 3000 port.');
});