mirror of
https://github.com/CHOMPStation2/CHOMPStation2.git
synced 2025-12-10 10:12:45 +00:00
52 lines
1.6 KiB
Python
52 lines
1.6 KiB
Python
import sys
|
|
import re
|
|
import os.path as path
|
|
|
|
# Usage: tools/bootstrap/python -m dm_annotator [filename]
|
|
# If filename is not provided, stdin is checked instead
|
|
|
|
def red(text):
|
|
return "\033[31m" + str(text) + "\033[0m"
|
|
|
|
def green(text):
|
|
return "\033[32m" + str(text) + "\033[0m"
|
|
|
|
def yellow(text):
|
|
return "\033[33m" + str(text) + "\033[0m"
|
|
|
|
def annotate(raw_output):
|
|
# Remove ANSI escape codes
|
|
raw_output = re.sub(r'(\x9B|\x1B\[)[0-?]*[ -\/]*[@-~]', '', raw_output)
|
|
|
|
print("::group::DreamChecker Output")
|
|
print(raw_output)
|
|
print("::endgroup::")
|
|
|
|
annotation_regex = r'(?P<filename>.*?), line (?P<line>\d+), column (?P<column>\d+):\s{1,2}(?P<type>error|warning): (?P<message>.*)'
|
|
has_issues = False
|
|
|
|
print("DM Code Annotations:")
|
|
for annotation in re.finditer(annotation_regex, raw_output):
|
|
print(f"::{annotation['type']} file={annotation['filename']},line={annotation['line']},col={annotation['column']}::{annotation['message']}")
|
|
has_issues = True
|
|
|
|
if not has_issues:
|
|
print(green("No DM issues found"))
|
|
|
|
def main():
|
|
if len(sys.argv) > 1:
|
|
if not path.exists(sys.argv[1]):
|
|
print(red(f"Error: Annotations file '{sys.argv[1]}' does not exist"))
|
|
sys.exit(1)
|
|
with open(sys.argv[1], 'r') as f:
|
|
annotate(f.read())
|
|
elif not sys.stdin.isatty():
|
|
annotate(sys.stdin.read())
|
|
else:
|
|
print(red("Error: No input provided"))
|
|
print("Usage: tools/bootstrap/python -m dm_annotator [filename]")
|
|
sys.exit(1)
|
|
|
|
if __name__ == '__main__':
|
|
main()
|