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

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

namespace slowmotion
{
	namespace
	{
		int delay = 0;
		utils::hook::detour com_timescale_msec_hook;

		uint64_t com_timescale_msec(const int msec)
		{
			if (delay <= 0)
			{
				return com_timescale_msec_hook.invoke<uint64_t>(msec);
			}
			else
			{
				delay -= msec;
				return 0;
			}
		}

		void scr_cmd_set_slow_motion()
		{
			if (game::Scr_GetNumParam() < 1)
				return;

			int duration = 1000;
			float end = 1.0f;
			const float start = game::Scr_GetFloat(0);

			if (game::Scr_GetNumParam() >= 2)
				end = game::Scr_GetFloat(1);

			if (game::Scr_GetNumParam() >= 3)
				duration = static_cast<int>(game::Scr_GetFloat(2) * 1000.0f);

			const auto _delay = (start > end) ? 150 : 0;

			game::SV_SetConfigstring(game::CS_TIMESCALE,
			                         utils::string::va("%i %i %g %g", *game::mp::gameTime, duration, start, end));
			game::Com_SetSlowMotion(start, end, duration - _delay);

			delay = _delay;

			for (auto i = 0; i < game::Dvar_FindVar("sv_maxclients")->current.integer; i++)
			{
				auto* client = &game::mp::svs->clients[i];
				client->nextSnapshotTime = game::mp::svs->time - 1;
			}
		}
	}

	class component final : public component_interface
	{
	public:
		void post_unpack() override
		{
			if (!game::environment::is_dedi()) return;

			utils::hook::jump(0x1403B4A10, scr_cmd_set_slow_motion);

			// Detour used here instead of call hook because Com_TimeScaleMsec is called from arxan encrypted function
			com_timescale_msec_hook.create(game::Com_TimeScaleMsec, com_timescale_msec);
		}
	};
}

REGISTER_COMPONENT(slowmotion::component)