server.js

/*
Copyright 2017 Amazon.com, Inc. or its affiliates. All Rights Reserved.

Licensed under the Amazon Software License (the "License"). You may not use this file except in compliance with the License. A copy of the License is located at

    http://aws.amazon.com/asl/

or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, express or implied. See the License for the specific language governing permissions and limitations under the License.
*/
var faker = require('faker');
var moment = require('moment');

var express = require('express');
var app = express();
var http = require('http').Server(app);
var io = require('socket.io')(http);
var port = process.env.PORT || 3000;

var cfenv = require('cfenv');
// default config for local redis (overridden when run on cloud foundry)
var appEnv = cfenv.getAppEnv({
  "vcap": {
    "application": {
      "name": "pad-redis-demo"
    },
    "services": {
      "local-redis": [
        {
          "credentials": {
            "hostname": "127.0.0.1",
            "password": "",
            "port": "6379"
          },
          "name": "test-redis",
          "tags": [ "redis" ]
        }
      ]
    }
  }
});

// Two ways to bind to a redis service:
// 1. Explicitly set the REDIS_SERVICE_NAME env variable to the name of the redis service (eg: test-redis)
// 2. Select the first service that contains a 'redis' tag
var redis_cred;
if (process.env.REDIS_SERVICE_NAME) {
  redis_cred = appEnv.getServiceCreds(process.env.REDIS_SERVICE_NAME);
} else {
  var found;
  var services = appEnv.getServices();
  for (var serviceName in services) {
    if (services.hasOwnProperty(serviceName)) {
      var service = services[serviceName];
      service.tags.some(function(tag) {
        if (tag == "redis") {
          found = service;
          return;
        }
      });
      if (found) {
        redis_cred = appEnv.getServiceCreds(found.name);
        break;
      }
    }
  }
}

if (!redis_cred) {
  console.log('No Redis service bound to this app.');
  process.exit(1);
}

// p-redis uses "host" but rediscloud uses "hostname"
var redis_address = "redis://:"+redis_cred["password"]+"@"+(redis_cred["host"] || redis_cred["hostname"])+":"+redis_cred["port"];

var Redis = require('ioredis');
var redis = new Redis(redis_address);
var redis_subscribers = {};
var channel_history_max = 10;

app.use(express.static('public'));
app.get('/health', function(request, response) {
    response.send('ok');
});

function add_redis_subscriber(subscriber_key) {
    var client = new Redis(redis_address);

    client.subscribe(subscriber_key);
    client.on('message', function(channel, message) {
        io.emit(subscriber_key, JSON.parse(message));
    });

    redis_subscribers[subscriber_key] = client;
}
add_redis_subscriber('messages');
add_redis_subscriber('member_add');
add_redis_subscriber('member_delete');

io.on('connection', function(socket) {
    var get_members = redis.hgetall('members').then(function(redis_members) {
        var members = {};
        for (var key in redis_members) {
            members[key] = JSON.parse(redis_members[key]);
        }
        return members;
    });

    var initialize_member = get_members.then(function(members) {
        if (members[socket.id]) {
            return members[socket.id];
        }

        var username = faker.fake("{{name.firstName}} {{name.lastName}}");
        var member = {
            socket: socket.id,
            username: username,
            avatar: "//api.adorable.io/avatars/30/" + username + '.png'
        };

        return redis.hset('members', socket.id, JSON.stringify(member)).then(function() {
            return member;
        });
    });

    // get the highest ranking messages (most recent) up to channel_history_max size
    var get_messages = redis.zrange('messages', -1 * channel_history_max, -1).then(function(result) {
        return result.map(function(x) {
            return JSON.parse(x);
        });
    });

    Promise.all([get_members, initialize_member, get_messages]).then(function(values) {
        var members = values[0];
        var member = values[1];
        var messages = values[2];

        io.emit('member_history', members);
        io.emit('message_history', messages);

        redis.publish('member_add', JSON.stringify(member));

        socket.on('send', function(message_text) {
            var date = moment.now();
            var message = JSON.stringify({
                date: date,
                username: member['username'],
                avatar: member['avatar'],
                message: message_text
            });

            redis.zadd('messages', date, message);
            redis.publish('messages', message);
        });

        socket.on('disconnect', function() {
            redis.hdel('members', socket.id);
            redis.publish('member_delete', JSON.stringify(socket.id));
        });
    }).catch(function(reason) {
        console.log('ERROR: ' + reason);
    });
});

http.listen(port, function() {
    console.log('Started server on port ' + port);
});