修正自定义多选相册的逻辑

MRFramework/MRFramework/UIControls/Album/MRAlbumDetailViewController.swift

@@ -71,12 +71,15 @@ open class MRAlbumDetailViewController: UIViewController {
     
     open override func viewWillAppear(_ animated: Bool) {
         super.viewWillAppear(animated)
-        navigationController?.setNavigationBarHidden(true, animated: false)
+//        navigationController?.setNavigationBarHidden(true, animated: false)
     }
     
     open override func viewWillDisappear(_ animated: Bool) {
         super.viewWillDisappear(animated)
-        navigationController?.setNavigationBarHidden(false, animated: false)
+//        navigationController?.setNavigationBarHidden(false, animated: false)
+        if state == .loading {
+            state = .none
+        }
     }
     
     open override var prefersStatusBarHidden: Bool {
@@ -175,8 +178,8 @@ extension MRAlbumDetailViewController {
             setupMainViews()
             setupLayouts()
             resizeImageView()
-        default:
-            break
+        case .none:
+            self.imageCache.stopCachingImagesForAllAssets()
         }
     }
     

MRFramework/MRFramework/UIControls/Album/MRAlbumListViewController.swift

@@ -150,23 +150,25 @@ open class MRAlbumListViewController: UIViewController {
     /// - Parameter cell: cell
     /// - Parameter indexPath: indexPath
     open func didClickOnCellCheck(_ cell: UICollectionViewCell, at indexPath: IndexPath) {
-        if let selected = assests?.object(at: indexPath.row) {
-            
-            if selectedAssests.contains(selected) {
-                selectedAssests.removeAll(where: { $0 == selected })
-                cell.isSelected = false
-                return
-            }
-            
-            if selectedAssests.count >= maxCount {
-                return
-            }
-            
-            selectedAssests.append(selected)
-            if !collectionView.allowsMultipleSelection {
-                confirm()
-            } else {
-                cell.isSelected = true
+        if maxCount > 0 {
+            if let selected = assests?.object(at: indexPath.row) {
+                
+                if selectedAssests.contains(selected) {
+                    selectedAssests.removeAll(where: { $0 == selected })
+                    cell.isSelected = false
+                    return
+                }
+                
+                if selectedAssests.count >= maxCount {
+                    return
+                }
+                
+                selectedAssests.append(selected)
+                if !collectionView.allowsMultipleSelection {
+                    confirm()
+                } else {
+                    cell.isSelected = true
+                }
             }
         }
     }
@@ -183,15 +185,13 @@ open class MRAlbumListViewController: UIViewController {
             let backItem = UIBarButtonItem(customView: button)
             navigationItem.leftBarButtonItem = backItem
             
-            if collectionView.allowsMultipleSelection {
-                let confirmBtn = UIButton(type: .custom)
-                confirmBtn.frame = CGRect(x: 0, y: 0, width: 50, height: 40)
-                confirmBtn.setTitle("确定", for: .normal)
-                confirmBtn.setTitleColor(.init(0x333333), for: .normal)
-                confirmBtn.addTarget(self, action: #selector(confirm), for: .touchUpInside)
-                let item = UIBarButtonItem(customView: confirmBtn)
-                navigationItem.rightBarButtonItem = item
-            }
+            let confirmBtn = UIButton(type: .custom)
+            confirmBtn.frame = CGRect(x: 0, y: 0, width: 50, height: 40)
+            confirmBtn.setTitle("确定", for: .normal)
+            confirmBtn.setTitleColor(.init(0x333333), for: .normal)
+            confirmBtn.addTarget(self, action: #selector(confirm), for: .touchUpInside)
+            let item = UIBarButtonItem(customView: confirmBtn)
+            navigationItem.rightBarButtonItem = item
         }
     }
     
@@ -205,7 +205,7 @@ open class MRAlbumListViewController: UIViewController {
             collectionView.allowsMultipleSelection = maxCount > 1
             let count = selectedAssests.count - maxCount
             if count > 0 {
-                selectedAssests.removeLast(count)
+                selectedAssests.removeFirst(count)
             }
         }
         collectionView.reloadData()
@@ -222,8 +222,22 @@ open class MRAlbumListViewController: UIViewController {
             vc.isSelected = selectedAssests.contains(assest)
             vc.imageCache = imageCache
             vc.confirm = { vc, isSelected in
+                var removedIndex = NSNotFound
                 if isSelected, !self.selectedAssests.contains(assest) {
-                    self.selectedAssests.append(assest)
+                    if self.selectedAssests.count == self.maxCount {
+                        let removedAssest = self.selectedAssests[0]
+                        if let assests = self.assests {
+                            removedIndex = assests.index(of: assest)
+                        }
+                        if removedIndex != NSNotFound,
+                            let cell = collectionView.cellForItem(at: IndexPath(row: removedIndex, section: 0)) {
+                            cell.isSelected = false
+                        }
+                        
+                        self.selectedAssests[0] = assest
+                    } else {
+                        self.selectedAssests.append(assest)
+                    }
                 } else if !isSelected {
                     self.selectedAssests.removeAll(where: { $0 == assest })
                 }
@@ -242,10 +256,14 @@ open class MRAlbumListViewController: UIViewController {
     
     /// 点击确定
     @objc open func confirm() {
-        ToastView.loading()
-        requestImages(with: selectedAssests, size: maxImageSize) { (images, count) in
-            ToastView.hide(animate: false)
-            self.didConfirm?(self, images, count)
+        if selectedAssests.count > 0 {
+            ToastView.loading()
+            requestImages(with: selectedAssests, size: maxImageSize) { (images, count) in
+                ToastView.hide(animate: false)
+                self.didConfirm?(self, images, count)
+            }
+        } else {
+            self.didCancel?(self)
         }
     }
 }