mirror of
https://github.com/ddnet/ddnet.git
synced 2024-10-30 04:38:18 +00:00
91 lines
2.4 KiB
Python
Executable file
91 lines
2.4 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
import argparse
|
|
|
|
os.chdir(os.path.dirname(__file__) + "/..")
|
|
|
|
def recursive_file_list(path):
|
|
result = []
|
|
for dirpath, _, filenames in os.walk(path):
|
|
result += [os.path.join(dirpath, filename) for filename in filenames]
|
|
return result
|
|
|
|
IGNORE_FILES = [
|
|
"src/engine/client/keynames.h",
|
|
"src/engine/keys.h",
|
|
]
|
|
IGNORE_DIRS = [
|
|
"src/game/generated",
|
|
"src/rust-bridge"
|
|
]
|
|
def filter_ignored(filenames):
|
|
result = []
|
|
for filename in filenames:
|
|
real_filename = os.path.realpath(filename)
|
|
if real_filename not in [os.path.realpath(ignore_file) for ignore_file in IGNORE_FILES] \
|
|
and not any(real_filename.startswith(os.path.realpath(subdir) + os.path.sep) for subdir in IGNORE_DIRS):
|
|
result.append(filename)
|
|
|
|
return result
|
|
|
|
def filter_cpp(filenames):
|
|
return [filename for filename in filenames
|
|
if any(filename.endswith(ext) for ext in ".c .cpp .h".split())]
|
|
|
|
def find_clang_format(version):
|
|
for binary in (
|
|
"clang-format",
|
|
f"clang-format-{version}",
|
|
f"/opt/clang-format-static/clang-format-{version}"):
|
|
try:
|
|
out = subprocess.check_output([binary, "--version"])
|
|
except FileNotFoundError:
|
|
continue
|
|
if f"clang-format version {version}." in out.decode("utf-8"):
|
|
return binary
|
|
print(f"Found no clang-format {version}")
|
|
sys.exit(-1)
|
|
|
|
clang_format_bin = find_clang_format(10)
|
|
|
|
def reformat(filenames):
|
|
for filename in filenames:
|
|
with open(filename, 'r+b') as f:
|
|
try:
|
|
f.seek(-1, os.SEEK_END)
|
|
if f.read(1) != b'\n':
|
|
f.write(b'\n')
|
|
except OSError:
|
|
f.seek(0)
|
|
subprocess.check_call([clang_format_bin, "-i"] + filenames)
|
|
|
|
def warn(filenames):
|
|
clang = subprocess.call([clang_format_bin, "-Werror", "--dry-run"] + filenames)
|
|
newline = 0
|
|
for filename in filenames:
|
|
with open(filename, 'rb') as f:
|
|
try:
|
|
f.seek(-1, os.SEEK_END)
|
|
if f.read(1) != b'\n':
|
|
print(filename + ": error: missing newline at EOF", file=sys.stderr)
|
|
newline = 1
|
|
except OSError:
|
|
f.seek(0)
|
|
return clang or newline
|
|
|
|
def main():
|
|
p = argparse.ArgumentParser(description="Check and fix style of changed files")
|
|
p.add_argument("-n", "--dry-run", action="store_true", help="Don't fix, only warn")
|
|
args = p.parse_args()
|
|
filenames = filter_ignored(filter_cpp(recursive_file_list("src")))
|
|
if not args.dry_run:
|
|
reformat(filenames)
|
|
else:
|
|
sys.exit(warn(filenames))
|
|
|
|
if __name__ == "__main__":
|
|
main()
|