From a90dd7210e1c85e2276b4464efc30932795c65cc Mon Sep 17 00:00:00 2001
From: BaerMitUmlaut <BaerMitUmlaut@users.noreply.github.com>
Date: Sat, 27 Oct 2018 22:13:40 +0200
Subject: [PATCH] Safemode - Fix indicator when switching units (#6612)

* Fix safe mode indicator when switching units

* Use DIK macro instead of magic number
---
 addons/safemode/XEH_postInit.sqf | 18 ++++++++----------
 1 file changed, 8 insertions(+), 10 deletions(-)

diff --git a/addons/safemode/XEH_postInit.sqf b/addons/safemode/XEH_postInit.sqf
index 81c1521c92..db922f9b35 100644
--- a/addons/safemode/XEH_postInit.sqf
+++ b/addons/safemode/XEH_postInit.sqf
@@ -1,15 +1,10 @@
 // by esteldunedain
 #include "script_component.hpp"
+#include "\a3\ui_f\hpp\defineDIKCodes.inc"
 
 if (!hasInterface) exitWith {};
 
-// IGNORE_PRIVATE_WARNING(_player)
-//["Soldier", {_player = ACE_player; if (currentWeapon _player in (_player getVariable [QGVAR(safedWeapons), []])) then {[false] call FUNC(setSafeModeVisual)}] call EFUNC(common,addInfoDisplayEventHandler);
-//@todo addEventHandler infoDisplayChanged with select 1 == "Soldier"
-
-// add keybinds
-["ACE3 Weapons", QGVAR(safeMode), localize LSTRING(SafeMode),
-{
+["ACE3 Weapons", QGVAR(safeMode), localize LSTRING(SafeMode), {
     // Conditions: canInteract
     if !([ACE_player, objNull, ["isNotEscorting", "isNotInside", "isNotSwimming"]] call EFUNC(common,canInteractWith)) exitWith {false};
     // Conditions: specific
@@ -18,6 +13,9 @@ if (!hasInterface) exitWith {};
     // Statement
     [ACE_player, currentWeapon ACE_player, currentMuzzle ACE_player] call FUNC(lockSafety);
     true
-},
-{false},
-[41, [false, true, false]], false] call CBA_fnc_addKeybind;
+}, {false}, [DIK_GRAVE, [false, true, false]], false] call CBA_fnc_addKeybind;
+
+["unit", {
+    private _weaponSafe = currentWeapon ACE_player in (ACE_player getVariable [QGVAR(safedWeapons), []]);
+    [!_weaponSafe] call FUNC(setSafeModeVisual);
+}] call CBA_fnc_addPlayerEventHandler;