summaryrefslogtreecommitdiffstats
path: root/taskcluster/gecko_taskgraph/transforms/release_deps.py
blob: e44af576eb3f3573435f9fe57fc316278a6406f9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
# 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/.
"""
Add dependencies to release tasks.
"""

from taskgraph.transforms.base import TransformSequence

PHASES = ["build", "promote", "push", "ship"]

transforms = TransformSequence()


@transforms.add
def add_dependencies(config, jobs):
    for job in jobs:
        dependencies = {}
        # Add any kind_dependencies_tasks with matching product as dependencies
        product = job.get("shipping-product")
        phase = job.get("shipping-phase")
        if product is None:
            continue

        required_signoffs = set(
            job.setdefault("attributes", {}).get("required_signoffs", [])
        )
        for dep_task in config.kind_dependencies_tasks.values():
            # Weed out unwanted tasks.
            # XXX we have run-on-projects which specifies the on-push behavior;
            # we need another attribute that specifies release promotion,
            # possibly which action(s) each task belongs in.

            # We can only depend on tasks in the current or previous phases
            dep_phase = dep_task.attributes.get("shipping_phase")
            if dep_phase and PHASES.index(dep_phase) > PHASES.index(phase):
                continue

            if dep_task.attributes.get("build_platform") and job.get(
                "attributes", {}
            ).get("build_platform"):
                if (
                    dep_task.attributes["build_platform"]
                    != job["attributes"]["build_platform"]
                ):
                    continue
            # Add matching product tasks to deps
            if (
                dep_task.task.get("shipping-product") == product
                or dep_task.attributes.get("shipping_product") == product
            ):
                dependencies[dep_task.label] = dep_task.label
                required_signoffs.update(
                    dep_task.attributes.get("required_signoffs", [])
                )

        job.setdefault("dependencies", {}).update(dependencies)
        if required_signoffs:
            job["attributes"]["required_signoffs"] = sorted(required_signoffs)

        yield job