Adding upstream version 110.0.1.upstream/110.0.1upstream

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

1 files changed, 213 insertions, 0 deletions

diff --git a/testing/mozharness/scripts/desktop_partner_repacks.py b/testing/mozharness/scripts/desktop_partner_repacks.py
new file mode 100755
index 0000000000..4f20663c73
--- /dev/null
+++ b/testing/mozharness/scripts/desktop_partner_repacks.py
@@ -0,0 +1,213 @@
+#!/usr/bin/env python
+# ***** BEGIN LICENSE BLOCK *****
+# 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/.
+# ***** END LICENSE BLOCK *****
+"""desktop_partner_repacks.py
+
+This script manages Desktop partner repacks for beta/release builds.
+"""
+import os
+import sys
+
+# load modules from parent dir
+sys.path.insert(1, os.path.dirname(sys.path[0]))
+
+from mozharness.base.log import FATAL
+from mozharness.base.python import VirtualenvMixin
+from mozharness.base.script import BaseScript
+from mozharness.mozilla.automation import AutomationMixin
+from mozharness.mozilla.secrets import SecretsMixin
+
+
+# DesktopPartnerRepacks {{{1
+class DesktopPartnerRepacks(AutomationMixin, BaseScript, VirtualenvMixin, SecretsMixin):
+    """Manages desktop partner repacks"""
+
+    actions = [
+        "get-secrets",
+        "setup",
+        "repack",
+        "summary",
+    ]
+    config_options = [
+        [
+            ["--version", "-v"],
+            {
+                "dest": "version",
+                "help": "Version of Firefox to repack",
+            },
+        ],
+        [
+            ["--build-number", "-n"],
+            {
+                "dest": "build_number",
+                "help": "Build number of Firefox to repack",
+            },
+        ],
+        [
+            ["--platform"],
+            {
+                "dest": "platform",
+                "help": "Platform to repack (e.g. linux64, macosx64, ...)",
+            },
+        ],
+        [
+            ["--partner", "-p"],
+            {
+                "dest": "partner",
+                "help": "Limit repackaging to partners matching this string",
+            },
+        ],
+        [
+            ["--taskid", "-t"],
+            {
+                "dest": "taskIds",
+                "action": "extend",
+                "help": "taskId(s) of upstream tasks for vanilla Firefox artifacts",
+            },
+        ],
+        [
+            ["--limit-locale", "-l"],
+            {
+                "dest": "limitLocales",
+                "action": "append",
+            },
+        ],
+    ]
+
+    def __init__(self):
+        # fxbuild style:
+        buildscript_kwargs = {
+            "all_actions": DesktopPartnerRepacks.actions,
+            "default_actions": DesktopPartnerRepacks.actions,
+            "config": {
+                "log_name": "partner-repacks",
+                "hashType": "sha512",
+                "workdir": "partner-repacks",
+            },
+        }
+        #
+
+        BaseScript.__init__(
+            self, config_options=self.config_options, **buildscript_kwargs
+        )
+
+    def _pre_config_lock(self, rw_config):
+        if os.getenv("REPACK_MANIFESTS_URL"):
+            self.info(
+                "Overriding repack_manifests_url to %s"
+                % os.getenv("REPACK_MANIFESTS_URL")
+            )
+            self.config["repack_manifests_url"] = os.getenv("REPACK_MANIFESTS_URL")
+        if os.getenv("UPSTREAM_TASKIDS"):
+            self.info("Overriding taskIds with %s" % os.getenv("UPSTREAM_TASKIDS"))
+            self.config["taskIds"] = os.getenv("UPSTREAM_TASKIDS").split()
+
+        if "version" not in self.config:
+            self.fatal("Version (-v) not supplied.")
+        if "build_number" not in self.config:
+            self.fatal("Build number (-n) not supplied.")
+        if "repo_file" not in self.config:
+            self.fatal("repo_file not supplied.")
+        if "repack_manifests_url" not in self.config:
+            self.fatal(
+                "repack_manifests_url not supplied in config or via REPACK_MANIFESTS_URL"
+            )
+        if "taskIds" not in self.config:
+            self.fatal("Need upstream taskIds from command line or in UPSTREAM_TASKIDS")
+
+    def query_abs_dirs(self):
+        if self.abs_dirs:
+            return self.abs_dirs
+        abs_dirs = super(DesktopPartnerRepacks, self).query_abs_dirs()
+        for directory in abs_dirs:
+            value = abs_dirs[directory]
+            abs_dirs[directory] = value
+        dirs = {}
+        dirs["abs_repo_dir"] = os.path.join(abs_dirs["abs_work_dir"], ".repo")
+        dirs["abs_partners_dir"] = os.path.join(abs_dirs["abs_work_dir"], "partners")
+        for key in dirs.keys():
+            if key not in abs_dirs:
+                abs_dirs[key] = dirs[key]
+        self.abs_dirs = abs_dirs
+        return self.abs_dirs
+
+    # Actions {{{
+    def _repo_cleanup(self):
+        self.rmtree(self.query_abs_dirs()["abs_repo_dir"])
+        self.rmtree(self.query_abs_dirs()["abs_partners_dir"])
+
+    def _repo_init(self, repo):
+        partial_env = {
+            "GIT_SSH_COMMAND": "ssh -oIdentityFile={}".format(self.config["ssh_key"])
+        }
+        status = self.run_command(
+            [
+                repo,
+                "init",
+                "--no-repo-verify",
+                "-u",
+                self.config["repack_manifests_url"],
+            ],
+            cwd=self.query_abs_dirs()["abs_work_dir"],
+            partial_env=partial_env,
+        )
+        if status:
+            return status
+        return self.run_command(
+            [repo, "sync", "--current-branch", "--no-tags"],
+            cwd=self.query_abs_dirs()["abs_work_dir"],
+            partial_env=partial_env,
+        )
+
+    def setup(self):
+        """setup step"""
+        repo = self.download_file(
+            self.config["repo_file"],
+            file_name="repo",
+            parent_dir=self.query_abs_dirs()["abs_work_dir"],
+            error_level=FATAL,
+        )
+        if not os.path.exists(repo):
+            self.fatal("Unable to download repo tool.")
+        self.chmod(repo, 0o755)
+        self.retry(
+            self._repo_init,
+            args=(repo,),
+            error_level=FATAL,
+            cleanup=self._repo_cleanup(),
+            good_statuses=[0],
+            sleeptime=5,
+        )
+
+    def repack(self):
+        """creates the repacks"""
+        repack_cmd = [
+            "./mach",
+            "python",
+            "python/mozrelease/mozrelease/partner_repack.py",
+            "-v",
+            self.config["version"],
+            "-n",
+            str(self.config["build_number"]),
+        ]
+        if self.config.get("platform"):
+            repack_cmd.extend(["--platform", self.config["platform"]])
+        if self.config.get("partner"):
+            repack_cmd.extend(["--partner", self.config["partner"]])
+        if self.config.get("taskIds"):
+            for taskId in self.config["taskIds"]:
+                repack_cmd.extend(["--taskid", taskId])
+        if self.config.get("limitLocales"):
+            for locale in self.config["limitLocales"]:
+                repack_cmd.extend(["--limit-locale", locale])
+
+        self.run_command(repack_cmd, cwd=os.environ["GECKO_PATH"], halt_on_failure=True)
+
+
+# main {{{
+if __name__ == "__main__":
+    partner_repacks = DesktopPartnerRepacks()
+    partner_repacks.run_and_exit()