1 files changed, 61 insertions, 0 deletions
diff --git a/taskcluster/gecko_taskgraph/transforms/release_deps.py b/taskcluster/gecko_taskgraph/transforms/release_deps.py
new file mode 100644
index 0000000000..e44af576eb
--- /dev/null
+++ b/taskcluster/gecko_taskgraph/transforms/release_deps.py
@@ -0,0 +1,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