71 lines
2.4 KiB
Plaintext
71 lines
2.4 KiB
Plaintext
|
#!/usr/bin/env python3
|
||
|
|
import functools
|
||
|
|
import os
|
||
|
|
from pprint import pprint
|
||
|
|
import re
|
||
|
|
import subprocess
|
||
|
|
import sys
|
||
|
|
|
||
|
|
class ValidationError(Exception): pass
|
||
|
|
|
||
|
|
def check_output(cmd, expect_process_error=False):
|
||
|
|
try:
|
||
|
|
return subprocess.check_output(cmd)
|
||
|
|
except Exception as e:
|
||
|
|
if not expect_process_error or not isinstance(e, subprocess.CalledProcessError):
|
||
|
|
print('Exception while running cmd %s' % (cmd,))
|
||
|
|
raise
|
||
|
|
|
||
|
|
def changes():
|
||
|
|
return check_output(('git', 'diff', '--no-ext-diff', '--cached', '-U0'))
|
||
|
|
|
||
|
|
def toplevel():
|
||
|
|
return check_output(('git', 'rev-parse', '--show-toplevel')).strip(b'\n').decode('utf-8')
|
||
|
|
|
||
|
|
def toList(fn):
|
||
|
|
return functools.wraps(fn)(lambda *args, **kwargs: list(fn(*args, **kwargs)))
|
||
|
|
|
||
|
|
top = toplevel()
|
||
|
|
|
||
|
|
clangformat = "/usr/local/clang_3_6_0/bin/clang-format"
|
||
|
|
|
||
|
|
### CLANG-FORMAT BEGIN
|
||
|
|
if os.path.exists(os.path.join(top, '.clang-format')):
|
||
|
|
reformatted_files = []
|
||
|
|
files_had_unstaged_edits = []
|
||
|
|
for filename in check_output(['git', 'diff-index', '--cached', '--name-only', 'HEAD']).decode('utf-8').splitlines():
|
||
|
|
if any(filename.endswith(ext) for ext in ('.cpp', '.h', '.hpp')):
|
||
|
|
file_path = os.path.join(top, filename)
|
||
|
|
|
||
|
|
with open(file_path, 'rb') as f:
|
||
|
|
staged = f.read()
|
||
|
|
formatted = check_output([clangformat, file_path])
|
||
|
|
|
||
|
|
if staged == formatted:
|
||
|
|
continue
|
||
|
|
|
||
|
|
try:
|
||
|
|
check_output(['git', 'diff', '--exit-code', file_path], expect_process_error=True)
|
||
|
|
except subprocess.CalledProcessError:
|
||
|
|
files_had_unstaged_edits.append(filename)
|
||
|
|
|
||
|
|
check_output([clangformat, '-i', file_path])
|
||
|
|
|
||
|
|
try:
|
||
|
|
check_output(['git', 'diff', '--exit-code', file_path], expect_process_error=True)
|
||
|
|
except subprocess.CalledProcessError:
|
||
|
|
reformatted_files.append(filename)
|
||
|
|
|
||
|
|
if reformatted_files:
|
||
|
|
print('These files were reformatted with clang-format, please stage the changes first:\n- %s'
|
||
|
|
% '\n- '.join(reformatted_files), file=sys.stderr)
|
||
|
|
if files_had_unstaged_edits:
|
||
|
|
print('\nThese files had unstaged edits before clang-format reformatting. Please stage desired parts and confirm the commit again.\n- %s'
|
||
|
|
% '\n- '.join(files_had_unstaged_edits), file=sys.stderr)
|
||
|
|
if reformatted_files or files_had_unstaged_edits:
|
||
|
|
exit(1)
|
||
|
|
### CLANG-FORMAT END
|
||
|
|
|
||
|
|
exit(0)
|
||
|
|
|