socket-chat/server.js at master · lokth/socket-chat · 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
/**
 * @author Lok Thapa
 */

var httpd = require('http').createServer(handler);
var io = require('socket.io').listen(httpd);
var fs = require('fs');

httpd.listen(4000);

function handler(req, res) {
	fs.readFile(__dirname + '/index.html', function(err, data) {
		if (err) {
			res.writeHead(500);
			return res.end('Error loading index.html');
		}
		res.writeHead(200);
		res.end(data);
	});
}

io.sockets.on('connection', function(socket) {

	socket.on('clientMessage', function(content) {
		socket.emit('serverMessage', 'You said: '+ content);
			socket.get('username', function(err, username) {
				if (!username) {
					username = socket.id;
				}
					socket.get('room',
							function(err, room) {
								if (err) {
									throw err;
								}
								var broadcast = socket.broadcast;
								var message = content;
								if (room) {
									broadcast.to(room);
								}
								broadcast.emit('serverMessage', username + ' said: ' + message);
					});
			});
	});


	socket.on('login', function(username) {
		socket.set('username', username, function(err) {
			if (err) {
				throw err;
			}
			socket.emit('serverMessage', 'Currently logged in as ' + username);
			socket.broadcast.emit('serverMessage', 'User ' + username + ' logged in');
		});
	});


	socket.on('disconnect', function() {
		socket.get('username', function(err, username) {
			if (!username) {
				username = socket.id;
			}
			socket.broadcast.emit('serverMessage', 'User ' + username + ' disconnected');
		});
	});


	socket.on('join', function(room) {
		socket.get('room', function(err, oldRoom) {
			if (err) {
				throw err;
			}
			socket.set('room', room, function(err) {
				if (err) {
					throw err;
				}
				socket.join(room);
				if (oldRoom) {
					socket.leave(oldRoom);
				}
				socket.get('username', function(err, username) {
					if (!username) {
						username = socket.id;
					}
				});
				socket.emit('serverMessage', 'You joined room ' + room);
				socket.get('username', function(err, username) {
					if (!username) {
						username = socket.id;
					}
					socket.broadcast.to(room).emit('serverMessage', 'User ' + username + ' joined this room');
				});
			});
		});
	});


	socket.emit('login');

});