1 files changed, 37 insertions, 0 deletions
diff --git a/taskcluster/taskgraph/actions/purge_caches.py b/taskcluster/taskgraph/actions/purge_caches.py
new file mode 100644
index 0000000000..9e1670750e
--- /dev/null
+++ b/taskcluster/taskgraph/actions/purge_caches.py
@@ -0,0 +1,37 @@
+# -*- coding: utf-8 -*-
+
+# 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/.
+
+from __future__ import absolute_import, print_function, unicode_literals
+
+import logging
+
+from taskgraph.util.taskcluster import purge_cache
+from .registry import register_callback_action
+from taskgraph.util import taskcluster
+
+logger = logging.getLogger(__name__)
+
+
+@register_callback_action(
+    title="Purge Worker Caches",
+    name="purge-cache",
+    symbol="purge-cache",
+    description=(
+        "Purge any caches associated with this task "
+        "across all workers of the same workertype as the task."
+    ),
+    order=450,
+    context=[{"worker-implementation": "docker-worker"}],
+)
+def purge_caches_action(parameters, graph_config, input, task_group_id, task_id):
+    task = taskcluster.get_task_definition(task_id)
+    if task["payload"].get("cache"):
+        for cache in task["payload"]["cache"]:
+            purge_cache(
+                task["provisionerId"], task["workerType"], cache, use_proxy=True
+            )
+    else:
+        logger.info("Task has no caches. Will not clear anything!")