BugFix: single selection mode #189

tilltue · Jul 17, 2019 · bdb091f · bdb091f
1 parent a8b8080
commit bdb091f
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 5 deletions.
diff --git a/TLPhotoPicker.podspec b/TLPhotoPicker.podspec
@@ -8,7 +8,7 @@
 
 Pod::Spec.new do |s|
   s.name             = 'TLPhotoPicker'
-  s.version          = '1.9.2'
+  s.version          = '1.9.3'
   s.summary          = 'multiple phassets picker for iOS lib. like facebook'
 
 # This description is used to generate tags and improve search results.

diff --git a/TLPhotoPicker/Classes/TLPhotosPickerViewController.swift b/TLPhotoPicker/Classes/TLPhotosPickerViewController.swift
@@ -247,15 +247,23 @@ open class TLPhotosPickerViewController: UIViewController {
         }
     }
 
+    private func findIndexAndReloadCells(phAsset: PHAsset) {
+        if
+            var index = self.focusedCollection?.fetchResult?.index(of: phAsset),
+            index != NSNotFound
+        {
+            index += (getfocusedIndex() == 0 && self.configure.usedCameraButton) ? 1 : 0
+            self.collectionView.reloadItems(at: [IndexPath(row: index, section: 0)])
+        }
+    }
+
     open func deselectWhenUsingSingleSelectedMode() {
         if
             self.configure.singleSelectedMode == true,
-            let selectedPHAsset = self.selectedAssets.first?.phAsset,
-            var index = self.focusedCollection?.fetchResult?.index(of: selectedPHAsset)
+            let selectedPHAsset = self.selectedAssets.first?.phAsset
         {
-            index += self.configure.usedCameraButton ? 1 : 0
             self.selectedAssets.removeAll()
-            self.collectionView.reloadItems(at: [IndexPath(row: index, section: 0)])
+            findIndexAndReloadCells(phAsset: selectedPHAsset)
         }
     }