48 lines
1.7 KiB
Python
48 lines
1.7 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 taskgraph.transforms.base import TransformSequence
|
|
|
|
from gecko_taskgraph.util.platforms import platform_family
|
|
|
|
transforms = TransformSequence()
|
|
|
|
|
|
@transforms.add
|
|
def set_build_schedules_optimization(config, tasks):
|
|
"""Set the `build` optimization based on the build platform."""
|
|
for task in tasks:
|
|
# don't add an optimization if there's already one defined
|
|
if "when" in task or "optimization" in task:
|
|
yield task
|
|
continue
|
|
|
|
schedules = []
|
|
if config.kind in (
|
|
"build-components",
|
|
"build-samples-browser",
|
|
"test-components",
|
|
):
|
|
# These are Android components builds and can only impact Fenix or Focus.
|
|
schedules = ["android", "fenix", "focus-android"]
|
|
|
|
elif config.kind in ("build-apk", "build-bundle", "test-apk", "ui-test-apk"):
|
|
# These are APK builds for Fenix or Focus
|
|
schedules = ["android"]
|
|
|
|
if "fenix" in task["name"]:
|
|
schedules.append("fenix")
|
|
elif "focus" in task["name"] or "klar" in task["name"]:
|
|
schedules.append("focus-android")
|
|
|
|
else:
|
|
family = platform_family(task["attributes"]["build_platform"])
|
|
schedules = [family]
|
|
|
|
if "android" not in family:
|
|
# These are not GeckoView builds, so are associated with Firefox.
|
|
schedules.append("firefox")
|
|
|
|
task["optimization"] = {"build": schedules}
|
|
yield task
|