forked from mirrors/gecko-dev
faa5410de1
# ignore-this-changeset Differential Revision: https://phabricator.services.mozilla.com/D162661
129 lines
5 KiB
Python
129 lines
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 partner attribution task into an actual task description.
|
|
"""
|
|
|
|
|
|
import json
|
|
import logging
|
|
from collections import defaultdict
|
|
|
|
from taskgraph.transforms.base import TransformSequence
|
|
|
|
from gecko_taskgraph.util.partners import (
|
|
apply_partner_priority,
|
|
check_if_partners_enabled,
|
|
generate_attribution_code,
|
|
get_partner_config_by_kind,
|
|
)
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
transforms = TransformSequence()
|
|
transforms.add(check_if_partners_enabled)
|
|
transforms.add(apply_partner_priority)
|
|
|
|
|
|
@transforms.add
|
|
def add_command_arguments(config, tasks):
|
|
enabled_partners = config.params.get("release_partners")
|
|
dependencies = {}
|
|
fetches = defaultdict(set)
|
|
attributions = []
|
|
release_artifacts = []
|
|
attribution_config = get_partner_config_by_kind(config, config.kind)
|
|
|
|
for partner_config in attribution_config.get("configs", []):
|
|
# we might only be interested in a subset of all partners, eg for a respin
|
|
if enabled_partners and partner_config["campaign"] not in enabled_partners:
|
|
continue
|
|
attribution_code = generate_attribution_code(
|
|
attribution_config["defaults"], partner_config
|
|
)
|
|
for platform in partner_config["platforms"]:
|
|
stage_platform = platform.replace("-shippable", "")
|
|
for locale in partner_config["locales"]:
|
|
# find the upstream, throw away locales we don't have, somehow. Skip ?
|
|
if locale == "en-US":
|
|
upstream_label = "repackage-signing-{platform}/opt".format(
|
|
platform=platform
|
|
)
|
|
upstream_artifact = "target.installer.exe"
|
|
else:
|
|
upstream_label = (
|
|
"repackage-signing-l10n-{locale}-{platform}/opt".format(
|
|
locale=locale, platform=platform
|
|
)
|
|
)
|
|
upstream_artifact = "{locale}/target.installer.exe".format(
|
|
locale=locale
|
|
)
|
|
if upstream_label not in config.kind_dependencies_tasks:
|
|
raise Exception(f"Can't find upstream task for {platform} {locale}")
|
|
upstream = config.kind_dependencies_tasks[upstream_label]
|
|
|
|
# set the dependencies to just what we need rather than all of l10n
|
|
dependencies.update({upstream.label: upstream.label})
|
|
|
|
fetches[upstream_label].add((upstream_artifact, stage_platform, locale))
|
|
|
|
artifact_part = "{platform}/{locale}/target.installer.exe".format(
|
|
platform=stage_platform, locale=locale
|
|
)
|
|
artifact = (
|
|
"releng/partner/{partner}/{sub_partner}/{artifact_part}".format(
|
|
partner=partner_config["campaign"],
|
|
sub_partner=partner_config["content"],
|
|
artifact_part=artifact_part,
|
|
)
|
|
)
|
|
# config for script
|
|
# TODO - generalise input & output ??
|
|
# add releng/partner prefix via get_artifact_prefix..()
|
|
attributions.append(
|
|
{
|
|
"input": f"/builds/worker/fetches/{artifact_part}",
|
|
"output": f"/builds/worker/artifacts/{artifact}",
|
|
"attribution": attribution_code,
|
|
}
|
|
)
|
|
release_artifacts.append(artifact)
|
|
|
|
# bail-out early if we don't have any attributions to do
|
|
if not attributions:
|
|
return
|
|
|
|
for task in tasks:
|
|
worker = task.get("worker", {})
|
|
worker["chain-of-trust"] = True
|
|
|
|
task.setdefault("dependencies", {}).update(dependencies)
|
|
task.setdefault("fetches", {})
|
|
for upstream_label, upstream_artifacts in fetches.items():
|
|
task["fetches"][upstream_label] = [
|
|
{
|
|
"artifact": upstream_artifact,
|
|
"dest": "{platform}/{locale}".format(
|
|
platform=platform, locale=locale
|
|
),
|
|
"extract": False,
|
|
"verify-hash": True,
|
|
}
|
|
for upstream_artifact, platform, locale in upstream_artifacts
|
|
]
|
|
worker.setdefault("env", {})["ATTRIBUTION_CONFIG"] = json.dumps(
|
|
attributions, sort_keys=True
|
|
)
|
|
worker["artifacts"] = [
|
|
{
|
|
"name": "releng/partner",
|
|
"path": "/builds/worker/artifacts/releng/partner",
|
|
"type": "directory",
|
|
}
|
|
]
|
|
task.setdefault("attributes", {})["release_artifacts"] = release_artifacts
|
|
task["label"] = config.kind
|
|
|
|
yield task
|