YimMenu/src/views/vehicle/view_spawn_vehicle.cpp

#include "fiber_pool.hpp"
#include "natives.hpp"
#include "services/gta_data/gta_data_service.hpp"
#include "services/model_preview/model_preview_service.hpp"
#include "util/vehicle.hpp"
#include "views/view.hpp"

namespace big
{
	void render_spawn_new_vehicle()
	{
		ImGui::SetWindowSize({0.f, (float)*g_pointers->m_gta.m_resolution_y}, ImGuiCond_Always);

		if (ImGui::Checkbox("PREVIEW"_T.data(), &g.spawn_vehicle.preview_vehicle))
		{
			if (!g.spawn_vehicle.preview_vehicle)
			{
				g_model_preview_service->stop_preview();
			}
		}
		ImGui::SameLine();
		components::command_checkbox<"spawnin">();
		ImGui::SameLine();
		components::command_checkbox<"spawnmaxed">();

		static char plate_buf[9] = {0};
		strncpy(plate_buf, g.spawn_vehicle.plate.c_str(), 9);

		ImGui::SetNextItemWidth(300.f);
		components::input_text_with_hint("PLATE"_T, "PLATE_NUMBER"_T, plate_buf, sizeof(plate_buf), ImGuiInputTextFlags_None, [] {
			g.spawn_vehicle.plate = plate_buf;
		});

		static int selected_class = -1;
		const auto& class_arr     = g_gta_data_service->vehicle_classes();

		ImGui::SetNextItemWidth(300.f);
		if (ImGui::BeginCombo("VEHICLE_CLASS"_T.data(),
		        selected_class == -1 ? "ALL"_T.data() : class_arr[selected_class].c_str()))
		{
			if (ImGui::Selectable("ALL"_T.data(), selected_class == -1))
			{
				selected_class = -1;
			}

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

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

			ImGui::EndCombo();
		}


		static char search[64];

		ImGui::SetNextItemWidth(300.f);
		components::input_text_with_hint("MODEL_NAME"_T, "SEARCH"_T, search, sizeof(search), ImGuiInputTextFlags_None);


		if (ImGui::BeginListBox("###vehicles", {300, static_cast<float>(*g_pointers->m_gta.m_resolution_y - 188 - 38 * 4)}))
		{
			if (self::veh)
			{
				static auto veh_hash = 0;

				g_fiber_pool->queue_job([] {
					veh_hash = ENTITY::GET_ENTITY_MODEL(self::veh);
				});

				if (veh_hash)
				{
					const auto& item = g_gta_data_service->vehicle_by_hash(veh_hash);

					components::selectable(std::vformat("SPAWN_VEHICLE_CURRENT_VEHICLE"_T, std::make_format_args(item.m_display_name)), false, [] {
						if (self::veh)
						{
							Vector3 spawn_location = vehicle::get_spawn_location(g.spawn_vehicle.spawn_inside, veh_hash);
							float spawn_heading = ENTITY::GET_ENTITY_HEADING(self::ped);

							auto owned_mods = vehicle::get_owned_mods_from_vehicle(self::veh);

							auto veh = vehicle::clone_from_owned_mods(owned_mods, spawn_location, spawn_heading);

							if (veh == 0)
							{
								g_notification_service->push_error("VEHICLE"_T.data(), "UNABLE_TO_SPAWN_VEHICLE"_T.data());
							}
							else
							{
								if (g.spawn_vehicle.spawn_maxed)
								{
									vehicle::max_vehicle(veh);
								}

								vehicle::set_plate(veh, plate_buf);

								if (g.spawn_vehicle.spawn_inside)
								{
									vehicle::teleport_into_vehicle(veh);
								}
							}
						}

						g_model_preview_service->stop_preview();
					});

					if (!g.spawn_vehicle.preview_vehicle || (g.spawn_vehicle.preview_vehicle && !ImGui::IsAnyItemHovered()))
					{
						g_model_preview_service->stop_preview();
					}
					else if (ImGui::IsItemHovered())
					{
						g_fiber_pool->queue_job([] {
							g_model_preview_service->show_vehicle(vehicle::get_owned_mods_from_vehicle(self::veh),
							    g.spawn_vehicle.spawn_maxed);
						});
					}
				}
			}

			const auto& item_arr = g_gta_data_service->vehicles();
			if (item_arr.size() > 0)
			{
				std::string lower_search = search;
				std::transform(lower_search.begin(), lower_search.end(), lower_search.begin(), tolower);

				for (auto& item : item_arr)
				{
					const auto& vehicle = item.second;

					std::string display_name         = vehicle.m_display_name;
					std::string display_manufacturer = vehicle.m_display_manufacturer;
					std::string clazz                = vehicle.m_vehicle_class;

					std::transform(display_name.begin(), display_name.end(), display_name.begin(), ::tolower);
					std::transform(display_manufacturer.begin(), display_manufacturer.end(), display_manufacturer.begin(), ::tolower);

					if ((selected_class == -1 || class_arr[selected_class] == clazz)
					    && (display_name.find(lower_search) != std::string::npos || display_manufacturer.find(lower_search) != std::string::npos))
					{
						ImGui::PushID(vehicle.m_hash);
						components::selectable(vehicle.m_display_name, false, [&vehicle] {
							const auto spawn_location =
							    vehicle::get_spawn_location(g.spawn_vehicle.spawn_inside, vehicle.m_hash);
							const auto spawn_heading = ENTITY::GET_ENTITY_HEADING(self::ped);

							const auto veh = vehicle::spawn(vehicle.m_hash, spawn_location, spawn_heading);

							if (veh == 0)
							{
								g_notification_service->push_error("VEHICLE"_T.data(), "UNABLE_TO_SPAWN_VEHICLE"_T.data());
							}
							else
							{
								if (g.spawn_vehicle.spawn_maxed)
								{
									vehicle::max_vehicle(veh);
								}

								vehicle::set_plate(veh, plate_buf);

								if (g.spawn_vehicle.spawn_inside)
								{
									vehicle::teleport_into_vehicle(veh);
								}
							}

							g_model_preview_service->stop_preview();
						});
						ImGui::PopID();

						if (!g.spawn_vehicle.preview_vehicle || (g.spawn_vehicle.preview_vehicle && !ImGui::IsAnyItemHovered()))
						{
							g_model_preview_service->stop_preview();
						}
						else if (ImGui::IsItemHovered())
						{
							g_model_preview_service->show_vehicle(vehicle.m_hash, g.spawn_vehicle.spawn_maxed);
						}
					}
				}
			}
			else
			{
				ImGui::Text("NO_VEHICLE_IN_REGISTRY"_T.data());
			}
			ImGui::EndListBox();
		}
	}

	void view::spawn_vehicle()
	{
		static int spawn_type = 0;
		ImGui::RadioButton("New", &spawn_type, 0);
		ImGui::SameLine();
		ImGui::RadioButton("Personal", &spawn_type, 1);
		ImGui::SameLine();
		ImGui::RadioButton("Persistent", &spawn_type, 2);

		switch (spawn_type)
		{
		case 0: render_spawn_new_vehicle(); break;
		case 1: view::pv(); break;
		case 2: view::persist_car(); break;
		}
	}
}