#include "fiber_pool.hpp"
#include "script.hpp"
#include "natives.hpp"
#include "pointers.hpp"
#include "views/view.hpp"
#include "util/ped.hpp"
#include "services/gta_data/gta_data_service.hpp"
#include "services/model_preview/model_preview_service.hpp"
#include "services/players/player_service.hpp"

#include <imgui_internal.h>


#define SPAWN_PED_ALL_WEAPONS -1
#define SPAWN_PED_NO_WEAPONS -2

namespace big
{
	Ped spawn_ped_at_location(
		const int selected_ped_type, 
		const char* ped_model_buf, 
		const Player selected_ped_player_id, 
		const Player selected_ped_for_player_id,
		const bool is_bodyguard
	) {
		Hash hash = 0;
		Ped clone = 0;
		Vector3 location;
		Player player;
		Ped player_ped;

		if (selected_ped_type == -2)
		{
			if (selected_ped_player_id == -1)
			{
				clone = self::ped;
				hash = ENTITY::GET_ENTITY_MODEL(clone);
			}
			else
			{
				auto plyr = g_player_service->get_by_id(selected_ped_player_id);
				if (plyr == nullptr || !plyr->is_valid() || !plyr->get_ped() || !plyr->get_ped()->m_navigation)
				{
					g_notification_service->push_error("Ped", "Invalid Online Player.");
					return 0;
				}

				clone = PLAYER::GET_PLAYER_PED_SCRIPT_INDEX(plyr->id());
				hash = ENTITY::GET_ENTITY_MODEL(clone);
			}
		}
		else
		{
			hash = rage::joaat(ped_model_buf);
		}


		if (selected_ped_for_player_id == -1)
		{
			location = self::pos;
			player = self::id;
			player_ped = self::ped;
		}
		else
		{
			auto plyr = g_player_service->get_by_id(selected_ped_for_player_id);
			if (plyr == nullptr || !plyr->is_valid() || !plyr->get_ped() || !plyr->get_ped()->m_navigation)
			{
				g_notification_service->push_error("Ped", "Invalid Online Player.");
				return 0;
			}

			auto player_pos = plyr->get_ped()->m_navigation->m_position;

			location.x = player_pos.x;
			location.y = player_pos.y;
			location.z = player_pos.z;
			player = plyr->id();
			player_ped = PLAYER::GET_PLAYER_PED_SCRIPT_INDEX(player);
		}

		location.x += 1.f;
		location.y += 1.f;
		Ped ped = ped::spawn(ePedType::PED_TYPE_ARMY, hash, clone, location, 0);

		if (ped == 0)
		{
			g_notification_service->push_error("Ped", "Failed to spawn model, did you give an incorrect model ? ");
			return 0;
		}

		if (is_bodyguard)
		{
			int player_group = PLAYER::GET_PLAYER_GROUP(player);

			PED::SET_PED_AS_GROUP_MEMBER(ped, player_group);
			PED::SET_PED_RELATIONSHIP_GROUP_HASH(ped, PED::GET_PED_RELATIONSHIP_GROUP_HASH(player_ped));
			PED::SET_PED_AS_GROUP_LEADER(player_ped, player_group);
			PED::SET_PED_CAN_BE_TARGETTED_BY_PLAYER(ped, player, true);
			PED::SET_PED_ARMOUR(ped, 100);
			ENTITY::SET_ENTITY_MAX_HEALTH(ped, 1000);
			ENTITY::SET_ENTITY_HEALTH(ped, 1000, 0);
			PED::SET_PED_COMBAT_ABILITY(ped, 100);
			PED::SET_PED_COMBAT_ATTRIBUTES(ped, 46, 1);
			PED::SET_PED_COMBAT_ATTRIBUTES(ped, 63, 0);

			TASK::CLEAR_PED_TASKS_IMMEDIATELY(ped);
			TASK::TASK_COMBAT_HATED_TARGETS_AROUND_PED(ped, 100.f, 0);
			PED::SET_PED_KEEP_TASK(ped, true);
		}

		return ped;
	}


	void spawn_ped_give_weapon(
		const Ped ped, 
		const int selected_ped_weapon_type, 
		const Hash selected_ped_weapon_hash
	) {
		if (selected_ped_weapon_type == SPAWN_PED_NO_WEAPONS)
		{
			return;
		}

		auto weapon_type_arr = g_gta_data_service->get_weapon_type_arr();
		for (auto& weapon : g_gta_data_service->get_weapon_arr())
		{
			if (
				selected_ped_weapon_type == SPAWN_PED_ALL_WEAPONS ||
				weapon.weapon_type == weapon_type_arr[selected_ped_weapon_type]
			) {
				if (
					selected_ped_weapon_hash == 0 ||
					weapon.hash == selected_ped_weapon_hash
				) {
					WEAPON::GIVE_WEAPON_TO_PED(ped, weapon.hash, 9999, false, selected_ped_weapon_hash != 0);
				}
			}
		}
	}

	void view::spawn_ped()
	{
		static int selected_ped_type = -2;
		static bool ped_model_dropdown_open = false;
		static char ped_model_buf[64];
		static Player selected_ped_player_id = -1;

		auto ped_type_arr = g_gta_data_service->get_ped_type_arr();
		auto ped_arr = g_gta_data_service->get_ped_arr();


		static int selected_ped_weapon_type = SPAWN_PED_ALL_WEAPONS;
		static Hash selected_ped_weapon_hash = 0;
		auto weapon_type_arr = g_gta_data_service->get_weapon_type_arr();
		auto weapon_arr = g_gta_data_service->get_weapon_arr();

		static Player selected_ped_for_player_id = -1;
		auto player_arr = g_player_service->players();

		if (!*g_pointers->m_is_session_started)
		{
			selected_ped_player_id = -1;
			selected_ped_for_player_id = -1;
		}
		else
		{
			if (g_player_service->get_by_id(selected_ped_player_id) == nullptr)
			{
				selected_ped_player_id = -1;
			}

			if (g_player_service->get_by_id(selected_ped_for_player_id) == nullptr)
			{
				selected_ped_for_player_id = -1;
			}
		}


		components::sub_title("Ped Model");
		{
			ImGui::BeginGroup();
			{
				ImGui::Text("Ped Type");

				ImGui::SetNextItemWidth(160.f);
				if (ImGui::BeginCombo(
					"##ped_type", 
					selected_ped_type == -1 ? "ALL" : 
					selected_ped_type == -2 ? "ONLINE PLAYER" : 
					ped_type_arr[selected_ped_type].c_str()
				)) {

					if (ImGui::Selectable("ONLINE PLAYER", selected_ped_type == -2))
					{
						selected_ped_type = -2;
					}

					if (ImGui::Selectable("ALL", selected_ped_type == -1))
					{
						selected_ped_type = -1;
					}

					for (int i = 0; i < ped_type_arr.size(); i++)
					{
						if (ImGui::Selectable(ped_type_arr[i].c_str(), selected_ped_type == i))
						{
							selected_ped_type = i;
							ped_model_buf[0] = 0;
						}

						if (selected_ped_type == i)
						{
							ImGui::SetItemDefaultFocus();
						}
					}

					ImGui::EndCombo();
				}
			}
			ImGui::EndGroup();

			ImGui::SameLine();

			if (selected_ped_type == -2)
			{
				ImGui::BeginGroup();
				{
					ImGui::Text("Player");

					ImGui::SetNextItemWidth(240.f);
					if (ImGui::BeginCombo(
						"##ped_player",
						selected_ped_player_id == -1 ?
						"Self" :
						g_player_service->get_by_id(selected_ped_player_id)->get_name()
					)) {
						if (ImGui::Selectable("Self", selected_ped_player_id == -1))
						{
							selected_ped_player_id = -1;
							g_model_preview_service->stop_preview();
						}
						else if (!g->spawn_ped.preview_ped || (g->spawn_ped.preview_ped && !ImGui::IsAnyItemHovered()))
						{
							g_model_preview_service->stop_preview();
						}
						else if (ImGui::IsItemHovered())
						{
							g_fiber_pool->queue_job([] {
								Ped ped = self::ped;
								Hash hash = ENTITY::GET_ENTITY_MODEL(ped);
								g_model_preview_service->show_ped(hash, ped);
								});
						}

						if (selected_ped_player_id == -1)
						{
							ImGui::SetItemDefaultFocus();
						}

						if (*g_pointers->m_is_session_started)
						{
							for (auto& item : player_arr)
							{
								auto plyr = item.second;
								auto plyr_id = plyr->id();

								ImGui::PushID(plyr_id);
								if (ImGui::Selectable(plyr->get_name(), selected_ped_player_id == plyr_id))
								{
									selected_ped_player_id = plyr_id;
									g_model_preview_service->stop_preview();
								}
								else if (!g->spawn_ped.preview_ped || (g->spawn_ped.preview_ped && !ImGui::IsAnyItemHovered()))
								{
									g_model_preview_service->stop_preview();
								}
								else if (ImGui::IsItemHovered())
								{
									g_fiber_pool->queue_job([plyr_id] {

										auto plyr = g_player_service->get_by_id(plyr_id);
										if (plyr)
										{
											Ped ped = PLAYER::GET_PLAYER_PED_SCRIPT_INDEX(plyr->id());
											Hash hash = ENTITY::GET_ENTITY_MODEL(ped);
											g_model_preview_service->show_ped(hash, ped);
										}
										});

								}
								ImGui::PopID();

								if (selected_ped_player_id == plyr_id)
								{
									ImGui::SetItemDefaultFocus();
								}
							}
						}

						ImGui::EndCombo();
					}
				}
				ImGui::EndGroup();
			}
			else
			{
				ImGui::BeginGroup();
				{
					ImGui::Text("Model Name");

					ImGui::SetNextItemWidth(240.f);
					components::input_text_with_hint(
						"##ped_model_name", "Model Name", 
						ped_model_buf, sizeof(ped_model_buf), ImGuiInputTextFlags_EnterReturnsTrue, 
						[] {
							ped_model_dropdown_open = false;
						}
					);
				}
				ImGui::EndGroup();

				// ped model dropdown
				{
					bool ped_model_dropdown_focused = ImGui::IsItemActive();

					if (ImGui::IsItemActivated())
					{
						ped_model_dropdown_open = true;
					}

					if (ped_model_dropdown_open)
					{
						bool is_open = true;
						bool item_hovered = false;

						std::string lower_search = ped_model_buf;
						std::transform(lower_search.begin(), lower_search.end(), lower_search.begin(), tolower);

						ImGui::SetNextWindowPos({ ImGui::GetItemRectMin().x, ImGui::GetItemRectMax().y });
						ImGui::SetNextWindowSize({ 300, 300 });
						if (ImGui::Begin("##player_model_popup", &is_open, ImGuiWindowFlags_NoTitleBar | ImGuiWindowFlags_NoMove | ImGuiWindowFlags_NoResize | ImGuiWindowFlags_Tooltip))
						{
							ImGui::BringWindowToDisplayFront(ImGui::GetCurrentWindow());
							ped_model_dropdown_focused |= ImGui::IsWindowFocused();

							for (auto& item : ped_arr)
							{
								std::string ped_type = item.ped_type;
								std::string name = item.name;

								std::transform(name.begin(), name.end(), name.begin(), tolower);

								if ((
									selected_ped_type == -1 || ped_type_arr[selected_ped_type] == ped_type
								) && (
									name.find(lower_search) != std::string::npos
								)) {

									bool selectable_highlighted = lower_search == name;
									bool selectable_clicked = ImGui::Selectable(item.name.c_str(), selectable_highlighted);
									ped_model_dropdown_focused |= ImGui::IsItemFocused();

									if (selectable_clicked)
									{
										strncpy(ped_model_buf, item.name.c_str(), 64);
										ped_model_dropdown_open = false;
										ped_model_dropdown_focused = false;
									}

									if (selectable_highlighted)
									{
										ImGui::SetItemDefaultFocus();
									}

									if (ImGui::IsItemHovered())
									{
										item_hovered = true;
										g_model_preview_service->show_ped(item.hash);
									}
								}
							}
							ImGui::End();
						}

						ped_model_dropdown_open = ped_model_dropdown_focused;

						if (!g->spawn_ped.preview_ped || (g->spawn_ped.preview_ped && (!item_hovered || !ped_model_dropdown_open)))
						{
							g_model_preview_service->stop_preview();
						}
					}
				}
			}
		}
		ImGui::Separator();


		components::sub_title("Weapon");
		{
			ImGui::BeginGroup();
			{
				ImGui::Text("Weapon Type");

				ImGui::SetNextItemWidth(160.f);
				if (ImGui::BeginCombo(
					"##ped_weapon_type", 
					selected_ped_weapon_type == SPAWN_PED_ALL_WEAPONS ? 
					"ALL" : 
					selected_ped_weapon_type == SPAWN_PED_NO_WEAPONS ?
					"NO WEAPONS" :
					weapon_type_arr[selected_ped_weapon_type].c_str()
				)) {
					if (ImGui::Selectable("ALL", selected_ped_weapon_type == SPAWN_PED_ALL_WEAPONS))
					{
						selected_ped_weapon_type = SPAWN_PED_ALL_WEAPONS;
					}

					if (selected_ped_weapon_hash == SPAWN_PED_ALL_WEAPONS)
					{
						ImGui::SetItemDefaultFocus();
					}

					if (ImGui::Selectable("NO WEAPONS", selected_ped_weapon_type == SPAWN_PED_NO_WEAPONS))
					{
						selected_ped_weapon_type = SPAWN_PED_NO_WEAPONS;
					}

					if (selected_ped_weapon_hash == SPAWN_PED_NO_WEAPONS)
					{
						ImGui::SetItemDefaultFocus();
					}

					for (int i = 0; i < weapon_type_arr.size(); i++)
					{
						if (ImGui::Selectable(weapon_type_arr[i].c_str(), selected_ped_weapon_type == i))
						{
							selected_ped_weapon_type = i;
							selected_ped_weapon_hash = 0;
						}

						if (selected_ped_weapon_type == i)
						{
							ImGui::SetItemDefaultFocus();
						}
					}

					ImGui::EndCombo();
				}
			}
			ImGui::EndGroup();

			ImGui::SameLine();

			ImGui::BeginGroup();
			{
				ImGui::Text("Weapon");

				ImGui::SetNextItemWidth(240.f);
				if (ImGui::BeginCombo(
					"##ped_weapon", 
					selected_ped_weapon_type == SPAWN_PED_NO_WEAPONS ?
					"NO WEAPONS" :
					selected_ped_weapon_hash == 0 ? 
					"ALL" : 
					g_gta_data_service->find_weapon_by_hash(selected_ped_weapon_hash).name.c_str()
				)) {
					if (selected_ped_weapon_type != SPAWN_PED_NO_WEAPONS)
					{
						if (ImGui::Selectable("ALL", selected_ped_weapon_hash == 0))
						{
							selected_ped_weapon_hash = 0;
						}

						if (selected_ped_weapon_hash == 0)
						{
							ImGui::SetItemDefaultFocus();
						}

						for (auto& weapon : weapon_arr)
						{
							if (
								selected_ped_weapon_type == SPAWN_PED_ALL_WEAPONS ||
								weapon.weapon_type == weapon_type_arr[selected_ped_weapon_type]
							) {
								if (ImGui::Selectable(weapon.name.c_str(), weapon.hash == selected_ped_weapon_hash))
								{
									selected_ped_weapon_hash = weapon.hash;
								}
							}

							if (selected_ped_weapon_hash == weapon.hash)
							{
								ImGui::SetItemDefaultFocus();
							}
						}
					}

					ImGui::EndCombo();
				}
			}
			ImGui::EndGroup();

		}
		ImGui::Separator();


		components::sub_title("Spawn For");
		{
			if (ImGui::BeginCombo(
				"##ped_for",
				selected_ped_for_player_id == -1 ?
				"Self" :
				g_player_service->get_by_id(selected_ped_for_player_id)->get_name()
			)) {
				if (ImGui::Selectable("Self", selected_ped_for_player_id == -1))
				{
					selected_ped_for_player_id = -1;
				}

				if (selected_ped_for_player_id == -1)
				{
					ImGui::SetItemDefaultFocus();
				}

				if (*g_pointers->m_is_session_started)
				{
					for (auto& [_, plyr] : player_arr)
					{
						auto plyr_id = plyr->id();

						ImGui::PushID(plyr_id);
						if (ImGui::Selectable(plyr->get_name(), selected_ped_for_player_id == plyr_id))
						{
							selected_ped_for_player_id = plyr_id;
						}
						ImGui::PopID();

						if (selected_ped_for_player_id == plyr_id)
						{
							ImGui::SetItemDefaultFocus();
						}
					}
				}

				ImGui::EndCombo();
			}
		}
		ImGui::Separator();


		if (ImGui::Checkbox("Preview", &g->spawn_ped.preview_ped))
		{
			if (!g->spawn_ped.preview_ped)
			{
				g_model_preview_service->stop_preview();
			}
		}

		components::button("Change Player Model", [] {
			if (selected_ped_type == -2)
			{
				if (selected_ped_player_id != -1)
				{
					auto plyr = g_player_service->get_by_id(selected_ped_player_id);
					if (plyr)
					{
						Ped ped = PLAYER::GET_PLAYER_PED_SCRIPT_INDEX(plyr->id());
						ped::steal_identity(ped);
					}
				}
				}
			else
			{
				if (!ped::change_player_model(rage::joaat(ped_model_buf)))
				{
					g_notification_service->push_error("Ped", "Failed to spawn model, did you give an incorrect model ? ");
					return;
				}
			}

			spawn_ped_give_weapon(self::ped, selected_ped_weapon_type, selected_ped_weapon_hash);
		});

		ImGui::SameLine();

		components::button("Spawn Ped", [] {
			Ped ped = spawn_ped_at_location(selected_ped_type, ped_model_buf, selected_ped_player_id, selected_ped_for_player_id, false);

			if (ped)
			{
				spawn_ped_give_weapon(ped, selected_ped_weapon_type, selected_ped_weapon_hash);
			}
		});

		ImGui::SameLine();

		components::button("Spawn Bodyguard", [] {
			Ped ped = spawn_ped_at_location(selected_ped_type, ped_model_buf, selected_ped_player_id, selected_ped_for_player_id, true);

			if (ped)
			{
				spawn_ped_give_weapon(ped, selected_ped_weapon_type, selected_ped_weapon_hash);
			}
		});
	}
}