|
| 1 | +// Setup basic express server |
| 2 | +var express = require('express'); |
| 3 | +var app = express(); |
| 4 | +var server = require('http').createServer(app); |
| 5 | +var io = require('socket.io')(server); |
| 6 | +var redis = require('socket.io-redis'); |
| 7 | +var port = process.env.PORT || 3000; |
| 8 | +var serverName = process.env.NAME || 'Unknown'; |
| 9 | + |
| 10 | +io.adapter(redis({ host: 'redis', port: 6379 })); |
| 11 | + |
| 12 | +server.listen(port, function () { |
| 13 | + console.log('Server listening at port %d', port); |
| 14 | + console.log('Hello, I\'m %s, how can I help?', serverName); |
| 15 | +}); |
| 16 | + |
| 17 | +// Routing |
| 18 | +app.use(express.static(__dirname + '/public')); |
| 19 | + |
| 20 | +// Health check |
| 21 | +app.head('/health', function (req, res) { |
| 22 | + res.sendStatus(200); |
| 23 | +}); |
| 24 | + |
| 25 | +// Chatroom |
| 26 | + |
| 27 | +var numUsers = 0; |
| 28 | + |
| 29 | +io.on('connection', function (socket) { |
| 30 | + socket.emit('my-name-is', serverName); |
| 31 | + |
| 32 | + var addedUser = false; |
| 33 | + |
| 34 | + // when the client emits 'new message', this listens and executes |
| 35 | + socket.on('new message', function (data) { |
| 36 | + // we tell the client to execute 'new message' |
| 37 | + socket.broadcast.emit('new message', { |
| 38 | + username: socket.username, |
| 39 | + message: data |
| 40 | + }); |
| 41 | + }); |
| 42 | + |
| 43 | + // when the client emits 'add user', this listens and executes |
| 44 | + socket.on('add user', function (username) { |
| 45 | + if (addedUser) return; |
| 46 | + |
| 47 | + // we store the username in the socket session for this client |
| 48 | + socket.username = username; |
| 49 | + ++numUsers; |
| 50 | + addedUser = true; |
| 51 | + socket.emit('login', { |
| 52 | + numUsers: numUsers |
| 53 | + }); |
| 54 | + // echo globally (all clients) that a person has connected |
| 55 | + socket.broadcast.emit('user joined', { |
| 56 | + username: socket.username, |
| 57 | + numUsers: numUsers |
| 58 | + }); |
| 59 | + }); |
| 60 | + |
| 61 | + // when the client emits 'typing', we broadcast it to others |
| 62 | + socket.on('typing', function () { |
| 63 | + socket.broadcast.emit('typing', { |
| 64 | + username: socket.username |
| 65 | + }); |
| 66 | + }); |
| 67 | + |
| 68 | + // when the client emits 'stop typing', we broadcast it to others |
| 69 | + socket.on('stop typing', function () { |
| 70 | + socket.broadcast.emit('stop typing', { |
| 71 | + username: socket.username |
| 72 | + }); |
| 73 | + }); |
| 74 | + |
| 75 | + // when the user disconnects.. perform this |
| 76 | + socket.on('disconnect', function () { |
| 77 | + if (addedUser) { |
| 78 | + --numUsers; |
| 79 | + |
| 80 | + // echo globally that this client has left |
| 81 | + socket.broadcast.emit('user left', { |
| 82 | + username: socket.username, |
| 83 | + numUsers: numUsers |
| 84 | + }); |
| 85 | + } |
| 86 | + }); |
| 87 | +}); |
0 commit comments