120 lines
4.2 KiB
Python
120 lines
4.2 KiB
Python
# 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/.
|
|
"""
|
|
Transform mac notarization tasks into actual task descriptions.
|
|
"""
|
|
|
|
from typing import Optional
|
|
|
|
from taskgraph.transforms.base import TransformSequence
|
|
from taskgraph.util.dependencies import get_primary_dependency
|
|
from taskgraph.util.schema import Schema
|
|
from taskgraph.util.treeherder import inherit_treeherder_from_dep, join_symbol
|
|
|
|
from gecko_taskgraph.transforms.task import TaskDescriptionSchema
|
|
|
|
|
|
class MacNotarizationDescriptionSchema(Schema, kw_only=True):
|
|
label: Optional[str] = None
|
|
treeherder: TaskDescriptionSchema.__annotations__["treeherder"] = None
|
|
shipping_phase: TaskDescriptionSchema.__annotations__["shipping_phase"] = None
|
|
worker: TaskDescriptionSchema.__annotations__["worker"] = None
|
|
worker_type: TaskDescriptionSchema.__annotations__["worker_type"] = None
|
|
task_from: TaskDescriptionSchema.__annotations__["task_from"] = None
|
|
attributes: TaskDescriptionSchema.__annotations__["attributes"] = None
|
|
dependencies: TaskDescriptionSchema.__annotations__["dependencies"] = None
|
|
run_on_repo_type: TaskDescriptionSchema.__annotations__["run_on_repo_type"] = None
|
|
|
|
|
|
transforms = TransformSequence()
|
|
|
|
|
|
@transforms.add
|
|
def remove_name(config, jobs):
|
|
for job in jobs:
|
|
if "name" in job:
|
|
del job["name"]
|
|
yield job
|
|
|
|
|
|
transforms.add_validate(MacNotarizationDescriptionSchema)
|
|
|
|
|
|
@transforms.add
|
|
def make_notarization_description(config, jobs):
|
|
for job in jobs:
|
|
dep_job = get_primary_dependency(config, job)
|
|
assert dep_job
|
|
|
|
dependencies = {dep_job.kind: dep_job.label}
|
|
build_platform = dep_job.attributes.get("build_platform")
|
|
|
|
job["worker"]["signing-type"] = "release-apple-notarization"
|
|
|
|
task = {
|
|
"label": job["label"],
|
|
"worker-type": job["worker-type"],
|
|
"worker": job["worker"],
|
|
"dependencies": dependencies,
|
|
"attributes": job.get("attributes", {}),
|
|
"run-on-repo-type": job.get("run-on-repo-type", ["git", "hg"]),
|
|
}
|
|
|
|
if config.kind == "openh264-notarization":
|
|
_configure_openh264(task, job, dep_job, build_platform)
|
|
elif config.kind == "geckodriver-mac-notarization":
|
|
_configure_geckodriver(task, job, dep_job, build_platform)
|
|
else:
|
|
raise Exception(f"Unsupported kind for mac_notarization: {config.kind}")
|
|
|
|
yield task
|
|
|
|
|
|
def _configure_openh264(task, job, dep_job, build_platform):
|
|
version = task["attributes"].get("openh264_version")
|
|
if not version:
|
|
raise Exception(f"openh264_version attribute missing from {dep_job.label}")
|
|
|
|
task["description"] = (
|
|
f"Mac notarization - OpenH264 plugin for build '{build_platform}'"
|
|
)
|
|
|
|
dep_th = dep_job.task.get("extra", {}).get("treeherder", {})
|
|
treeherder = inherit_treeherder_from_dep(job, dep_job)
|
|
treeherder.setdefault(
|
|
"symbol",
|
|
join_symbol(dep_th.get("groupSymbol", "?"), "BN"),
|
|
)
|
|
task["treeherder"] = treeherder
|
|
|
|
task["worker"]["upstream-artifacts"] = [
|
|
{
|
|
"taskId": {"task-reference": "<openh264-signing>"},
|
|
"taskType": "signing",
|
|
"paths": [
|
|
f"private/openh264/openh264-v{version}-{build_platform}.zip",
|
|
],
|
|
"formats": ["apple_notarization_openh264_plugin"],
|
|
}
|
|
]
|
|
|
|
task["run-on-projects"] = dep_job.attributes.get("run_on_projects")
|
|
|
|
|
|
def _configure_geckodriver(task, job, dep_job, build_platform):
|
|
treeherder = job.get("treeherder", {})
|
|
dep_treeherder = dep_job.task.get("extra", {}).get("treeherder", {})
|
|
treeherder.setdefault(
|
|
"platform", dep_job.task.get("extra", {}).get("treeherder-platform")
|
|
)
|
|
treeherder.setdefault("tier", dep_treeherder.get("tier", 1))
|
|
treeherder.setdefault("kind", "build")
|
|
task["treeherder"] = treeherder
|
|
|
|
task["description"] = f"Mac notarization - Geckodriver for build '{build_platform}'"
|
|
|
|
platform = build_platform.rsplit("-", 1)[0]
|
|
task["run-on-projects"] = ["mozilla-central"]
|
|
task["index"] = {"product": "geckodriver", "job-name": platform}
|