bot: Support hook triggered by pulse message from code-coverage-artifacts task

Author: marco-c

bot/code_coverage_bot/cli.py

@@ -14,15 +14,18 @@
 from code_coverage_tools.log import init_logger
 
 
-def setup_cli(ask_repository=True, ask_revision=True):
+def setup_cli(parameters=True):
     """
     Setup CLI options parser and taskcluster bootstrap
     """
     parser = argparse.ArgumentParser(description="Mozilla Code Coverage Bot")
-    if ask_repository:
+    if parameters:
         parser.add_argument("--repository", default=os.environ.get("REPOSITORY"))
-    if ask_revision:
+
+        parser.add_argument("--task-group-id", default=os.environ.get("TASK_GROUP_ID"))
+
         parser.add_argument("--revision", default=os.environ.get("REVISION"))
+
     parser.add_argument(
         "--cache-root", required=True, help="Cache root, used to pull changesets"
     )
@@ -50,6 +53,18 @@ def setup_cli(ask_repository=True, ask_revision=True):
     parser.add_argument("--taskcluster-access-token", help="Taskcluster Access token")
     args = parser.parse_args()
 
+    if parameters:
+        if args.task_group_id and (args.repository or args.revision):
+            parser.error(
+                "Provide either --task-group-id or --repository/--revision, not both."
+            )
+
+        if args.repository and not args.revision:
+            parser.error("--repository requires --revision")
+
+        if not args.task_group_id and not args.repository:
+            parser.error("Provide either --task-group-id or --repository/--revision.")
+
     # Auth on Taskcluster
     taskcluster_config.auth(args.taskcluster_client_id, args.taskcluster_access_token)
 

bot/code_coverage_bot/hooks/cron.py

@@ -61,6 +61,6 @@ def run(self) -> None:
 
 def main() -> None:
     logger.info("Starting code coverage bot for cron")
-    args = setup_cli(ask_revision=False, ask_repository=False)
+    args = setup_cli(parameters=False)
     hook = CronHook(args.task_name_filter, args.cache_root, args.working_dir)
     hook.run()

bot/code_coverage_bot/hooks/crontrigger.py

@@ -50,6 +50,6 @@ def run(self) -> None:
 
 def main() -> None:
     logger.info("Starting code coverage bot for crontrigger")
-    args = setup_cli(ask_revision=False, ask_repository=False)
+    args = setup_cli(parameters=False)
     hook = CronTriggerHook(args.task_name_filter, None, args.working_dir)
     hook.run()

bot/code_coverage_bot/hooks/repo.py

@@ -11,6 +11,7 @@
 import structlog
 
 from code_coverage_bot import config
+from code_coverage_bot import taskcluster
 from code_coverage_bot import hgmo
 from code_coverage_bot import uploader
 from code_coverage_bot.cli import setup_cli
@@ -229,14 +230,22 @@ def main():
     logger.info("Starting code coverage bot for repository")
     args = setup_cli()
 
+    if args.repository:
+        repository = args.repository
+        revision = args.revision
+    else:
+        decision_task = taskcluster.get_task_details(args.task_group_id)
+        repository = decision_task["payload"]["env"]["GECKO_HEAD_REPOSITORY"]
+        revision = decision_task["payload"]["env"]["GECKO_HEAD_REV"]
+
     hooks = {
         config.MOZILLA_CENTRAL_REPOSITORY: MozillaCentralHook,
         config.TRY_REPOSITORY: TryHook,
     }
-    hook_class = hooks.get(args.repository)
-    assert hook_class is not None, f"Unsupported repository {args.repository}"
+    hook_class = hooks.get(repository)
+    assert hook_class is not None, f"Unsupported repository {repository}"
 
     hook = hook_class(
-        args.revision, args.task_name_filter, args.cache_root, args.working_dir
+        revision, args.task_name_filter, args.cache_root, args.working_dir
     )
     hook.run()