1 files changed, 43 insertions, 0 deletions
diff --git a/taskcluster/taskgraph/transforms/upload_generated_sources.py b/taskcluster/taskgraph/transforms/upload_generated_sources.py
new file mode 100644
index 0000000000..e8a53dac9f
--- /dev/null
+++ b/taskcluster/taskgraph/transforms/upload_generated_sources.py
@@ -0,0 +1,43 @@
+# 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 the upload-generated-files task description template,
+taskcluster/ci/upload-generated-sources/kind.yml, into an actual task description.
+"""
+
+from __future__ import absolute_import, print_function, unicode_literals
+
+from taskgraph.transforms.base import TransformSequence
+
+
+transforms = TransformSequence()
+
+
+@transforms.add
+def add_task_info(config, jobs):
+    for job in jobs:
+        dep_task = job["primary-dependency"]
+        del job["primary-dependency"]
+
+        # Add a dependency on the build task.
+        job["dependencies"] = {"build": dep_task.label}
+        # Label the job to match the build task it's uploading from.
+        job["label"] = dep_task.label.replace("build-", "upload-generated-sources-")
+        # Copy over some bits of metdata from the build task.
+        dep_th = dep_task.task["extra"]["treeherder"]
+        job.setdefault("attributes", {})
+        job["attributes"]["build_platform"] = dep_task.attributes.get("build_platform")
+        if dep_task.attributes.get("shippable"):
+            job["attributes"]["shippable"] = True
+        plat = "{}/{}".format(
+            dep_th["machine"]["platform"], dep_task.attributes.get("build_type")
+        )
+        job["treeherder"]["platform"] = plat
+        job["treeherder"]["tier"] = dep_th["tier"]
+        if dep_th["symbol"] != "N":
+            job["treeherder"]["symbol"] = "Ugs{}".format(dep_th["symbol"])
+        job["run-on-projects"] = dep_task.attributes.get("run_on_projects")
+        job["optimization"] = dep_task.optimization
+
+        yield job