mirror of
https://github.com/neondatabase/neon.git
synced 2025-12-22 21:59:59 +00:00
56 lines
1.4 KiB
Python
Executable File
56 lines
1.4 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
from typing import List
|
|
import subprocess
|
|
import sys
|
|
|
|
RED = "\033[0;31m"
|
|
GREEN = "\033[0;33m"
|
|
CYAN = "\033[0;36m"
|
|
NC = "\033[0m" # No Color
|
|
|
|
|
|
def fix_inplace() -> bool:
|
|
if len(sys.argv) == 2 and sys.argv[1] == "--fix-inplace":
|
|
return True
|
|
return False
|
|
|
|
|
|
def get_commit_files() -> List[str]:
|
|
files = subprocess.check_output(
|
|
"git diff --cached --name-only --diff-filter=ACM".split()
|
|
)
|
|
return files.decode().splitlines()
|
|
|
|
|
|
def check(name: str, suffix: str, cmd: str, changed_files: List[str]):
|
|
print(f"Checking: {name} ", end="")
|
|
applicable_files = list(
|
|
filter(lambda fname: fname.strip().endswith(suffix), changed_files)
|
|
)
|
|
if not applicable_files:
|
|
print(f"{CYAN}[NOT APPLICABLE]{NC}")
|
|
return
|
|
|
|
cmd = f'{cmd} {" ".join(applicable_files)}'
|
|
res = subprocess.run(cmd.split(), capture_output=True)
|
|
if res.returncode != 0:
|
|
print(f"{RED}[FAILED]{NC}")
|
|
print("Please inspect the output below and run make fmt to fix automatically\n")
|
|
print(res.stdout.decode())
|
|
exit(1)
|
|
print(f"{GREEN}[OK]{NC}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
files = get_commit_files()
|
|
rustfmt = "rustfmt --edition=2018"
|
|
if not fix_inplace():
|
|
rustfmt += " --check"
|
|
check(
|
|
name="rustfmt",
|
|
suffix=".rs",
|
|
cmd=rustfmt,
|
|
changed_files=files,
|
|
)
|