From 0e2edf6bb60893b7b8edf54539a90c1aeb9e3083 Mon Sep 17 00:00:00 2001
From: mrmr1993
Date: Tue, 2 Jun 2015 16:12:40 +0100
Subject: Move key-only code from FindMode.exit to its keydown handler

---
 content_scripts/vimium_frontend.coffee | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

(limited to 'content_scripts')

diff --git a/content_scripts/vimium_frontend.coffee b/content_scripts/vimium_frontend.coffee
index eb94b6e8..86af6350 100644
--- a/content_scripts/vimium_frontend.coffee
+++ b/content_scripts/vimium_frontend.coffee
@@ -747,9 +747,11 @@ class FindMode extends Mode
         if event.keyCode == keyCodes.backspace || event.keyCode == keyCodes.deleteKey
           HUD.hide()
           @exit()
+          new PostFindMode if findModeQueryHasResults
         else if event.keyCode == keyCodes.enter
           handleEnterForFindMode()
           @exit()
+          new PostFindMode if findModeQueryHasResults
         else if event.keyCode == keyCodes.upArrow
           if rawQuery = FindModeHistory.getQuery @historyIndex + 1
             @historyIndex += 1
@@ -760,14 +762,13 @@ class FindMode extends Mode
           rawQuery = if 0 <= @historyIndex then FindModeHistory.getQuery @historyIndex else @partialQuery
           HUD.showFindMode rawQuery
         else if KeyboardUtils.isEscape event
-          @exit event
+          @exit()
+          handleEscapeForFindMode()
+          new PostFindMode if findModeQueryHasResults
 
   exit: (event) ->
     super()
-    handleEscapeForFindMode() if event?.type == "keydown" and KeyboardUtils.isEscape event
-    handleEscapeForFindMode() if event?.type == "click"
-    if findModeQueryHasResults and event?.type != "click"
-      new PostFindMode
+    handleEscapeForFindMode() if event
 
 window.performFindInPlace = ->
   # Restore the selection.  That way, we're always searching forward from the same place, so we find the right
-- 
cgit v1.2.3