rename_headers.py 2.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475
  1. #!/usr/bin/env python3
  2. #
  3. # This script renames SDL headers in the specified paths
  4. import argparse
  5. import pathlib
  6. import re
  7. def do_include_replacements(paths):
  8. replacements = [
  9. ( re.compile(r"(?:[\"<])(?:SDL2/)?SDL_image.h(?:[\">])"), r"<SDL3_image/SDL_image.h>" ),
  10. ( re.compile(r"(?:[\"<])(?:SDL2/)?SDL_mixer.h(?:[\">])"), r"<SDL3_mixer/SDL_mixer.h>" ),
  11. ( re.compile(r"(?:[\"<])(?:SDL2/)?SDL_net.h(?:[\">])"), r"<SDL3_net/SDL_net.h>" ),
  12. ( re.compile(r"(?:[\"<])(?:SDL2/)?SDL_rtf.h(?:[\">])"), r"<SDL3_rtf/SDL_rtf.h>" ),
  13. ( re.compile(r"(?:[\"<])(?:SDL2/)?SDL_ttf.h(?:[\">])"), r"<SDL3_ttf/SDL_ttf.h>" ),
  14. ( re.compile(r"(?:[\"<])(?:SDL2/)?SDL_gamecontroller.h(?:[\">])"), r"<SDL3/SDL_gamepad.h>" ),
  15. ( re.compile(r"(?:[\"<])(?:SDL2/)?begin_code.h(?:[\">])"), r"<SDL3/SDL_begin_code.h>" ),
  16. ( re.compile(r"(?:[\"<])(?:SDL2/)?close_code.h(?:[\">])"), r"<SDL3/SDL_close_code.h>" ),
  17. ( re.compile(r"(?:[\"<])(?:SDL2/)?(SDL[_a-z0-9]*\.h)(?:[\">])"), r"<SDL3/\1>" )
  18. ]
  19. for entry in paths:
  20. path = pathlib.Path(entry)
  21. if not path.exists():
  22. print("{} does not exist, skipping".format(entry))
  23. continue
  24. replace_headers_in_path(path, replacements)
  25. def replace_headers_in_file(file, replacements):
  26. try:
  27. with file.open("r", encoding="UTF-8", newline="") as rfp:
  28. original = rfp.read()
  29. contents = original
  30. for regex, replacement in replacements:
  31. contents = regex.sub(replacement, contents)
  32. if contents != original:
  33. with file.open("w", encoding="UTF-8", newline="") as wfp:
  34. wfp.write(contents)
  35. except UnicodeDecodeError:
  36. print("%s is not text, skipping" % file)
  37. except Exception as err:
  38. print("%s" % err)
  39. def replace_headers_in_dir(path, replacements):
  40. for entry in path.glob("*"):
  41. if entry.is_dir():
  42. replace_headers_in_dir(entry, replacements)
  43. else:
  44. print("Processing %s" % entry)
  45. replace_headers_in_file(entry, replacements)
  46. def replace_headers_in_path(path, replacements):
  47. if path.is_dir():
  48. replace_headers_in_dir(path, replacements)
  49. else:
  50. replace_headers_in_file(path, replacements)
  51. def main():
  52. parser = argparse.ArgumentParser(fromfile_prefix_chars='@', description="Rename #include's for SDL3.")
  53. parser.add_argument("args", metavar="PATH", nargs="*", help="Input source file")
  54. args = parser.parse_args()
  55. try:
  56. do_include_replacements(args.args)
  57. except Exception as e:
  58. print(e)
  59. return 1
  60. if __name__ == "__main__":
  61. raise SystemExit(main())