taskcluster/android_taskgraph/worker_types.py

   1 # This Source Code Form is subject to the terms of the Mozilla Public
   2 # License, v. 2.0. If a copy of the MPL was not distributed with this
   3 # file, You can obtain one at http://mozilla.org/MPL/2.0/.
   4
   5
   6 from gecko_taskgraph.transforms.task import payload_builder
   7 from taskgraph.util.schema import taskref_or_string
   8 from voluptuous import Any, Optional, Required
   9
  10
  11 @payload_builder(
  12     "scriptworker-beetmover",
  13     schema={
  14         Required("action"): str,
  15         Required("version"): str,
  16         Required("artifact-map"): [
  17             {
  18                 Required("paths"): {
  19                     Any(str): {
  20                         Required("destinations"): [str],
  21                     },
  22                 },
  23                 Required("taskId"): taskref_or_string,
  24             }
  25         ],
  26         Required("beetmover-application-name"): str,
  27         Required("bucket"): str,
  28         Required("upstream-artifacts"): [
  29             {
  30                 Required("taskId"): taskref_or_string,
  31                 Required("taskType"): str,
  32                 Required("paths"): [str],
  33             }
  34         ],
  35     },
  36 )
  37 def build_scriptworker_beetmover_payload(config, task, task_def):
  38     worker = task["worker"]
  39
  40     task_def["tags"]["worker-implementation"] = "scriptworker"
  41
  42     # Needed by beetmover-scriptworker
  43     for map_ in worker["artifact-map"]:
  44         map_["locale"] = "en-US"
  45         for path_config in map_["paths"].values():
  46             path_config["checksums_path"] = ""
  47
  48     task_def["payload"] = {
  49         "artifactMap": worker["artifact-map"],
  50         "releaseProperties": {"appName": worker.pop("beetmover-application-name")},
  51         "upstreamArtifacts": worker["upstream-artifacts"],
  52         "version": worker["version"],
  53     }
  54
  55     scope_prefix = config.graph_config["scriptworker"]["scope-prefix"]
  56     task_def["scopes"].extend(
  57         [
  58             "{}:beetmover:action:{}".format(scope_prefix, worker["action"]),
  59             "{}:beetmover:bucket:{}".format(scope_prefix, worker["bucket"]),
  60         ]
  61     )
  62
  63
  64 @payload_builder(
  65     "scriptworker-pushapk",
  66     schema={
  67         Required("upstream-artifacts"): [
  68             {
  69                 Required("taskId"): taskref_or_string,
  70                 Required("taskType"): str,
  71                 Required("paths"): [str],
  72             }
  73         ],
  74         Required("certificate-alias"): str,
  75         Required("channel"): str,
  76         Required("commit"): bool,
  77         Required("product"): str,
  78         Required("dep"): bool,
  79     },
  80 )
  81 def build_push_apk_payload(config, task, task_def):
  82     worker = task["worker"]
  83
  84     task_def["tags"]["worker-implementation"] = "scriptworker"
  85
  86     task_def["payload"] = {
  87         "certificate_alias": worker["certificate-alias"],
  88         "channel": worker["channel"],
  89         "commit": worker["commit"],
  90         "upstreamArtifacts": worker["upstream-artifacts"],
  91     }
  92
  93     scope_prefix = config.graph_config["scriptworker"]["scope-prefix"]
  94     task_def["scopes"].append(
  95         "{}:googleplay:product:{}{}".format(
  96             scope_prefix, worker["product"], ":dep" if worker["dep"] else ""
  97         )
  98     )
  99
 100
 101 @payload_builder(
 102     "scriptworker-shipit",
 103     schema={
 104         Required("release-name"): str,
 105     },
 106 )
 107 def build_shipit_payload(config, task, task_def):
 108     worker = task["worker"]
 109
 110     task_def["tags"]["worker-implementation"] = "scriptworker"
 111
 112     task_def["payload"] = {"release_name": worker["release-name"]}
 113
 114
 115 @payload_builder(
 116     "scriptworker-tree",
 117     schema={
 118         Optional("upstream-artifacts"): [
 119             {
 120                 Optional("taskId"): taskref_or_string,
 121                 Optional("taskType"): str,
 122                 Optional("paths"): [str],
 123             }
 124         ],
 125         Required("bump"): bool,
 126         Optional("bump-files"): [str],
 127         Optional("push"): bool,
 128         Optional("branch"): str,
 129     },
 130 )
 131 def build_version_bump_payload(config, task, task_def):
 132     worker = task["worker"]
 133     task_def["tags"]["worker-implementation"] = "scriptworker"
 134
 135     scopes = task_def.setdefault("scopes", [])
 136     scope_prefix = f"project:mobile:{config.params['project']}:treescript:action"
 137     task_def["payload"] = {}
 138
 139     if worker["bump"]:
 140         if not worker["bump-files"]:
 141             raise Exception("Version Bump requested without bump-files")
 142
 143         bump_info = {}
 144         bump_info["next_version"] = config.params["next_version"]
 145         bump_info["files"] = worker["bump-files"]
 146         task_def["payload"]["version_bump_info"] = bump_info
 147         scopes.append(f"{scope_prefix}:version_bump")
 148
 149     if worker["push"]:
 150         task_def["payload"]["push"] = True
 151
 152     if worker.get("force-dry-run"):
 153         task_def["payload"]["dry_run"] = True
 154
 155     if worker.get("branch"):
 156         task_def["payload"]["branch"] = worker["branch"]