# 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 calls to the DTE test suite in GHA.
"""

from taskgraph.transforms.base import TransformSequence
from taskgraph.util.taskcluster import get_artifact_path

transforms = TransformSequence()


@transforms.add
def update_env(config, tasks):
    for task in tasks:
        name = task["name"]
        if "win" in name:
            input_key = "win_installer_link"
            artifact = "target.zip"
        elif "linux" in name:
            input_key = "linux_tarball_link"
            artifact = "target.tar.xz"
        elif "macos" in name:
            input_key = "mac_installer_link"
            artifact = "target.dmg"

        task["worker"]["env"] |= {
            "INSTALLER_LINK": {
                "artifact-reference": f"<build/{get_artifact_path(task, artifact)}>"
            },
            "INPUT_KEY": input_key,
        }
        yield task