Various fixes

submodules/TelegramUI/Sources/ContactMultiselectionController.swift

@@ -78,6 +78,7 @@ class ContactMultiselectionControllerImpl: ViewController, ContactMultiselection
     private var initialPeersDisposable: Disposable?
     private let options: [ContactListAdditionalOption]
     private let filters: [ContactListFilter]
+    private let limit: Int32?
     
     init(_ params: ContactMultiselectionControllerParams) {
         self.params = params
@@ -85,6 +86,7 @@ class ContactMultiselectionControllerImpl: ViewController, ContactMultiselection
         self.mode = params.mode
         self.options = params.options
         self.filters = params.filters
+        self.limit = params.limit
         self.presentationData = self.context.sharedContext.currentPresentationData.with { $0 }
         
         self.titleView = CounterContollerTitleView(theme: self.presentationData.theme)
@@ -228,6 +230,7 @@ class ContactMultiselectionControllerImpl: ViewController, ContactMultiselection
             self?.presentingViewController?.dismiss(animated: true, completion: nil)
         }
         
+        let limit = self.limit
         self.contactsNode.openPeer = { [weak self] peer in
             if let strongSelf = self, case let .peer(peer, _, _) = peer {
                 var updatedCount: Int?
@@ -261,8 +264,8 @@ class ContactMultiselectionControllerImpl: ViewController, ContactMultiselection
                         }
                     }
                 case let .chats(chatsNode):
-                    chatsNode.updateState { state in
-                        var state = state
+                    chatsNode.updateState { initialState in
+                        var state = initialState
                         if state.selectedPeerIds.contains(peer.id) {
                             state.selectedPeerIds.remove(peer.id)
                             removedTokenId = peer.id
@@ -271,6 +274,12 @@ class ContactMultiselectionControllerImpl: ViewController, ContactMultiselection
                             state.selectedPeerIds.insert(peer.id)
                         }
                         updatedCount = state.selectedPeerIds.count
+                        if let limit = limit, let count = updatedCount, count > limit {
+                            updatedCount = nil
+                            removedTokenId = nil
+                            addedToken = nil
+                            return initialState
+                        }
                         var updatedState = ContactListNodeGroupSelectionState()
                         for peerId in state.selectedPeerIds {
                             updatedState = updatedState.withToggledPeerId(.peer(peerId))