Only replace widgets that were actually blocked.

Instead of hypothetically blocked (which doesn't account
for page context, for instance).

src/js/webrequest.js

@@ -529,6 +529,8 @@ function _isTabAnExtension(tabId) {
 /**
  * Provides the widget replacing content script with list of widgets to replace.
  *
+ * @param {Integer} tab_id the ID of the tab we're replacing widgets in
+ *
  * @returns {Object} dict containing the complete list of widgets
  * as well as a mapping to indicate which ones should be replaced
  */
@@ -544,7 +546,7 @@ let getWidgetBlockList = (function () {
     { key: "allow_once" },
   ];
 
-  return function () {
+  return function (tab_id) {
     // A mapping of individual SocialWidget objects to boolean values that determine
     // whether the content script should replace that tracker's button/widget
     var widgetsToReplace = {};
@@ -561,13 +563,19 @@ let getWidgetBlockList = (function () {
     badger.widgetList.forEach(function (widget) {
       // replace blocked widgets only
       // and only if the widget is not on the 'do not replace' list
-      const replace = !badger.getSettings().getItem('widgetReplacementExceptions').includes(widget.name);
-      const action = badger.storage.getBestAction(widget.domain);
+      let replace = !badger.getSettings().getItem('widgetReplacementExceptions').includes(widget.name);
+
+      if (replace && badger.tabData.hasOwnProperty(tab_id)) {
+        if (badger.tabData[tab_id].origins.hasOwnProperty(widget.domain)) {
+          const action = badger.tabData[tab_id].origins[widget.domain];
+          replace = (
+            action == constants.BLOCK ||
+            action == constants.USER_BLOCK
+          );
+        }
+      }
 
-      widgetsToReplace[widget.name] = replace && (
-        action == constants.BLOCK ||
-        action == constants.USER_BLOCK
-      );
+      widgetsToReplace[widget.name] = replace;
     });
 
     return {
@@ -679,7 +687,7 @@ function dispatcher(request, sender, sendResponse) {
 
   case "checkReplaceButton": {
     if (badger.isPrivacyBadgerEnabled(window.extractHostFromURL(sender.tab.url)) && badger.isWidgetReplacementEnabled()) {
-      let widgetBlockList = getWidgetBlockList();
+      let widgetBlockList = getWidgetBlockList(sender.tab.id);
       sendResponse(widgetBlockList);
     }