Modes; yet more minor tweeks.

1 files changed, 6 insertions, 0 deletions

diff --git a/content_scripts/mode.coffee b/content_scripts/mode.coffee
index b3cca56c..b81a4ede 100644
--- a/content_scripts/mode.coffee
+++ b/content_scripts/mode.coffee
@@ -67,6 +67,8 @@ class Mode
     console.log @count, "create:", @name
 
     @handlers = []
+    @exitHandlers = []
+
     @push
       keydown: @keydown
       keypress: @keypress
@@ -78,9 +80,13 @@ class Mode
   push: (handlers) ->
     @handlers.push handlerStack.push handlers
 
+  onExit: (handler) ->
+    @exitHandlers.push handler
+
   exit: ->
     if @modeIsActive
       console.log @count, "exit:", @name
+      handler() for handler in @exitHandlers
       handlerStack.remove handlerId for handlerId in @handlers
       Mode.modes = Mode.modes.filter (mode) => mode != @
       Mode.updateBadge()