// ============================================================= //
// nebuild
// Copyright (C) 2024-2025, Amlal El Mahrouss, all rights reserved.
// ============================================================= //

#include <BuildKit/JSONManifestBuilder.h>

using JSON = nlohmann::json;

namespace FS = std::filesystem;

using namespace NeBuild;

/// @brief Builds a JSON target from a JSON file.
/// @param arg_sz filename size (must be 1 or greater).
/// @param arg_val filename path (must be a valid JSON file).
/// @retval true building has succeeded.
/// @retval false fail to build, see error message.
bool JSONManifestBuilder::BuildTarget(const std::string& argv_val, const bool dry_run) {
  std::string path;

  if (argv_val.empty()) {
    NeBuild::Logger::info() << "nebuild: error: file path is empty" << std::endl;
    return false;
  } else {
    path = argv_val;

    if (!FS::exists(path)) {
      NeBuild::Logger::info() << "nebuild: error: file '" << path << "' does not exist"
                              << std::endl;
      return false;
    }
  }

  try {
    std::ifstream json(path);

    if (!json.good()) {
      NeBuild::Logger::info() << "nebuild: error: file '" << path << "' is not a valid JSON"
                              << std::endl;
      return false;
    }

    JSON json_obj = JSON::parse(json);

    std::string compiler = json_obj["compiler_path"].get<std::string>();

    JSON header_search_path = json_obj["headers_path"];
    JSON sources_files      = json_obj["sources_path"];

    std::string command = compiler + " ";

    for (auto& sources : sources_files) {
      command += sources.get<std::string>() + " ";
    }

    for (auto& headers : header_search_path) {
      command += "-I" + headers.get<std::string>() + " ";
    }

    JSON macros_list = json_obj["cpp_macros"];

    for (auto& macro : macros_list) {
      command += "-D" + macro.get<std::string>() + " ";
    }

    JSON compiler_flags = json_obj["compiler_flags"];

    for (auto& flag : compiler_flags) {
      command += flag.get<std::string>() + " ";
    }

    if (json_obj["compiler_std"].is_string())
      command += "-std=" + json_obj["compiler_std"].get<std::string>() + " ";

    command += "-o " + json_obj["output_name"].get<std::string>();

    auto target = json_obj["output_name"].get<std::string>();

    NeBuild::Logger::info() << "output path: " << target << "\n";
    NeBuild::Logger::info() << "command: " << command << "\n";

    try {
      if (json_obj["dry_run"].get<bool>()) return true;
    } catch (...) {
    }

    if (dry_run) {
      return true;
    }

    auto ret_exec = std::system(command.c_str());

    if (ret_exec > 0) {
      NeBuild::Logger::info() << "error: exit with message: " << std::strerror(ret_exec) << ""
                              << std::endl;
      return false;
    }

    try {
      if (json_obj["run_after_build"].get<bool>()) {
        if (target.ends_with(".so")) {
          NeBuild::Logger::info()
              << "error: can't open dynamic library, it mayn't have an entrypoint" << std::endl;

          return true;
        } else if (target.ends_with(".dylib") || target.ends_with(".dll")) {
          std::ifstream     file = std::ifstream(target);
          std::stringstream ss;

          ss << file.rdbuf();

          if (ss.str()[0] == 'O' && ss.str()[1] == 'p' && ss.str()[2] == 'e' && ss.str()[3] == 'n')
            NeBuild::Logger::info()
                << "error: can't open PEF dynamic library, it mayn't contain an entrypoint"
                << std::endl;
          else if (ss.str()[0] == 'n' && ss.str()[1] == 'e' && ss.str()[2] == 'p' &&
                   ss.str()[3] == 'O')
            NeBuild::Logger::info()
                << "error: can't open FEP dynamic library, it mayn't contain an entrypoint"
                << std::endl;
          else if (ss.str()[0] == 'M' && ss.str()[1] == 'Z')
            NeBuild::Logger::info()
                << "error: can't open MZ dynamic library, it mayn't contain an entrypoint"
                << std::endl;
          else if (ss.str()[0] == 0x7F && ss.str()[1] == 'E') {
            NeBuild::Logger::info()
                << "error: can't open ELF dynamic library, it mayn't contain an entrypoint"
                << std::endl;
          }

          return true;
        }

#if defined(NEBUILD_WINDOWS)
        std::system((".\\" + target).c_str());
#else
        std::system(("./" + target).c_str());
#endif
      }
    } catch (...) {
      return true;
    }
  } catch (std::runtime_error& err) {
    NeBuild::Logger::info() << "error: " << err.what() << std::endl;

    return false;
  }

  return true;
}

const char* JSONManifestBuilder::BuildSystem() {
  return "NeBuild (JSON)";
}