Merge pull request #683 from zrn-ns/feature/add_option_to_disable_pre…

…view_on_longpressing_cell Add option to disable preview on long-pressing cell
zhangao0086 · Feb 6, 2021 · 9a00af4 · 9a00af4
2 parents e61f49b + 05e880c
commit 9a00af4
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 5 deletions.
diff --git a/Sources/DKImagePickerController/DKImagePickerControllerBaseUIDelegate.swift b/Sources/DKImagePickerController/DKImagePickerControllerBaseUIDelegate.swift
@@ -268,6 +268,10 @@ open class DKImagePickerControllerBaseUIDelegate: NSObject, DKImagePickerControl
         return DKAssetGroupCell.self
     }
 
+    open func needsToShowPreviewOnLongPress() -> Bool {
+        return true
+    }
+
     open func imagePickerControllerSelectGroupButton(_ imagePickerController: DKImagePickerController, selectedGroup: DKAssetGroup) -> UIButton {
         let button = self.createSelectGroupButtonIfNeeded()
         let groupsCount = self.imagePickerController.groupDataManager.groupIds?.count ?? 0

diff --git a/Sources/DKImagePickerController/View/DKAssetGroupDetailVC.swift b/Sources/DKImagePickerController/View/DKAssetGroupDetailVC.swift
@@ -577,11 +577,14 @@ open class DKAssetGroupDetailVC: UIViewController,
                 cell.thumbnailImage = image
             }
         }
-
-        cell.longPressBlock = { [weak self, weak cell] in
-            guard let strongSelf = self, let strongCell = cell else { return }
-
-            strongSelf.showGallery(from: strongCell)
+
+        if let imagePickerController = imagePickerController,
+            imagePickerController.UIDelegate.needsToShowPreviewOnLongPress() {
+
+            cell.longPressBlock = { [weak self, weak cell] in
+                guard let strongSelf = self, let strongCell = cell else { return }
+                strongSelf.showGallery(from: strongCell)
+            }
         }
     }