0
0
mirror of https://github.com/rbock/sqlpp11.git synced 2024-12-27 16:41:08 +08:00
sqlpp11/pre-commit

71 lines
2.4 KiB
Plaintext
Raw Permalink Normal View History

#!/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)