1 files changed, 40 insertions, 0 deletions
diff --git a/server/ws.js b/server/ws.js
new file mode 100644
index 0000000..860319a
--- /dev/null
+++ b/server/ws.js
@@ -0,0 +1,40 @@
+
+var server = require('socket.io')
+
+var ws = module.exports = {}
+
+var rooms = {}
+
+ws.listen = function(app){
+  ws.io = server(app)
+}
+ws.add = function(name){
+  if (name in rooms) { return }
+  var room = {}
+  room.users = {}
+  
+  var ns = base.io.of('/' + name)
+  
+  ns.on('connection', function(socket){
+    
+    var username
+  
+    socket.on('join', function(data){
+      username = data.username
+      if (room.users[username]) {
+        // already connected?
+        room.users[username].disconnect()
+        room.users[username] = socket
+      }
+    })
+    
+    socket.on('message', function(data){
+      ns.emit('message', data)
+    })
+    
+    socket.on('disconnect', function(){
+      delete room.users[username]
+    })
+
+  })
+}