1 files changed, 110 insertions, 0 deletions
diff --git a/toolkit/modules/ResetProfile.sys.mjs b/toolkit/modules/ResetProfile.sys.mjs
new file mode 100644
index 0000000000..01a85549cf
--- /dev/null
+++ b/toolkit/modules/ResetProfile.sys.mjs
@@ -0,0 +1,110 @@
+/* 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/. */
+
+import { AppConstants } from "resource://gre/modules/AppConstants.sys.mjs";
+import { XPCOMUtils } from "resource://gre/modules/XPCOMUtils.sys.mjs";
+
+const lazy = {};
+
+XPCOMUtils.defineLazyGetter(lazy, "MigrationUtils", () => {
+  // MigrationUtils is currently only available in browser builds.
+  if (AppConstants.MOZ_BUILD_APP != "browser") {
+    return undefined;
+  }
+
+  try {
+    let { MigrationUtils } = ChromeUtils.importESModule(
+      "resource:///modules/MigrationUtils.sys.mjs"
+    );
+    return MigrationUtils;
+  } catch (e) {
+    console.error(`Unable to load MigrationUtils.sys.mjs: ${e}`);
+  }
+  return undefined;
+});
+
+const MOZ_APP_NAME = AppConstants.MOZ_APP_NAME;
+
+export var ResetProfile = {
+  /**
+   * Check if reset is supported for the currently running profile.
+   *
+   * @return boolean whether reset is supported.
+   */
+  resetSupported() {
+    if (Services.policies && !Services.policies.isAllowed("profileRefresh")) {
+      return false;
+    }
+
+    // Reset is only supported if the self-migrator used for reset exists.
+    if (
+      !lazy.MigrationUtils ||
+      !lazy.MigrationUtils.migratorExists(MOZ_APP_NAME)
+    ) {
+      return false;
+    }
+
+    // We also need to be using a profile the profile manager knows about.
+    let profileService = Cc[
+      "@mozilla.org/toolkit/profile-service;1"
+    ].getService(Ci.nsIToolkitProfileService);
+    let currentProfileDir = Services.dirsvc.get("ProfD", Ci.nsIFile);
+    for (let profile of profileService.profiles) {
+      if (profile.rootDir && profile.rootDir.equals(currentProfileDir)) {
+        return true;
+      }
+    }
+    return false;
+  },
+
+  /**
+   * Ask the user if they wish to restart the application to reset the profile.
+   */
+  async openConfirmationDialog(window) {
+    let win = window;
+    // If we are, for instance, on an about page, get the chrome window to
+    // access its gDialogBox.
+    if (win.docShell.chromeEventHandler) {
+      win = win.browsingContext?.topChromeWindow;
+    }
+
+    let params = {
+      learnMore: false,
+      reset: false,
+    };
+
+    if (win.gDialogBox) {
+      await win.gDialogBox.open(
+        "chrome://global/content/resetProfile.xhtml",
+        params
+      );
+    } else {
+      win.openDialog(
+        "chrome://global/content/resetProfile.xhtml",
+        null,
+        "modal,centerscreen,titlebar",
+        params
+      );
+    }
+
+    if (params.learnMore) {
+      win.openTrustedLinkIn(
+        "https://support.mozilla.org/kb/refresh-firefox-reset-add-ons-and-settings",
+        "tab"
+      );
+      return;
+    }
+
+    if (!params.reset) {
+      return;
+    }
+
+    // Set the reset profile environment variable.
+    Services.env.set("MOZ_RESET_PROFILE_RESTART", "1");
+
+    Services.startup.quit(
+      Ci.nsIAppStartup.eForceQuit | Ci.nsIAppStartup.eRestart
+    );
+  },
+};