Merge pull request #34 from Liquidibrium/master

Removed dependency on window
yjs · Apr 24, 2022 · 3037783 · 3037783
2 parents c059b86 + 017b9e8
commit 3037783
Showing 1 changed file with 13 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 () {
@@ -422,6 +428,11 @@ export class Room {
 
   destroy () {
     this.disconnect()
+    if (typeof window !== 'undefined') {
+      window.removeEventListener('beforeunload', this._beforeUnloadHandler)
+    } else if (typeof process !== 'undefined') {
+      process.off('exit', this._beforeUnloadHandler)
+    }
   }
 }