#!/usr/bin/env python3

import shutil
from pathlib import Path

from upstream_utils import Lib


def copy_upstream_src(wpilib_root: Path):
    wpiutil = wpilib_root / "wpiutil"

    # Copy header into allwpilib
    dest_filename = (
        wpiutil / f"src/main/native/thirdparty/argparse/include/wpi/util/argparse.hpp"
    )
    shutil.copyfile("include/argparse/argparse.hpp", dest_filename)
    # Rename namespace from argparse to wpi
    with open(dest_filename) as f:
        content = f.read()
    content = content.replace("namespace argparse", "namespace wpi::util")
    content = content.replace("argparse::", "wpi::util::")
    content = content.replace("ARGPARSE_", "WPI_")
    with open(dest_filename, "w") as f:
        f.write(content)


def main():
    name = "argparse_lib"
    url = "https://github.com/p-ranav/argparse"
    # master on 2024-09-11
    tag = "fd13c2859131ab463e617a5a8abcc69eb7e1d897"

    expected = Lib(name, url, tag, copy_upstream_src)
    expected.main()


if __name__ == "__main__":
    main()
