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

#include "game/game.hpp"
#include "scheduler.hpp"

#include <utils/hook.hpp>

namespace dedicated_patches
{
	namespace
	{
		void scr_are_textures_loaded_stub([[maybe_unused]] game::scriptInstance_t inst)
		{
			game::Scr_AddInt(game::SCRIPTINSTANCE_SERVER, 1);
		}

		game::eNetworkModes get_online_mode_stub()
		{
			return game::MODE_NETWORK_ONLINE;
		}

		bool is_online_stub()
		{
			return true;
		}

		bool is_mod_loaded_stub()
		{
			return false;
		}

		void patch_is_mod_loaded_checks()
		{
			const std::vector<uintptr_t> is_mod_loaded_addresses =
			{
				{ 0x1401DA1B5_g },
				{ 0x14024D4A0_g },
				{ 0x14024D669_g },
				{ 0x14024D939_g },
				{ 0x14024DC64_g },
				{ 0x14024E13A_g },
				{ 0x14024E5A3_g },
				{ 0x14024FFB9_g },
				{ 0x140251E9E_g },
				{ 0x140253680_g },
				{ 0x140257BF6_g },
				{ 0x1402D25E2_g }, // unsure if needed to patch
				{ 0x1402D296D_g }, // unsure
				{ 0x1402D58E9_g }, // unsure
				{ 0x1402DC5BB_g }, // unsure
				{ 0x140468374_g },
				{ 0x14046B796_g }
			};

			for (const auto& address : is_mod_loaded_addresses)
			{
				utils::hook::call(address, is_mod_loaded_stub);
			}
		}
	}

	struct component final : server_component
	{
		void post_unpack() override
		{
			// Fix infinite loop
			utils::hook::jump(0x1402E86B0_g, scr_are_textures_loaded_stub);

			// Online classes
			utils::hook::jump(0x1405003E0_g, get_online_mode_stub);
			utils::hook::jump(0x1405003B0_g, get_online_mode_stub);

			// Progression / Ranked
			utils::hook::jump(0x140500A50_g, is_online_stub);
			utils::hook::jump(0x140500980_g, is_online_stub);
			utils::hook::jump(0x1402565D0_g, is_online_stub);
			patch_is_mod_loaded_checks();

			scheduler::once([]()
			{
				game::Com_SessionMode_SetNetworkMode(game::MODE_NETWORK_ONLINE);
				game::Com_SessionMode_SetGameMode(game::MODE_GAME_MATCHMAKING_PLAYLIST);
			}, scheduler::pipeline::main, 1s);
		}
	};
}

REGISTER_COMPONENT(dedicated_patches::component)