#!/usr/bin/env python

# Copyright 2023, Thomas Atkinson

#

# SPDX-License-Identifier: Apache-2.0

#

# Licensed under the Apache License, Version 2.0 the "License";

# you may not use this file except in compliance with the License.

# You may obtain a copy of the License at

#

# http://www.apache.org/licenses/LICENSE-2.0

#

# Unless required by applicable law or agreed to in writing, software

# distributed under the License is distributed on an "AS IS" BASIS,

# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

# See the License for the specific language governing permissions and

# limitations under the License.

import argparse

import os

import sys

from shutil import which

from subprocess import check_output

# Get the file extension

def get_ext(file_path):

file_name = os.path.basename(file_path)

file_name, file_ext = os.path.splitext(file_name)

return file_ext

class terminal_colors:

SUCCESS = "\033[92m"

INFO = "\033[94m"

WARNING = "\033[33m"

ERROR = "\033[91m"

END = "\033[0m"

if __name__ == "__main__":

argument_parser = argparse.ArgumentParser(

description="Format C/C++ files using clang-format"

)

argument_parser.add_argument(

"branch",

type=str,

default="main",

nargs="?",

help="Branch from which to compute the diff",

)

args = argument_parser.parse_args()

if len(sys.argv) == 1:

argument_parser.print_help(sys.stderr)

sys.exit(1)

files = None

if not which("git"):

print(terminal_colors.ERROR + "Missing git" + terminal_colors.END)

sys.exit(1)

if not which("clang-format"):

print(terminal_colors.ERROR + "Missing clang-format" + terminal_colors.END)

sys.exit(1)

out = check_output(["git", "diff", args.branch, "--name-only"])

check_files = [".h", ".hpp", ".cpp"]

files = out.decode("utf-8").split("\n")

files = [f for f in files if f and get_ext(f) in check_files]

if files and len(files) > 0:

print(terminal_colors.INFO + "Formatting files:" + terminal_colors.END)

for f in files:

print(terminal_colors.INFO + " " + f + terminal_colors.END)

print()

for f in files:

if os.path.isfile(f):

check_output(["clang-format", "-i", f])

else:

print(terminal_colors.INFO + "No files to format" + terminal_colors.END)