server.js

"use strict";

/*!
 * WebSocket Realtime Transport - Server
 * Copyright(c) 2013 Owen Barnes <owen@socketstream.org>
 * MIT Licensed
 */

var WebSocketServer = require('ws').Server;

module.exports = function(options) {

  options = options || {};

  var clientCount = 0;
  var clients = {};

  return function(server) {

    var ws = new WebSocketServer({port: server.port});

    ws.on('connection', function(socket) {

      // Assign SocketID
      socket.id = String(++clientCount);

      clients[socket.id] = socket;

      server.events.emit('connect', socket.id);

      socket.on('message', function(message) {

        var request = {
          message:    message,
          transport:  'ws',
          socketId:   socket.id,
          clientIp:   socket.upgradeReq.headers.host.split(':')[0]
        };

        server.process(request);

      });

      socket.on('close', function() {
        server.events.emit('disconnect', socket.id);
        delete clients[socket.id];
      });

    });

    return {

      sendToSocketId: function(socketId, msg) {
        clients[socketId].send(msg);
      },

      broadcast: function(msg){
        for (var socketId in clients) {
          clients[socketId].send(msg);
        }
      }

    };

  };
};