[upstream_utils] Rework upstream_utils scripts (#6829)

upstream_utils/memory.py

@@ -0,0 +1,111 @@
+#!/usr/bin/env python3
+
+import os
+import shutil
+
+from upstream_utils import (
+    get_repo_root,
+    clone_repo,
+    comment_out_invalid_includes,
+    walk_if,
+    copy_to,
+    Lib,
+)
+
+
+def run_source_replacements(memory_files):
+    for wpi_file in memory_files:
+        with open(wpi_file) as f:
+            content = f.read()
+
+        # Fix #includes
+        content = content.replace('include "', 'include "wpi/memory/')
+        content = content.replace(
+            "wpi/memory/free_list_utils.hpp", "free_list_utils.hpp"
+        )
+
+        with open(wpi_file, "w") as f:
+            f.write(content)
+
+
+def run_header_replacements(memory_files):
+    for wpi_file in memory_files:
+        if "detail" not in wpi_file:
+            continue
+        with open(wpi_file) as f:
+            content = f.read()
+
+        # Fix #includes
+        content = content.replace('include "config.hpp', 'include "../config.hpp')
+
+        with open(wpi_file, "w") as f:
+            f.write(content)
+
+
+def run_global_replacements(memory_files):
+    for wpi_file in memory_files:
+        with open(wpi_file) as f:
+            content = f.read()
+
+        # Rename namespace from foonathan to wpi
+        content = content.replace("namespace foonathan", "namespace wpi")
+        content = content.replace("foonathan::", "wpi::")
+        content = content.replace("FOONATHAN_", "WPI_")
+
+        # Fix #includes
+        content = content.replace('include "foonathan', 'include "wpi')
+
+        with open(wpi_file, "w") as f:
+            f.write(content)
+
+
+def copy_upstream_src(wpilib_root):
+    wpiutil = os.path.join(wpilib_root, "wpiutil")
+
+    # Delete old install
+    for d in [
+        "src/main/native/thirdparty/memory/src",
+        "src/main/native/thirdparty/memory/include",
+    ]:
+        shutil.rmtree(os.path.join(wpiutil, d), ignore_errors=True)
+
+    # Copy sources
+    src_files = walk_if("src", lambda dp, f: f.endswith(".cpp") or f.endswith(".hpp"))
+    src_files = copy_to(
+        src_files, os.path.join(wpiutil, "src/main/native/thirdparty/memory")
+    )
+    run_global_replacements(src_files)
+    run_source_replacements(src_files)
+
+    # Copy headers
+    os.chdir(os.path.join("include", "foonathan"))
+    include_files = walk_if(".", lambda dp, f: f.endswith(".hpp"))
+    include_files = copy_to(
+        include_files,
+        os.path.join(wpiutil, "src/main/native/thirdparty/memory/include/wpi"),
+    )
+    os.chdir(os.path.join("..", ".."))
+    run_global_replacements(include_files)
+    run_header_replacements(include_files)
+
+    # Copy config_impl.hpp
+    shutil.copyfile(
+        os.path.join(wpilib_root, "upstream_utils/memory_files/config_impl.hpp"),
+        os.path.join(
+            wpiutil,
+            "src/main/native/thirdparty/memory/include/wpi/memory/config_impl.hpp",
+        ),
+    )
+
+
+def main():
+    name = "memory"
+    url = "https://github.com/foonathan/memory"
+    tag = "v0.7-3"
+
+    memory = Lib(name, url, tag, [], copy_upstream_src)
+    memory.main()
+
+
+if __name__ == "__main__":
+    main()