Single media in single album deselect crash fix

niralishaha25 · Oct 1, 2024 · 736fc0f · 736fc0f
1 parent 3c0f2b9
commit 736fc0f
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/Source/Pages/Gallery/YPLibraryVC+CollectionView.swift b/Source/Pages/Gallery/YPLibraryVC+CollectionView.swift
@@ -68,8 +68,12 @@ extension YPLibraryVC {
 
             // Replace the current selected image with the previously selected one
             if let previouslySelectedIndexPath = selectedIndexPaths.last {
-                v.collectionView.deselectItem(at: indexPath, animated: false)
-                v.collectionView.selectItem(at: previouslySelectedIndexPath, animated: false, scrollPosition: [])
+                if(v.collectionView.indexPathsForSelectedItems?.contains(indexPath) ?? false){
+                    v.collectionView.deselectItem(at: indexPath, animated: false)
+                }
+                if(v.collectionView.indexPathsForVisibleItems.contains(previouslySelectedIndexPath)){
+                    v.collectionView.selectItem(at: previouslySelectedIndexPath, animated: false, scrollPosition: [])
+                }
                 currentlySelectedIndex = previouslySelectedIndexPath.row
                 changeAsset(mediaManager.getAsset(at: previouslySelectedIndexPath.row))
             }