Repair connectivity correctly v1

toju-app/src/app/infrastructure/realtime/peer-connection-manager/recovery/peer-recovery.ts

@@ -1,5 +1,6 @@
 import {
   CONNECTION_STATE_CONNECTED,
+  DATA_CHANNEL_RECOVERY_GRACE_MS,
   DATA_CHANNEL_STATE_OPEN,
   P2P_TYPE_VOICE_STATE_REQUEST,
   PEER_DISCONNECT_GRACE_MS,
@@ -27,6 +28,7 @@ export function removePeer(
   const preserveReconnectState = options?.preserveReconnectState === true;
 
   clearPeerDisconnectGraceTimer(state, peerId);
+  clearDataChannelRecoveryTimer(state, peerId);
 
   if (!preserveReconnectState) {
     clearPeerReconnectTimer(state, peerId);
@@ -56,6 +58,7 @@ export function removePeer(
 export function closeAllPeers(state: PeerConnectionManagerState): void {
   clearAllPeerReconnectTimers(state);
   clearAllPeerDisconnectGraceTimers(state);
+  clearAllDataChannelRecoveryTimers(state);
   clearAllPingTimers(state);
 
   state.activePeerConnections.forEach((peerData) => {
@@ -106,6 +109,18 @@ export function clearPeerDisconnectGraceTimer(
   }
 }
 
+export function clearDataChannelRecoveryTimer(
+  state: PeerConnectionManagerState,
+  peerId: string
+): void {
+  const timer = state.dataChannelRecoveryTimers.get(peerId);
+
+  if (timer) {
+    clearTimeout(timer);
+    state.dataChannelRecoveryTimers.delete(peerId);
+  }
+}
+
 /** Cancel all pending peer reconnect timers and clear the tracker. */
 export function clearAllPeerReconnectTimers(state: PeerConnectionManagerState): void {
   state.peerReconnectTimers.forEach((timer) => clearInterval(timer));
@@ -118,6 +133,85 @@ export function clearAllPeerDisconnectGraceTimers(state: PeerConnectionManagerSt
   state.peerDisconnectGraceTimers.clear();
 }
 
+export function clearAllDataChannelRecoveryTimers(state: PeerConnectionManagerState): void {
+  state.dataChannelRecoveryTimers.forEach((timer) => clearTimeout(timer));
+  state.dataChannelRecoveryTimers.clear();
+}
+
+export function scheduleDataChannelRecovery(
+  context: PeerConnectionManagerContext,
+  peerId: string,
+  channel: RTCDataChannel,
+  reason: string,
+  handlers: RecoveryHandlers
+): void {
+  const { logger, state } = context;
+  const peerData = state.activePeerConnections.get(peerId);
+
+  if (!peerData || peerData.dataChannel !== channel)
+    return;
+
+  if (channel.readyState === DATA_CHANNEL_STATE_OPEN)
+    return;
+
+  if (state.dataChannelRecoveryTimers.has(peerId))
+    return;
+
+  logger.warn('[data-channel] Control channel unavailable; waiting before reconnect', {
+    channelLabel: channel.label,
+    peerId,
+    readyState: channel.readyState,
+    reason
+  });
+
+  const timer = setTimeout(() => {
+    state.dataChannelRecoveryTimers.delete(peerId);
+
+    const latestPeerData = state.activePeerConnections.get(peerId);
+
+    if (!latestPeerData || latestPeerData.dataChannel !== channel)
+      return;
+
+    if (latestPeerData.dataChannel?.readyState === DATA_CHANNEL_STATE_OPEN)
+      return;
+
+    logger.warn('[data-channel] Control channel did not recover; selecting repair path', {
+      channelLabel: channel.label,
+      connectionState: latestPeerData.connection.connectionState,
+      peerId,
+      readyState: latestPeerData.dataChannel?.readyState ?? null,
+      reason
+    });
+
+    if (latestPeerData.connection.connectionState === CONNECTION_STATE_CONNECTED) {
+      if (latestPeerData.isInitiator && handlers.replaceDataChannel(peerId, channel)) {
+        logger.info('[data-channel] Replaced control channel without recreating media transport', {
+          peerId,
+          reason
+        });
+
+        return;
+      }
+
+      if (!latestPeerData.isInitiator) {
+        logger.info('[data-channel] Waiting for initiator to replace control channel; preserving media transport', {
+          peerId,
+          reason
+        });
+
+        return;
+      }
+    }
+
+    trackDisconnectedPeer(state, peerId);
+    handlers.removePeer(peerId, { preserveReconnectState: true });
+    attemptPeerReconnect(context, peerId, handlers);
+    schedulePeerReconnect(context, peerId, handlers);
+  }, DATA_CHANNEL_RECOVERY_GRACE_MS);
+
+  state.dataChannelRecoveryTimers.set(peerId, timer);
+}
+
 export function schedulePeerDisconnectRecovery(
   context: PeerConnectionManagerContext,
   peerId: string,