#include "hooking.hpp"
#include "gta_util.hpp"

namespace big
{
	void format_string(std::string_view player_name, std::string_view protection_type, bool should_log, bool should_notify)
	{
		if (should_log)
			LOG(WARNING) << "BLOCKED_SCRIPT_EVENT From: " << player_name << " Event Type: " << protection_type;

		if (should_notify)
			g_notification_service->push_warning("Script Event Protection",
				fmt::format("From: {}\nEvent Type: {}", player_name.data(), protection_type.data())
			);
	}

	bool hooks::scripted_game_event(CScriptedGameEvent* scripted_game_event, CNetGamePlayer* player)
	{
		const auto args = scripted_game_event->m_args;

		const auto hash = static_cast<eRemoteEvent>(args[0]);
		const auto player_name = player->get_name();

		const auto& notify = g->notifications.script_event_handler;

		switch (hash)
		{
		case eRemoteEvent::Bounty:
			if (g->protections.script_events.bounty)
			{
				format_string(player_name, "Bounty", notify.bounty.log, notify.bounty.notify);

				return true;
			}
			break;
		case eRemoteEvent::CeoBan:
			if (g->protections.script_events.ceo_ban)
			{
				format_string(player_name, "Ceo Ban", notify.ceo_ban.log, notify.ceo_ban.notify);

				return true;
			}
			break;
		case eRemoteEvent::CeoKick:
			if (g->protections.script_events.ceo_kick)
			{
				format_string(player_name, "Ceo Kick", notify.ceo_kick.log, notify.ceo_kick.notify);

				return true;
			}
			break;
		case eRemoteEvent::CeoMoney:
			if (g->protections.script_events.ceo_money)
			{
				format_string(player_name, "Ceo Money", notify.ceo_money.log, notify.ceo_money.notify);

				return true;
			}
			break;
		case eRemoteEvent::ClearWantedLevel:
			if (g->protections.script_events.clear_wanted_level)
			{
				format_string(player_name, "Clear Wanted Level", notify.clear_wanted_level.log, notify.clear_wanted_level.notify);

				return true;
			}
			break;
		case eRemoteEvent::Crash:
			if (g->protections.script_events.crash)
			{
				format_string(player_name, "TSE Crash", notify.crash.log, notify.crash.notify);

				return true;
			}
			break;
		case eRemoteEvent::FakeDeposit:
			if (g->protections.script_events.fake_deposit)
			{
				format_string(player_name, "Fake Deposit", notify.fake_deposit.log, notify.fake_deposit.notify);

				return true;
			}
			break;
		case eRemoteEvent::ForceMission:
			if (g->protections.script_events.force_mission)
			{
				format_string(player_name, "Force Mission", notify.force_mission.log, notify.force_mission.notify);

				return true;
			}
			break;
		case eRemoteEvent::GtaBanner:
			if (g->protections.script_events.gta_banner)
			{
				format_string(player_name, "GTA Banner", notify.gta_banner.log, notify.gta_banner.notify);

				return true;
			}
			break;
		case eRemoteEvent::MCTeleport:
			if (g->protections.script_events.mc_teleport)
			{
				format_string(player_name, "MC Teleport", notify.mc_teleport.log, notify.mc_teleport.notify);

				return true;
			}
			break;
		case eRemoteEvent::NetworkBail:
			if (g->protections.script_events.network_bail)
			{
				format_string(player_name, "Network Bail", notify.network_bail.log, notify.network_bail.notify);

				return true;
			}
			break;
		case eRemoteEvent::PersonalVehicleDestroyed:
			if (g->protections.script_events.personal_vehicle_destroyed)
			{
				format_string(player_name, "Personal Vehicle Destroyed", notify.personal_vehicle_destroyed.log, notify.personal_vehicle_destroyed.notify);

				return true;
			}
			break;
		case eRemoteEvent::RemoteOffradar:
			if (g->protections.script_events.remote_off_radar)
			{
				format_string(player_name, "Off Radar", notify.remote_off_radar.log, notify.remote_off_radar.notify);

				return true;
			}
			break;
		case eRemoteEvent::RotateCam:
			if (g->protections.script_events.crash && args[2] == 537560473) {
				format_string(player_name, "Crash - Rotate Cam", notify.crash.log, notify.crash.notify);
				return true;
			}

			if (g->protections.script_events.rotate_cam)
			{
				if (CNetworkPlayerMgr* player_mgr = gta_util::get_network_player_mgr(); player_mgr != nullptr)
					if (args[2] == player_mgr->m_local_net_player->m_player_id)
						format_string(player_name, "Rotate Cam", notify.rotate_cam.log, notify.rotate_cam.notify);

				return true;
			}
			break;
		case eRemoteEvent::SendToCayoPerico:
			if (g->protections.script_events.send_to_location)
			{
				format_string(player_name, "Send to Cayo Perico", notify.send_to_location.log, notify.send_to_location.notify);

				return true;
			}
			break;
		case eRemoteEvent::SendToCutscene:
			if (g->protections.script_events.send_to_cutscene)
			{
				format_string(player_name, "Send to Cutscene", notify.send_to_cutscene.log, notify.send_to_cutscene.notify);

				return true;
			}
			break;
		case eRemoteEvent::SendToLocation:
		{
			bool known_location = false;

			if (args[2] == 0 && args[3] == 0)
			{
				if (args[4] == 4 && args[5] == 0)
				{
					known_location = true;

					if (g->protections.script_events.send_to_location)
					{
						format_string(player_name, "Send to Beach", notify.send_to_location.log, notify.send_to_location.notify);

						return true;
					}
				}
				else if ((args[4] == 3 || args[4] == 4) && args[5] == 1)
				{
					known_location = true;

					if (g->protections.script_events.send_to_location)
					{
						format_string(player_name, "Send to Cayo Perico", notify.send_to_location.log, notify.send_to_location.notify);

						return true;
					}
				}
			}

			if (!known_location)
			{
				format_string(player_name, "TSE Freeze", notify.tse_freeze.log, notify.tse_freeze.notify);

				return true;
			}
			break;
		}
		case eRemoteEvent::SoundSpam:
			if (g->protections.script_events.sound_spam)
			{
				format_string(player_name, "Sound Spamn", notify.sound_spam.log, notify.sound_spam.notify);

				return true;
			}
			break;
		case eRemoteEvent::Spectate:
			if (g->protections.script_events.spectate)
			{
				format_string(player_name, "Spectate", notify.spectate.log, notify.spectate.notify);

				return true;
			}
			break;
		case eRemoteEvent::Teleport:
			if (g->protections.script_events.force_teleport)
			{
				format_string(player_name, "Apartment Invite", notify.force_teleport.log, notify.force_teleport.notify);

				return true;
			}
			break;
		case eRemoteEvent::TransactionError:
			if (g->protections.script_events.transaction_error)
			{
				format_string(player_name, "Transaction Error", notify.transaction_error.log, notify.transaction_error.notify);

				return true;
			}
			break;
		case eRemoteEvent::VehicleKick:
			if (g->protections.script_events.vehicle_kick)
			{
				format_string(player_name, "Vehicle Kick", notify.vehicle_kick.log, notify.vehicle_kick.notify);

				return true;
			}
			break;
		case eRemoteEvent::Unknown1:
			if (g->protections.script_events.crash && args[2] >= 32) {
				format_string(player_name, "Crash - #" + std::to_string(args[0]), notify.crash.log, notify.crash.notify);

				return true;
			}
			break;
		case eRemoteEvent::Unknown2:
			if (g->protections.script_events.crash && (args[2] >= 62 || args[3] >= 32)) {
				format_string(player_name, "Crash - #" + std::to_string(args[0]), notify.crash.log, notify.crash.notify);

				return true;
			}
			break;
		case eRemoteEvent::Unknown3:
			if (g->protections.script_events.crash && args[2] >= 62) {
				format_string(player_name, "Crash - #" + std::to_string(args[0]), notify.crash.log, notify.crash.notify);

				return true;
			}
			break;
		case eRemoteEvent::Unknown4:
		case eRemoteEvent::Unknown5:
			if (g->protections.script_events.crash && args[2] >= 20) {
				format_string(player_name, "Crash - #" + std::to_string(args[0]), notify.crash.log, notify.crash.notify);

				return true;
			}
			break;
		case eRemoteEvent::Unknown6:
			if (g->protections.script_events.crash) {
				format_string(player_name, "Crash - #" + std::to_string(args[0]), notify.crash.log, notify.crash.notify);

				return true;
			}
			break;
		case eRemoteEvent::SEC3:
			if (
				g->protections.script_events.crash &&
				(
					(args[2] <= 115831 || args[2] >= 9999449) ||
					(args[3] <= -1 || args[3] >= 1) ||
					(args[4] <= -1 || args[4] >= 3) ||
					(args[5] <= -1 || args[5] >= 101) ||
					(args[6] <= -1 || args[6] >= 3) ||
					(args[7] <= -1 || args[7] >= 1)
				)
			) {
				format_string(player_name, "Crash - #" + std::to_string(args[0]), notify.crash.log, notify.crash.notify);

				return true;
			}
			break;
		case eRemoteEvent::SEC6:
			if (
				g->protections.script_events.crash &&
				(
					(args[2] <= -1 || args[2] >= 1) ||
					(args[3] <= -2 || args[3] >= 0) ||
					(args[4] <= 3 || args[4] >= 5) ||
					(args[5] <= 172 || args[5] >= 174) ||
					(args[6] <= 20 || args[6] >= 510) ||
					(args[7] <= 62 || args[7] >= 64) ||
					(args[11] <= -1 || args[11] >= 1) ||
					(args[12] <= -1 || args[12] >= 1) ||
					(args[13] <= -1 || args[13] >= 1)
				)
			) {
				format_string(player_name, "Crash - #" + std::to_string(args[0]), notify.crash.log, notify.crash.notify);

				return true;
			}
			break;
		case eRemoteEvent::SEC7:
			if (g->protections.script_events.crash) {
				format_string(player_name, "Crash - #" + std::to_string(args[0]), notify.crash.log, notify.crash.notify);

				return true;
			}
			break;
		case eRemoteEvent::SEC8:
			if (g->protections.script_events.crash) {
				format_string(player_name, "Crash - #" + std::to_string(args[0]), notify.crash.log, notify.crash.notify);

				return true;
			}
			break;
		case eRemoteEvent::SEC9:
			if (
				g->protections.script_events.crash &&
				(
					(args[2] <= -1986324736 || args[2] >= 1747413822) ||
					(args[2] > -9999999 || args[2] < 77777777) ||
					(args[3] <= -1986324736 || args[3] >= 1777712108) ||
					(args[3] > -9999999 || args[3] < 77777777) ||
					(args[4] <= -1673857408 || args[4] >= 1780088064) ||
					(args[4] > -9999999 || args[4] < 77777777) ||
					(args[6] <= (0LL - 2588888790LL) || args[6] >= 2100146067)
				)
			) {
				format_string(player_name, "Crash - #" + std::to_string(args[0]), notify.crash.log, notify.crash.notify);

				return true;
			}
			break;
		case eRemoteEvent::SEC10:
			if (g->protections.script_events.crash) {
				format_string(player_name, "Crash - #" + std::to_string(args[0]), notify.crash.log, notify.crash.notify);

				return true;
			}
			break;
		case eRemoteEvent::SEC11:
			if (g->protections.script_events.crash) {
				format_string(player_name, "Crash - #" + std::to_string(args[0]), notify.crash.log, notify.crash.notify);

				return true;
			}
			break;
		case eRemoteEvent::SEC12:
			if (
				g->protections.script_events.crash &&
				(
					(args[2] <= -1 || args[2] >= 50) ||
					(args[3] <= -1 || args[3] >= 50)
				)
			) {
				format_string(player_name, "Crash - #" + std::to_string(args[0]), notify.crash.log, notify.crash.notify);

				return true;
			}
			break;
		case eRemoteEvent::SEC13:
			if (g->protections.script_events.crash) {
				format_string(player_name, "Crash - #" + std::to_string(args[0]), notify.crash.log, notify.crash.notify);

				return true;
			}
			break;
		case eRemoteEvent::Spaghettios:
			if (g->protections.script_events.crash) {
				format_string(player_name, "Crash - #" + std::to_string(args[0]), notify.crash.log, notify.crash.notify);

				return true;
			}
			break;
		case eRemoteEvent::ForceMission2:
			if (g->protections.script_events.force_mission)
			{
				format_string(player_name, "Force Mission", notify.force_mission.log, notify.force_mission.notify);

				return true;
			}
			break;
		case eRemoteEvent::SHKick:
			if (g->protections.script_events.network_bail)
			{
				format_string(player_name, "Network Bail", notify.network_bail.log, notify.network_bail.notify);

				return true;
			}
		}


		if (g->debug.script_event_logging)
		{
			LOG(INFO) << "== Begin of Script Event ==";
			LOG(INFO) << "Player: " << player->get_name();
			LOG(INFO) << "Hash/Arg #0: " << (int)hash;

			for (std::size_t i = 1; i < sizeof(args); i++)
				LOG(INFO) << "Arg #" << i << ": " << args[i];

			LOG(INFO) << "== End of Script Event ==";
		}

		return false;
	}
}