Updated DropUpMenu.swift

- added `dropUpMenuWillDismiss` delegate method
teambition · Apr 12, 2017 · d92ff5a · d92ff5a
1 parent ba302fc
commit d92ff5a
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/DropdownMenu/DropUpMenu.swift b/DropdownMenu/DropUpMenu.swift
@@ -20,6 +20,7 @@ public protocol DropUpMenuDelegate: class {
     func dropUpMenu(_ dropUpMenu: DropUpMenu, cellForRowAt indexPath: IndexPath) -> UITableViewCell?
     func dropUpMenu(_ dropUpMenu: DropUpMenu, didSelectRowAt indexPath: IndexPath)
     func dropUpMenuCancel(_ dropUpMenu: DropUpMenu)
+    func dropUpMenuWillDismiss(_ dropUpMenu: DropdownMenu)
 }
 
 public extension DropUpMenuDelegate {
@@ -32,6 +33,9 @@ public extension DropUpMenuDelegate {
 
     func dropUpMenuCancel(_ dropUpMenu: DropUpMenu) {
     }
+
+    func dropUpMenuWillDismiss(_ dropUpMenu: DropUpMenu) {
+    }
 }
 
 private let screenRect = UIScreen.main.bounds
@@ -181,6 +185,7 @@ open class DropUpMenu: UIView {
     }
 
     open func hideMenu(isSelectAction: Bool = false) {
+        delegate?.dropUpMenuWillDismiss(self)
         UIView.animate(withDuration: animateDuration, animations: {
             self.backgroundColor = self.backgroudBeginColor
             self.tableView.frame.origin.y = screenRect.height - self.bottomOffsetY