#include "build.hpp"

std::filesystem::path OBJ_DIR = "obj";
std::filesystem::path INC_DIR = "inc";
std::filesystem::path SRC_DIR = "src";
std::filesystem::path RAYLIB_DIR = "raylib";
std::filesystem::path BUILD_FILE = "main";

bool web_build = false;

std::string cpp_compiler = "g++";
std::string c_compiler = "gcc";
std::string opt_flags = "-ggdb";
std::vector<std::string> command;
std::vector<std::string> RAYINCLUDE = {"-Iraylib/src", "-Iraylib/src/external/glfw/include", "-Iraylib/src/external/glfw/deps/mingw"};

void compile_raylib_dir()
{
  std::filesystem::path out_dir = OBJ_DIR / RAYLIB_DIR;
  std::filesystem::path ray_src_dir = RAYLIB_DIR / "src";
  std::filesystem::create_directory(out_dir);
  std::vector<std::filesystem::path> ray_srcs = {"rcore.c", "rshapes.c", "rtextures.c", "rtext.c", "utils.c", "rmodels.c", "raudio.c"};
  std::vector<std::string> ray_flags = {"-DPLATFORM_WEB", "-DGRAPHICS_API_OPENGL_ES2"};

  if (!web_build)
  {
    ray_srcs.push_back("rglfw.c");
    ray_flags = {"-D_GNU_SOURCE", "-DPLATFORM_DESKTOP", "-DGRAPHICS_API_OPENGL_33"};
  }

  command.clear();
  command.push_back(c_compiler);
  command.push_back("-c");
  command.push_back("");
  command.push_back("-o");
  command.push_back("");

  command.insert(command.end(), ray_flags.begin(), ray_flags.end());
  command.push_back(opt_flags);
  command.insert(command.end(), RAYINCLUDE.begin(), RAYINCLUDE.end());

  std::filesystem::path out, src;

  for (size_t i = 0; i < ray_srcs.size(); i++)
  {
    src = ray_src_dir / ray_srcs[i];
    out = out_dir / ray_srcs[i].replace_extension(".o");
    if (check_if_rebuild(src, out))
    {
      command[2] = src.c_str();
      command[4] = out.c_str();
      nob_cmd_run_sync(command);
    }
  }
}

bool compile_src_dir()
{
  nob_directory src_dir = get_all_files_in_dir(SRC_DIR);
  nob_directory obj_dir;

  for (auto &&i : src_dir.files)
  {
    std::filesystem::path tmp = i;
    obj_dir.files.push_back(OBJ_DIR / tmp.replace_extension(".o"));
  }

  for (auto &&i : src_dir.dirs)
  {
    obj_dir.dirs.push_back(OBJ_DIR / i);
  }

  for (auto &&i : obj_dir.dirs)
  {
    std::filesystem::create_directory(i);
  }

  command.clear();
  command.push_back(cpp_compiler);
  command.push_back("-c");
  command.push_back("");
  command.push_back("-o");
  command.push_back("");
  command.push_back(opt_flags);
  command.insert(command.end(), RAYINCLUDE.begin(), RAYINCLUDE.end());
  command.push_back("-Iinc");

  command.push_back("-std=c++23");
  command.push_back("-Wall");
  command.push_back("-Werror");
  bool build = false;
  for (size_t i = 0; i < src_dir.files.size(); i++)
  {
    if (check_if_rebuild(src_dir.files[i], obj_dir.files[i]))
    {
      command[2] = src_dir.files[i];
      command[4] = obj_dir.files[i];
      if (!nob_cmd_run_sync(command))
        return false;

      build = true;
    }
  }
  return build;
}

void compile_obj_dir()
{
  nob_directory obj_dir = get_all_files_in_dir(OBJ_DIR);

  command.clear();
  command.push_back(cpp_compiler);

  for (auto &&i : obj_dir.files)
  {
    command.push_back(i);
  }

  command.push_back("-o");
  command.push_back(BUILD_FILE);
  nob_cmd_run_sync(command);
}

int main(int argc, char const *argv[])
{
  if (rebuild_my_self(__FILE__, argv[0]))
    return 0;

  std::filesystem::create_directory(OBJ_DIR);

  if (!std::filesystem::is_directory(RAYLIB_DIR))
  {
    command = {"git", "clone", "--depth", "1", "--branch", "5.0", "git@github.com:raysan5/raylib.git"};
    nob_cmd_run_sync(command);
  }
  if (!std::filesystem::is_directory(RAYLIB_DIR))
    return 0;
  compile_raylib_dir();

  if (compile_src_dir())
    compile_obj_dir();

  command = {"./main"};
  nob_cmd_run_sync(command);

  return 0;
}