Fix manager.addAction will crash, and other fixes

Speed up dismissal duration
Add dismiss handler
Add handler for each action

Example/Example/View Controllers/RootViewController.swift

@@ -53,9 +53,12 @@ class RootViewController: UITableViewController {
         // Since `UIBarButtonItem` is not a subclass of `UIView`, we need to
         // know the view's frame to make the relative position work
         controller.setBarButtonItemForSourceView(barButtonItem)
-
         controller.delegate = self
 
+        controller.dismissalHandler = { selected in
+            print("Selected: \(selected ? "Yep" : "Nope")")
+        }
+
         // Present menu controller
         present(controller, animated: true, completion: nil)
     }

PopMenu/Classes/PopMenuManager.swift

@@ -32,6 +32,9 @@ final public class PopMenuManager: NSObject {
     /// Determines whether to dismiss menu after an action is selected.
     public var popMenuShouldDismissOnSelection: Bool = true
 
+    /// The dismissal handler for pop menu.
+    public var popMenuDismissalHandler: ((Bool) -> Void)?
+
     /// Appearance for passing on to pop menu.
     public let popMenuAppearance: PopMenuAppearance
 
@@ -47,6 +50,7 @@ final public class PopMenuManager: NSObject {
         popMenu.delegate = popMenuDelegate
         popMenu.appearance = popMenuAppearance
         popMenu.shouldDismissOnSelection = popMenuShouldDismissOnSelection
+        popMenu.dismissalHandler = popMenuDismissalHandler
 
         if let barButtonItem = barButtonItem {
             popMenu.setBarButtonItemForSourceView(barButtonItem)
@@ -60,7 +64,11 @@ final public class PopMenuManager: NSObject {
 
     /// Pass a new action to pop menu.
     public func addAction(_ action: PopMenuAction) {
-        popMenu.addAction(action)
+        if let popMenu = popMenu {
+            popMenu.addAction(action)
+        } else {
+            actions.append(action)
+        }
     }
 
 }

PopMenu/Transitions/PopMenuDismissAnimationController.swift

@@ -20,7 +20,7 @@ final public class PopMenuDismissAnimationController: NSObject, UIViewController
 
     /// Duration of the transition.
     public func transitionDuration(using transitionContext: UIViewControllerContextTransitioning?) -> TimeInterval {
-        return 0.142
+        return 0.0982
     }
 
     /// Animate PopMenuViewController custom transition.

PopMenu/Transitions/PopMenuPresentAnimationController.swift

@@ -20,7 +20,7 @@ final public class PopMenuPresentAnimationController: NSObject, UIViewController
 
     /// Duration of the transition.
     public func transitionDuration(using transitionContext: UIViewControllerContextTransitioning?) -> TimeInterval {
-        return 0.142
+        return 0.138
     }
 
     /// Animate PopMenuViewController custom transition.

PopMenu/View Controller & Views/PopMenuAction.swift

@@ -23,6 +23,9 @@ import UIKit
     /// The initial color of the action.
     var color: Color? { get }
 
+    /// The handler of action.
+    var handler: PopMenuActionHandler? { get }
+
     /// Left padding when texts-only.
     static var textLeftPadding: CGFloat { get }
 
@@ -49,6 +52,9 @@ import UIKit
 
     /// Called when the action gets selected.
     @objc optional func actionSelected(animated: Bool)
+
+    /// Type alias for selection handler.
+    typealias PopMenuActionHandler = (PopMenuAction) -> Void
 
 }
 
@@ -67,6 +73,9 @@ public class PopMenuDefaultAction: NSObject, PopMenuAction {
     /// Color of action.
     public let color: Color?
 
+    /// Handler of action when selected.
+    public let handler: PopMenuActionHandler?
+
     // MARK: - Computed Properties
 
     /// Text color of the label.
@@ -142,10 +151,12 @@ public class PopMenuDefaultAction: NSObject, PopMenuAction {
 
     // MARK: - Initializer
 
-    public init(title: String? = nil, image: UIImage? = nil, color: Color? = nil) {
+    /// Initializer.
+    public init(title: String? = nil, image: UIImage? = nil, color: Color? = nil, handler: PopMenuActionHandler? = nil) {
         self.title = title
         self.image = image
         self.color = color
+        self.handler = handler
 
         view = UIView()
     }
@@ -196,12 +207,16 @@ public class PopMenuDefaultAction: NSObject, PopMenuAction {
 
     /// When the action is selected.
     public func actionSelected(animated: Bool) {
+        // Trigger handler.
+        handler?(self)
+
+        // Animate selection
         guard animated else { return }
 
         DispatchQueue.main.async {
             UIView.animate(withDuration: 0.175, animations: {
-                self.view.transform = CGAffineTransform.identity.scaledBy(x: 0.92, y: 0.92)
-                self.view.backgroundColor = self.backgroundColor.withAlphaComponent(0.2)
+                self.view.transform = CGAffineTransform.identity.scaledBy(x: 0.915, y: 0.915)
+                self.view.backgroundColor = self.backgroundColor.withAlphaComponent(0.18)
             }, completion: {
                 if $0 {
                     UIView.animate(withDuration: 0.175, animations: {

PopMenu/View Controller & Views/PopMenuViewController.swift

@@ -66,6 +66,9 @@ final public class PopMenuViewController: UIViewController {
     /// Determines whether the pan gesture is enabled on the actions.
     public var shouldEnablePanGesture: Bool = true
 
+    /// Handler for when the menu is dismissed.
+    public var dismissalHandler: ((Bool) -> Void)?
+
     // MARK: - Constraints
 
     private(set) var contentLeftConstraint: NSLayoutConstraint!
@@ -95,7 +98,7 @@ final public class PopMenuViewController: UIViewController {
 
     /// Max content width allowed for the content to stretch to.
     fileprivate let maxContentWidth: CGFloat = UIScreen.main.bounds.size.width * 0.9
-        
+
     // MARK: - View Life Cycle
 
     public convenience init(sourceView: UIView? = nil, actions: [PopMenuAction], appearance: PopMenuAppearance? = nil) {
@@ -426,7 +429,10 @@ extension PopMenuViewController {
     @objc fileprivate func backgroundViewDidTap(_ gesture: UITapGestureRecognizer) {
         guard gesture.isEqual(tapGestureForDismissal), !touchedInsideContent(location: gesture.location(in: view)) else { return }
 
-        dismiss(animated: true, completion: nil)
+        dismiss(animated: true) {
+            // No selection made.
+            self.dismissalHandler?(false)
+        }
     }
 
     /// When the menu action gets tapped.
@@ -493,13 +499,17 @@ extension PopMenuViewController {
     fileprivate func actionDidSelect(at index: Int, animated: Bool = true) {
         let action = actions[index]
         action.actionSelected?(animated: animated)
-
+        // Generate haptics
         Haptic.impact(.medium).generate()
-
+        // Notify delegate
         delegate?.popMenuDidSelectItem?(self, at: index)
 
+        // Should dismiss or not
         if shouldDismissOnSelection {
-            dismiss(animated: true, completion: nil)
+            dismiss(animated: true) {
+                // Selection made.
+                self.dismissalHandler?(true)
+            }
         }
     }