mirror of
https://github.com/Aurorastation/Aurora.3.git
synced 2026-02-07 23:19:18 +00:00
49d1ea41a6
Ports and adapts TG's maplint, not all checks it has are present, this is just the first iteration. Updated python dependency to 3.11.9. Updated bootstrap. Fixed various maps to not fail maplint.
124 lines
3.8 KiB
Python
124 lines
3.8 KiB
Python
import argparse
|
|
import glob
|
|
import pathlib
|
|
import traceback
|
|
import yaml
|
|
|
|
from . import dmm, lint
|
|
from .error import MaplintError
|
|
|
|
def green(text):
|
|
return "\033[32m" + str(text) + "\033[0m"
|
|
|
|
def red(text):
|
|
return "\033[31m" + str(text) + "\033[0m"
|
|
|
|
def process_dmm(map_filename, lints: dict[str, lint.Lint]) -> list[MaplintError]:
|
|
problems: list[MaplintError] = []
|
|
|
|
with open(map_filename, "r", encoding="utf-8") as file:
|
|
try:
|
|
map_data = dmm.parse_dmm(file)
|
|
except MaplintError as error:
|
|
problems.append(error)
|
|
# No structured data to lint.
|
|
return problems
|
|
|
|
for lint_name, lint in lints.items():
|
|
try:
|
|
problems.extend(lint.run(map_data))
|
|
except KeyboardInterrupt:
|
|
raise
|
|
except Exception:
|
|
problems.append(MaplintError(
|
|
f"An exception occurred, this is either a bug in maplint or a bug in a lint. \n{traceback.format_exc()}",
|
|
lint_name,
|
|
))
|
|
|
|
return problems
|
|
|
|
def print_error(message: str, filename: str, line_number: int, github_error_style: bool):
|
|
if github_error_style:
|
|
print(f"::error file={filename},line={line_number},title=DMM Linter::{message}")
|
|
else:
|
|
print(red(f"- Error parsing {filename} (line {line_number}): {message}"))
|
|
|
|
def print_maplint_error(error: MaplintError, github_error_style: bool):
|
|
print_error(
|
|
f"{f'(in pop {error.pop_id}) ' if error.pop_id else ''}{f'(at {error.coordinates}) ' if error.coordinates else ''}{error}" + (f"\n {error.help}" if error.help is not None else ""),
|
|
error.file_name,
|
|
error.line_number,
|
|
github_error_style,
|
|
)
|
|
|
|
def main(args):
|
|
any_failed = False
|
|
github_error_style = args.github
|
|
|
|
lints: dict[str, lint.Lint] = {}
|
|
|
|
lint_base = pathlib.Path(__file__).parent.parent / "lints"
|
|
lint_filenames = []
|
|
if args.lints is None:
|
|
lint_filenames = lint_base.glob("*.yml")
|
|
else:
|
|
lint_filenames = [lint_base / f"{lint_name}.yml" for lint_name in args.lints]
|
|
|
|
for lint_filename in lint_filenames:
|
|
try:
|
|
lints[lint_filename] = lint.Lint(yaml.safe_load(lint_filename.read_text()))
|
|
except MaplintError as error:
|
|
print_maplint_error(error, github_error_style)
|
|
any_failed = True
|
|
except Exception:
|
|
print_error("Error loading lint file.", lint_filename, 1, github_error_style)
|
|
traceback.print_exc()
|
|
any_failed = True
|
|
|
|
for map_filename in (args.maps or glob.glob("_maps/**/*.dmm", recursive = True) + glob.glob("maps/**/*.dmm", recursive = True)):
|
|
print(map_filename, end = " ")
|
|
|
|
success = True
|
|
all_failures: list[MaplintError] = []
|
|
|
|
try:
|
|
problems = process_dmm(map_filename, lints)
|
|
if len(problems) > 0:
|
|
success = False
|
|
all_failures.extend(problems)
|
|
except KeyboardInterrupt:
|
|
raise
|
|
except Exception:
|
|
success = False
|
|
|
|
all_failures.append(MaplintError(
|
|
f"An exception occurred, this is either a bug in maplint or a bug in a lint.' {traceback.format_exc()}",
|
|
map_filename,
|
|
))
|
|
|
|
if success:
|
|
print(green("OK"))
|
|
else:
|
|
print(red("X"))
|
|
any_failed = True
|
|
|
|
for failure in all_failures:
|
|
print_maplint_error(failure, github_error_style)
|
|
|
|
if any_failed:
|
|
exit(1)
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(
|
|
prog = "maplint",
|
|
description = "Checks for common errors in maps.",
|
|
)
|
|
|
|
parser.add_argument("maps", nargs = "*")
|
|
parser.add_argument("--lints", nargs = "*")
|
|
parser.add_argument("--github", action='store_true')
|
|
|
|
args = parser.parse_args()
|
|
|
|
main(args)
|