Source code
Revision control
Copy as Markdown
Other Tools
# 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
import copy
from taskgraph.transforms.base import TransformSequence
from taskgraph.util.taskcluster import get_artifact_prefix
from gecko_taskgraph.util.declarative_artifacts import get_geckoview_upstream_artifacts
transforms = TransformSequence()
MOZ_ANDROID_FAT_AAR_ENV_MAP = {
"android-arm-shippable": "MOZ_ANDROID_FAT_AAR_ARMEABI_V7A",
"android-arm-shippable-lite": "MOZ_ANDROID_FAT_AAR_ARMEABI_V7A",
"android-aarch64-shippable": "MOZ_ANDROID_FAT_AAR_ARM64_V8A",
"android-aarch64-shippable-lite": "MOZ_ANDROID_FAT_AAR_ARM64_V8A",
"android-x86-shippable": "MOZ_ANDROID_FAT_AAR_X86",
"android-x86-shippable-lite": "MOZ_ANDROID_FAT_AAR_X86",
"android-x86_64-shippable": "MOZ_ANDROID_FAT_AAR_X86_64",
"android-x86_64-shippable-lite": "MOZ_ANDROID_FAT_AAR_X86_64",
"android-arm-opt": "MOZ_ANDROID_FAT_AAR_ARMEABI_V7A",
"android-aarch64-opt": "MOZ_ANDROID_FAT_AAR_ARM64_V8A",
"android-x86-opt": "MOZ_ANDROID_FAT_AAR_X86",
"android-x86_64-opt": "MOZ_ANDROID_FAT_AAR_X86_64",
}
@transforms.add
def set_fetches_and_locations(config, jobs):
"""Set defaults, including those that differ per worker implementation"""
for job in jobs:
dependencies = copy.deepcopy(job["dependencies"])
for platform, label in dependencies.items():
job["dependencies"] = {"build": label}
aar_location = _get_aar_location(config, job, platform)
prefix = get_artifact_prefix(job)
if not prefix.endswith("/"):
prefix = prefix + "/"
if aar_location.startswith(prefix):
aar_location = aar_location[len(prefix) :]
job.setdefault("fetches", {}).setdefault(platform, []).append(
{
"artifact": aar_location,
"extract": False,
}
)
aar_file_name = aar_location.split("/")[-1]
env_var = MOZ_ANDROID_FAT_AAR_ENV_MAP[platform]
job["worker"]["env"][env_var] = aar_file_name
job["dependencies"] = dependencies
yield job
def _get_aar_location(config, job, platform):
artifacts_locations = []
for package in job["attributes"]["maven_packages"]:
artifacts_locations += get_geckoview_upstream_artifacts(
config, job, package, platform=platform
)
aar_locations = [
path for path in artifacts_locations[0]["paths"] if path.endswith(".aar")
]
if len(aar_locations) != 1:
raise ValueError(f"Only a single AAR must be given. Got: {aar_locations}")
return aar_locations[0]