twitst4tz

index.js (5421B)

---

 
 /**
  * Module dependencies.
  */
 
 var Emitter = require('events').EventEmitter;
 
 /**
  * Module exports.
  */
 
 module.exports = Adapter;
 
 /**
  * Memory adapter constructor.
  *
  * @param {Namespace} nsp
  * @api public
  */
 
 function Adapter(nsp){
   this.nsp = nsp;
   this.rooms = {};
   this.sids = {};
   this.encoder = nsp.server.encoder;
 }
 
 /**
  * Inherits from `EventEmitter`.
  */
 
 Adapter.prototype.__proto__ = Emitter.prototype;
 
 /**
  * Adds a socket to a room.
  *
  * @param {String} socket id
  * @param {String} room name
  * @param {Function} callback
  * @api public
  */
 
 Adapter.prototype.add = function(id, room, fn){
   return this.addAll(id, [ room ], fn);
 };
 
 /**
  * Adds a socket to a list of room.
  *
  * @param {String} socket id
  * @param {String} rooms
  * @param {Function} callback
  * @api public
  */
 
 Adapter.prototype.addAll = function(id, rooms, fn){
   for (var i = 0; i < rooms.length; i++) {
     var room = rooms[i];
     this.sids[id] = this.sids[id] || {};
     this.sids[id][room] = true;
     this.rooms[room] = this.rooms[room] || Room();
     this.rooms[room].add(id);
   }
   if (fn) process.nextTick(fn.bind(null, null));
 };
 
 /**
  * Removes a socket from a room.
  *
  * @param {String} socket id
  * @param {String} room name
  * @param {Function} callback
  * @api public
  */
 
 Adapter.prototype.del = function(id, room, fn){
   if (this.sids[id]) delete this.sids[id][room];
 
   if (this.rooms.hasOwnProperty(room)) {
     this.rooms[room].del(id);
     if (this.rooms[room].length === 0) delete this.rooms[room];
   }
 
   if (fn) process.nextTick(fn.bind(null, null));
 };
 
 /**
  * Removes a socket from all rooms it's joined.
  *
  * @param {String} socket id
  * @param {Function} callback
  * @api public
  */
 
 Adapter.prototype.delAll = function(id, fn){
   var rooms = this.sids[id];
   if (rooms) {
     for (var room in rooms) {
       if (this.rooms.hasOwnProperty(room)) {
         this.rooms[room].del(id);
         if (this.rooms[room].length === 0) delete this.rooms[room];
       }
     }
   }
   delete this.sids[id];
 
   if (fn) process.nextTick(fn.bind(null, null));
 };
 
 /**
  * Broadcasts a packet.
  *
  * Options:
  *  - `flags` {Object} flags for this packet
  *  - `except` {Array} sids that should be excluded
  *  - `rooms` {Array} list of rooms to broadcast to
  *
  * @param {Object} packet object
  * @api public
  */
 
 Adapter.prototype.broadcast = function(packet, opts){
   var rooms = opts.rooms || [];
   var except = opts.except || [];
   var flags = opts.flags || {};
   var packetOpts = {
     preEncoded: true,
     volatile: flags.volatile,
     compress: flags.compress
   };
   var ids = {};
   var self = this;
   var socket;
 
   packet.nsp = this.nsp.name;
   this.encoder.encode(packet, function(encodedPackets) {
     if (rooms.length) {
       for (var i = 0; i < rooms.length; i++) {
         var room = self.rooms[rooms[i]];
         if (!room) continue;
         var sockets = room.sockets;
         for (var id in sockets) {
           if (sockets.hasOwnProperty(id)) {
             if (ids[id] || ~except.indexOf(id)) continue;
             socket = self.nsp.connected[id];
             if (socket) {
               socket.packet(encodedPackets, packetOpts);
               ids[id] = true;
             }
           }
         }
       }
     } else {
       for (var id in self.sids) {
         if (self.sids.hasOwnProperty(id)) {
           if (~except.indexOf(id)) continue;
           socket = self.nsp.connected[id];
           if (socket) socket.packet(encodedPackets, packetOpts);
         }
       }
     }
   });
 };
 
 /**
  * Gets a list of clients by sid.
  *
  * @param {Array} explicit set of rooms to check.
  * @param {Function} callback
  * @api public
  */
 
 Adapter.prototype.clients = function(rooms, fn){
   if ('function' == typeof rooms){
     fn = rooms;
     rooms = null;
   }
 
   rooms = rooms || [];
 
   var ids = {};
   var sids = [];
   var socket;
 
   if (rooms.length) {
     for (var i = 0; i < rooms.length; i++) {
       var room = this.rooms[rooms[i]];
       if (!room) continue;
       var sockets = room.sockets;
       for (var id in sockets) {
         if (sockets.hasOwnProperty(id)) {
           if (ids[id]) continue;
           socket = this.nsp.connected[id];
           if (socket) {
             sids.push(id);
             ids[id] = true;
           }
         }
       }
     }
   } else {
     for (var id in this.sids) {
       if (this.sids.hasOwnProperty(id)) {
         socket = this.nsp.connected[id];
         if (socket) sids.push(id);
       }
     }
   }
 
   if (fn) process.nextTick(fn.bind(null, null, sids));
 };
 
 /**
  * Gets the list of rooms a given client has joined.
  *
  * @param {String} socket id
  * @param {Function} callback
  * @api public
  */
 Adapter.prototype.clientRooms = function(id, fn){
   var rooms = this.sids[id];
   if (fn) process.nextTick(fn.bind(null, null, rooms ? Object.keys(rooms) : null));
 };
 
 /**
 * Room constructor.
 *
 * @api private
 */
 
 function Room(){
   if (!(this instanceof Room)) return new Room();
   this.sockets = {};
   this.length = 0;
 }
 
 /**
  * Adds a socket to a room.
  *
  * @param {String} socket id
  * @api private
  */
 
 Room.prototype.add = function(id){
   if (!this.sockets.hasOwnProperty(id)) {
     this.sockets[id] = true;
     this.length++;
   }
 };
 
 /**
  * Removes a socket from a room.
  *
  * @param {String} socket id
  * @api private
  */
 
 Room.prototype.del = function(id){
   if (this.sockets.hasOwnProperty(id)) {
     delete this.sockets[id];
     this.length--;
   }
 };