Seperated _beforeUnloadHandler for NodeJS

yjs · Apr 24, 2022 · c6a88a1 · c6a88a1
1 parent c059b86
commit c6a88a1
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/y-webrtc.js b/src/y-webrtc.js
@@ -359,12 +359,18 @@ export class Room {
       broadcastRoomMessage(this, encoding.toUint8Array(encoderAwareness))
     }
 
-    window.addEventListener('beforeunload', () => {
+    this._beforeUnloadHandler = () => {
       awarenessProtocol.removeAwarenessStates(this.awareness, [doc.clientID], 'window unload')
       rooms.forEach(room => {
         room.disconnect()
       })
-    })
+    }
+
+    if (typeof window !== 'undefined') {
+      window.addEventListener('beforeunload', this._beforeUnloadHandler)
+    } else if (typeof process !== 'undefined') {
+      process.on('exit', this._beforeUnloadHandler)
+    }
   }
 
   connect () {