Updated DropdownMenu.swift

- added `dropdownMenuWillDismiss` delegate method
teambition · Apr 12, 2017 · ba302fc · ba302fc
1 parent 273fe3c
commit ba302fc
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/DropdownMenu/DropdownMenu.swift b/DropdownMenu/DropdownMenu.swift
@@ -12,12 +12,14 @@ public protocol DropdownMenuDelegate: class {
     func dropdownMenu(_ dropdownMenu: DropdownMenu, cellForRowAt indexPath: IndexPath) -> UITableViewCell?
     func dropdownMenu(_ dropdownMenu: DropdownMenu, didSelectRowAt indexPath: IndexPath)
     func dropdownMenuCancel(_ dropdownMenu: DropdownMenu)
+    func dropdownMenuWillDismiss(_ dropdownMenu: DropdownMenu)
 }
 
 public extension DropdownMenuDelegate {
     func dropdownMenu(_ dropdownMenu: DropdownMenu, cellForRowAt indexPath: IndexPath) -> UITableViewCell? { return nil }
     func dropdownMenu(_ dropdownMenu: DropdownMenu, didSelectRowAt indexPath: IndexPath) { }
     func dropdownMenuCancel(_ dropdownMenu: DropdownMenu) { }
+    func dropdownMenuWillDismiss(_ dropdownMenu: DropdownMenu) { }
 }
 
 open class DropdownMenu: UIView {
@@ -234,6 +236,7 @@ open class DropdownMenu: UIView {
     }
 
     open func hideMenu(isSelectAction: Bool = false) {
+        delegate?.dropdownMenuWillDismiss(self)
         UIView.animate(withDuration: animateDuration, animations: {
             self.backgroundColor = self.backgroudBeginColor
             self.tableView.frame.origin.y = -self.tableViewHeight