#!/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,
)


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 main():
    upstream_root = clone_repo("https://github.com/foonathan/memory", "v0.7-2")
    wpilib_root = get_repo_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
    os.chdir(upstream_root)
    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(upstream_root, "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",
        ),
    )


if __name__ == "__main__":
    main()