1 files changed, 75 insertions, 0 deletions
diff --git a/comm/taskcluster/comm_taskgraph/target_tasks.py b/comm/taskcluster/comm_taskgraph/target_tasks.py
new file mode 100644
index 0000000000..cfb758200d
--- /dev/null
+++ b/comm/taskcluster/comm_taskgraph/target_tasks.py
@@ -0,0 +1,75 @@
+#  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/.
+
+import logging
+
+from gecko_taskgraph.target_tasks import (
+    _target_task,
+    _try_task_config,
+    filter_on_platforms,
+    filter_out_shippable,
+    filter_out_shipping_phase,
+    standard_filter,
+)
+
+from comm_taskgraph.try_option_syntax import _try_cc_option_syntax
+
+logger = logging.getLogger(__name__)
+
+
+@_target_task("comm_searchfox_index")
+def target_tasks_searchfox(full_task_graph, parameters, graph_config):
+    """Select tasks required for indexing Thunderbird for Searchfox web site each day"""
+    return [
+        "searchfox-linux64-searchfox/debug",
+        "searchfox-macosx64-searchfox/debug",
+        "searchfox-win64-searchfox/debug",
+    ]
+
+
+@_target_task("comm_central_tasks")
+def target_tasks_default(full_task_graph, parameters, graph_config):
+    """Target the tasks which have indicated they should be run on this project
+    via the `run_on_projects` attributes."""
+    return [
+        l
+        for l, t in full_task_graph.tasks.items()
+        if standard_filter(t, parameters) and filter_out_shipping_phase(t, parameters)
+    ]
+
+
+@_target_task("try_cc_tasks")
+def target_tasks_try(full_task_graph, parameters, graph_config):
+    try_mode = parameters["try_mode"]
+    if try_mode == "try_task_config":
+        return _try_task_config(full_task_graph, parameters, graph_config)
+    elif try_mode == "try_option_syntax":
+        return _try_cc_option_syntax(full_task_graph, parameters, graph_config)
+    else:
+        # With no try mode, we schedule nothing, allowing the user to add tasks
+        # later via treeherder.
+        return []
+
+
+@_target_task("ash_tasks")
+def target_tasks_ash(full_task_graph, parameters, graph_config):
+    run_for_platforms = (
+        "linux64",
+        "macosx64",
+        "win64",
+    )
+
+    def filter_source_test(task):
+        return task.kind == "source-test" and task.attributes.get("always_target", False)
+
+    def _filter(task, _parameters):
+        return all(
+            [
+                filter_on_platforms(task, run_for_platforms) or filter_source_test(task),
+                standard_filter(task, _parameters),
+                filter_out_shippable(task),
+            ]
+        )
+
+    return [l for l, t in full_task_graph.tasks.items() if _filter(t, parameters)]