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

#include "command.hpp"
#include "console.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::serverTime - 1;
			}
		}

		void set_code_time_scale(float timescale)
		{
			assert(timescale > 0.0f);
			*game::com_codeTimeScale = timescale;
		}

		void com_timescale_f(const command::params& params)
		{
			if (params.size() != 2)
			{
				console::info("timescale <rate>\n");
				return;
			}

			const auto timescale = static_cast<float>(std::atof(params.get(1)));
			console::info("timescale set to %f\n", timescale);
			set_code_time_scale(timescale);
		}
	}

	class component final : public component_interface
	{
	public:
		void post_unpack() override
		{
			command::add("timescale", com_timescale_f);

			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(0x140415D50, com_timescale_msec);
		}
	};
}

REGISTER_COMPONENT(slowmotion::component)