Some refactoring for integration with frontend

app/chat/chat_socket.js

@@ -1,4 +1,5 @@
 var User = require('../models/user');  
+var ChatRecord = require('../models/chat_record');
 var config = require('../../config/main');
 var jwt = require('jsonwebtoken');
 var jwtAuth = require('socketio-jwt-auth');
@@ -24,11 +25,62 @@ module.exports = function(server) {
       return done(null, user);
     });
   }));
-  
+
   io.on('connection', function(socket) {
+    ChatRecord.remove({}, function() {
+
+    })
     let user = socket.request.user;
 
-    console.log('user ' + user._id + ' connected to the chat');
+    socket.on('subscribe', function(heroId) {
+      socket.join(heroId);
+      socket.room = heroId;
+
+      let notification = new ChatRecord({
+        heroId: heroId, 
+        type: 'notification', 
+        from: user._id, 
+        message: 'joined the chat room'
+      });
+
+      socket.to(heroId).emit('new message', notification);
+      notification.save();
+    });
+
+    socket.on('get user info', function(userId, fn) {
+      User.findOne({_id: userId}, function(err, user){
+        if(user) {
+          fn({_id: userId, name: user.name});
+        }
+      });
+    })
+
+    socket.on('retrieve messages', function(heroId, fn) {
+      ChatRecord.find({heroId: heroId}, function(err, records){
+        fn(records);
+        console.log(records);
+      });
+    });
+
+    socket.on('new message', function(msg, fn){
+      let message = new ChatRecord(msg);
+      socket.to(message.heroId).emit('new message', message);
+      message.save();
+      fn(message);
+    });
+
+    socket.on('disconnect', function() {
+      let notification = new ChatRecord({
+        heroId: socket.room, 
+        type: 'notification', 
+        from: user._id, 
+        message: 'left the chat room'
+      });
+
+      socket.to(socket.room).emit('new message', notification);
+      notification.save();
+    });
+
   });
 };
 

app/models/chat_record.js

@@ -0,0 +1,27 @@
+const mongoose = require('mongoose');
+
+// Schema defines how the user's data will be stored in MongoDB
+const ChatRecordSchema = new mongoose.Schema({
+  heroId: {
+    type: String,
+    required: true
+    // chat room id = hero id
+  },
+  type: {
+      type: String, 
+      enum: ['message', 'notification'],
+      default: 'message'
+  },
+  from: {
+    type: String, 
+    required: true
+  }, 
+  message: {
+      type: String,
+      required: true
+  }
+}, {
+    timestamps: true
+});
+
+module.exports = mongoose.model('ChatRecord', ChatRecordSchema);

app/routes/user_routes.js

@@ -24,7 +24,7 @@ module.exports = function(app) {
         return res.status(400).json({ success: false, message: err.message });
       }
 
-      return res.status(201).json({ success: true, message: 'Successfully created new user.', userInfo: generateUserInfo(newUser) });
+      return res.status(201).json({ success: true, message: 'Successfully created new user.' });
     });
   });
 

server.js

@@ -17,7 +17,6 @@ app.use(cors());
 
 // Log requests to console
 app.use(morgan('dev'));
-app.use('/', express.static(__dirname + '/www'));
 
 // Connect to database
 mongoose.connect(config.database);