1 files changed, 141 insertions, 0 deletions
diff --git a/python/mozrelease/mozrelease/mach_commands.py b/python/mozrelease/mozrelease/mach_commands.py
new file mode 100644
index 0000000000..e7c8da59fe
--- /dev/null
+++ b/python/mozrelease/mozrelease/mach_commands.py
@@ -0,0 +1,141 @@
+# -*- 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/.
+
+
+import logging
+import sys
+
+from mach.decorators import Command, CommandArgument, SubCommand
+from mozilla_version.gecko import GeckoVersion
+
+
+@Command(
+    "release",
+    category="release",
+    description="Task that are part of the release process.",
+)
+def release(command_context):
+    """
+    The release subcommands all relate to the release process.
+    """
+
+
+@SubCommand(
+    "release",
+    "buglist",
+    description="Generate list of bugs since the last release.",
+)
+@CommandArgument(
+    "--version",
+    required=True,
+    type=GeckoVersion.parse,
+    help="The version being built.",
+)
+@CommandArgument("--product", required=True, help="The product being built.")
+@CommandArgument("--repo", help="The repo being built.")
+@CommandArgument("--revision", required=True, help="The revision being built.")
+def buglist(command_context, version, product, revision, repo):
+    setup_logging(command_context)
+    from mozrelease.buglist_creator import create_bugs_url
+
+    print(
+        create_bugs_url(
+            product=product,
+            current_version=version,
+            current_revision=revision,
+            repo=repo,
+        )
+    )
+
+
+@SubCommand(
+    "release",
+    "send-buglist-email",
+    description="Send an email with the bugs since the last release.",
+)
+@CommandArgument(
+    "--address",
+    required=True,
+    action="append",
+    dest="addresses",
+    help="The email address to send the bug list to "
+    "(may be specified more than once.",
+)
+@CommandArgument(
+    "--version",
+    type=GeckoVersion.parse,
+    required=True,
+    help="The version being built.",
+)
+@CommandArgument("--product", required=True, help="The product being built.")
+@CommandArgument("--repo", required=True, help="The repo being built.")
+@CommandArgument("--revision", required=True, help="The revision being built.")
+@CommandArgument("--build-number", required=True, help="The build number")
+@CommandArgument("--task-group-id", help="The task group of the build.")
+def buglist_email(command_context, **options):
+    setup_logging(command_context)
+    from mozrelease.buglist_creator import email_release_drivers
+
+    email_release_drivers(**options)
+
+
+@SubCommand(
+    "release",
+    "push-scriptworker-canary",
+    description="Push tasks to try, to test new scriptworker deployments.",
+)
+@CommandArgument(
+    "--address",
+    required=True,
+    action="append",
+    dest="addresses",
+    help="The email address to send notifications to "
+    "(may be specified more than once).",
+)
+@CommandArgument(
+    "--scriptworker",
+    required=True,
+    action="append",
+    dest="scriptworkers",
+    help="Scriptworker to run canary for (may be specified more than once).",
+)
+@CommandArgument(
+    "--ssh-key-secret",
+    required=False,
+    help="Taskcluster secret with ssh-key to use for hg.mozilla.org",
+)
+def push_scriptworker_canary(command_context, scriptworkers, addresses, ssh_key_secret):
+    setup_logging(command_context)
+    from mozrelease.scriptworker_canary import push_canary
+
+    push_canary(
+        scriptworkers=scriptworkers,
+        addresses=addresses,
+        ssh_key_secret=ssh_key_secret,
+    )
+
+
+def setup_logging(command_context, quiet=False, verbose=True):
+    """
+    Set up Python logging for all loggers, sending results to stderr (so
+    that command output can be redirected easily) and adding the typical
+    mach timestamp.
+    """
+    # remove the old terminal handler
+    old = command_context.log_manager.replace_terminal_handler(None)
+
+    # re-add it, with level and fh set appropriately
+    if not quiet:
+        level = logging.DEBUG if verbose else logging.INFO
+        command_context.log_manager.add_terminal_logging(
+            fh=sys.stderr,
+            level=level,
+            write_interval=old.formatter.write_interval,
+            write_times=old.formatter.write_times,
+        )
+
+    # all of the taskgraph logging is unstructured logging
+    command_context.log_manager.enable_unstructured()