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

#include "game/game.hpp"
#include "game/utils.hpp"

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

#include "command.hpp"
#include "client_command.hpp"

namespace chat
{
	namespace
	{
		const game::dvar_t* g_deadChat;

		void cmd_say_f(game::gentity_s* ent, const command::params_sv& params)
		{
			if (params.size() < 2)
			{
				return;
			}

			int mode = 0;
			if (params[0] == "say_team"s)
			{
				mode = 1;
			}

			const auto p = params.join(1);
			game::Scr_AddString(game::SCRIPTINSTANCE_SERVER, p.data() + 1); // Skip special char
			game::Scr_Notify_Canon(ent, game::Scr_CanonHash(params[0]), 1);

			game::G_Say(ent, nullptr, mode, p.data());
		}

		void cmd_chat_f(game::gentity_s* ent, const command::params_sv& params)
		{
			auto p = params.join(1);

			// Not a mistake! + 2 is necessary for the GSC script to receive only the actual chat text
			game::Scr_AddString(game::SCRIPTINSTANCE_SERVER, p.data() + 2);
			game::Scr_Notify_Canon(ent, game::Scr_CanonHash(params[0]), 1);

			utils::hook::invoke<void>(0x140298E70_g, ent, p.data());
		}

		uint64_t* divert_xuid_to_client_num_stub(int, const int client_num, int)
		{
			static thread_local uint64_t value;
			// zero xuid is invalid, so increase the clientnum to prevent 0 values
			value = static_cast<uint64_t>(client_num) + 1;
			return &value;
		}

		void send_chat_message(int client_num, const std::string& text)
		{
			game::SV_GameSendServerCommand(client_num, game::SV_CMD_RELIABLE_0, utils::string::va("v \"%Iu %d %d %s\"", -1, 0, 0, text.data()));
		}

		// This function has probably a different name
		void g_say_to_stub(utils::hook::assembler& a)
		{
			const auto no_dead_chat = a.newLabel();

			// game's code
			a.mov(rax, qword_ptr(rbx));

			a.push(rax);

			a.mov(rax, qword_ptr(reinterpret_cast<std::uintptr_t>(&g_deadChat)));
			a.mov(al, byte_ptr(rax, 0x28)); // dvar_t.current.value.enabled
			a.test(al, al);

			a.pop(rax);

			a.je(no_dead_chat);

			a.jmp(0x140299061_g);

			a.bind(no_dead_chat);
			a.cmp(dword_ptr(rax, 0x16AE0), 0x0); // game's code
			a.jmp(0x14029905B_g);
		}

		void cl_handle_chat(char* dest, size_t dest_size, const char* src)
		{
			game::I_strcpy(dest, dest_size, src);
			printf("%s\n", dest);
		}
	}

	const char* get_client_name(const uint64_t xuid)
	{
		if (xuid == 0xFFFFFFFF)
		{
			return "Server";
		}

		if (xuid < 19 && !game::is_server())
		{
			char buffer[256]{};
			game::CL_GetClientName(0, static_cast<int>(xuid - 1), buffer, sizeof(buffer), true);

			return utils::string::va("%s\n", buffer);
		}

		return "Unknown Soldier";
	}

	class component final : public generic_component
	{
	public:
		void post_unpack() override
		{
			utils::hook::call(game::select(0x141974B04, 0x14029908A), divert_xuid_to_client_num_stub);

			if (game::is_server())
			{
				client_command::add("say", cmd_say_f);
				client_command::add("say_team", cmd_say_f);

				client_command::add("chat", cmd_chat_f);

				// Overwrite say command
				utils::hook::jump(0x14052A6C0_g, +[]
				{
					if (!game::is_server_running())
					{
						printf("Server is not running\n");
						return;
					}

					const command::params params{};
					const auto text = params.join(1);

					send_chat_message(-1, text);
					printf("Server: %s\n", text.data());
				});

				// Overwrite tell command
				utils::hook::jump(0x14052A7E0_g, +[]
				{
					if (!game::is_server_running())
					{
						printf("Server is not running\n");
						return;
					}

					const command::params params{};
					if (params.size() < 2)
					{
						return;
					}

					const auto client = atoi(params[1]);
					const auto text = params.join(2);

					send_chat_message(client, text);
					printf("Server -> %i: %s\n", client, text.data());
				});

				// Kill say fallback
				utils::hook::set<uint8_t>(0x1402FF987_g, 0xEB);

				g_deadChat = game::register_dvar_bool("g_deadChat", false, game::DVAR_NONE, "Allow dead players to chat with living players");
				utils::hook::jump(0x140299051_g, utils::hook::assemble(g_say_to_stub));
			}
			else
			{
				// Ignore some check that suppresses the chat
				utils::hook::nop(0x141DEA9BD_g, 2);

				// Add chat history to the in-game console
				utils::hook::call(0x141DEAA0F_g, cl_handle_chat); // I_strcpy
			}
		}
	};
}

REGISTER_COMPONENT(chat::component)
