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
from typing import Optional
from taskgraph.transforms.base import TransformSequence
from taskgraph.util.schema import (
Schema,
optionally_keyed_by,
resolve_keyed_by,
validate_schema,
)
MOZCONFIG_PLATFORMS = (
"android-aarch64",
"android-arm",
"android-x86_64",
"ios",
"ios-sim",
"linux64",
"linux64-aarch64",
"macosx64",
"macosx64-aarch64",
"win32",
"win64",
"win64-aarch64",
)
class MozconfigSchema(Schema, kw_only=True):
variant: optionally_keyed_by("release-type", str, use_msgspec=True)
platform: Optional[str] = None
app: Optional[str] = None
transforms = TransformSequence()
def default_platform(name):
build_platform = name.split("/")[0]
matches = [
platform
for platform in MOZCONFIG_PLATFORMS
if build_platform == platform or build_platform.startswith(platform + "-")
]
return max(matches, key=len, default=None)
@transforms.add
def set_mozconfig(config, tasks):
for task in tasks:
mozconfig = task.pop("mozconfig", None)
if mozconfig is None:
yield task
continue
name = task["name"]
validate_schema(MozconfigSchema, mozconfig, f"In mozconfig of {name!r}:")
resolve_keyed_by(
mozconfig,
"variant",
item_name=name,
**{"release-type": config.params["release_type"]},
)
platform = mozconfig.get("platform") or default_platform(name)
if platform is None:
raise Exception(
f"{name} needs mozconfig.platform, its name does not start with "
f"one of {MOZCONFIG_PLATFORMS}"
)
app = mozconfig.get("app")
if app is None:
if platform.startswith("android-"):
app = "mobile/android"
else:
app = config.graph_config["product-dir"]
task.setdefault("worker", {}).setdefault("env", {})["MOZCONFIG"] = (
f"{app}/config/mozconfigs/{platform}/{mozconfig['variant']}"
)
yield task