Fixed tabState reseting bug

Author: gsingh93

src/background/BackgroundScript.ts

@@ -4,9 +4,11 @@
 
 module BackgroundScript {
     var tabStates;
+
     export function getTabStateManager() {
         return tabStates;
     }
+
     function init() {
         tabStates = new TabStateManager();
         addTabStateListeners(tabStates);
@@ -32,10 +34,9 @@ module BackgroundScript {
             var id = sender.tab.id;
             // Reset the tab state whenever it is loaded/reloaded
             if (request.event == "loaded") {
-                if (tabStates.exists(id)) {
+                if (!tabStates.exists(id)) { // New tab
                     tabStates.resetState(id);
-                } else {
-                    // Don't change the query, only reset searching
+                } else { // Same tab, refresh or new URL
                     tabStates.set(id, "searching", false);
                 }
             }

src/popup/popup.ts

@@ -8,7 +8,8 @@ module Popup {
         var tabStates = BackgroundInterface.getTabStateManager();
 
         // In most cases the map entry will already be initialized. However, there
-        // may be a few edge cases where we need to initialize it ourselves.
+        // may be a few edge cases where we need to initialize it ourselves,
+        // like if we enable/reload the extension
         if (!tabStates.exists(id)) {
             Utils.log("ID doesn't exist. Initializing entry.")
             tabStates.resetState(id);