mirror of
https://github.com/OpenRCT2/OpenRCT2.git
synced 2024-08-29 21:46:05 +02:00
361 lines
11 KiB
Python
Executable file
361 lines
11 KiB
Python
Executable file
#!/usr/bin/env python
|
|
"""A wrapper script around clang-format, suitable for linting multiple files
|
|
and to use for continuous integration.
|
|
|
|
This is an alternative API for the clang-format command line.
|
|
It runs over multiple files and directories in parallel.
|
|
A diff output is produced and a sensible exit code is returned.
|
|
|
|
"""
|
|
|
|
from __future__ import print_function, unicode_literals
|
|
|
|
import argparse
|
|
import codecs
|
|
import difflib
|
|
import fnmatch
|
|
import io
|
|
import multiprocessing
|
|
import os
|
|
import signal
|
|
import subprocess
|
|
import sys
|
|
import traceback
|
|
|
|
from functools import partial
|
|
|
|
DEFAULT_EXTENSIONS = 'c,h,C,H,cpp,hpp,cc,hh,c++,h++,cxx,hxx'
|
|
|
|
|
|
class ExitStatus:
|
|
SUCCESS = 0
|
|
DIFF = 1
|
|
TROUBLE = 2
|
|
|
|
|
|
def list_files(files, recursive=False, extensions=None, exclude=None):
|
|
if extensions is None:
|
|
extensions = []
|
|
if exclude is None:
|
|
exclude = []
|
|
|
|
out = []
|
|
for file in files:
|
|
if recursive and os.path.isdir(file):
|
|
for dirpath, dnames, fnames in os.walk(file):
|
|
fpaths = [os.path.join(dirpath, fname) for fname in fnames]
|
|
for pattern in exclude:
|
|
# os.walk() supports trimming down the dnames list
|
|
# by modifying it in-place,
|
|
# to avoid unnecessary directory listings.
|
|
dnames[:] = [
|
|
x for x in dnames
|
|
if
|
|
not fnmatch.fnmatch(os.path.join(dirpath, x), pattern)
|
|
]
|
|
fpaths = [
|
|
x for x in fpaths if not fnmatch.fnmatch(x, pattern)
|
|
]
|
|
for f in fpaths:
|
|
ext = os.path.splitext(f)[1][1:]
|
|
if ext in extensions:
|
|
out.append(f)
|
|
else:
|
|
out.append(file)
|
|
return out
|
|
|
|
|
|
def make_diff(file, original, reformatted):
|
|
return list(
|
|
difflib.unified_diff(
|
|
original,
|
|
reformatted,
|
|
fromfile='{}\t(original)'.format(file),
|
|
tofile='{}\t(reformatted)'.format(file),
|
|
n=3))
|
|
|
|
|
|
class DiffError(Exception):
|
|
def __init__(self, message, errs=None):
|
|
super(DiffError, self).__init__(message)
|
|
self.errs = errs or []
|
|
|
|
|
|
class UnexpectedError(Exception):
|
|
def __init__(self, message, exc=None):
|
|
super(UnexpectedError, self).__init__(message)
|
|
self.formatted_traceback = traceback.format_exc()
|
|
self.exc = exc
|
|
|
|
def run_clang_format_version(args):
|
|
invocation = [args.clang_format_executable, "--version"]
|
|
|
|
encoding_py3 = {}
|
|
if sys.version_info[0] >= 3:
|
|
encoding_py3['encoding'] = 'utf-8'
|
|
|
|
try:
|
|
proc = subprocess.Popen(
|
|
invocation,
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE,
|
|
universal_newlines=True,
|
|
**encoding_py3)
|
|
except OSError as exc:
|
|
raise DiffError(str(exc))
|
|
proc_stdout = proc.stdout
|
|
proc_stderr = proc.stderr
|
|
if sys.version_info[0] < 3:
|
|
# make the pipes compatible with Python 3,
|
|
# reading lines should output unicode
|
|
encoding = 'utf-8'
|
|
proc_stdout = codecs.getreader(encoding)(proc_stdout)
|
|
proc_stderr = codecs.getreader(encoding)(proc_stderr)
|
|
# hopefully the stderr pipe won't get full and block the process
|
|
outs = list(proc_stdout.readlines())
|
|
errs = list(proc_stderr.readlines())
|
|
proc.wait()
|
|
if proc.returncode:
|
|
raise DiffError("clang-format exited with status {}: '{}'".format(
|
|
proc.returncode, file), errs)
|
|
print(outs[0])
|
|
return
|
|
|
|
def run_clang_format_diff_wrapper(args, file):
|
|
try:
|
|
ret = run_clang_format_diff(args, file)
|
|
return ret
|
|
except DiffError:
|
|
raise
|
|
except Exception as e:
|
|
raise UnexpectedError('{}: {}: {}'.format(file, e.__class__.__name__,
|
|
e), e)
|
|
|
|
|
|
def run_clang_format_diff(args, file):
|
|
try:
|
|
with io.open(file, 'r', encoding='utf-8') as f:
|
|
original = f.readlines()
|
|
except IOError as exc:
|
|
raise DiffError(str(exc))
|
|
invocation = [args.clang_format_executable, file]
|
|
|
|
# Use of utf-8 to decode the process output.
|
|
#
|
|
# Hopefully, this is the correct thing to do.
|
|
#
|
|
# It's done due to the following assumptions (which may be incorrect):
|
|
# - clang-format will returns the bytes read from the files as-is,
|
|
# without conversion, and it is already assumed that the files use utf-8.
|
|
# - if the diagnostics were internationalized, they would use utf-8:
|
|
# > Adding Translations to Clang
|
|
# >
|
|
# > Not possible yet!
|
|
# > Diagnostic strings should be written in UTF-8,
|
|
# > the client can translate to the relevant code page if needed.
|
|
# > Each translation completely replaces the format string
|
|
# > for the diagnostic.
|
|
# > -- http://clang.llvm.org/docs/InternalsManual.html#internals-diag-translation
|
|
#
|
|
# It's not pretty, due to Python 2 & 3 compatibility.
|
|
encoding_py3 = {}
|
|
if sys.version_info[0] >= 3:
|
|
encoding_py3['encoding'] = 'utf-8'
|
|
|
|
try:
|
|
proc = subprocess.Popen(
|
|
invocation,
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE,
|
|
universal_newlines=True,
|
|
**encoding_py3)
|
|
except OSError as exc:
|
|
raise DiffError(str(exc))
|
|
proc_stdout = proc.stdout
|
|
proc_stderr = proc.stderr
|
|
if sys.version_info[0] < 3:
|
|
# make the pipes compatible with Python 3,
|
|
# reading lines should output unicode
|
|
encoding = 'utf-8'
|
|
proc_stdout = codecs.getreader(encoding)(proc_stdout)
|
|
proc_stderr = codecs.getreader(encoding)(proc_stderr)
|
|
# hopefully the stderr pipe won't get full and block the process
|
|
outs = list(proc_stdout.readlines())
|
|
errs = list(proc_stderr.readlines())
|
|
proc.wait()
|
|
if proc.returncode:
|
|
raise DiffError("clang-format exited with status {}: '{}'".format(
|
|
proc.returncode, file), errs)
|
|
return make_diff(file, original, outs), errs
|
|
|
|
|
|
def bold_red(s):
|
|
return '\x1b[1m\x1b[31m' + s + '\x1b[0m'
|
|
|
|
|
|
def colorize(diff_lines):
|
|
def bold(s):
|
|
return '\x1b[1m' + s + '\x1b[0m'
|
|
|
|
def cyan(s):
|
|
return '\x1b[36m' + s + '\x1b[0m'
|
|
|
|
def green(s):
|
|
return '\x1b[32m' + s + '\x1b[0m'
|
|
|
|
def red(s):
|
|
return '\x1b[31m' + s + '\x1b[0m'
|
|
|
|
for line in diff_lines:
|
|
if line[:4] in ['--- ', '+++ ']:
|
|
yield bold(line)
|
|
elif line.startswith('@@ '):
|
|
yield cyan(line)
|
|
elif line.startswith('+'):
|
|
yield green(line)
|
|
elif line.startswith('-'):
|
|
yield red(line)
|
|
else:
|
|
yield line
|
|
|
|
|
|
def print_diff(diff_lines, use_color):
|
|
if use_color:
|
|
diff_lines = colorize(diff_lines)
|
|
if sys.version_info[0] < 3:
|
|
sys.stdout.writelines((l.encode('utf-8') for l in diff_lines))
|
|
else:
|
|
sys.stdout.writelines(diff_lines)
|
|
|
|
|
|
def print_trouble(prog, message, use_colors):
|
|
error_text = 'error:'
|
|
if use_colors:
|
|
error_text = bold_red(error_text)
|
|
print("{}: {} {}".format(prog, error_text, message), file=sys.stderr)
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description=__doc__)
|
|
parser.add_argument(
|
|
'--clang-format-executable',
|
|
metavar='EXECUTABLE',
|
|
help='path to the clang-format executable',
|
|
default='clang-format')
|
|
parser.add_argument(
|
|
'--extensions',
|
|
help='comma separated list of file extensions (default: {})'.format(
|
|
DEFAULT_EXTENSIONS),
|
|
default=DEFAULT_EXTENSIONS)
|
|
parser.add_argument(
|
|
'-r',
|
|
'--recursive',
|
|
action='store_true',
|
|
help='run recursively over directories')
|
|
parser.add_argument('files', metavar='file', nargs='+')
|
|
parser.add_argument(
|
|
'-q',
|
|
'--quiet',
|
|
action='store_true')
|
|
parser.add_argument(
|
|
'-j',
|
|
metavar='N',
|
|
type=int,
|
|
default=0,
|
|
help='run N clang-format jobs in parallel'
|
|
' (default number of cpus + 1)')
|
|
parser.add_argument(
|
|
'--color',
|
|
default='auto',
|
|
choices=['auto', 'always', 'never'],
|
|
help='show colored diff (default: auto)')
|
|
parser.add_argument(
|
|
'-e',
|
|
'--exclude',
|
|
metavar='PATTERN',
|
|
action='append',
|
|
default=[],
|
|
help='exclude paths matching the given glob-like pattern(s)'
|
|
' from recursive search')
|
|
|
|
args = parser.parse_args()
|
|
|
|
run_clang_format_version(args)
|
|
|
|
# use default signal handling, like diff return SIGINT value on ^C
|
|
# https://bugs.python.org/issue14229#msg156446
|
|
signal.signal(signal.SIGINT, signal.SIG_DFL)
|
|
try:
|
|
signal.SIGPIPE
|
|
except AttributeError:
|
|
# compatibility, SIGPIPE does not exist on Windows
|
|
pass
|
|
else:
|
|
signal.signal(signal.SIGPIPE, signal.SIG_DFL)
|
|
|
|
colored_stdout = False
|
|
colored_stderr = False
|
|
if args.color == 'always':
|
|
colored_stdout = True
|
|
colored_stderr = True
|
|
elif args.color == 'auto':
|
|
colored_stdout = sys.stdout.isatty()
|
|
colored_stderr = sys.stderr.isatty()
|
|
|
|
retcode = ExitStatus.SUCCESS
|
|
files = list_files(
|
|
args.files,
|
|
recursive=args.recursive,
|
|
exclude=args.exclude,
|
|
extensions=args.extensions.split(','))
|
|
|
|
if not files:
|
|
return
|
|
|
|
njobs = args.j
|
|
if njobs == 0:
|
|
njobs = multiprocessing.cpu_count() + 1
|
|
njobs = min(len(files), njobs)
|
|
|
|
if njobs == 1:
|
|
# execute directly instead of in a pool,
|
|
# less overhead, simpler stacktraces
|
|
it = (run_clang_format_diff_wrapper(args, file) for file in files)
|
|
pool = None
|
|
else:
|
|
pool = multiprocessing.Pool(njobs)
|
|
it = pool.imap_unordered(
|
|
partial(run_clang_format_diff_wrapper, args), files)
|
|
while True:
|
|
try:
|
|
outs, errs = next(it)
|
|
except StopIteration:
|
|
break
|
|
except DiffError as e:
|
|
print_trouble(parser.prog, str(e), use_colors=colored_stderr)
|
|
retcode = ExitStatus.TROUBLE
|
|
sys.stderr.writelines(e.errs)
|
|
except UnexpectedError as e:
|
|
print_trouble(parser.prog, str(e), use_colors=colored_stderr)
|
|
sys.stderr.write(e.formatted_traceback)
|
|
retcode = ExitStatus.TROUBLE
|
|
# stop at the first unexpected error,
|
|
# something could be very wrong,
|
|
# don't process all files unnecessarily
|
|
if pool:
|
|
pool.terminate()
|
|
break
|
|
else:
|
|
sys.stderr.writelines(errs)
|
|
if outs == []:
|
|
continue
|
|
if not args.quiet:
|
|
print_diff(outs, use_color=colored_stdout)
|
|
if retcode == ExitStatus.SUCCESS:
|
|
retcode = ExitStatus.DIFF
|
|
return retcode
|
|
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(main())
|