#!/usr/bin/env python3

import shutil
from pathlib import Path

from upstream_utils import Lib, walk_cwd_and_copy_if


def copy_upstream_src(wpilib_root: Path):
    stb = wpilib_root / "thirdparty/imgui_suite/stb"

    # Delete old install
    for d in ["include", "cpp"]:
        shutil.rmtree(stb / d, ignore_errors=True)

    # Copy files
    walk_cwd_and_copy_if(
        lambda dp, f: f == "stb_image.h",
        stb / "include",
    )

    (stb / "cpp").mkdir(parents=True)

    with open(stb / "cpp/stb_image.cpp", "w") as f:
        f.write(
            """#define STBI_WINDOWS_UTF8
#define STB_IMAGE_IMPLEMENTATION

#ifdef __GNUC__
#pragma GCC diagnostic ignored "-Wsign-compare"
#pragma GCC diagnostic ignored "-Wunused-but-set-variable"
#endif

#include "stb_image.h"
"""
        )


def main():
    name = "stb"
    url = "https://github.com/nothings/stb.git"
    tag = "c9064e317699d2e495f36ba4f9ac037e88ee371a"

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


if __name__ == "__main__":
    main()