Bastien Orivel
aaa1869325
The android_taskgraph version was a slightly simpler version of the gecko_taskgraph one (missing some resolve_keyed_by, not_for_locales, ...) but none of the difference matter as they're optional extra features that the android version doesn't use. In the end, the only change I made was that check for `en-US` to bypass `all_locales: false`, since android uses `multi` as the sentinel value for that. Differential Revision: https://phabricator.services.mozilla.com/D288898
53 lines
2.0 KiB
Python
53 lines
2.0 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/.
|
|
|
|
from gecko_taskgraph.util.scriptworker import generate_beetmover_upstream_artifacts
|
|
from taskgraph.transforms.base import TransformSequence
|
|
from taskgraph.util.dependencies import get_dependencies
|
|
|
|
transforms = TransformSequence()
|
|
|
|
|
|
def _get_task_type(dep_kind):
|
|
if dep_kind.startswith("build-"):
|
|
return "build"
|
|
elif dep_kind.startswith("signing-"):
|
|
return "signing"
|
|
return dep_kind
|
|
|
|
|
|
@transforms.add
|
|
def build_upstream_artifacts(config, tasks):
|
|
for task in tasks:
|
|
worker_definition = {
|
|
"upstream-artifacts": [],
|
|
}
|
|
if "artifact_map" in task["attributes"]:
|
|
# Beetmover-apk tasks use declarative artifacts.
|
|
locale = task["attributes"].get("locale")
|
|
build_type = task["attributes"]["build-type"]
|
|
worker_definition["upstream-artifacts"] = (
|
|
generate_beetmover_upstream_artifacts(config, task, build_type, locale)
|
|
)
|
|
else:
|
|
only_archs = task.pop("only-archs", [])
|
|
for dep in get_dependencies(config, task):
|
|
paths = list(dep.attributes.get("artifacts", {}).values())
|
|
paths.extend([
|
|
apk_metadata["name"]
|
|
for arch, apk_metadata in dep.attributes.get("apks", {}).items()
|
|
if not only_archs or arch in only_archs
|
|
])
|
|
if dep.attributes.get("aab"):
|
|
paths.extend([dep.attributes.get("aab")])
|
|
if paths:
|
|
worker_definition["upstream-artifacts"].append({
|
|
"taskId": {"task-reference": f"<{dep.kind}>"},
|
|
"taskType": _get_task_type(dep.kind),
|
|
"paths": sorted(paths),
|
|
})
|
|
|
|
task.setdefault("worker", {}).update(worker_definition)
|
|
yield task
|