1 files changed, 77 insertions, 0 deletions
diff --git a/toolkit/modules/ResetProfile.jsm b/toolkit/modules/ResetProfile.jsm
new file mode 100644
index 0000000000..98079fc17a
--- /dev/null
+++ b/toolkit/modules/ResetProfile.jsm
@@ -0,0 +1,77 @@
+/* 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";
+
+var EXPORTED_SYMBOLS = ["ResetProfile"];
+
+const { Services } = ChromeUtils.import("resource://gre/modules/Services.jsm");
+const { AppConstants } = ChromeUtils.import(
+  "resource://gre/modules/AppConstants.jsm"
+);
+
+const MOZ_APP_NAME = AppConstants.MOZ_APP_NAME;
+const MOZ_BUILD_APP = AppConstants.MOZ_BUILD_APP;
+
+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.
+    let migrator =
+      "@mozilla.org/profile/migrator;1?app=" +
+      MOZ_BUILD_APP +
+      "&type=" +
+      MOZ_APP_NAME;
+    if (!(migrator in Cc)) {
+      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.
+   */
+  openConfirmationDialog(window) {
+    // Prompt the user to confirm.
+    let params = {
+      reset: false,
+    };
+    window.browsingContext.topChromeWindow.openDialog(
+      "chrome://global/content/resetProfile.xhtml",
+      null,
+      "modal,centerscreen,titlebar",
+      params
+    );
+    if (!params.reset) {
+      return;
+    }
+
+    // Set the reset profile environment variable.
+    let env = Cc["@mozilla.org/process/environment;1"].getService(
+      Ci.nsIEnvironment
+    );
+    env.set("MOZ_RESET_PROFILE_RESTART", "1");
+
+    Services.startup.quit(
+      Ci.nsIAppStartup.eForceQuit | Ci.nsIAppStartup.eRestart
+    );
+  },
+};