mirror of https://github.com/zeldaret/tp.git
52 lines
1.8 KiB
Python
52 lines
1.8 KiB
Python
import click
|
|
import os
|
|
import re
|
|
import sys
|
|
|
|
@click.command()
|
|
@click.option('--path', required=True, help='Path to the header file or directory.')
|
|
def modify_headers(path):
|
|
if os.path.isfile(path):
|
|
process_file(path)
|
|
elif os.path.isdir(path):
|
|
for root, _, files in os.walk(path):
|
|
for file in files:
|
|
if file.endswith('.h'):
|
|
file_path = os.path.join(root, file)
|
|
process_file(file_path)
|
|
else:
|
|
sys.exit(f'Error: Path {path} is neither a file nor a directory.')
|
|
|
|
def process_file(file_path):
|
|
try:
|
|
with open(file_path, 'r') as file:
|
|
lines = file.readlines()
|
|
|
|
class_decl_pattern = re.compile(r'class\s+(\w+)\s*:')
|
|
doxygen_comment_pattern = re.compile(r'/\*\*\s+@ingroup actors-npcs\s+@class\s+\w+\s+@brief', re.MULTILINE)
|
|
|
|
for i, line in enumerate(lines):
|
|
match = class_decl_pattern.search(line)
|
|
if match:
|
|
class_name = match.group(1)
|
|
# Check if Doxygen comment is missing
|
|
if not doxygen_comment_pattern.search(''.join(lines[max(0, i-3):i])):
|
|
doxygen_comment = f"""/**\n * @ingroup actors-unsorted\n * @class {class_name}\n * @brief\n *\n * @details\n *\n */\n"""
|
|
lines.insert(i, doxygen_comment)
|
|
|
|
# Stop after processing the first class declaration
|
|
break
|
|
|
|
with open(file_path, 'w') as file:
|
|
file.writelines(lines)
|
|
|
|
print(f'Processed {file_path} successfully.')
|
|
|
|
except FileNotFoundError:
|
|
sys.exit(f'Error: File {file_path} not found.')
|
|
except Exception as e:
|
|
sys.exit(f'Error processing file {file_path}: {e}')
|
|
|
|
if __name__ == '__main__':
|
|
modify_headers()
|