From e5f458fd79f1f6df67cb4e2fe2bc6c3ffa96032c Mon Sep 17 00:00:00 2001
From: Daniel Lugo <danlugo92@gmail.com>
Date: Wed, 11 Nov 2020 17:50:28 -0400
Subject: [PATCH] contact sockets

---
 src/sockets.js | 140 +++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 140 insertions(+)

diff --git a/src/sockets.js b/src/sockets.js
index e09da0be..432ee300 100644
--- a/src/sockets.js
+++ b/src/sockets.js
@@ -16,6 +16,7 @@ const {
   isAuthenticated
 } = require('../services/gunDB/Mediator')
 const { deepDecryptIfNeeded } = require('../services/gunDB/rpc')
+const GunEvents = require('../services/gunDB/contact-api/events')
 /**
  * @typedef {import('../services/gunDB/Mediator').SimpleSocket} SimpleSocket
  * @typedef {import('../services/gunDB/contact-api/SimpleGUN').ValidDataValue} ValidDataValue
@@ -491,5 +492,144 @@ module.exports = (
     }
   )
 
+  // TODO: do this through rpc
+
+  const emptyUnsub = () => {}
+
+  let chatsUnsub = emptyUnsub
+
+  io.of('chats').on('connect', async socket => {
+    try {
+      if (!isAuthenticated()) {
+        logger.info(
+          'not authenticated in gun for chats socket, will send NOT_AUTH'
+        )
+        socket.emit(Common.Constants.ErrorCode.NOT_AUTH)
+
+        return
+      }
+
+      logger.info('now checking token for chats socket')
+      const { token } = socket.handshake.query
+      const isAuth = await isValidToken(token)
+
+      if (!isAuth) {
+        logger.warn('invalid token for chats socket')
+        socket.emit(Common.Constants.ErrorCode.NOT_AUTH)
+        return
+      }
+
+      if (chatsUnsub !== emptyUnsub) {
+        logger.error(
+          'Tried to set chats socket twice, this might be due to an app restart and the old socket not being recycled by socket.io in time, will disable the older subscription, which means the old socket wont work and data will be sent to this new socket instead'
+        )
+        chatsUnsub()
+        chatsUnsub = emptyUnsub
+      }
+
+      GunEvents.onChats(chats => {
+        socket.emit(Common.Constants.Event.ON_CHATS, chats)
+      })
+
+      socket.on('disconnect', () => {
+        chatsUnsub()
+        chatsUnsub = emptyUnsub
+      })
+    } catch (e) {
+      logger.error('Error inside chats socket connect: ' + e.message)
+      socket.emit('$error', e.message)
+    }
+  })
+
+  let sentReqsUnsub = emptyUnsub
+
+  io.of('sentReqs').on('connect', async socket => {
+    try {
+      if (!isAuthenticated()) {
+        logger.info(
+          'not authenticated in gun for sentReqs socket, will send NOT_AUTH'
+        )
+        socket.emit(Common.Constants.ErrorCode.NOT_AUTH)
+
+        return
+      }
+
+      logger.info('now checking token for sentReqs socket')
+      const { token } = socket.handshake.query
+      const isAuth = await isValidToken(token)
+
+      if (!isAuth) {
+        logger.warn('invalid token for sentReqs socket')
+        socket.emit(Common.Constants.ErrorCode.NOT_AUTH)
+        return
+      }
+
+      if (sentReqsUnsub !== emptyUnsub) {
+        logger.error(
+          'Tried to set sentReqs socket twice, this might be due to an app restart and the old socket not being recycled by socket.io in time, will disable the older subscription, which means the old socket wont work and data will be sent to this new socket instead'
+        )
+        sentReqsUnsub()
+        sentReqsUnsub = emptyUnsub
+      }
+
+      GunEvents.onSimplerSentRequests(sentReqs => {
+        socket.emit(Common.Constants.Event.ON_SENT_REQUESTS, sentReqs)
+      })
+
+      socket.on('disconnect', () => {
+        sentReqsUnsub()
+        sentReqsUnsub = emptyUnsub
+      })
+    } catch (e) {
+      logger.error('Error inside sentReqs socket connect: ' + e.message)
+      socket.emit('$error', e.message)
+    }
+  })
+
+  let receivedReqsUnsub = emptyUnsub
+
+  io.of('receivedReqs').on('connect', async socket => {
+    try {
+      if (!isAuthenticated()) {
+        logger.info(
+          'not authenticated in gun for receivedReqs socket, will send NOT_AUTH'
+        )
+        socket.emit(Common.Constants.ErrorCode.NOT_AUTH)
+
+        return
+      }
+
+      logger.info('now checking token for receivedReqs socket')
+      const { token } = socket.handshake.query
+      const isAuth = await isValidToken(token)
+
+      if (!isAuth) {
+        logger.warn('invalid token for receivedReqs socket')
+        socket.emit(Common.Constants.ErrorCode.NOT_AUTH)
+        return
+      }
+
+      if (receivedReqsUnsub !== emptyUnsub) {
+        logger.error(
+          'Tried to set receivedReqs socket twice, this might be due to an app restart and the old socket not being recycled by socket.io in time, will disable the older subscription, which means the old socket wont work and data will be sent to this new socket instead'
+        )
+        receivedReqsUnsub()
+        receivedReqsUnsub = emptyUnsub
+      }
+
+      GunEvents.onSimplerReceivedRequests(receivedReqs => {
+        socket.emit(Common.Constants.Event.ON_RECEIVED_REQUESTS, receivedReqs)
+      })
+
+      socket.on('disconnect', () => {
+        receivedReqsUnsub()
+        receivedReqsUnsub = emptyUnsub
+      })
+    } catch (e) {
+      logger.error('Error inside receivedReqs socket connect: ' + e.message)
+      socket.emit('$error', e.message)
+    }
+  })
+
   return io
 }