End select mode after hitting delete

// FREEBIE
2018-03-26 16:22:57 -04:00 · 2018-03-26 16:22:57 -04:00 · 405edaa125
parent 6e3de94e99
commit 405edaa125
1 changed files with 5 additions and 0 deletions
--- a/Signal/src/ViewControllers/MediaTileViewController.swift
+++ b/Signal/src/ViewControllers/MediaTileViewController.swift
@ -482,6 +482,10 @@ public class MediaTileViewController: UICollectionViewController, MediaGalleryDa

    @objc
    func didCancelSelect(_ sender: Any) {
+        endSelectMode()
+    }
+
+    func endSelectMode() {
        isInBatchSelectMode = false

        guard let collectionView = self.collectionView else {
@ -537,6 +541,7 @@ public class MediaTileViewController: UICollectionViewController, MediaGalleryDa

        let deleteAction = UIAlertAction(title: confirmationTitle, style: .destructive) { _ in
            mediaGalleryDataSource.delete(items: items)
+            self.endSelectMode()
        }

        let actionSheet = UIAlertController(title: nil, message: nil, preferredStyle: .actionSheet)