#include "Archive/PakRegistry.h"
#include "File/ResourcePath.h"
#include "Utility/GGHashMgr.h"
#include <cstring>
#include <shlwapi.h>

// OFFSET: 0x0057BC20
// FUNCTION: fn_02596
std::string MakeResourcePath(std::string filename) {
  std::string path = std::string("data\\") + filename;

  if (static_cast<int>(path.find(":")) >= 1) {
    path = path.substr(path.find(":") - 1);
  }

  std::size_t duplicate;
  while ((duplicate = path.find("//")) != std::string::npos) {
    path.replace(duplicate, 2, "/", 1);
  }

  return path;
}

// OFFSET: 0x0057BB50
// FUNCTION: fn_02595
bool ResourceExists(const char* filename, int looseOnly) {
  std::string path = MakeResourcePath(std::string(filename));
  return g_pakRegistry.Exists(path.c_str(), static_cast<char>(looseOnly) != 0);
}

// OFFSET: 0x00AE9EE0
// FUNCTION: fn_18113
void* PakRegistry::ReadLooseResource(const char* filename, std::uint32_t* size) {
  HANDLE file = CreateFileA(filename, GENERIC_READ, FILE_SHARE_READ, nullptr, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, nullptr);
  if (file == INVALID_HANDLE_VALUE) {
    return nullptr;
  }

  LARGE_INTEGER fileSize;
  GetFileSizeEx(file, &fileSize);
  *size = fileSize.LowPart;
  if (!size) {
    CloseHandle(file);
    return nullptr;
  }

  void* data = VirtualAlloc(nullptr, *size, MEM_COMMIT, PAGE_READWRITE);
  if (!data) {
    VirtualFree(nullptr, 0, MEM_RELEASE);
    CloseHandle(file);
    return nullptr;
  }

  DWORD read;
  ReadFile(file, data, *size, &read, nullptr);
  CloseHandle(file);

  const char* extension = PathFindExtensionA(filename);
  if (!extension || std::strcmp(extension, ".ebm") != 0) {
    g_ggHashMgr->VerifyFile(filename);
  }

  return data;
}