From b0f56ca439af45b62b23efd8c19c3838945f21f4 Mon Sep 17 00:00:00 2001
From: Stephen Blott
Date: Tue, 6 Jan 2015 10:52:13 +0000
Subject: Modes; yet more minor tweeks.

---
 content_scripts/mode.coffee | 6 ++++++
 1 file changed, 6 insertions(+)

(limited to 'content_scripts/mode.coffee')

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()
-- 
cgit v1.2.3