1 files changed, 42 insertions, 0 deletions
diff --git a/browser/base/content/aboutRestartRequired.js b/browser/base/content/aboutRestartRequired.js
new file mode 100644
index 0000000000..e29242c383
--- /dev/null
+++ b/browser/base/content/aboutRestartRequired.js
@@ -0,0 +1,42 @@
+/* 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/. */
+
+/* eslint-env mozilla/remote-page */
+
+var AboutRestartRequired = {
+  /* Only do autofocus if we're the toplevel frame; otherwise we
+     don't want to call attention to ourselves!  The key part is
+     that autofocus happens on insertion into the tree, so we
+     can remove the button, add @autofocus, and reinsert the
+     button.
+  */
+  addAutofocus() {
+    if (window.top == window) {
+      var button = document.getElementById("restart");
+      var parent = button.parentNode;
+      button.remove();
+      button.setAttribute("autofocus", "true");
+      parent.insertAdjacentElement("afterbegin", button);
+    }
+  },
+  restart() {
+    Services.startup.quit(
+      Ci.nsIAppStartup.eRestart | Ci.nsIAppStartup.eAttemptQuit
+    );
+  },
+  init() {
+    this.addAutofocus();
+  },
+};
+
+AboutRestartRequired.init();
+
+let restartButton = document.getElementById("restart");
+restartButton.onclick = function() {
+  AboutRestartRequired.restart();
+};
+
+// Dispatch this event so tests can detect that we finished loading the page.
+let event = new CustomEvent("AboutRestartRequiredLoad", { bubbles: true });
+document.dispatchEvent(event);