[elpa] externals/frog-menu ca08068 13/43: Add cleanup handler option

[Clemens Radermacher]

branch: externals/frog-menu
commit ca080688ad850567950cd209a4a7e7b311f30df1
Author: Clemens Radermacher <address@hidden>
Commit: Clemens Radermacher <address@hidden>

    Add cleanup handler option
---
 frog-menu.el | 23 ++++++++++++++++++++---
 1 file changed, 20 insertions(+), 3 deletions(-)

diff --git a/frog-menu.el b/frog-menu.el
index 5b8fae4..ce94b08 100644
--- a/frog-menu.el
+++ b/frog-menu.el
@@ -79,7 +79,11 @@ When defining a new menu type, handlers need to be added for
 
 `frog-menu-display-handler-alist'
 
-`frog-menu-query-handler-alist'."
+`frog-menu-query-handler-alist'
+
+and optionally to
+
+`frog-menu-cleanup-handler-alist'."
   :type 'symbol)
 
 
@@ -119,6 +123,14 @@ value. If the user exited the query return nil."
   :type '(alist :key-type symbol
                 :value-type function))
 
+(defcustom frog-menu-cleanup-handler-alist
+  '((avy-posframe . posframe-hide))
+  "Maps `frog-menu-type' to a cleanup handler.
+
+The cleanup handler receives the displayed buffer as argument."
+  :type '(alist :key-type symbol
+                :value-type function))
+
 (defcustom frog-menu-after-init-hook '()
   "Frog menu init hook.
 
@@ -364,7 +376,6 @@ gets hidden after the query."
          (pos (avy--process
                candidates
                (avy--style-fn avy-style))))
-    (posframe-hide buffer)
     (cond ((number-or-marker-p pos)
            ;; string
            (with-current-buffer buffer
@@ -408,7 +419,13 @@ RETURN will be the returned value if KEY is pressed."
          (candidates (funcall dhandler buf))
          (qhandler (cdr (assq frog-menu-type
                               frog-menu-query-handler-alist)))
-         (res (funcall qhandler candidates actions buf)))
+         (cuhandler (cdr (assq frog-menu-type
+                               frog-menu-cleanup-handler-alist)))
+         (res nil))
+    (unwind-protect
+        (setq res (funcall qhandler candidates actions buf))
+      (when cuhandler
+        (funcall cuhandler buf)))
     res))