Bug 1725404 - [taskgraph] Re-use 'action-callback' subcommand args from main.py, r=bhearsum

Differential Revision: https://phabricator.services.mozilla.com/D122526

taskcluster/mach_commands.py

@@ -319,116 +319,25 @@ class MachCommands(MachCommandBase):
         "taskgraph",
         "action-callback",
         description="Run action callback used by action tasks",
-    )
+        parser=partial(get_taskgraph_command_parser, "action-callback"),
-    @CommandArgument(
-        "--root",
-        "-r",
-        default="taskcluster/ci",
-        help="root of the taskgraph definition relative to topsrcdir",
     )
     def action_callback(self, command_context, **options):
-        from taskgraph.actions import trigger_action_callback
+        self.setup_logging(command_context)
-        from taskgraph.actions.util import get_parameters
+        taskgraph_commands["action-callback"].func(options)
-
-        try:
-            self.setup_logging(command_context)
-
-            # the target task for this action (or null if it's a group action)
-            task_id = json.loads(os.environ.get("ACTION_TASK_ID", "null"))
-            # the target task group for this action
-            task_group_id = os.environ.get("ACTION_TASK_GROUP_ID", None)
-            input = json.loads(os.environ.get("ACTION_INPUT", "null"))
-            callback = os.environ.get("ACTION_CALLBACK", None)
-            root = options["root"]
-
-            parameters = get_parameters(task_group_id)
-
-            return trigger_action_callback(
-                task_group_id=task_group_id,
-                task_id=task_id,
-                input=input,
-                callback=callback,
-                parameters=parameters,
-                root=root,
-                test=False,
-            )
-        except Exception:
-            traceback.print_exc()
-            sys.exit(1)
 
     @SubCommand(
         "taskgraph",
         "test-action-callback",
         description="Run an action callback in a testing mode",
-    )
+        parser=partial(get_taskgraph_command_parser, "test-action-callback"),
-    @CommandArgument(
-        "--root",
-        "-r",
-        default="taskcluster/ci",
-        help="root of the taskgraph definition relative to topsrcdir",
-    )
-    @CommandArgument(
-        "--parameters",
-        "-p",
-        default="project=mozilla-central",
-        help="parameters file (.yml or .json; see "
-        "`taskcluster/docs/parameters.rst`)`",
-    )
-    @CommandArgument(
-        "--task-id", default=None, help="TaskId to which the action applies"
-    )
-    @CommandArgument(
-        "--task-group-id", default=None, help="TaskGroupId to which the action applies"
-    )
-    @CommandArgument("--input", default=None, help="Action input (.yml or .json)")
-    @CommandArgument(
-        "callback", default=None, help="Action callback name (Python function name)"
     )
     def test_action_callback(self, command_context, **options):
-        import taskgraph.actions
+        self.setup_logging(command_context)
-        import taskgraph.parameters
-        from taskgraph.util import yaml
 
-        def load_data(filename):
+        if not options["parameters"]:
-            with open(filename) as f:
+            options["parameters"] = "project=mozilla-central"
-                if filename.endswith(".yml"):
-                    return yaml.load_stream(f)
-                elif filename.endswith(".json"):
-                    return json.load(f)
-                else:
-                    raise Exception("unknown filename {}".format(filename))
 
-        try:
+        taskgraph_commands["test-action-callback"].func(options)
-            self.setup_logging(command_context)
-            task_id = options["task_id"]
-
-            if options["input"]:
-                input = load_data(options["input"])
-            else:
-                input = None
-
-            parameters = taskgraph.parameters.load_parameters_file(
-                options["parameters"],
-                strict=False,
-                # FIXME: There should be a way to parameterize this.
-                trust_domain="gecko",
-            )
-            parameters.check()
-
-            root = options["root"]
-
-            return taskgraph.actions.trigger_action_callback(
-                task_group_id=options["task_group_id"],
-                task_id=task_id,
-                input=input,
-                callback=options["callback"],
-                parameters=parameters,
-                root=root,
-                test=True,
-            )
-        except Exception:
-            traceback.print_exc()
-            sys.exit(1)
 
     def setup_logging(self, command_context, quiet=False, verbose=True):
         """