chore(notifications): Live Notifications

- use socket to emit on notification [Finishes #]
andela · Jan 13, 2020 · 455fda4 · 455fda4
1 parent c4c1cad
commit 455fda4
Show file tree

Hide file tree

Showing 9 changed files with 96 additions and 8 deletions.
diff --git a/src/controllers/accommodationController.js b/src/controllers/accommodationController.js
@@ -167,6 +167,7 @@ export default class AccommodationController {
             const newBooking = await models.booking.create(bookingData);
 
             await notifSender(
+              req,
               'Accommodation Booked',
               newBooking,
               accommodation[0].owner,
@@ -280,7 +281,7 @@ export default class AccommodationController {
         await bookingHelper.updateAccomodation(req, remainingSpace);
       }
 
-      await notifSender(subject, booking, booking.userId, APP_URL_BACKEND, activity, 'booking', host);
+      await notifSender(req, subject, booking, booking.userId, APP_URL_BACKEND, activity, 'booking', host);
       return responseHelper(res, responseMessage, null, 200);
     }
     return responseHelper(res, strings.accommodations.error.BOOKING_NOT_FOUND, null, 404);

diff --git a/src/controllers/chatController.js b/src/controllers/chatController.js
@@ -130,4 +130,5 @@ const viewPrivateChats = async (req, res) => {
     ? EMPTY_HISTORY : CHAT_HISTORY, privateChats);
 };
 
+
 export default { chat, viewChats , viewPrivateChats};
diff --git a/src/controllers/commentsController.js b/src/controllers/commentsController.js
@@ -29,7 +29,7 @@ export default class CommentsController {
         userNotiified,
         `A comment has been made on request ${requestId}. Click here to view: ${APP_URL_BACKEND}/api/v1/requests/${requestId}.`,
       );
-      await notifSaver(notification);
+      await notifSaver(req, notification);
       const {
         deleted, userId,
         ...data

diff --git a/src/controllers/destinationController.js b/src/controllers/destinationController.js
@@ -24,7 +24,7 @@ export default class destinationController {
 
     const { lineManager } = user.payload;
 
-    await notifSender('Request Created', request, lineManager, APP_URL_BACKEND, 'created', 'request', host);
+    await notifSender(req, 'Request Created', request, lineManager, APP_URL_BACKEND, 'created', 'request', host);
 
     return Utilities.responseHelper(
       res,

diff --git a/src/controllers/requestController.js b/src/controllers/requestController.js
@@ -95,7 +95,7 @@ export default class requestController {
 
       request = request[1][0].dataValues;
 
-      await notifSender(subject, request, request.userId, APP_URL_BACKEND, activity, 'request', host);
+      await notifSender(req, subject, request, request.userId, APP_URL_BACKEND, activity, 'request', host);
       return responseHelper(res, responseMessage, null, 200);
     }
     return responseHelper(res, strings.requests.NOT_FOUND, null, 404);
@@ -123,7 +123,7 @@ export default class requestController {
 
       const request = await allSearch({ id });
       const requestData = request[0].dataValues;
-      await notifSender('Request Edited', requestData, lineManager, APP_URL_BACKEND, 'edited', 'request');
+      await notifSender(req, 'Request Edited', requestData, lineManager, APP_URL_BACKEND, 'edited', 'request');
       return responseUtil(res, 200, strings.request.success.SUCCESS_UPDATE_REQUEST, request);
     } catch (error) {
       return res.status(500).json({ error: 'Something wrong' });

diff --git a/src/helpers/notifSender.js b/src/helpers/notifSender.js
@@ -16,7 +16,9 @@ const transporter = nodemailer.createTransport({
   }
 });
 
-const notifSender = async (subject, object, userId, APP_URL_BACKEND, activity, table, host) => {
+const notifSender = async (
+  req, subject, object, userId, APP_URL_BACKEND, activity, table, host
+) => {
   try {
     const user = await findUser({ id: userId });
     let URL;
@@ -33,7 +35,7 @@ const notifSender = async (subject, object, userId, APP_URL_BACKEND, activity, t
     if (user.appNotif) {
       const notification = await notifBuilder(table, object.id, userId, activityMessage);
       try {
-        await notifSaver(notification);
+        await notifSaver(req, notification);
       } catch (error) {
         return error;
       }

diff --git a/src/helpers/socketNotif.js b/src/helpers/socketNotif.js
@@ -0,0 +1,55 @@
+import jwt from 'express-jwt';
+import io from '../index';
+
+const socketNotif = notification => {
+  const onlineUsers = [];
+  const getLoggedInUser = (token, socket) => {
+    if (token === null) {
+      socket.emit('connection_error', 'Token not present');
+    }
+    try {
+      const user = jwt.verify(token, process.env.JWT_SECRET);
+      const userObject = {
+        socketId: socket.id,
+        userId: user.payload.id,
+        email: user.payload.email
+      };
+
+      const filter = onlineUsers.filter(client => client.userId === user.payload.id);
+
+      if (filter.length === 0) {
+        onlineUsers.push(userObject);
+      } else {
+        onlineUsers.forEach(client => {
+          if (client.userId === user.payload.id) {
+            client.socketId = socket.id;
+          }
+        });
+      }
+      return user;
+    } catch (error) {
+      socket.emit('connection_error', error.message);
+    }
+  };
+
+  try {
+    io.use((socket, next) => {
+      const { token } = socket.request._query;
+      const user = getLoggedInUser(token, socket);
+      socket.user = user;
+      next();
+    });
+  } catch (error) {
+    socket.emit('connection_error', error.message);
+  }
+
+  if (!receiverId) {
+    io.emit(entity, data);
+  } else if (connectedClients[receiverId.toString()]) {
+    connectedClients[receiverId.toString()].forEach(element => {
+      io.to(element).emit(entity, data);
+    });
+  }
+};
+
+export default socketNotif;
diff --git a/src/index.js b/src/index.js
@@ -61,4 +61,24 @@ app.use('*', (req, res) => {
 // eslint-disable-next-line no-console
 http.listen(port, () => console.log(`Barefoot Nomad is runnig server on port ${port}...`));
 
+// const connectedClients = {};
+// io.use((socket, next) => {
+//   const { token } = socket.handshake.query;
+//   const userData = validateToken(token);
+//   if (userData) {
+//     const clientKey = Number.parseInt(userData.userId, 10);
+//     connectedClients[clientKey] = connectedClients[clientKey] || [];
+//     connectedClients[clientKey].push(socket.id);
+//   }
+//   next();
+// });
+
+// app.use((req, res, next) => {
+//   req.io = io;
+//   req.connectedClients = connectedClients;
+//   next();
+// });
+
+export { io };
+
 export default app;
diff --git a/src/services/notifServices.js b/src/services/notifServices.js
@@ -1,10 +1,19 @@
 /* eslint-disable  require-jsdoc */
 import moment from 'moment';
 import models from '../database/models';
+import socketNotif from '../helpers/socketNotif';
+import { io } from '../index';
 
 export default class notifServices {
-  static async notifSaver(notif) {
+  static async notifSaver(req, notif) {
     const notification = await notif.save();
+    try {
+      // socketNotif(notification.userNotified, notification, req.connectedClients, req.io, 'notification');
+      io.emit('notification', notification);
+      console.log('sockectNotif ===> ', io.emit('notification', notification));
+    } catch (error) {
+      // console.log('sockectNotif error ===> ', error);
+    }
     return notification;
   }