WIP feat: Terminates realtime on client logout

src/lib/realtime.js

@@ -5,6 +5,8 @@ const APPS_DOCTYPE = 'io.cozy.apps'
 /**
  * Initialize realtime sockets
  *
+ * Returns function that closes the realtime and unsubscribe handlers
+ *
  * @private
  * @param {object}
  * @returns {Promise}
@@ -30,13 +32,22 @@ function initializeRealtime({ getApp, onCreate, onDelete, cozyClient }) {
     }
   }
 
+  let realtime
   try {
-    const realtime = new CozyRealtime({ cozyClient })
+    realtime = new CozyRealtime({ cozyClient })
     realtime.subscribe('created', APPS_DOCTYPE, handleAppCreation)
     realtime.subscribe('deleted', APPS_DOCTYPE, handleAppRemoval)
   } catch (error) {
     console.warn(`Cannot initialize realtime in Cozy-bar: ${error.message}`)
   }
+
+  return () => {
+    if (!realtime) {
+      return
+    }
+    realtime.unsubscribe('created', APPS_DOCTYPE, handleAppCreation)
+    realtime.unsubscribe('deleted', APPS_DOCTYPE, handleAppRemoval)
+  }
 }
 
 export default initializeRealtime

src/lib/stack-client.js

@@ -353,13 +353,23 @@ const getSettingsAppURL = function() {
  * @returns {Promise}
  */
 const init = function({ cozyClient: client, onCreate, onDelete }) {
-  cozyClient = client
-  if (!cozyClient.isLogged) return
-  initializeRealtime({
-    getApp,
-    onCreate,
-    onDelete,
-    cozyClient
+  cozyClient = client // Setting global variable
+  let terminateRealtime
+  const onLogin = () => {
+    terminateRealtime = initializeRealtime({
+      getApp,
+      onCreate,
+      onDelete,
+      cozyClient
+    })
+  }
+  if (cozyClient.isLogged) {
+    onLogin()
+  }
+  cozyClient.once('logout', () => {
+    if (terminateRealtime) {
+      terminateRealtime()
+    }
   })
 }
 

test/lib/stack-client/stack-client.init.spec.js

@@ -4,7 +4,6 @@ import CozyClient from 'cozy-client'
 import initializeRealtime from 'lib/realtime'
 
 jest.mock('lib/realtime')
-initializeRealtime.mockResolvedValue(Promise.resolve())
 
 const { init } = stack
 
@@ -30,6 +29,15 @@ describe('stack client', () => {
       await init(params)
     }
 
+    beforeEach(() => {
+      isLogged = undefined
+      isPublic = undefined
+      initializeRealtime.mockReset().mockImplementation(() => {
+        terminateRealtime = jest.fn()
+        return terminateRealtime
+      })
+    })
+
     afterAll(() => {
       jest.restoreAllMocks()
     })
@@ -49,5 +57,13 @@ describe('stack client', () => {
       expect(initializeRealtime).toHaveBeenCalled()
       expect(initializeRealtime.mock.calls[0][0].cozyClient).toBe(cozyClient)
     })
+
+    it('should terminate realtime if the user logs out', async  () => {
+      isLogged = true
+      await setup()
+      expect(terminateRealtime).not.toHaveBeenCalled()
+      cozyClient.emit('logout')
+      expect(terminateRealtime).toHaveBeenCalled()
+    })
   })
 })