From 43a97878ce14b72f0981164f87f2e35e14151312 Mon Sep 17 00:00:00 2001
From: Daniel Baumann <daniel.baumann@progress-linux.org>
Date: Sun, 7 Apr 2024 11:22:09 +0200
Subject: Adding upstream version 110.0.1.

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
---
 .../extensions/parent/ext-normandyAddonStudy.js    | 86 ++++++++++++++++++++++
 1 file changed, 86 insertions(+)
 create mode 100644 browser/components/extensions/parent/ext-normandyAddonStudy.js

(limited to 'browser/components/extensions/parent/ext-normandyAddonStudy.js')

diff --git a/browser/components/extensions/parent/ext-normandyAddonStudy.js b/browser/components/extensions/parent/ext-normandyAddonStudy.js
new file mode 100644
index 0000000000..24bf08c2a3
--- /dev/null
+++ b/browser/components/extensions/parent/ext-normandyAddonStudy.js
@@ -0,0 +1,86 @@
+/* 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 { AddonStudies } = ChromeUtils.import(
+  "resource://normandy/lib/AddonStudies.jsm"
+);
+const { ClientID } = ChromeUtils.importESModule(
+  "resource://gre/modules/ClientID.sys.mjs"
+);
+
+ChromeUtils.defineModuleGetter(
+  this,
+  "AddonManager",
+  "resource://gre/modules/AddonManager.jsm"
+);
+
+this.normandyAddonStudy = class extends ExtensionAPI {
+  getAPI(context) {
+    let { extension } = context;
+
+    return {
+      normandyAddonStudy: {
+        /**
+         * Returns a study object for the current study.
+         *
+         * @returns {Study}
+         */
+        async getStudy() {
+          const studies = await AddonStudies.getAll();
+          return studies.find(study => study.addonId === extension.id);
+        },
+
+        /**
+         * Marks the study as ended and then uninstalls the addon.
+         *
+         * @param {string} reason Why the study is ending
+         */
+        async endStudy(reason) {
+          const study = await this.getStudy();
+
+          // Mark the study as ended
+          await AddonStudies.markAsEnded(study, reason);
+
+          // Uninstall the addon
+          const addon = await AddonManager.getAddonByID(study.addonId);
+          if (addon) {
+            await addon.uninstall();
+          }
+        },
+
+        /**
+         * Returns an object with metadata about the client which may
+         * be required for constructing survey URLs.
+         *
+         * @returns {object}
+         */
+        async getClientMetadata() {
+          return {
+            updateChannel: Services.appinfo.defaultUpdateChannel,
+            fxVersion: Services.appinfo.version,
+            clientID: await ClientID.getClientID(),
+          };
+        },
+
+        onUnenroll: new EventManager({
+          context,
+          name: "normandyAddonStudy.onUnenroll",
+          register: fire => {
+            const listener = async reason => {
+              await fire.async(reason);
+            };
+
+            AddonStudies.addUnenrollListener(extension.id, listener);
+
+            return () => {
+              AddonStudies.removeUnenrollListener(extension.id, listener);
+            };
+          },
+        }).api(),
+      },
+    };
+  }
+};
-- 
cgit v1.2.3