Merge pull request #304 from xuning0/bugfix/store

Fix selected assets are not removed after the corresponding photo is deleted
mikaoj · Oct 20, 2020 · db9bd79 · db9bd79
2 parents 7ac53be + be9180c
commit db9bd79
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/Sources/Scene/Assets/AssetsViewController.swift b/Sources/Scene/Assets/AssetsViewController.swift
@@ -217,7 +217,9 @@ extension AssetsViewController: PHPhotoLibraryChangeObserver {
                         let removedItems = removed.map { IndexPath(item: $0, section:0) }
                         let removedSelections = self.collectionView.indexPathsForSelectedItems?.filter { return removedItems.contains($0) }
                         removedSelections?.forEach {
-                            self.delegate?.assetsViewController(self, didDeselectAsset: changes.fetchResultBeforeChanges.object(at: $0.row))
+                            let removedAsset = changes.fetchResultBeforeChanges.object(at: $0.row)
+                            self.store.remove(removedAsset)
+                            self.delegate?.assetsViewController(self, didDeselectAsset: removedAsset)
                         }
                         self.collectionView.deleteItems(at: removedItems)
                     }