import glob
from pprint import pprint

source_path = "/home/jacques/blender-git/blender/source"

files_to_check = set()
for filepath in glob.iglob(source_path + "/**/**", recursive=True):
    if filepath.endswith(".hh") or filepath.endswith(".h"):
        files_to_check.add(filepath)

ignored_guard_names = {
    "__RNA_ACCESS_H__",
    "__RNA_TYPES_H__",
    "__BLI_TASK_H__",
    "__BLI_ENDIAN_SWITCH_H__",
    "__BLI_MEMORY_UTILS_H__",
    "__PY_CAPI_UTILS_H__",
    "__BLI_UTILDEFINES_H__",
}

for filepath in sorted(files_to_check):
    with open(filepath) as fs:
        text = fs.read()

    lines = text.splitlines()

    for ifndef_line, line in enumerate(lines):
        if line.startswith("#ifndef __"):
            break
    else:
        continue

    define_line = ifndef_line + 1
    guard_name = lines[define_line].split(" ")[1]

    if guard_name in ignored_guard_names:
        continue

    for endif_line, line in reversed(list(enumerate(lines))):
        if line.startswith("#endif"):
            break

    lines[ifndef_line] = "#pragma once"
    lines[define_line] = ""
    lines[endif_line] = ""

    print(filepath)

    new_text = "\n".join(lines)
    with open(filepath, "w") as fs:
        fs.write(new_text)