Revision control
Copy as Markdown
# 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
from datetime import datetime
from voluptuous import Required
from taskgraph.util.schema import taskref_or_string
from taskgraph.transforms.task import payload_builder
@payload_builder(
"scriptworker-signing",
schema={
Required("max-run-time"): int,
Required("cert"): str,
Required("upstream-artifacts"): [{
Required("taskId"): taskref_or_string,
Required("taskType"): str,
Required("paths"): [str],
Required("formats"): [str],
}]
}
)
def build_scriptworker_signing_payload(config, task, task_def):
worker = task["worker"]
task_def["tags"]["worker-implementation"] = "scriptworker"
task_def["payload"] = {
"maxRunTime": worker["max-run-time"],
"upstreamArtifacts": worker["upstream-artifacts"],
}
formats = set()
for artifacts in worker["upstream-artifacts"]:
formats.update(artifacts["formats"])
scope_prefix = config.graph_config["scriptworker"]["scope-prefix"]
task_def["scopes"].append(
"{}:signing:cert:{}".format(scope_prefix, worker["cert"])
)
task_def["scopes"].extend([
f"{scope_prefix}:signing:format:{signing_format}" for signing_format in sorted(formats)
])
@payload_builder(
"scriptworker-beetmover",
schema={
Required("action"): str,
Required("bucket"): str,
Required("max-run-time"): int,
Required("version"): str,
Required("app-name"): str,
Required("upstream-artifacts"): [{
Required("taskId"): taskref_or_string,
Required("taskType"): str,
Required("paths"): [str],
}],
Required("artifact-map"): [{
Required("task-id"): taskref_or_string,
Required("locale"): str,
Required("paths"): {str: dict},
}],
}
)
def build_scriptworker_beetmover_payload(config, task, task_def):
worker = task["worker"]
task_def["tags"]["worker-implementation"] = "scriptworker"
task_def["payload"] = {
"maxRunTime": worker["max-run-time"],
"upstreamArtifacts": worker["upstream-artifacts"],
"artifactMap": [{
"taskId": entry["task-id"],
"locale": entry["locale"],
"paths": entry["paths"],
} for entry in worker["artifact-map"]],
"version": worker["version"],
"releaseProperties": {
"appName": worker["app-name"],
}
}
if worker["action"] != "push-to-maven":
task_def["payload"]["upload_date"] = int(datetime.now().timestamp())
task_def["payload"]["releaseProperties"].update({
"appVersion": worker["version"],
"branch": config.params["head_ref"],
"buildid": config.params["moz_build_date"],
"hashType": "sha512",
"platform": task["attributes"]["build-type"],
})
for artifact in task_def["payload"]["upstreamArtifacts"]:
artifact["locale"] = "multi"
for map_ in task_def["payload"]["artifactMap"]:
for path_config in map_["paths"].values():
path_config["checksums_path"] = ""
scope_prefix = config.graph_config["scriptworker"]["scope-prefix"]
task_def["scopes"].append(
f"{scope_prefix}:beetmover:bucket:{worker['bucket']}"
)
task_def["scopes"].append(
f"{scope_prefix}:beetmover:action:{worker['action']}"
)