slight tweak of the context menu interface

lkzhao · Aug 3, 2021 · 017a62c · 017a62c
1 parent 4a8e297
commit 017a62c
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/Sources/UIComponent/Components/TappableView.swift b/Sources/UIComponent/Components/TappableView.swift
@@ -49,7 +49,7 @@ open class TappableView: ComponentView {
     }
   }
 
-  public var contextMenuProvider: UIContextMenuActionProvider? {
+  public var contextMenuProvider: ((TappableView) -> UIMenu?)? {
     didSet {
       if contextMenuProvider != nil {
         addInteraction(contextMenuInteraction)
@@ -106,7 +106,10 @@ open class TappableView: ComponentView {
 
 extension TappableView: UIContextMenuInteractionDelegate {
   public func contextMenuInteraction(_ interaction: UIContextMenuInteraction, configurationForMenuAtLocation location: CGPoint) -> UIContextMenuConfiguration? {
-    return UIContextMenuConfiguration(identifier: nil, previewProvider: nil, actionProvider: contextMenuProvider)
+    return UIContextMenuConfiguration(identifier: nil, previewProvider: nil) { [weak self] _ in
+      guard let self = self else { return nil }
+      return self.contextMenuProvider?(self)
+    }
   }
 }