Fix peer updating

Author: rfree18

Sources/MultipeerKit/Internal API/MultipeerConnection.swift

@@ -120,17 +120,15 @@ extension MultipeerConnection: MCSessionDelegate {
 
         let handler = invitationCompletionHandlers[peerID]
 
-        defer { invitationCompletionHandlers[peerID] = nil }
-
         DispatchQueue.main.async {
             switch state {
             case .connected:
                 handler?(.success(peer))
-
+                self.invitationCompletionHandlers[peerID] = nil
                 self.didConnectToPeer?(peer)
             case .notConnected:
                 handler?(.failure(MultipeerError(localizedDescription: "Failed to connect to peer.")))
-
+                self.invitationCompletionHandlers[peerID] = nil
                 self.didDisconnectFromPeer?(peer)
             case .connecting:
                 break

Sources/MultipeerKit/Public API/MultipeerConfiguration.swift

@@ -20,7 +20,7 @@ public struct MultipeerConfiguration {
     /// Configures security-related aspects of the multipeer connection.
     public struct Security {
 
-        public typealias InvitationHandler = (Peer, Data?, (Bool) -> Void) -> Void
+        public typealias InvitationHandler = (Peer, Data?, @escaping (Bool) -> Void) -> Void
 
         /// An array of information that can be used to identify the peer to other nearby peers.
         ///

Sources/MultipeerKit/Public API/MultipeerTransceiver.swift

@@ -151,7 +151,7 @@ public final class MultipeerTransceiver {
     }
 
     private func handlePeerRemoved(_ peer: Peer) {
-        guard let idx = availablePeers.firstIndex(of: peer) else { return }
+        guard let idx = availablePeers.firstIndex(where: { $0.underlyingPeer == peer.underlyingPeer }) else { return }
 
         availablePeers.remove(at: idx)
     }
@@ -165,7 +165,7 @@ public final class MultipeerTransceiver {
     }
 
     private func setConnected(_ connected: Bool, on peer: Peer) {
-        guard let idx = availablePeers.firstIndex(of: peer) else { return }
+        guard let idx = availablePeers.firstIndex(where: { $0.underlyingPeer == peer.underlyingPeer }) else { return }
 
         var mutablePeer = availablePeers[idx]
         mutablePeer.isConnected = connected