format.py

#!/usr/bin/env python3
import argparse
import glob
import multiprocessing
import re
import subprocess
import sys

from concurrent.futures import ProcessPoolExecutor

CLANG_VER = 16

FORMAT_OPTS = "-i -style=file"
TIDY_OPTS = "-p . --fix --fix-errors"

COMPILER_OPTS = "-std=gnu11"


parser = argparse.ArgumentParser(description="Format files using clang-format and clang-tidy")
parser.add_argument("-j", "--jobs", type=int, help="Number of concurrent jobs to run")


def get_clang(program: str) -> str:
    try:
        subprocess.run(
            f"clang-{program}-{CLANG_VER} --version".split(),
            check=True,
            stdout=subprocess.DEVNULL,
            stderr=subprocess.DEVNULL,
        )
        return f"clang-{program}-{CLANG_VER}"
    except FileNotFoundError:
        out = subprocess.run(f"clang-{program} --version".split(), check=True, capture_output=True)
        ver_re = re.compile(r"version (\d+)(\.\d+\.\d+)")
        match = re.search(ver_re, out.stdout.decode())
        if match:
            ver = int(match.group(1))
            if ver >= CLANG_VER:
                return f"clang-{program}"
    return ""


def format_file(files: list[str]) -> None:
    subprocess.run(
        f"{clang_format} {FORMAT_OPTS} {' '.join(files)}".split(),
        check=True,
        stdout=subprocess.DEVNULL,
        stderr=subprocess.DEVNULL,
    )


def tidy_file(files: list[str]) -> None:
    subprocess.run(
        f"{clang_tidy} {TIDY_OPTS} {' '.join(files)} -- {COMPILER_OPTS}".split(),
        check=True,
        stdout=subprocess.DEVNULL,
        stderr=subprocess.DEVNULL,
    )


def main(jobs: int) -> None:
    file_list = glob.glob("src/**/*.[c,h]", recursive=True)
    num_jobs = jobs if jobs and jobs > 0 else multiprocessing.cpu_count()

    print(f"Formatting {len(file_list)} files with {num_jobs} jobs...")
    if num_jobs == 1:
        format_file(file_list)
        tidy_file(file_list)
    else:
        with ProcessPoolExecutor(max_workers=num_jobs) as executor:
            # Split file list into num_jobs chunks where each chunk is the same size
            chunks = [
                file_list[i : i + len(file_list) // num_jobs + 1]
                for i in range(0, len(file_list), len(file_list) // num_jobs + 1)
            ]
            executor.map(format_file, chunks)
            executor.map(tidy_file, chunks)
    print("Done formatting all files.")


if __name__ == "__main__":
    try:
        clang_format = get_clang("format")
    except FileNotFoundError:
        sys.stderr.write(f"error: clang-format {CLANG_VER} not found\n")
        sys.exit(1)

    try:
        clang_tidy = get_clang("tidy")
    except FileNotFoundError:
        sys.stderr.write(f"error: clang-tidy {CLANG_VER} not found\n")
        sys.exit(1)

    args = parser.parse_args()
    main(args.jobs)