gecko-dev/tools/mach_commands.py

# 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 argparse
from datetime import datetime, timedelta
import logging
from operator import itemgetter
import sys

from mach.decorators import (
    CommandArgument,
    CommandProvider,
    Command,
    SubCommand,
)

from mozbuild.base import MachCommandBase, MozbuildObject


def _get_busted_bugs(payload):
    import requests

    payload = dict(payload)
    payload["include_fields"] = "id,summary,last_change_time,resolution"
    payload["blocks"] = 1543241
    response = requests.get("https://bugzilla.mozilla.org/rest/bug", payload)
    response.raise_for_status()
    return response.json().get("bugs", [])


@CommandProvider
class BustedProvider(MachCommandBase):
    @Command(
        "busted",
        category="misc",
        description="Query known bugs in our tooling, and file new ones.",
    )
    def busted_default(self, command_context):
        unresolved = _get_busted_bugs({"resolution": "---"})
        creation_time = datetime.now() - timedelta(days=15)
        creation_time = creation_time.strftime("%Y-%m-%dT%H-%M-%SZ")
        resolved = _get_busted_bugs({"creation_time": creation_time})
        resolved = [bug for bug in resolved if bug["resolution"]]
        all_bugs = sorted(
            unresolved + resolved, key=itemgetter("last_change_time"), reverse=True
        )
        if all_bugs:
            for bug in all_bugs:
                print(
                    "[%s] Bug %s - %s"
                    % (
                        "UNRESOLVED"
                        if not bug["resolution"]
                        else "RESOLVED - %s" % bug["resolution"],
                        bug["id"],
                        bug["summary"],
                    )
                )
        else:
            print("No known tooling issues found.")

    @SubCommand("busted", "file", description="File a bug for busted tooling.")
    @CommandArgument(
        "against",
        help=(
            "The specific mach command that is busted (i.e. if you encountered "
            "an error with `mach build`, run `mach busted file build`). If "
            "the issue is not connected to any particular mach command, you "
            "can also run `mach busted file general`."
        ),
    )
    def busted_file(self, command_context, against):
        import webbrowser

        if (
            against != "general"
            and against not in self._mach_context.commands.command_handlers
        ):
            print(
                "%s is not a valid value for `against`. `against` must be "
                "the name of a `mach` command, or else the string "
                '"general".' % against
            )
            return 1

        if against == "general":
            product = "Firefox Build System"
            component = "General"
        else:
            import inspect
            import mozpack.path as mozpath

            # Look up the file implementing that command, then cross-refernce
            # moz.build files to get the product/component.
            handler = self._mach_context.commands.command_handlers[against]
            method = getattr(handler.cls, handler.method)
            sourcefile = mozpath.relpath(inspect.getsourcefile(method), self.topsrcdir)
            reader = self.mozbuild_reader(config_mode="empty")
            try:
                res = reader.files_info([sourcefile])[sourcefile]["BUG_COMPONENT"]
                product, component = res.product, res.component
            except TypeError:
                # The file might not have a bug set.
                product = "Firefox Build System"
                component = "General"

        uri = (
            "https://bugzilla.mozilla.org/enter_bug.cgi?"
            "product=%s&component=%s&blocked=1543241" % (product, component)
        )
        webbrowser.open_new_tab(uri)


MACH_PASTEBIN_DURATIONS = {
    "onetime": "onetime",
    "hour": "3600",
    "day": "86400",
    "week": "604800",
    "month": "2073600",
}

EXTENSION_TO_HIGHLIGHTER = {
    ".hgrc": "ini",
    "Dockerfile": "docker",
    "Makefile": "make",
    "applescript": "applescript",
    "arduino": "arduino",
    "bash": "bash",
    "bat": "bat",
    "c": "c",
    "clojure": "clojure",
    "cmake": "cmake",
    "coffee": "coffee-script",
    "console": "console",
    "cpp": "cpp",
    "cs": "csharp",
    "css": "css",
    "cu": "cuda",
    "cuda": "cuda",
    "dart": "dart",
    "delphi": "delphi",
    "diff": "diff",
    "django": "django",
    "docker": "docker",
    "elixir": "elixir",
    "erlang": "erlang",
    "go": "go",
    "h": "c",
    "handlebars": "handlebars",
    "haskell": "haskell",
    "hs": "haskell",
    "html": "html",
    "ini": "ini",
    "ipy": "ipythonconsole",
    "ipynb": "ipythonconsole",
    "irc": "irc",
    "j2": "django",
    "java": "java",
    "js": "js",
    "json": "json",
    "jsx": "jsx",
    "kt": "kotlin",
    "less": "less",
    "lisp": "common-lisp",
    "lsp": "common-lisp",
    "lua": "lua",
    "m": "objective-c",
    "make": "make",
    "matlab": "matlab",
    "md": "_markdown",
    "nginx": "nginx",
    "numpy": "numpy",
    "patch": "diff",
    "perl": "perl",
    "php": "php",
    "pm": "perl",
    "postgresql": "postgresql",
    "py": "python",
    "rb": "rb",
    "rs": "rust",
    "rst": "rst",
    "sass": "sass",
    "scss": "scss",
    "sh": "bash",
    "sol": "sol",
    "sql": "sql",
    "swift": "swift",
    "tex": "tex",
    "typoscript": "typoscript",
    "vim": "vim",
    "xml": "xml",
    "xslt": "xslt",
    "yaml": "yaml",
    "yml": "yaml",
}


def guess_highlighter_from_path(path):
    """Return a known highlighter from a given path

    Attempt to select a highlighter by checking the file extension in the mapping
    of extensions to highlighter. If that fails, attempt to pass the basename of
    the file. Return `_code` as the default highlighter if that fails.
    """
    import os

    _name, ext = os.path.splitext(path)

    if ext.startswith("."):
        ext = ext[1:]

    if ext in EXTENSION_TO_HIGHLIGHTER:
        return EXTENSION_TO_HIGHLIGHTER[ext]

    basename = os.path.basename(path)

    return EXTENSION_TO_HIGHLIGHTER.get(basename, "_code")


PASTEMO_MAX_CONTENT_LENGTH = 250 * 1024 * 1024

PASTEMO_URL = "https://paste.mozilla.org/api/"

MACH_PASTEBIN_DESCRIPTION = """
Command line interface to paste.mozilla.org.

Takes either a filename whose content should be pasted, or reads
content from standard input. If a highlighter is specified it will
be used, otherwise the file name will be used to determine an
appropriate highlighter.
"""


@CommandProvider
class PastebinProvider(MachCommandBase):
    @Command("pastebin", category="misc", description=MACH_PASTEBIN_DESCRIPTION)
    @CommandArgument(
        "--list-highlighters",
        action="store_true",
        help="List known highlighters and exit",
    )
    @CommandArgument(
        "--highlighter", default=None, help="Syntax highlighting to use for paste"
    )
    @CommandArgument(
        "--expires",
        default="week",
        choices=sorted(MACH_PASTEBIN_DURATIONS.keys()),
        help="Expire paste after given time duration (default: %(default)s)",
    )
    @CommandArgument(
        "--verbose",
        action="store_true",
        help="Print extra info such as selected syntax highlighter",
    )
    @CommandArgument(
        "path",
        nargs="?",
        default=None,
        help="Path to file for upload to paste.mozilla.org",
    )
    def pastebin(
        self, command_context, list_highlighters, highlighter, expires, verbose, path
    ):
        import requests

        def verbose_print(*args, **kwargs):
            """Print a string if `--verbose` flag is set"""
            if verbose:
                print(*args, **kwargs)

        # Show known highlighters and exit.
        if list_highlighters:
            lexers = set(EXTENSION_TO_HIGHLIGHTER.values())
            print("Available lexers:\n" "    - %s" % "\n    - ".join(sorted(lexers)))
            return 0

        # Get a correct expiry value.
        try:
            verbose_print("Setting expiry from %s" % expires)
            expires = MACH_PASTEBIN_DURATIONS[expires]
            verbose_print("Using %s as expiry" % expires)
        except KeyError:
            print(
                "%s is not a valid duration.\n"
                "(hint: try one of %s)"
                % (expires, ", ".join(MACH_PASTEBIN_DURATIONS.keys()))
            )
            return 1

        data = {
            "format": "json",
            "expires": expires,
        }

        # Get content to be pasted.
        if path:
            verbose_print("Reading content from %s" % path)
            try:
                with open(path, "r") as f:
                    content = f.read()
            except IOError:
                print("ERROR. No such file %s" % path)
                return 1

            lexer = guess_highlighter_from_path(path)
            if lexer:
                data["lexer"] = lexer
        else:
            verbose_print("Reading content from stdin")
            content = sys.stdin.read()

        # Assert the length of content to be posted does not exceed the maximum.
        content_length = len(content)
        verbose_print("Checking size of content is okay (%d)" % content_length)
        if content_length > PASTEMO_MAX_CONTENT_LENGTH:
            print(
                "Paste content is too large (%d, maximum %d)"
                % (content_length, PASTEMO_MAX_CONTENT_LENGTH)
            )
            return 1

        data["content"] = content

        # Highlight as specified language, overwriting value set from filename.
        if highlighter:
            verbose_print("Setting %s as highlighter" % highlighter)
            data["lexer"] = highlighter

        try:
            verbose_print("Sending request to %s" % PASTEMO_URL)
            resp = requests.post(PASTEMO_URL, data=data)

            # Error code should always be 400.
            # Response content will include a helpful error message,
            # so print it here (for example, if an invalid highlighter is
            # provided, it will return a list of valid highlighters).
            if resp.status_code >= 400:
                print("Error code %d: %s" % (resp.status_code, resp.content))
                return 1

            verbose_print("Pasted successfully")

            response_json = resp.json()

            verbose_print("Paste highlighted as %s" % response_json["lexer"])
            print(response_json["url"])

            return 0
        except Exception as e:
            print("ERROR. Paste failed.")
            print("%s" % e)
        return 1


class PypiBasedTool:
    """
    Helper for loading a tool that is hosted on pypi. The package is expected
    to expose a `mach_interface` module which has `new_release_on_pypi`,
    `parser`, and `run` functions.
    """

    def __init__(self, module_name, pypi_name=None):
        self.name = module_name
        self.pypi_name = pypi_name or module_name

    def _import(self):
        # Lazy loading of the tools mach interface.
        # Note that only the mach_interface module should be used from this file.
        import importlib

        try:
            return importlib.import_module("%s.mach_interface" % self.name)
        except ImportError:
            return None

    def create_parser(self, subcommand=None):
        # Create the command line parser.
        # If the tool is not installed, or not up to date, it will
        # first be installed.
        cmd = MozbuildObject.from_environment()
        cmd.activate_virtualenv()
        tool = self._import()
        if not tool:
            # The tool is not here at all, install it
            cmd.virtualenv_manager.install_pip_package(self.pypi_name)
            print(
                "%s was installed. please re-run your"
                " command. If you keep getting this message please "
                " manually run: 'pip install -U %s'." % (self.pypi_name, self.pypi_name)
            )
        else:
            # Check if there is a new release available
            release = tool.new_release_on_pypi()
            if release:
                print(release)
                # there is one, so install it. Note that install_pip_package
                # does not work here, so just run pip directly.
                cmd.virtualenv_manager._run_pip(
                    ["install", "%s==%s" % (self.pypi_name, release)]
                )
                print(
                    "%s was updated to version %s. please"
                    " re-run your command." % (self.pypi_name, release)
                )
            else:
                # Tool is up to date, return the parser.
                if subcommand:
                    return tool.parser(subcommand)
                else:
                    return tool.parser()
        # exit if we updated or installed mozregression because
        # we may have already imported mozregression and running it
        # as this may cause issues.
        sys.exit(0)

    def run(self, **options):
        tool = self._import()
        tool.run(options)


def mozregression_create_parser():
    # Create the mozregression command line parser.
    # if mozregression is not installed, or not up to date, it will
    # first be installed.
    loader = PypiBasedTool("mozregression")
    return loader.create_parser()


@CommandProvider
class MozregressionCommand(MachCommandBase):
    @Command(
        "mozregression",
        category="misc",
        description=("Regression range finder for nightly" " and inbound builds."),
        parser=mozregression_create_parser,
    )
    def run(self, command_context, **options):
        self.activate_virtualenv()
        mozregression = PypiBasedTool("mozregression")
        mozregression.run(**options)


@CommandProvider
class NodeCommands(MachCommandBase):
    @Command(
        "node",
        category="devenv",
        description="Run the NodeJS interpreter used for building.",
    )
    @CommandArgument("args", nargs=argparse.REMAINDER)
    def node(self, command_context, args):
        from mozbuild.nodeutil import find_node_executable

        # Avoid logging the command
        self.log_manager.terminal_handler.setLevel(logging.CRITICAL)

        node_path, _ = find_node_executable()

        return self.run_process(
            [node_path] + args,
            pass_thru=True,  # Allow user to run Node interactively.
            ensure_exit_code=False,  # Don't throw on non-zero exit code.
        )

    @Command(
        "npm",
        category="devenv",
        description="Run the npm executable from the NodeJS used for building.",
    )
    @CommandArgument("args", nargs=argparse.REMAINDER)
    def npm(self, command_context, args):
        from mozbuild.nodeutil import find_npm_executable

        # Avoid logging the command
        self.log_manager.terminal_handler.setLevel(logging.CRITICAL)

        npm_path, _ = find_npm_executable()

        return self.run_process(
            [npm_path, "--scripts-prepend-node-path=auto"] + args,
            pass_thru=True,  # Avoid eating npm output/error messages
            ensure_exit_code=False,  # Don't throw on non-zero exit code.
        )


def logspam_create_parser(subcommand):
    # Create the logspam command line parser.
    # if logspam is not installed, or not up to date, it will
    # first be installed.
    loader = PypiBasedTool("logspam", "mozilla-log-spam")
    return loader.create_parser(subcommand)


from functools import partial


@CommandProvider
class LogspamCommand(MachCommandBase):
    @Command(
        "logspam",
        category="misc",
        description=("Warning categorizer for treeherder test runs."),
    )
    def logspam(self, command_context):
        pass

    @SubCommand("logspam", "report", parser=partial(logspam_create_parser, "report"))
    def report(self, command_context, **options):
        self.activate_virtualenv()
        logspam = PypiBasedTool("logspam")
        logspam.run(command="report", **options)

    @SubCommand("logspam", "bisect", parser=partial(logspam_create_parser, "bisect"))
    def bisect(self, command_context, **options):
        self.activate_virtualenv()
        logspam = PypiBasedTool("logspam")
        logspam.run(command="bisect", **options)

    @SubCommand("logspam", "file", parser=partial(logspam_create_parser, "file"))
    def create(self, command_context, **options):
        self.activate_virtualenv()
        logspam = PypiBasedTool("logspam")
        logspam.run(command="file", **options)