#include <std_include.hpp>
#include "loader/component_loader.hpp"

#include "game/game.hpp"

#include "command.hpp"
#include "game_console.hpp"
#include "scheduler.hpp"

#include <utils/hook.hpp>
#include <utils/io.hpp>

namespace mods
{
	class component final : public component_interface
	{
	public:
		void post_unpack() override
		{
			if (!utils::io::directory_exists("mods"))
			{
				utils::io::create_directory("mods");
			}

			command::add("loadmod", [](const command::params& params)
			{
				if (params.size() < 2)
				{
					game_console::print(game_console::con_type_info, "Usage: loadmod mods/<modname>");
					return;
				}

				if (::game::SV_Loaded())
				{
					game_console::print(game_console::con_type_error, "Cannot load mod while in-game!\n");
					game::CG_GameMessage(0, "^1Cannot unload mod while in-game!");
					return;
				}

				const auto path = params.get(1);
				game_console::print(game_console::con_type_info, "Loading mod %s\n", path);

				if (!utils::io::directory_exists(path))
				{
					game_console::print(game_console::con_type_error, "Mod %s not found!\n", path);
					return;
				}

				game::mod_folder = path;

				scheduler::once([]()
				{
					command::execute("lui_restart", true);
				}, scheduler::pipeline::renderer);
			});

			command::add("unloadmod", [](const command::params& params)
			{
				if (game::mod_folder.empty())
				{
					game_console::print(game_console::con_type_info, "No mod loaded\n");
					return;
				}

				if (::game::SV_Loaded())
				{
					game_console::print(game_console::con_type_error, "Cannot unload mod while in-game!\n");
					game::CG_GameMessage(0, "^1Cannot unload mod while in-game!");
					return;
				}

				game_console::print(game_console::con_type_info, "Unloading mod %s\n", game::mod_folder.data());
				game::mod_folder.clear();

				scheduler::once([]()
				{
					command::execute("lui_restart", true);
				}, scheduler::pipeline::renderer);
			});
		}
	};
}

REGISTER_COMPONENT(mods::component)