#include "backend/looped_command.hpp"
#include "natives.hpp"
#include "util/vehicle.hpp"
#include "gta/vehicle_values.hpp"

namespace big
{
	class keep_vehicle_repaired : looped_command
	{
		using looped_command::looped_command;

		virtual void on_tick() override
		{
			Vehicle veh = self::veh;

			if (veh == 0 || VEHICLE::GET_PED_IN_VEHICLE_SEAT(self::veh, SEAT_DRIVER, FALSE) != self::ped)
			{
				return;
			}

			if (*g_pointers->m_gta.m_is_session_started)
			{
				if (!entity::take_control_of(veh, 0))
				{
					return;
				}
			}

			if (VEHICLE::GET_DOES_VEHICLE_HAVE_DAMAGE_DECALS(veh))
			{
				if (!g.vehicle.keep_vehicle_clean)
				{
					VEHICLE::SET_VEHICLE_DIRT_LEVEL(veh, 0.f);
				}

				// Rear window
				constexpr int rear_window_index = 7;
				if (!VEHICLE::IS_VEHICLE_WINDOW_INTACT(veh, rear_window_index))
				{
					VEHICLE::FIX_VEHICLE_WINDOW(veh, rear_window_index);
				}

				g_pointers->m_gta.m_decal_manager_remove(g_pointers->m_gta.m_decal_manager, g_pointers->m_gta.m_handle_to_ptr(veh), -1, 0, 0x0001E000);

				if (!g.vehicle.god_mode)
				{
					VEHICLE::SET_VEHICLE_DEFORMATION_FIXED(veh);
				}

				vehicle::repair_engine_from_water(veh);

				const auto door_count = VEHICLE::GET_NUMBER_OF_VEHICLE_DOORS(veh);
				for (int i = 0; i < door_count; i++)
				{
					if (VEHICLE::IS_VEHICLE_DOOR_DAMAGED(veh, i))
					{
						VEHICLE::SET_VEHICLE_FIXED(veh);
						return;
					}
				}

				for (int i = 0; i <= 14; i++)
				{
					if (VEHICLE::DOES_EXTRA_EXIST(veh, i) && VEHICLE::IS_VEHICLE_EXTRA_TURNED_ON(veh, i) && VEHICLE::IS_EXTRA_BROKEN_OFF(veh, i))
					{
						VEHICLE::SET_VEHICLE_FIXED(veh);
						return;
					}
				}

				if (VEHICLE::IS_VEHICLE_BUMPER_BOUNCING(veh, TRUE) || VEHICLE::IS_VEHICLE_BUMPER_BOUNCING(veh, FALSE) || VEHICLE::GET_VEHICLE_NUM_OF_BROKEN_LOOSEN_PARTS(veh) > 0)
				{
					VEHICLE::SET_VEHICLE_FIXED(veh);
					return;
				}
			}
		}
	};

	keep_vehicle_repaired
	    g_keep_vehicle_repaired("keepfixed", "KEEP_VEHICLE_FIXED", "KEEP_VEHICLE_FIXED_DESC", g.vehicle.keep_vehicle_repaired);
}