Tie staging environment to strict blocking policies

core/main.js

@@ -41,7 +41,8 @@ main._initializeOptions = function () {
             options = optionDefaults; // Restore option defaults.
         }
 
-        if (options.enforceStaging === true) {
+        if (options.enforceStaging === true ||
+            options.blockMissing === true) {
 
             // Improve coverage at the cost of website stability.
             stateManager.setExtensionEnvironment('staging');

core/messenger.js

@@ -30,6 +30,14 @@ messenger._handleMessageReceived = function (message, sender, sendResponse) {
     topic = message.topic;
     value = message.value;
 
+    if (topic === 'extension:set-environment') {
+
+        stateManager.setExtensionEnvironment(value);
+        sendResponse({value});
+
+        return MessageResponse.SYNCHRONOUS;
+    }
+
     if (topic === 'tab:fetch-injections') {
 
         sendResponse({'value': stateManager.tabs[value].injections});

pages/options/options.js

@@ -205,11 +205,23 @@ options._onOptionChanged = function ({target}) {
 
     if (optionKey === Setting.BLOCK_MISSING) {
 
+        let preferredEnvironment = 'stable';
+
         if (optionValue === true) {
+
             options._renderBlockMissingNotice();
+            preferredEnvironment = 'staging';
+
         } else {
             options._hideBlockMissingNotice();
         }
+
+        let message = {
+            'topic': 'extension:set-environment',
+            'value': preferredEnvironment
+        };
+
+        chrome.runtime.sendMessage(message);
     }
 
     if (optionKey === Setting.DISABLE_PREFETCH) {