1 files changed, 123 insertions, 0 deletions
diff --git a/toolkit/components/normandy/lib/NormandyAddonManager.jsm b/toolkit/components/normandy/lib/NormandyAddonManager.jsm
new file mode 100644
index 0000000000..965d721008
--- /dev/null
+++ b/toolkit/components/normandy/lib/NormandyAddonManager.jsm
@@ -0,0 +1,123 @@
+/* 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, {
+  AddonManager: "resource://gre/modules/AddonManager.jsm",
+  PromiseUtils: "resource://gre/modules/PromiseUtils.jsm",
+});
+
+var EXPORTED_SYMBOLS = ["NormandyAddonManager"];
+
+const NormandyAddonManager = {
+  async downloadAndInstall({
+    createError,
+    extensionDetails,
+    applyNormandyChanges,
+    undoNormandyChanges,
+    onInstallStarted,
+    reportError,
+  }) {
+    const {
+      extension_id,
+      hash,
+      hash_algorithm,
+      version,
+      xpi,
+    } = extensionDetails;
+
+    const downloadDeferred = PromiseUtils.defer();
+    const installDeferred = PromiseUtils.defer();
+
+    const install = await AddonManager.getInstallForURL(xpi, {
+      hash: `${hash_algorithm}:${hash}`,
+      telemetryInfo: { source: "internal" },
+    });
+
+    const listener = {
+      onInstallStarted(cbInstall) {
+        const versionMatches = cbInstall.addon.version === version;
+        const idMatches = cbInstall.addon.id === extension_id;
+
+        if (!versionMatches || !idMatches) {
+          installDeferred.reject(createError("metadata-mismatch"));
+          return false; // cancel the installation, server metadata does not match downloaded add-on
+        }
+
+        if (onInstallStarted) {
+          return onInstallStarted(cbInstall, installDeferred);
+        }
+
+        return true;
+      },
+
+      onDownloadFailed() {
+        downloadDeferred.reject(
+          createError("download-failure", {
+            detail: AddonManager.errorToString(install.error),
+          })
+        );
+      },
+
+      onDownloadEnded() {
+        downloadDeferred.resolve();
+        return false; // temporarily pause installation for Normandy bookkeeping
+      },
+
+      onInstallFailed() {
+        installDeferred.reject(
+          createError("install-failure", {
+            detail: AddonManager.errorToString(install.error),
+          })
+        );
+      },
+
+      onInstallEnded() {
+        installDeferred.resolve();
+      },
+    };
+
+    install.addListener(listener);
+
+    // Download the add-on
+    try {
+      install.install();
+      await downloadDeferred.promise;
+    } catch (err) {
+      reportError(err);
+      install.removeListener(listener);
+      throw err;
+    }
+
+    // Complete any book-keeping
+    try {
+      await applyNormandyChanges(install);
+    } catch (err) {
+      reportError(err);
+      install.removeListener(listener);
+      install.cancel();
+      throw err;
+    }
+
+    // Finish paused installation
+    try {
+      install.install();
+      await installDeferred.promise;
+    } catch (err) {
+      reportError(err);
+      install.removeListener(listener);
+      await undoNormandyChanges();
+      throw err;
+    }
+
+    install.removeListener(listener);
+
+    return [install.addon.id, install.addon.version];
+  },
+};