#pragma once
#include "entity.hpp"
#include "gta/enums.hpp"
#include "natives.hpp"
#include "pointers.hpp"
#include "script_global.hpp"
#include "services/players/player_service.hpp"
#include "pools.hpp"

namespace big::train
{
	inline int get_closest_train()
	{
		for (auto veh : pools::get_all_vehicles())
		{
			if (veh->m_model_info->m_hash == "freight"_J)
				return g_pointers->m_gta.m_ptr_to_handle(veh);
		}
		return 0;
	}

	inline void hijack_train()
	{
		auto train = get_closest_train();

		if (train != 0)
		{
			if (auto ped = VEHICLE::GET_PED_IN_VEHICLE_SEAT(train, -1, true))
				TASK::CLEAR_PED_TASKS_IMMEDIATELY(ped);

			PED::SET_PED_INTO_VEHICLE(PLAYER::PLAYER_PED_ID(), train, -1);
			g_notification_service.push_success("HIJACK_TRAIN"_T.data(), "HIJACK_TRAIN_FOUND_TRAIN"_T.data());
		}
		else
		{
			g_notification_service.push_warning("HIJACK_TRAIN"_T.data(), "HIJACK_TRAIN_NOTFOUND_TRAIN"_T.data());
		}
	}

	inline void delete_train()
	{
		if (!self::veh && get_closest_train() != 0)
		{
			VEHICLE::DELETE_ALL_TRAINS();
		}
		else
		{
			g_notification_service.push_warning("HIJACK_TRAIN"_T.data(), "HIJACK_TRAIN_NOTFOUND_TRAIN"_T.data());
		}
	}

	inline void exit_train()
	{
		if (self::veh && VEHICLE::IS_THIS_MODEL_A_TRAIN(ENTITY::GET_ENTITY_MODEL(self::veh)))
			TASK::CLEAR_PED_TASKS_IMMEDIATELY(PLAYER::PLAYER_PED_ID());
	}

	inline void set_train_speed(float value)
	{
		if (PED::IS_PED_IN_ANY_VEHICLE(PLAYER::PLAYER_PED_ID(), false))
			VEHICLE::SET_TRAIN_CRUISE_SPEED(get_closest_train(), value);
	}
}