From d4c43d8f9095325b41544ad7811cc131c1b186f1 Mon Sep 17 00:00:00 2001
From: Stephen Blott
Date: Fri, 2 Jan 2015 14:33:18 +0000
Subject: Modes; show keyqueue in badge.

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

(limited to 'content_scripts')

diff --git a/content_scripts/vimium_frontend.coffee b/content_scripts/vimium_frontend.coffee
index 7ce3e988..fb6199bf 100644
--- a/content_scripts/vimium_frontend.coffee
+++ b/content_scripts/vimium_frontend.coffee
@@ -125,7 +125,7 @@ initializePreDomReady = ->
     # Overriding updateBadgeForMode() from Mode.updateBadgeForMode().
     updateBadgeForMode: (badge) ->
       handlerStack.alwaysPropagate =>
-        badge.badge ||= @badge
+        badge.badge ||= if keyQueue then keyQueue else @badge
         badge.badge = "" unless isEnabledForUrl
 
   # Initialize the scroller. The scroller install a key handler, and this is next on the handler stack,
@@ -165,7 +165,9 @@ initializePreDomReady = ->
       Mode.updateBadge()
       return { enabled: isEnabledForUrl, passKeys: passKeys }
     setState: setState
-    currentKeyQueue: (request) -> passKeysMode.setState request
+    currentKeyQueue: (request) ->
+      keyQueue = request.keyQueue
+      passKeysMode.setState request
 
   chrome.runtime.onMessage.addListener (request, sender, sendResponse) ->
     # In the options page, we will receive requests from both content and background scripts. ignore those
-- 
cgit v1.2.3