#include <std_include.hpp>
#include "launcher/launcher.hpp"
#include "loader/loader.hpp"
#include "loader/module_loader.hpp"
#include "game/game.hpp"
#include "loader/binary_loader.hpp"

//#define GENERATE_DIFFS

void exit_hook(const int code)
{
	module_loader::pre_destroy();
	exit(code);
}

int main()
{
	FARPROC entry_point = nullptr;

	try
	{
#ifdef GENERATE_DIFFS
		binary_loader::create();
		return 0;
#endif

		module_loader::post_start();

		launcher launcher;
		const auto mode = launcher.run();

		if (mode == launcher::mode::none) return 0;

		loader loader(mode);
		loader.set_import_resolver([](const std::string& module, const std::string& function) -> FARPROC
		{
			if (module == "steam_api.dll")
			{
				return utils::nt::module().get_proc<FARPROC>(function);
			}
			else if (function == "ExitProcess")
			{
				return FARPROC(exit_hook);
			}

			return nullptr;
		});

		entry_point = loader.load({});
		if (!entry_point) throw std::runtime_error("Unable to load inject binary into memory");

		game::initialize(mode);
		module_loader::post_load();
	}
	catch (std::exception& e)
	{
		MessageBoxA(nullptr, e.what(), "ERROR", MB_ICONERROR);
		return 1;
	}

	return entry_point();
}