From 7524fa408ce85555751d2f596e856b2203d5cbf9 Mon Sep 17 00:00:00 2001
From: Alex Kontos <MrAlex94@users.noreply.github.com>
Date: Thu, 14 Jul 2022 15:43:17 +0100
Subject: [PATCH]  [WFX-344] Add update check disabling.

---
 browser/components/BrowserGlue.jsm            |  1 +
 browser/components/preferences/main.inc.xhtml |  3 ++
 browser/components/preferences/main.js        | 33 +++++++++++++++++--
 browser/modules/AppUpdater.jsm                | 11 +++++++
 toolkit/mozapps/update/UpdateService.jsm      |  9 +++++
 waterfox/browser/app/profile/000-waterfox.js  |  1 +
 6 files changed, 56 insertions(+), 2 deletions(-)

diff --git a/browser/components/BrowserGlue.jsm b/browser/components/BrowserGlue.jsm
index 0481565accab9..39a1450390c5a 100644
--- a/browser/components/BrowserGlue.jsm
+++ b/browser/components/BrowserGlue.jsm
@@ -1279,6 +1279,7 @@ BrowserGlue.prototype = {
     // check for update if our build is old
     if (
       AppConstants.MOZ_UPDATER &&
+      Services.prefs.getBoolPref("app.update.enabled") &&
       Services.prefs.getBoolPref("app.update.checkInstallTime")
     ) {
       let buildID = Services.appinfo.appBuildID;
diff --git a/browser/components/preferences/main.inc.xhtml b/browser/components/preferences/main.inc.xhtml
index 0deb010bbefdb..3d7e002d91155 100644
--- a/browser/components/preferences/main.inc.xhtml
+++ b/browser/components/preferences/main.inc.xhtml
@@ -620,6 +620,9 @@
       <radio id="manualDesktop"
              value="false"
              data-l10n-id="update-application-check-choose"/>
+      <radio id="disabledDesktop"
+             value="disabled"
+             data-l10n-id="update-application-manual"/>
     </radiogroup>
     <hbox id="updateSettingCrossUserWarningDesc" hidden="true">
       <hbox class="info-icon-container">
diff --git a/browser/components/preferences/main.js b/browser/components/preferences/main.js
index 9bb3e6ec3d5d3..627e63440af24 100644
--- a/browser/components/preferences/main.js
+++ b/browser/components/preferences/main.js
@@ -44,6 +44,9 @@ const ICON_URL_APP =
 // was set by us to a custom handler icon and CSS should not try to override it.
 const APP_ICON_ATTR_NAME = "appHandlerIcon";
 
+// WATERFOX
+const PREF_UPDATE_ENABLED = "app.update.enabled";
+
 Preferences.addAll([
   // Startup
   { id: "browser.startup.page", type: "int" },
@@ -162,6 +165,11 @@ Preferences.addAll([
 
   // Media
   { id: "media.hardwaremediakeys.enabled", type: "bool" },
+
+  // WATERFOX
+  // Enable auto update checking
+  { id: "app.update.enabled", type: "bool" },
+
 ]);
 
 if (AppConstants.HAVE_SHELL_SERVICE) {
@@ -663,6 +671,8 @@ var gMainPane = {
         // Start with no option selected since we are still reading the value
         document.getElementById("autoDesktop").removeAttribute("selected");
         document.getElementById("manualDesktop").removeAttribute("selected");
+        // WATERFOX
+        document.getElementById("disabledDesktop").removeAttribute("selected");
         // Start reading the correct value from the disk
         this.readUpdateAutoPref();
         setEventListener("updateRadioGroup", "command", event => {
@@ -1877,7 +1887,10 @@ var gMainPane = {
 
       radiogroup.disabled = true;
       let enabled = await UpdateUtils.getAppUpdateAutoEnabled();
-      radiogroup.value = enabled;
+      // WATERFOX
+      // If user sets app.update.enabled to false, set value to disabled, else use enabled
+      let manualUpdates = Services.prefs.getBoolPref(PREF_UPDATE_ENABLED, true);
+      radiogroup.value = !manualUpdates ? "disabled" : enabled;
       radiogroup.disabled = false;
 
       this.maybeDisableBackgroundUpdateControls();
@@ -1902,6 +1915,12 @@ var gMainPane = {
       try {
         await UpdateUtils.setAppUpdateAutoEnabled(updateAutoValue);
         await _disableTimeOverPromise;
+        // WATERFOX
+        // Ensure enabled pref is updated based on radiogroup value
+        Services.prefs.setBoolPref(
+          PREF_UPDATE_ENABLED,
+          radiogroup.value != "disabled"
+        );
         radiogroup.disabled = false;
       } catch (error) {
         Cu.reportError(error);
@@ -2111,7 +2130,17 @@ var gMainPane = {
       if (aData != "true" && aData != "false") {
         throw new Error("Invalid preference value for app.update.auto");
       }
-      document.getElementById("updateRadioGroup").value = aData;
+      // WATERFOX
+      // Going from Auto to Disable needs to be correctly reflected here
+      // At this point app.update.enabled has not yet been set, so we
+      // want to set disabled if it is going from true->false and is
+      // currently true
+      let manualUpdates = Services.prefs.getBoolPref(
+        "app.update.enabled",
+        true
+      );
+      document.getElementById("updateRadioGroup").value =
+        manualUpdates && !(aData === "true") ? "disabled" : aData;
       this.maybeDisableBackgroundUpdateControls();
     } else if (aTopic == BACKGROUND_UPDATE_CHANGED_TOPIC) {
       if (!AppConstants.MOZ_UPDATER || !AppConstants.MOZ_UPDATE_AGENT) {
diff --git a/browser/modules/AppUpdater.jsm b/browser/modules/AppUpdater.jsm
index 591b804b8dcef..b14b7ec8f7762 100644
--- a/browser/modules/AppUpdater.jsm
+++ b/browser/modules/AppUpdater.jsm
@@ -65,6 +65,11 @@ class AppUpdater {
       return;
     }
 
+    if (this.updateDisabled) {
+      this._setStatus(AppUpdater.STATUS.NEVER_CHECKED);
+      return;
+    }
+
     if (this.updateDisabledByPolicy) {
       this._setStatus(AppUpdater.STATUS.UPDATE_DISABLED_BY_POLICY);
       return;
@@ -175,6 +180,10 @@ class AppUpdater {
     );
   }
 
+  get updateDisabled() {
+    return !Services.prefs.getBoolPref("app.update.enabled", true);
+  }
+
   // true when updating has been disabled by enterprise policy
   get updateDisabledByPolicy() {
     return Services.policies && !Services.policies.isAllowed("appUpdate");
@@ -463,6 +472,8 @@ class AppUpdater {
     if (!this._status) {
       if (!AppConstants.MOZ_UPDATER || this.updateDisabledByPackage) {
         this._status = AppUpdater.STATUS.NO_UPDATER;
+      } else if (this.updateDisabled) {
+        this._status = AppUpdater.STATUS.NEVER_CHECKED;
       } else if (this.updateDisabledByPolicy) {
         this._status = AppUpdater.STATUS.UPDATE_DISABLED_BY_POLICY;
       } else if (this.isReadyForRestart) {
diff --git a/toolkit/mozapps/update/UpdateService.jsm b/toolkit/mozapps/update/UpdateService.jsm
index a2ceea06cdd69..c1d4c21618a69 100644
--- a/toolkit/mozapps/update/UpdateService.jsm
+++ b/toolkit/mozapps/update/UpdateService.jsm
@@ -3647,6 +3647,15 @@ UpdateService.prototype = {
    * See nsIUpdateService.idl
    */
   get canUsuallyCheckForUpdates() {
+    var enabled = Services.prefs.getBoolPref("app.update.enabled", true);
+    if (!enabled) {
+      LOG(
+        "gCanCheckForUpdates - unable to automatically check for updates, " +
+          "the preference is disabled."
+      );
+      return false;
+    }
+
     if (this.disabledByPolicy) {
       LOG(
         "UpdateService.canUsuallyCheckForUpdates - unable to automatically check " +
diff --git a/waterfox/browser/app/profile/000-waterfox.js b/waterfox/browser/app/profile/000-waterfox.js
index a94514b95e3da..d757fb995b0fa 100644
--- a/waterfox/browser/app/profile/000-waterfox.js
+++ b/waterfox/browser/app/profile/000-waterfox.js
@@ -18,6 +18,7 @@
 pref("accessibility.support.url", "https://www.waterfox.net/support/accessibility-services")
 pref("app.support.baseURL", "https://www.waterfox.net/support/%OS%/");
 pref("app.update.badgeWaitTime", 0);
+pref("app.update.enabled", true);
 pref("app.update.notifyDuringDownload", true);
 pref("app.update.promptWaitTime", 3600);
 pref("app.update.url.override", "", sticky);