1 files changed, 101 insertions, 0 deletions
diff --git a/tools/lint/yamllint_/__init__.py b/tools/lint/yamllint_/__init__.py
new file mode 100644
index 0000000000..2244fabd3f
--- /dev/null
+++ b/tools/lint/yamllint_/__init__.py
@@ -0,0 +1,101 @@
+# 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 os
+import re
+import sys
+from collections import defaultdict
+
+from mozbuild.base import MozbuildObject
+
+topsrcdir = MozbuildObject.from_environment().topsrcdir
+
+from mozlint import result
+from mozlint.pathutils import get_ancestors_by_name
+from mozlint.util.implementation import LintProcess
+
+YAMLLINT_FORMAT_REGEX = re.compile("(.*):(.*):(.*): \[(error|warning)\] (.*) \((.*)\)$")
+
+results = []
+
+
+class YAMLLintProcess(LintProcess):
+    def process_line(self, line):
+        try:
+            match = YAMLLINT_FORMAT_REGEX.match(line)
+            abspath, line, col, level, message, code = match.groups()
+        except AttributeError:
+            print("Unable to match yaml regex against output: {}".format(line))
+            return
+
+        res = {
+            "path": os.path.relpath(str(abspath), self.config["root"]),
+            "message": str(message),
+            "level": "error",
+            "lineno": line,
+            "column": col,
+            "rule": code,
+        }
+
+        results.append(result.from_config(self.config, **res))
+
+
+def get_yamllint_version():
+    from yamllint import APP_VERSION
+
+    return APP_VERSION
+
+
+def run_process(config, cmd):
+    proc = YAMLLintProcess(config, cmd)
+    proc.run()
+    try:
+        proc.wait()
+    except KeyboardInterrupt:
+        proc.kill()
+
+
+def gen_yamllint_args(cmdargs, paths=None, conf_file=None):
+    args = cmdargs[:]
+    if isinstance(paths, str):
+        paths = [paths]
+    if conf_file and conf_file != "default":
+        return args + ["-c", conf_file] + paths
+    return args + paths
+
+
+def lint(files, config, **lintargs):
+    log = lintargs["log"]
+
+    log.debug("Version: {}".format(get_yamllint_version()))
+
+    cmdargs = [
+        sys.executable,
+        os.path.join(topsrcdir, "mach"),
+        "python",
+        "--",
+        "-m",
+        "yamllint",
+        "-f",
+        "parsable",
+    ]
+    log.debug("Command: {}".format(" ".join(cmdargs)))
+
+    config = config.copy()
+    config["root"] = lintargs["root"]
+
+    # Run any paths with a .yamllint file in the directory separately so
+    # it gets picked up. This means only .yamllint files that live in
+    # directories that are explicitly included will be considered.
+    paths_by_config = defaultdict(list)
+    for f in files:
+        conf_files = get_ancestors_by_name(".yamllint", f, config["root"])
+        paths_by_config[conf_files[0] if conf_files else "default"].append(f)
+
+    for conf_file, paths in paths_by_config.items():
+        run_process(
+            config, gen_yamllint_args(cmdargs, conf_file=conf_file, paths=paths)
+        )
+
+    return results