mirror of
https://github.com/mozilla/gecko-dev.git
synced 2025-11-03 09:48:38 +02:00
cb5035f10c
...and along the way, move merge-automation config into the `merge-automation` kind to get rid of the current split-brain situation where it lives in two places unnecessarily. Differential Revision: https://phabricator.services.mozilla.com/D252028
84 lines
2.5 KiB
Python
84 lines
2.5 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 the update generation task into an actual task description.
|
|
"""
|
|
|
|
from taskgraph.transforms.base import TransformSequence
|
|
from taskgraph.util.schema import resolve_keyed_by
|
|
|
|
transforms = TransformSequence()
|
|
|
|
|
|
@transforms.add
|
|
def handle_keyed_by(config, tasks):
|
|
"""Resolve fields that can be keyed by platform, etc."""
|
|
if "merge_config" not in config.params:
|
|
return
|
|
merge_config = config.params["merge_config"]
|
|
fields = [
|
|
"routes",
|
|
"worker.push",
|
|
"scopes",
|
|
"worker-type",
|
|
"worker.l10n-bump-info",
|
|
"worker.lando-repo",
|
|
"worker.matrix-rooms",
|
|
"worker.actions",
|
|
]
|
|
for task in tasks:
|
|
for field in fields:
|
|
resolve_keyed_by(
|
|
task,
|
|
field,
|
|
item_name=task["name"],
|
|
**{
|
|
"project": config.params["project"],
|
|
"release-type": config.params["release_type"],
|
|
"behavior": merge_config["behavior"],
|
|
"level": config.params["level"],
|
|
}
|
|
)
|
|
|
|
yield task
|
|
|
|
|
|
@transforms.add
|
|
def update_labels(config, tasks):
|
|
for task in tasks:
|
|
merge_config = config.params["merge_config"]
|
|
task["label"] = "merge-{}".format(merge_config["behavior"])
|
|
treeherder = task.get("treeherder", {})
|
|
treeherder["symbol"] = "Rel({})".format(merge_config["behavior"])
|
|
task["treeherder"] = treeherder
|
|
yield task
|
|
|
|
|
|
@transforms.add
|
|
def add_payload_config(config, tasks):
|
|
for task in tasks:
|
|
if "merge_config" not in config.params:
|
|
break
|
|
merge_config = config.params["merge_config"]
|
|
worker = task["worker"]
|
|
|
|
assert len(worker["actions"][0].keys()) == 1
|
|
action_name = list(worker["actions"][0].keys())[0]
|
|
|
|
# Override defaults, useful for testing.
|
|
for field in [
|
|
"from-repo",
|
|
"from-branch",
|
|
"to-repo",
|
|
"to-branch",
|
|
"fetch-version-from",
|
|
"lando-repo",
|
|
]:
|
|
if merge_config.get(field):
|
|
worker["actions"][0][action_name][field] = merge_config[field]
|
|
|
|
worker["force-dry-run"] = merge_config["force-dry-run"]
|
|
if merge_config.get("push"):
|
|
worker["push"] = merge_config["push"]
|
|
yield task
|