add message sending

api/v1/gateway/index.js

@@ -14,6 +14,7 @@ class GatewayServer {
 }
 
 GatewayServer.prototype.authDisconnect = function(socket, callback) {
+    console.log('[E] [gateway] [handshake] User disconnected due to failed authentication');
     socket.isConnected = false;
     socket.disconnect();
     socket.disconnect(true);
@@ -22,12 +23,12 @@ GatewayServer.prototype.authDisconnect = function(socket, callback) {
 
 GatewayServer.prototype.eventSetup = function() {
     this._gateway.use((socket, callback) => {
-        console.log('[*] [gateway] User authentication attempt');
+        console.log('[*] [gateway] [handshake] User authentication attempt');
         socket.isConnected = false;
 
         setTimeout(() => {
             if (socket.isConnected) return;
-            console.log('[*] [gateway] User still not connected after timeout, removing...');
+            console.log('[*] [gateway] [handshake] User still not connected after timeout, removing...');
             socket.disconnect();
             socket.disconnect(true);
         }, config.gatewayStillNotConnectedTimeoutMS);
@@ -54,17 +55,33 @@ GatewayServer.prototype.eventSetup = function() {
             if (permissionLevel < config.roleMap.USER) return this.authDisconnect(socket, callback);
 
             socket.username = data.username;
-            console.log(`[*] [gateway] User ${data.username} has successfully authenticated`);
+            console.log(`[*] [gateway] [handshake] User ${data.username} has successfully authenticated`);
             return callback();
         });
     });
 
     this._gateway.on('connection', (socket) => {
-        console.log(`[*] [gateway] User ${socket.username} connected, sending hello and waiting for yoo...`);
+        console.log(`[*] [gateway] [handshake] User ${socket.username} connected, sending hello and waiting for yoo...`);
         socket.emit('hello');
         socket.once('yoo', () => {
-            console.log(`[*] [gateway] Got yoo from ${socket.username}, connection is finally completed!`);
+            console.log(`[*] [gateway] [handshake] Got yoo from ${socket.username}, connection is finally completed!`);
             socket.isConnected = true;
+
+            socket.on('message', ({ categoryId, content }) => {
+                // TODO: URGENT: Check if the category exists and if the user has access to it (access coming soon)
+                socket.to(categoryId).emit('message', {
+                    username: socket.username,
+                    categoryId: categoryId,
+                    content: content
+                });
+            });
+
+            socket.on('subscribe', (categories) => {
+                for (let v of categories) {
+                    // TODO: URGENT: Check if the category exists and if the user has access to it (access coming soon)
+                    socket.join(v);
+                }    
+            });
         });
     });
 };

app/resources/js/app.js

@@ -76,6 +76,7 @@ GatewayConnection.prototype.connect = function(token) {
     this.socket.on('hello', () => {
         console.log('[*] [gateway] [handshake] Got hello from server, sending yoo...');
         this.socket.emit('yoo');
+        this.isConnected = true;
         this.onConnect('CONNECT_RECEIVED_HELLO');
     });
 
@@ -96,6 +97,19 @@ GatewayConnection.prototype.connect = function(token) {
     });
 };
 
+GatewayConnection.prototype.sendMessage = function(categoryId, content) {
+    if (!this.isConnected) return;
+
+    this.socket.emit('message', { categoryId, content });
+};
+
+GatewayConnection.prototype.subscribeToCategoryChat = function(categoryId) {
+    if (!this.isConnected) return;
+
+    this.socket.emit('subscribe', [categoryId]);
+};
+
+
 const app = new Vue({
     el: '#app',
     data: {
@@ -204,8 +218,17 @@ const app = new Vue({
                 this.resetSnackbarButton();
                 this.notification('ERROR: You have been disconnected from the gateway. Realtime features such as chat will not work and unexpected errors may occur.');
             };
+            this.gateway.onConnect = () => {
+                this.gateway.socket.on('message', ({ username, categoryId, content }) => {
+                    this.showMessageNotification(username, content);
+                })
+            };
             this.gateway.connect(this.loggedInUser.token);
         },
+        showMessageNotification: async function(username, content) {
+            this.resetSnackbarButton();
+            this.notification(`${username}: ${content}`);
+        },
         button: function(text, click) {
             this.cardButtons.push({ text, click });
         },
@@ -397,7 +420,10 @@ const app = new Vue({
                 this.cardButtons = [];
 
                 this.button('Chat', (post) => {
-                    this.browse(post);
+                    if (post._id) {
+                        this.gateway.subscribeToCategoryChat(post._id);
+                        this.gateway.sendMessage(post._id, 'yoooo');
+                    }
                 });
                 this.button('View', (post) => {
                     this.browse(post);