#include "Script/LuaInstance.h"
#include "File/File.h"

extern "C" {
#include <lauxlib.h>
#include <lualib.h>
}
#include <cstdlib>
#include <cstring>
#include <windows.h>

// OFFSET: 0x005221E0
// FUNCTION: fn_01123
CLuaInstance::CLuaInstance() : m_state(luaL_newstate()) {
  if (this->m_state) {
    luaL_openlibs(this->m_state);
    lua_register(this->m_state, "MessageBox", LuaMessageBox);
  }
}

// OFFSET: 0x00522290
// FUNCTION: fn_01125
CLuaInstance::~CLuaInstance() {
  if (this->m_state) {
    lua_close(this->m_state);
    this->m_state = nullptr;
  }
}

// OFFSET: 0x005223D0
// FUNCTION: fn_01128
bool CLuaInstance::Execute(const std::string& name) {
  lua_getglobal(this->m_state, name.c_str());

  if (lua_pcall(this->m_state, 0, 0, 0) == 0) {
    return true;
  }

  // STATIC: 0x010D9630
  // STRING: st_00223
  const char* error = lua_tostring(this->m_state, -1);
  MessageBoxA(nullptr, error, "Lua Error", MB_OK);

  return false;
}

// OFFSET: 0x00522460
// FUNCTION: fn_01129
int CLuaInstance::LuaMessageBox(lua_State* state) {
  const std::string text = lua_tostring(state, -1);
  MessageBoxA(nullptr, text.c_str(), "CLuaInstance", MB_OK);
  return 0;
}

// OFFSET: 0x005225F0
// FUNCTION: fn_01130
bool CLuaInstance::LoadFile(const std::string& path, bool showError) {
  if (luaL_loadfile(this->m_state, path.c_str()) == 0 && lua_pcall(this->m_state, 0, LUA_MULTRET, 0) == 0) {
    return true;
  }

  if (showError) {
    MessageBoxA(nullptr, lua_tostring(this->m_state, -1), path.c_str(), MB_OK);
  }

  return false;
}

// OFFSET: 0x00522660
// FUNCTION: fn_01131
bool CLuaInstance::LoadFile(const std::string& path) {
  CFile file;
  if (!file.Open(path.c_str())) {
    return false;
  }

  const std::uint32_t size = file.GetSize();
  char* buffer = static_cast<char*>(std::malloc(size + 1));
  if (!buffer) {
    file.Close();
    return false;
  }

  std::memset(buffer, 0, size + 1);
  file.Read(buffer, size);
  file.Close();

  if (luaL_loadbuffer(this->m_state, buffer, size, path.c_str()) == 0 && lua_pcall(this->m_state, 0, 0, 0) == 0) {
    std::free(buffer);
    return true;
  }

  const char* error = lua_tostring(this->m_state, -1);
  MessageBoxA(nullptr, error, path.c_str(), MB_OK);
  lua_pop(this->m_state, 1);
  std::free(buffer);

  return false;
}

// OFFSET: 0x00522870
// FUNCTION: fn_01132
void CLuaInstance::SetGlobal(const std::string& name, bool value) {
  lua_pushboolean(this->m_state, value);
  lua_setglobal(this->m_state, name.c_str());
}

// OFFSET: 0x005228B0
// FUNCTION: fn_01133
void CLuaInstance::SetGlobal(const std::string& name, int value) {
  lua_pushinteger(this->m_state, value);
  lua_setglobal(this->m_state, name.c_str());
}

// OFFSET: 0x005228F0
// FUNCTION: fn_01134
void CLuaInstance::SetGlobal(const std::string& name, double value) {
  lua_pushnumber(this->m_state, value);
  lua_setglobal(this->m_state, name.c_str());
}

// OFFSET: 0x00E27850
// FUNCTION: fn_23942
int CLuaInstance::ReadStrings(std::vector<std::string>& values, const char* global) {
  lua_getglobal(this->m_state, global);
  int type = lua_type(this->m_state, -1);
  if (type != LUA_TTABLE) {
    lua_pop(this->m_state, 1);
    return (type != LUA_TNIL) + 1;
  }

  values.resize(lua_objlen(this->m_state, -1));
  lua_pushnil(this->m_state);
  std::size_t index = 0;

  while (lua_next(this->m_state, -2)) {
    type = lua_type(this->m_state, -1);
    if (type != LUA_TSTRING) {
      lua_pop(this->m_state, 3);
      return (type != LUA_TNIL) + 1;
    }

    values[index++] = lua_tostring(this->m_state, -1);
    lua_pop(this->m_state, 1);
  }

  lua_pop(this->m_state, 1);
  return 0;
}