#include "services/custom_teleport/custom_teleport_service.hpp"
#include "util/math.hpp"
#include "util/teleport.hpp"
#include "views/view.hpp"

namespace big
{
	telelocation get_location_player_is_closest_to()
	{
		if (!g_local_player || !g_local_player->m_navigation || g_custom_teleport_service.all_saved_locations.empty())
			return {};

		Vector3 transformed_vector = Vector3(g_local_player->m_navigation->get_position()->x,
		    g_local_player->m_navigation->get_position()->y,
		    g_local_player->m_navigation->get_position()->z);

		float distance = 500;
		telelocation closest_location{};
		//saved_locations_filtered_list can be used to get a joint list of all categories when the filter is empty.
		for (auto& loc : g_custom_teleport_service.saved_locations_filtered_list())
		{
			float new_distance = math::distance_between_vectors(transformed_vector, {loc.x, loc.y, loc.z});

			if (new_distance < distance)
				closest_location = loc, distance = new_distance;
		}

		return closest_location;
	}

	float get_distance_to_telelocation(telelocation t)
	{
		return math::distance_between_vectors(Vector3(t.x,t.y,t.z), Vector3(g_local_player->m_navigation->get_position()->x,
		    g_local_player->m_navigation->get_position()->y,
		    g_local_player->m_navigation->get_position()->z));
	}

	void view::custom_teleport()
	{
		ImGui::BeginGroup();
		static std::string new_location_name;
		static std::string category = "Default";
		static telelocation deletion_telelocation;
		static std::string filter{};

		if (!std::string(deletion_telelocation.name).empty())
			ImGui::OpenPopup("##deletelocation");

		if (ImGui::BeginPopupModal("##deletelocation", nullptr, ImGuiWindowFlags_NoTitleBar | ImGuiWindowFlags_NoResize | ImGuiWindowFlags_NoMove))
		{
			ImGui::Text("Are you sure you want to delete %s?", deletion_telelocation.name);

			ImGui::Spacing();

			if (ImGui::Button("Yes"))
			{
				g_custom_teleport_service.delete_saved_location(category, deletion_telelocation.name);
				deletion_telelocation.name = "";
				ImGui::CloseCurrentPopup();
			}
			ImGui::SameLine();
			if (ImGui::Button("No"))
			{
				deletion_telelocation.name = "";
				ImGui::CloseCurrentPopup();
			}

			ImGui::EndPopup();
		}

		ImGui::PushItemWidth(300);
		components::input_text_with_hint("Category", "Category", &category);
		components::input_text_with_hint("Location name", "New location", &new_location_name);
		ImGui::PopItemWidth();

		components::button("Save current location", [] {
			if (g_custom_teleport_service.get_saved_location_by_name(new_location_name))
				g_notification_service->push_warning("Custom Teleport", std::format("Location with the name {} already exists", new_location_name));
			else
			{
				telelocation teleport_location;
				Entity teleport_entity = self::ped;
				if (self::veh != 0)
					teleport_entity = self::veh;
				auto coords                  = ENTITY::GET_ENTITY_COORDS(teleport_entity, TRUE);
				teleport_location.name       = new_location_name;
				teleport_location.x          = coords.x;
				teleport_location.y          = coords.y;
				teleport_location.z          = coords.z;
				teleport_location.yaw        = ENTITY::GET_ENTITY_HEADING(teleport_entity);
				teleport_location.pitch      = CAM::GET_GAMEPLAY_CAM_RELATIVE_PITCH();
				teleport_location.roll       = CAM::GET_GAMEPLAY_CAM_RELATIVE_HEADING();
				g_custom_teleport_service.save_new_location(category, teleport_location);
			}
		});

		ImGui::Separator();

		components::small_text("Double click to teleport\nShift click to delete");

		ImGui::Spacing();
		components::input_text_with_hint("##filter", "Search", &filter);

		ImGui::BeginGroup();
		components::small_text("Categories");
		if (ImGui::BeginListBox("##categories", {250, static_cast<float>(*g_pointers->m_gta.m_resolution_y * 0.5)}))
		{
			for (auto& l : g_custom_teleport_service.all_saved_locations | std::ranges::views::keys)
			{
				if (ImGui::Selectable(l.data(), l == category))
				{
					category = l;
				}
			}
			ImGui::EndListBox();
		}
		ImGui::EndGroup();
		ImGui::SameLine();
		ImGui::BeginGroup();
		components::small_text("Locations");
		if (ImGui::BeginListBox("##telelocations", {250, static_cast<float>(*g_pointers->m_gta.m_resolution_y * 0.5)}))
		{
			if (g_custom_teleport_service.all_saved_locations.find(category)
			    != g_custom_teleport_service.all_saved_locations.end())
			{
				std::vector<telelocation> current_list{};

				if (!filter.empty())
					current_list = g_custom_teleport_service.saved_locations_filtered_list(filter);
				else
					current_list = g_custom_teleport_service.all_saved_locations.at(category);

				for (const auto& l : current_list)
				{
					if (ImGui::Selectable(l.name.data(), l.name == get_location_player_is_closest_to().name, ImGuiSelectableFlags_AllowDoubleClick))
					{
						if (GetAsyncKeyState(VK_SHIFT) & 0x8000)
						{
							deletion_telelocation = l;
						}
						else
						{
							if (ImGui::IsMouseDoubleClicked(0))
							{
								g_fiber_pool->queue_job([l] {
									teleport::teleport_player_to_coords(g_player_service->get_self(), {l.x, l.y, l.z}, {l.yaw, l.pitch, l.roll});
								});
							}
						}
					}
					if (ImGui::IsItemHovered())
					{
						ImGui::BeginTooltip();
						if (l.name.length() > 27)
							ImGui::Text(l.name.data());
						ImGui::Text("Distance: %f", get_distance_to_telelocation(l));
						ImGui::EndTooltip();
					}
				}
			}

			ImGui::EndListBox();
		}

		ImGui::EndGroup();

		ImGui::EndGroup();
	}
}