daveo-webrtc/webrtc.io.js

var io = require('socket.io');

var rooms = {};

module.exports.listen = function() {
  // delegate all arguments to socket.io.listen
  var manager = io.listen.apply(io, arguments);

  manager.sockets.on('connection', function(socket) {
    // TODO: let you join multiple rooms
    socket.on('join room', function(room) {
      // initialize room as an empty array
      var connections = rooms[room] = rooms[room] || [];

      socket.join(room);

      // tell everyone else in the room about the new peer
      socket.broadcast.to(room)
        .emit('new peer connected', { socketId: socket.id });

      connections.push(socket);

      // pass array of connection ids except for peer's to peer
      var connectionsId = [];
      for (var i = 0, len = connections.length; i < len; i++) {
        var id = connections[i].id;

        if (id !== socket.id) {
          connectionsId.push(id);
        }
      }

      socket.emit('get peers', {
        connections: connectionsId
      });

      // remove connection from array and tell everyone else about the
      // disconnect
      socket.on('disconnect', function() {
        var connections = rooms[room];
        for (var i = 0; i < connections.length; i++) {
          var id = connections[i].id;

          if (id == socket.id) {
            connections.splice(i, 1);
            i--;
            socket.broadcast.to(room).emit('remove peer connected', {
              socketId: socket.id
            });
          }
        }
      });

      socket.on('receive ice candidate', function(data) {
        var soc = getSocket(room, data.socketId);

        if (soc) {
          soc.emit('receive ice candidate', {
            label: data.label,
            candidate: data.candidate,
            socketId: socket.id
          });
        }
      });

      socket.on('send offer', function(data) {
        var soc = getSocket(room, data.socketId);

        if (soc) {
          soc.emit('receive offer', {
            sdp: data.sdp,
            socketId: socket.id
          });
        }
      });

      socket.on('send answer', function(data) {
        var soc = getSocket(room, data.socketId);

        if (soc) {
          soc.emit('receive answer', {
            sdp: data.sdp,
            socketId: socket.id
          });
        }
      });
    });


  });

  return manager;
}

function getSocket(room, id) {
  var connections = rooms[room];

  if (!connections) {
    // TODO: Or error, or customize
    return;
  }

  for (var i = 0; i < connections.length; i++) {
    var socket = connections[i];
    if (id === socket.id) {
      return socket;
    }
  }
}