// Vehicle Service Point by Axe Cop

private ["_folder","_servicePointClasses","_maxDistance","_actionTitleFormat","_actionCostsFormat","_costsFree","_message","_messageShown","_refuel_enable","_refuel_costs","_refuel_updateInterval","_refuel_amount","_repair_enable","_repair_costs","_repair_repairTime","_rearm_enable","_rearm_costs","_rearm_magazineCount","_lastVehicle","_lastRole","_fnc_removeActions","_fnc_getCosts","_fnc_actionTitle","_fnc_isArmed","_fnc_getWeapons","_fnc_getMagazines"];

// ---------------- CONFIG START ----------------

// -- Single Currency Prices -- //
_coinsRepairAir = 5000;
_coinsRepairVehicles = 2000;
_coinsRepairTank = 20000;
_coinsRepairM113 = 75000;
_coinsRepairF35 = 100000;

_coinsRefuelVehicles = 0;
_coinsRefuelAir = 0;



//Rearm
_coinsRearmSUV = 20000;
_coinsRearmAir = 20000;
_coinsRearmVehicles = 10000;
_coinsRearmLB = 75000;
_coinsRearmVodnik = 50000;
_coinsRearmHMMWVSOV = 50000;
_coinsRearmHMMWVM2 = 60000;
_coinsRearmLRCZ = 30000;
_coinsRearmLRMG = 40000;
_coinsRearmUAZMG = 20000;
_coinsRearmBTR40 = 30000;
_coinsRearmHMMWVMK19 = 80000;
_coinsRearmHMMWVMK19D = 80000;
_coinsRearmTank = 100000;

// -- End Single Currency Prices -- //

// general settings
_folder = "Custom\service_point\"; // folder where the service point scripts are saved, relative to the mission file
_servicePointClasses = dayz_fuelpumparray; // service point classes (can be house, vehicle and unit classes)
_maxDistance = 10; // maximum distance from a service point for the options to be shown
_actionTitleFormat = "%1 (%2)"; // text of the vehicle menu, %1 = action name (Refuel, Repair, Rearm), %2 = costs (see format below)
_actionCostsFormat = "%2 %1"; // %1 = item name, %2 = item count
_costsFree = "free"; // text for no costs
_message = "Vehicle Service Point nearby"; // message to be shown when in range of a service point (set to "" to disable)

// refuel settings
_refuel_enable = true; // enable or disable the refuel option
_refuel_costs = [
	["AllVehicles",[CurrencyName,_coinsRefuelVehicles]],

	["Air",[CurrencyName,_coinsRefuelAir]]

	];
_refuel_updateInterval = 1; // update interval (in seconds)
_refuel_amount = 0.05; // amount of fuel to add with every update (in percent)

// repair settings
_repair_enable = true; // enable or disable the repair option
_repair_costs = [
	["Air",[CurrencyName,_coinsRepairAir]], // [1,"ItemGoldBar10oz",1]
	["Tank",[CurrencyName,_coinsRepairTank]], //
	["AllVehicles",[CurrencyName,_coinsRepairVehicles]], // 2 Gold for all other vehicles
	["M1133_MEV_EP1",[Currencyname,_coinsRepairM113]],
	["F35B",[CurrencyName,_coinsRepairF35]],
	["HMMWV_MK19",[CurrencyName,_coinsRearmHMMWVMK19]],
	["HMMWV_MK19_DES_EP1",[CurrencyName,_coinsRearmHMMWVMK19D]]
];
_repair_repairTime = 2; // time needed to repair each damaged part (in seconds)

// rearm settings
_rearm_enable = true; // enable or disable the rearm option
//_blockedWeaponNames=["S-5","Hydra","CRV7"]; // weapon names you wish to exclude from rearming.  Leave empty [] to allow all
_blockedWeaponNames=[]; // Weapon names you wish to exclude from rearming.  Leave empty [] to allow all
_blockedAmmoNames = [ // Ammo names you wish to exclude from rearming. Leave empty [] to allow all
	"192Rnd_57mm",
	"128Rnd_57mm",
	"1200Rnd_762x51_M240",
	"SmokeLauncherMag",
	"60Rnd_CMFlareMagazine",
	"120Rnd_CMFlareMagazine",
	"240Rnd_CMFlareMagazine",
	"120Rnd_CMFlare_Chaff_Magazine",
	"240Rnd_CMFlare_Chaff_Magazine",
	"4Rnd_Ch29",
	"80Rnd_80mm",
	"80Rnd_S8T",
	"150Rnd_30mmAP_2A42",
	"150Rnd_30mmHE_2A42",
	"38Rnd_FFAR",
	"12Rnd_CRV7",
	"1500Rnd_762x54_PKT",
	"2000Rnd_762x54_PKT",
	"150Rnd_30mmAP_2A42",
	"150Rnd_30mmHE_2A42",
	"230Rnd_30mmAP_2A42",
	"230Rnd_30mmHE_2A42",
	"4000Rnd_762x51_M134",
	"300Rnd_25mm_GAU12",
	"2Rnd_GBU12",
	"2Rnd_Sidewinder_F35",
	"14Rnd_57mm",
	"64Rnd_57mm"

	];
_rearm_costs = [

	["Car",[CurrencyName,_coinsRearmVehicles]],
	["Air",[CurrencyName,_coinsRearmAir]], //
	["Tank",[CurrencyName,_coinsRearmTank]], //
	["AllVehicles",[CurrencyName,_coinsRearmVehicles]], // 1 10oz Gold for all other vehicles
	["AH6J_DZE",[CurrencyName,_coinsRearmLB]],
	["GAZ_Vodnik_DZE",[CurrencyName,_coinsRearmVodnik]],
	["HMMWV_M998A2_SOV_DES_EP1_DZE",[CurrencyName,_coinsRearmHMMWVSOV]],
	["HMMWV_M1151_M2_CZ_DES_EP1_DZE",[CurrencyName,_coinsRearmHMMWVM2]],
	["LandRover_Special_CZ_EP1_DZE",[CurrencyName,_coinsRearmLRCZ]],
	["LandRover_MG_TK_EP1_DZE",[CurrencyName,_coinsRearmLRMG]],
	["UAZ_MG_TK_EP1_DZE",[CurrencyName,_coinsRearmUAZMG]],
	["BTR40_MG_TK_INS_EP1",[CurrencyName,_coinsRearmBTR40]]
];
_rearm_magazineCount = 1; // amount of magazines to be added to the vehicle weapon

// ----------------- CONFIG END -----------------

call compile preprocessFileLineNumbers (_folder + "ac_functions.sqf");
_lastVehicle = objNull;
_lastRole = [];

SP_refuel_action = -1;
SP_repair_action = -1;
SP_rearm_actions = [];

_messageShown = false;

_fnc_removeActions = {
	if (isNull _lastVehicle) exitWith {};
	_lastVehicle removeAction SP_refuel_action;
	SP_refuel_action = -1;
	_lastVehicle removeAction SP_repair_action;
	SP_repair_action = -1;
	{
		_lastVehicle removeAction _x;
	} forEach SP_rearm_actions;
	SP_rearm_actions = [];
	_lastVehicle = objNull;
	_lastRole = [];
};

_fnc_getCosts = {
	private ["_vehicle","_costs","_cost"];
	_vehicle = _this select 0;
	_costs = _this select 1;
	_cost = [];
	{
		private "_typeName";
		_typeName = _x select 0;
		if (_vehicle isKindOf _typeName) exitWith {
			_cost = _x select 1;
		};
	} forEach _costs;
	_cost
};

_fnc_actionTitle = {
	private ["_actionName","_costs","_costsText","_actionTitle"];
	_actionName = _this select 0;
	_costs = _this select 1;
	_costsText = _costsFree;
	if (count _costs == 2) then {
		private ["_itemName","_itemCount","_displayName"];
		_itemName = _costs select 0;
		_itemCount = _costs select 1;
		_displayName = _itemName;
		_costsText = format [_actionCostsFormat, _displayName, _itemCount];
	};
	_actionTitle = format [_actionTitleFormat, _actionName, _costsText];
	_actionTitle
};

_fnc_isArmed = {
	private ["_role","_armed"];
	_role = _this;
	_armed = count _role > 1;
	_armed
};

_fnc_getWeapons = {
	private ["_vehicle","_role","_weapons","_magazineNumber","_badAmmo","_badWeapon","_magazines","_weapon"];
	_vehicle = _this select 0;
	_role = _this select 1;
	_weapons = [];
	if (count _role > 1) then {
		private ["_turret","_weaponsTurret"];
		_turret = _role select 1;
		_weaponsTurret = _vehicle weaponsTurret _turret;
		{
			private "_weaponName";
			_weaponName = getText (configFile >> "CfgWeapons" >> _x >> "displayName");
			// block ammo types
			_badWeapon = _weaponName in _blockedWeaponNames;
			if (!_badWeapon) then {

				_weapon = _x;
				// get all ammo types for this weapon
				_magazines = [_weapon] call _fnc_getMagazines;

				// loop through all ammo types and add them to our list
				{
					_badAmmo = _x in _blockedAmmoNames;
					// check to see if our ammo is prohibited
					if (!_badAmmo) then {
						// add one entry to weapons per ammo type.
						_weapons set [count _weapons, [_weapon, _weaponName, _turret, _x]];
					};

				} foreach _magazines;
			};

		} forEach _weaponsTurret;
	} else {
		private ["_turret","_weaponsTurret","_badAmmo","_badWeapon","_magazines","_weapon"];
		_turret = [-1];
		_weaponsTurret = vehicle player weaponsTurret [-1];
		{
			private "_weaponName";
			_weaponName = getText (configFile >> "CfgWeapons" >> _x >> "displayName");

			// block ammo types
			_badWeapon = _weaponName in _blockedWeaponNames;
			if (!_badWeapon) then {

				_weapon = _x;
				// get all ammo types for this weapon
				_magazines = [_weapon] call _fnc_getMagazines;

				// loop through all ammo types and add them to our list
				{
					// check to see if our ammo is prohibited
					_badAmmo = _x in _blockedAmmoNames;
					if (!_badAmmo) then {
			_weapons set [count _weapons, [_weapon, _weaponName, _turret, _x]];
		} forEach _weaponsTurret;
	};
	_weapons
};

while {true} do {
	private ["_vehicle","_inVehicle"];
	_vehicle = vehicle player;
	_inVehicle = _vehicle != player;
	if (local _vehicle && _inVehicle) then {
		private ["_pos","_servicePoints","_inRange"];
		_pos = getPosATL _vehicle;
		_servicePoints = (nearestObjects [_pos, _servicePointClasses, _maxDistance]) - [_vehicle];
		_inRange = count _servicePoints > 0;
		if (_inRange) then {
			private ["_servicePoint","_role","_actionCondition","_costs","_actionTitle"];
			_servicePoint = _servicePoints select 0;
			if (assignedDriver _vehicle == player) then {
				_role = ["Driver", [-1]];
			} else {
				_role = assignedVehicleRole player;
			};
			if (((str _role) != (str _lastRole)) || (_vehicle != _lastVehicle)) then {
				// vehicle or seat changed
				call _fnc_removeActions;
			};
			_lastVehicle = _vehicle;
			_lastRole = _role;
			_actionCondition = "vehicle _this == _target && local _target";
			if (SP_refuel_action < 0 && _refuel_enable) then {
				_costs = [_vehicle, _refuel_costs] call _fnc_getCosts;
				_actionTitle = ["Refuel", _costs] call _fnc_actionTitle;
				SP_refuel_action = _vehicle addAction [_actionTitle, _folder + "service_point_refuel.sqf", [_servicePoint, _costs, _refuel_updateInterval, _refuel_amount], -1, false, true, "", _actionCondition];
			};
			if (SP_repair_action < 0 && _repair_enable) then {
				_costs = [_vehicle, _repair_costs] call _fnc_getCosts;
				_actionTitle = ["Repair", _costs] call _fnc_actionTitle;
				SP_repair_action = _vehicle addAction [_actionTitle, _folder + "service_point_repair.sqf", [_servicePoint, _costs, _repair_repairTime], -1, false, true, "", _actionCondition];
			};
			if ((_role call _fnc_isArmed) && (count SP_rearm_actions == 0) && _rearm_enable) then {
				private ["_weapons"];
				_costs = [_vehicle, _rearm_costs] call _fnc_getCosts;
				_weapons = [_vehicle, _role] call _fnc_getWeapons;
				{
					private "_weaponName";
					_weaponName = _x select 1;
					_actionTitle = [format["Rearm %1", _weaponName], _costs] call _fnc_actionTitle;
					SP_rearm_action = _vehicle addAction [_actionTitle, _folder + "service_point_rearm.sqf", [_servicePoint, _costs, _rearm_magazineCount, _x], -1, false, true, "", _actionCondition];
					SP_rearm_actions set [count SP_rearm_actions, SP_rearm_action];
				} forEach _weapons;
			};
			if (!_messageShown && _message != "") then {
				_messageShown = true;
				_vehicle vehicleChat _message;
			};
		} else {
			call _fnc_removeActions;
			_messageShown = false;
		};
	} else {
		call _fnc_removeActions;
		_messageShown = false;
	};
	sleep 2;
};