YimMenu/src/services/context_menu/context_menu_service.hpp

#pragma once
#include "natives.hpp"
#include "util/entity.hpp"
#include "util/ped.hpp"
#include "util/teleport.hpp"
#include "services/vehicle/persist_car_service.hpp"
#include "backend/command.hpp"
#include "backend/player_command.hpp"
#include "services/gta_data/gta_data_service.hpp"

namespace big
{
	struct context_option
	{
		std::string name;
		std::function<void()> command;
	};

	struct s_context_menu
	{
		ContextEntityType type;
		int current_option = 0;
		ImVec2 menu_size   = {};
		std::vector<context_option> options;
	};

	struct model_bounding_box_screen_space
	{
		ImVec2 edge1, edge2, edge3, edge4;
		ImVec2 edge5, edge6, edge7, edge8;
	};

	class context_menu_service final
	{
	private:
		void fill_model_bounding_box_screen_space();
		static double distance_to_middle_of_screen(const rage::fvector2& screen_pos);

	public:
		context_menu_service();
		~context_menu_service();

		context_menu_service(const context_menu_service&)                = delete;
		context_menu_service(context_menu_service&&) noexcept            = delete;
		context_menu_service& operator=(const context_menu_service&)     = delete;
		context_menu_service& operator=(context_menu_service&&) noexcept = delete;

		bool enabled = false;
		s_context_menu* get_context_menu();
		void get_entity_closest_to_screen_center();
		void load_shared();

		static void disable_control_action_loop();
		static void context_menu();

		Entity m_handle;
		rage::fwEntity* m_pointer{};
		model_bounding_box_screen_space m_model_bounding_box_screen_space;

		s_context_menu vehicle_menu{ContextEntityType::VEHICLE,
		    0,
		    {},
		    {{"KILL ENGINE",
		         [this] {
			         if (entity::take_control_of(m_handle))
			         {
				         VEHICLE::SET_VEHICLE_ENGINE_HEALTH(m_handle, 0.f);
				         VEHICLE::SET_VEHICLE_ENGINE_ON(m_handle, false, true, false);
			         }
			         else
				         g_notification_service->push_warning("Toxic", "Failed to take control of vehicle.");
		         }},
		        {"COPY VEHICLE",
		            [this] {
						Vehicle v = persist_car_service::clone_ped_car(PLAYER::PLAYER_PED_ID(), m_handle);
			            script::get_current()->yield();
			            PED::SET_PED_INTO_VEHICLE(PLAYER::PLAYER_PED_ID(), v, -1);
		            }},
		        {"BOOST",
		            [this] {
			            if (entity::take_control_of(m_handle))
							VEHICLE::SET_VEHICLE_FORWARD_SPEED(m_handle, 79);
			            else
				            g_notification_service->push_warning("Toxic", "Failed to take control of vehicle.");
			            
		            }},
		        {"LAUNCH",
		            [this] {
			            if (entity::take_control_of(m_handle))
				            ENTITY::APPLY_FORCE_TO_ENTITY(m_handle, 1, 0.f, 0.f, 50000.f, 0.f, 0.f, 0.f, 0, 0, 1, 1, 0, 1);
			            else
				            g_notification_service->push_warning("Toxic", "Failed to take control of vehicle.");
		            }},
		        {"EJECT",
		            [this] {
			            if (ped::get_player_from_ped(VEHICLE::GET_PED_IN_VEHICLE_SEAT(m_handle, -1, 0)) != NULL)
			            {
				            static player_command* command = (player_command*)command::get(rage::consteval_joaat("vehkick"));
				            command->call(ped::get_player_from_ped(VEHICLE::GET_PED_IN_VEHICLE_SEAT(m_handle, -1, 0)), {});
			            }
			           
				        TASK::CLEAR_PED_TASKS_IMMEDIATELY(VEHICLE::GET_PED_IN_VEHICLE_SEAT(m_handle, -1, 0));
				        TASK::CLEAR_PED_TASKS_IMMEDIATELY(m_handle);
			            
		            }},
		        {"DELETE",
		            [this] {
			            if (entity::take_control_of(m_handle))
			            {
				            entity::delete_entity(m_handle);
			            }
		            }},
		        {"TP INTO", [this] {
			         teleport::into_vehicle(m_handle);
		         }}}};

		s_context_menu ped_menu{ContextEntityType::PED, 0, {}, {
		   {"DISARM",
		       [this] {
					for (auto& [_, weapon] : g_gta_data_service->weapons())
						WEAPON::REMOVE_WEAPON_FROM_PED(m_handle, weapon.m_hash);
		       }},
		   {"RAGDOLL",
		      [this] {
			         PED::SET_PED_TO_RAGDOLL(m_handle, 2000, 2000, 0, 0, 0, 0);
		            }},
		   {"DANCE", [this] {
			         ped::ped_play_animation(m_handle, "mini@strip_club@private_dance@part1", "priv_dance_p1");
		    }}
		}};

		s_context_menu object_menu{ContextEntityType::OBJECT, 0, {}, {}};

		s_context_menu player_menu{ContextEntityType::PLAYER, 0, {}, {
			{"STEAL IDENTITY", [this] {
				ped::steal_identity(m_handle);
		     }},
		    {"BREAKUP KICK", [this] {
			         static player_command* command = (player_command*)command::get(rage::consteval_joaat("breakup"));
			         command->call(ped::get_player_from_ped(m_handle), {});
		     }},
		    {"KICK", [this] {
			     static player_command* command = (player_command*)command::get(rage::consteval_joaat("nfkick"));
			     static player_command* command1 = (player_command*)command::get(rage::consteval_joaat("shkick"));
			     static player_command* command2 = (player_command*)command::get(rage::consteval_joaat("endkick"));
			     command->call(ped::get_player_from_ped(m_handle), {});
			     command1->call(ped::get_player_from_ped(m_handle), {});
			     command2->call(ped::get_player_from_ped(m_handle), {});
		     }},
		    {"DISARM", [this] {
			     static player_command* command = (player_command*)command::get(rage::consteval_joaat("remweaps"));
			         command->call(ped::get_player_from_ped(m_handle), {});
		    }},
		    {"RAGDOLL", [this] {
			     static player_command* command = (player_command*)command::get(rage::consteval_joaat("ragdoll"));
			     command->call(ped::get_player_from_ped(m_handle), {});
		     }}
		}};

		s_context_menu shared_menu{ContextEntityType::SHARED,
		    0,
		    {},
		    {{"EXPLODE",
		         [this] {
			         rage::fvector3 pos = *m_pointer->m_navigation->get_position();
			         FIRE::ADD_EXPLOSION(pos.x, pos.y, pos.z, 1, 1000, 1, 0, 1, 0);
		         }},
		        {"TP TO",
		            [this] {
			            rage::fvector3 pos = *m_pointer->m_navigation->get_position();
			            teleport::to_coords({pos.x, pos.y, pos.z});
		            }},
		        {"ENFLAME",
		            [this] {
			            Vector3 pos = ENTITY::GET_ENTITY_COORDS(m_handle, TRUE);
			            FIRE::START_ENTITY_FIRE(m_handle);
			            FIRE::START_SCRIPT_FIRE(pos.x, pos.y, pos.z, 25, TRUE);
			            MISC::SHOOT_SINGLE_BULLET_BETWEEN_COORDS(pos.x, pos.y, pos.z, pos.x, pos.y, pos.z, 1.f, 0, 615608432, PLAYER::PLAYER_PED_ID(), TRUE, TRUE, 1.f);
		            }},
		        {"COPY HASH", [this] {
			         ImGui::SetClipboardText(std::format("0x{:08X}", (rage::joaat_t)m_pointer->m_model_info->m_hash).c_str());
			         g_notification_service->push("Context Menu",
			             std::format("Copy hash 0x{:08X}", (rage::joaat_t)m_pointer->m_model_info->m_hash).c_str());
		         }}}};

		std::unordered_map<ContextEntityType, s_context_menu> options = {{ContextEntityType::VEHICLE, vehicle_menu}, {ContextEntityType::PLAYER, player_menu}, {ContextEntityType::PED, ped_menu}, {ContextEntityType::SHARED, shared_menu}, {ContextEntityType::OBJECT, object_menu}};
	};

	inline context_menu_service* g_context_menu_service{};
}