1 files changed, 120 insertions, 0 deletions
diff --git a/tools/phabricator/mach_commands.py b/tools/phabricator/mach_commands.py
new file mode 100644
index 0000000000..6de99d9292
--- /dev/null
+++ b/tools/phabricator/mach_commands.py
@@ -0,0 +1,120 @@
+# 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, unicode_literals
+
+import mozfile
+from mach.decorators import CommandProvider, Command, CommandArgument
+from mozbuild.base import MachCommandBase
+
+
+@CommandProvider
+class PhabricatorCommandProvider(MachCommandBase):
+    @Command(
+        "install-moz-phab",
+        category="misc",
+        description="Install patch submission tool.",
+    )
+    @CommandArgument(
+        "--force",
+        "-f",
+        action="store_true",
+        help="Force installation even if already installed.",
+    )
+    def install_moz_phab(self, force=False):
+        import logging
+        import os
+        import re
+        import subprocess
+        import sys
+
+        existing = mozfile.which("moz-phab")
+        if existing and not force:
+            self.log(
+                logging.ERROR,
+                "already_installed",
+                {},
+                "moz-phab is already installed in %s." % existing,
+            )
+            sys.exit(1)
+
+        # pip3 is part of Python since 3.4, however some distros choose to
+        # remove core components from languages, so show a useful error message
+        # if pip3 is missing.
+        pip3 = mozfile.which("pip3")
+        if not pip3:
+            self.log(
+                logging.ERROR,
+                "pip3_not_installed",
+                {},
+                "`pip3` is not installed. Try installing it with your system "
+                "package manager.",
+            )
+            sys.exit(1)
+
+        command = [pip3, "install", "--upgrade", "MozPhab"]
+
+        if (
+            sys.platform.startswith("linux")
+            or sys.platform.startswith("openbsd")
+            or sys.platform.startswith("dragonfly")
+            or sys.platform.startswith("freebsd")
+        ):
+            # On all Linux and BSD distros we consider doing a user installation.
+            platform_prefers_user_install = True
+
+        elif sys.platform.startswith("darwin"):
+            # On MacOS we require brew or ports, which work better without --user.
+            platform_prefers_user_install = False
+
+        elif sys.platform.startswith("win32") or sys.platform.startswith("msys"):
+            # Likewise for Windows we assume a system level install is preferred.
+            platform_prefers_user_install = False
+
+        else:
+            # Unsupported, default to --user.
+            self.log(
+                logging.WARNING,
+                "unsupported_platform",
+                {},
+                "Unsupported platform (%s), assuming per-user installation is "
+                "preferred." % sys.platform,
+            )
+            platform_prefers_user_install = True
+
+        if platform_prefers_user_install and not os.environ.get("VIRTUAL_ENV"):
+            # Virtual environments don't see user packages, so only perform a user
+            # installation if we're not within one.
+            command.append("--user")
+
+        self.log(logging.INFO, "run", {}, "Installing moz-phab")
+        subprocess.run(command)
+
+        # There isn't an elegant way of determining the CLI location of a pip-installed package.
+        # The viable mechanism used here is to:
+        # 1. Get the list of info about the installed package via pip
+        # 2. Parse out the install location. This gives us the python environment in which the
+        #    package is installed
+        # 3. Parse out the relative location of the cli script
+        # 4. Join the two paths, and execute the script at that location
+
+        info = subprocess.check_output(
+            [pip3, "show", "-f", "MozPhab"], universal_newlines=True
+        )
+        mozphab_package_location = re.compile(r"Location: (.*)").search(info).group(1)
+        potential_cli_paths = re.compile(r"([^\s]*moz-phab(?:\.exe)?)").findall(info)
+
+        if len(potential_cli_paths) != 1:
+            self.log(
+                logging.WARNING,
+                "no_mozphab_console_script",
+                {},
+                "Could not find the CLI script for moz-phab. Skipping install-certificate step.",
+            )
+            sys.exit(1)
+
+        console_script = os.path.realpath(
+            os.path.join(mozphab_package_location, potential_cli_paths[0])
+        )
+        subprocess.run([console_script, "install-certificate"])