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 ++++++
 content_scripts/mode_find.coffee   | 3 ++-
 content_scripts/mode_insert.coffee | 4 ++--
 3 files changed, 10 insertions(+), 3 deletions(-)

(limited to 'content_scripts')

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()
diff --git a/content_scripts/mode_find.coffee b/content_scripts/mode_find.coffee
index 0937c510..85cdc6c5 100644
--- a/content_scripts/mode_find.coffee
+++ b/content_scripts/mode_find.coffee
@@ -10,7 +10,8 @@ class PostFindMode extends InsertModeBlocker
   constructor: (findModeAnchorNode) ->
     element = document.activeElement
 
-    super PostFindMode, element, {name: "post-find"}
+    super element,
+      name: "post-find"
 
     return @exit() unless element and findModeAnchorNode
 
diff --git a/content_scripts/mode_insert.coffee b/content_scripts/mode_insert.coffee
index 2b0ebb45..24442928 100644
--- a/content_scripts/mode_insert.coffee
+++ b/content_scripts/mode_insert.coffee
@@ -87,9 +87,9 @@ class InsertModeTrigger extends Mode
 # mode on focusable elements.
 # If @element is provided, then don't suppress focus events, and suppress keydown events only on @element.
 class InsertModeBlocker extends SingletonMode
-  constructor: (singleton=InsertModeBlocker, @element=null, options={}) ->
+  constructor: (@element=null, options={}) ->
     options.name ||= "insert-blocker"
-    super singleton, options
+    super InsertModeBlocker, options
 
     unless @element?
       @push
-- 
cgit v1.2.3