Pacification status effect and Pax (#10796)

2022-09-05 22:52:58 -04:00
parent 263912f08c
commit a29e86a896
8 changed files with 90 additions and 0 deletions
--- a/Content.Shared/CombatMode/Pacification/PacificationSystem.cs
+++ b/Content.Shared/CombatMode/Pacification/PacificationSystem.cs
@@ -0,0 +1,44 @@
+using Content.Shared.CombatMode;
+using Content.Shared.Actions;
+
+namespace Content.Shared.CombatMode.Pacification
+{
+    public sealed class PacificationSystem : EntitySystem
+    {
+        [Dependency] private readonly SharedActionsSystem _actionsSystem = default!;
+        public override void Initialize()
+        {
+            base.Initialize();
+            SubscribeLocalEvent<PacifiedComponent, ComponentInit>(OnInit);
+            SubscribeLocalEvent<PacifiedComponent, ComponentShutdown>(OnShutdown);
+        }
+
+        private void OnInit(EntityUid uid, PacifiedComponent component, ComponentInit args)
+        {
+            if (!TryComp<SharedCombatModeComponent>(uid, out var combatMode))
+                return;
+
+            if (combatMode.DisarmAction != null)
+            {
+                _actionsSystem.SetToggled(combatMode.DisarmAction, false);
+                _actionsSystem.SetEnabled(combatMode.DisarmAction, false);
+            }
+            if (combatMode.CombatToggleAction != null)
+            {
+                combatMode.IsInCombatMode = false;
+                _actionsSystem.SetEnabled(combatMode.CombatToggleAction, false);
+            }
+        }
+
+        private void OnShutdown(EntityUid uid, PacifiedComponent component, ComponentShutdown args)
+        {
+            if (!TryComp<SharedCombatModeComponent>(uid, out var combatMode))
+                return;
+
+            if (combatMode.DisarmAction != null)
+                _actionsSystem.SetEnabled(combatMode.DisarmAction, true);
+            if (combatMode.CombatToggleAction != null)
+                _actionsSystem.SetEnabled(combatMode.CombatToggleAction, true);
+        }
+    }
+}