fix: listen to peer connection change

radixdlt · Feb 5, 2024 · 4a1eecf · 4a1eecf
1 parent 0394f36
commit 4a1eecf
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 5 deletions.
diff --git a/src/connector/webrtc/peer-connection-client.ts b/src/connector/webrtc/peer-connection-client.ts
@@ -65,16 +65,17 @@ export const PeerConnectionClient = (input: {
     )
   }
 
-  const onIceConnectionStateChange = () => {
+  const onPeerConnectionState = () => {
     logger?.debug(
-      `🕸🧊 onIceConnectionStateChange: ${peerConnection.iceConnectionState}`,
+      `🕸🧊 onPeerConnectionStateChange: ${peerConnection.connectionState}`,
     )
+    subjects.peerConnectionStateSubject.next(peerConnection.connectionState)
   }
 
   peerConnection.onnegotiationneeded = onNegotiationNeeded
   peerConnection.onsignalingstatechange = onSignalingStateChange
   peerConnection.onicegatheringstatechange = onIceGatheringStateChange
-  peerConnection.oniceconnectionstatechange = onIceConnectionStateChange
+  peerConnection.onconnectionstatechange = onPeerConnectionState
 
   const setLocalDescription = (description: RTCSessionDescriptionInit) =>
     ResultAsync.fromPromise(

diff --git a/src/connector/webrtc/subjects.ts b/src/connector/webrtc/subjects.ts
@@ -27,4 +27,7 @@ export const WebRtcSubjects = () => ({
   remoteAnswerSubject: new BehaviorSubject<
     RTCSessionDescriptionInit | undefined
   >(undefined),
+  peerConnectionStateSubject: new BehaviorSubject<
+    RTCPeerConnectionState | undefined
+  >(undefined),
 })
diff --git a/src/connector/webrtc/webrtc-client.ts b/src/connector/webrtc/webrtc-client.ts
@@ -212,9 +212,10 @@ export const WebRtcClient = (input: {
   )
 
   subscriptions.add(
-    subjects.iceConnectionStateSubject
+    subjects.peerConnectionStateSubject
       .pipe(
-        filter((state) => state === 'disconnected'),
+        filter((state): state is RTCPeerConnectionState => !!state),
+        filter((state) => ['disconnected', 'closed', 'failed'].includes(state)),
         tap(() => {
           restart()
         }),