1 files changed, 80 insertions, 0 deletions
diff --git a/toolkit/components/normandy/lib/ShieldPreferences.jsm b/toolkit/components/normandy/lib/ShieldPreferences.jsm
new file mode 100644
index 0000000000..c29303f3af
--- /dev/null
+++ b/toolkit/components/normandy/lib/ShieldPreferences.jsm
@@ -0,0 +1,80 @@
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+"use strict";
+
+const { XPCOMUtils } = ChromeUtils.import(
+  "resource://gre/modules/XPCOMUtils.jsm"
+);
+
+XPCOMUtils.defineLazyModuleGetters(this, {
+  Services: "resource://gre/modules/Services.jsm",
+  BranchedAddonStudyAction:
+    "resource://normandy/actions/BranchedAddonStudyAction.jsm",
+  AddonStudies: "resource://normandy/lib/AddonStudies.jsm",
+  CleanupManager: "resource://normandy/lib/CleanupManager.jsm",
+  PreferenceExperiments: "resource://normandy/lib/PreferenceExperiments.jsm",
+});
+
+var EXPORTED_SYMBOLS = ["ShieldPreferences"];
+
+const NS_PREFBRANCH_PREFCHANGE_TOPIC_ID = "nsPref:changed"; // from modules/libpref/nsIPrefBranch.idl
+const PREF_OPT_OUT_STUDIES_ENABLED = "app.shield.optoutstudies.enabled";
+
+/**
+ * Handles Shield-specific preferences, including their UI.
+ */
+var ShieldPreferences = {
+  init() {
+    // Watch for changes to the Opt-out pref
+    Services.prefs.addObserver(PREF_OPT_OUT_STUDIES_ENABLED, this);
+
+    CleanupManager.addCleanupHandler(() => {
+      Services.prefs.removeObserver(PREF_OPT_OUT_STUDIES_ENABLED, this);
+    });
+  },
+
+  observe(subject, topic, data) {
+    switch (topic) {
+      case NS_PREFBRANCH_PREFCHANGE_TOPIC_ID:
+        this.observePrefChange(data);
+        break;
+    }
+  },
+
+  async observePrefChange(prefName) {
+    let prefValue;
+    switch (prefName) {
+      // If the opt-out pref changes to be false, disable all current studies.
+      case PREF_OPT_OUT_STUDIES_ENABLED: {
+        prefValue = Services.prefs.getBoolPref(PREF_OPT_OUT_STUDIES_ENABLED);
+        if (!prefValue) {
+          const action = new BranchedAddonStudyAction();
+          const studyPromises = (await AddonStudies.getAll()).map(study => {
+            if (!study.active) {
+              return null;
+            }
+            return action.unenroll(study.recipeId, "general-opt-out");
+          });
+
+          const experimentPromises = (await PreferenceExperiments.getAll()).map(
+            experiment => {
+              if (experiment.expired) {
+                return null;
+              }
+              return PreferenceExperiments.stop(experiment.slug, {
+                reason: "general-opt-out",
+              });
+            }
+          );
+
+          const allPromises = studyPromises
+            .concat(experimentPromises)
+            .map(p => p && p.catch(err => Cu.reportError(err)));
+          await Promise.all(allPromises);
+        }
+        break;
+      }
+    }
+  },
+};