Fix according to review (separate vars for affected channels/groups)

pubnub · Sep 17, 2024 · 32d0e57 · 32d0e57
1 parent b3a3bb6
commit 32d0e57
Showing 1 changed file with 15 additions and 3 deletions.
diff --git a/Sources/PubNub/KMP/KMPPubNub+Listeners.swift b/Sources/PubNub/KMP/KMPPubNub+Listeners.swift
@@ -38,32 +38,44 @@ extension KMPPubNub {
 
       let category: KMPConnectionStatusCategory
       let errorIfAny: Error?
+      let affectedChannels: [String]
+      let affectedGroups: [String]
 
       switch newStatus {
       case .connected:
         category = .connected
         errorIfAny = nil
+        affectedChannels = pubnub.subscribedChannels
+        affectedGroups = pubnub.subscribedChannelGroups
       case .disconnected:
         category = .disconnected
         errorIfAny = nil
+        affectedChannels = pubnub.subscribedChannels
+        affectedGroups = pubnub.subscribedChannelGroups
       case let .disconnectedUnexpectedly(error):
         category = .disconnectedUnexpectedly
         errorIfAny = error
-      case .subscriptionChanged:
+        affectedChannels = pubnub.subscribedChannels
+        affectedGroups = pubnub.subscribedChannelGroups
+      case let .subscriptionChanged(channels, groups):
         category = .subscriptionChanged
         errorIfAny = nil
+        affectedChannels = channels
+        affectedGroups = groups
       case let .connectionError(error):
         category = .connectionError
         errorIfAny = error
+        affectedChannels = pubnub.subscribedChannels
+        affectedGroups = pubnub.subscribedChannelGroups
       }
 
       listener.onStatusChange?(
         KMPConnectionStatus(
           category: category,
           error: errorIfAny,
           currentTimetoken: NSNumber(value: pubnub.previousTimetoken ?? 0),
-          affectedChannels: Set(pubnub.subscribedChannels),
-          affectedChannelGroups: Set(pubnub.subscribedChannelGroups)
+          affectedChannels: Set(affectedChannels),
+          affectedChannelGroups: Set(affectedGroups)
         )
       )
     })