Merge pull request #86 from apptekstudios/master

Ensure willDisappear closure is called before removing contentView
huri000 · Aug 18, 2018 · 1ec36cd · 1ec36cd
2 parents 8e18267 + 8487c08
commit 1ec36cd
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/Source/Infra/EKContentView.swift b/Source/Infra/EKContentView.swift
@@ -460,6 +460,9 @@ class EKContentView: UIView {
     func removePromptly(keepWindow: Bool = true) {
         outDispatchWorkItem?.cancel()
         entryDelegate?.changeToInactive(withAttributes: attributes, pushOut: false)
+
+        // Execute willDisappear action if needed
+        self.contentView.content.attributes.lifecycleEvents.willDisappear?()
         removeFromSuperview(keepWindow: keepWindow)
     }
 
@@ -666,6 +669,8 @@ extension EKContentView {
         UIView.animate(withDuration: duration, delay: 0, usingSpringWithDamping: 1, initialSpringVelocity: 4, options: [.allowUserInteraction, .beginFromCurrentState], animations: {
             self.translateOut(withType: type)
         }, completion: { finished in
+            // Execute willDisappear action if needed
+            self.contentView.content.attributes.lifecycleEvents.willDisappear?()
             self.removeFromSuperview(keepWindow: false)
         })
     }