Generic clothing speed modifiers + hardsuit slowdown (#7094)

2022-03-13 23:03:49 -07:00
parent 1652498830
commit 8005cf31bb
8 changed files with 136 additions and 34 deletions
--- a/Content.Shared/Clothing/ClothingSpeedModifierSystem.cs
+++ b/Content.Shared/Clothing/ClothingSpeedModifierSystem.cs
@@ -0,0 +1,70 @@
+using Content.Shared.Movement.EntitySystems;
+using Robust.Shared.Containers;
+using Robust.Shared.GameStates;
+
+namespace Content.Shared.Clothing;
+
+public sealed class ClothingSpeedModifierSystem : EntitySystem
+{
+    [Dependency] private readonly MovementSpeedModifierSystem _movementSpeed = default!;
+    [Dependency] private readonly SharedContainerSystem _container = default!;
+
+    public override void Initialize()
+    {
+        base.Initialize();
+
+        SubscribeLocalEvent<ClothingSpeedModifierComponent, ComponentGetState>(OnGetState);
+        SubscribeLocalEvent<ClothingSpeedModifierComponent, ComponentHandleState>(OnHandleState);
+        SubscribeLocalEvent<ClothingSpeedModifierComponent, RefreshMovementSpeedModifiersEvent>(OnRefreshMoveSpeed);
+    }
+
+    // Public API
+
+    public void SetClothingSpeedModifierEnabled(EntityUid uid, bool enabled, ClothingSpeedModifierComponent? component = null)
+    {
+        if (!Resolve(uid, ref component, false))
+            return;
+
+        if (component.Enabled != enabled)
+        {
+            component.Enabled = enabled;
+            Dirty(component);
+
+            // inventory system will automatically hook into the event raised by this and update accordingly
+            if (_container.TryGetContainingContainer(uid, out var container))
+            {
+                _movementSpeed.RefreshMovementSpeedModifiers(container.Owner);
+            }
+        }
+    }
+
+    // Event handlers
+
+    private void OnGetState(EntityUid uid, ClothingSpeedModifierComponent component, ref ComponentGetState args)
+    {
+        args.State = new ClothingSpeedModifierComponentState(component.WalkModifier, component.SprintModifier, component.Enabled);
+    }
+
+    private void OnHandleState(EntityUid uid, ClothingSpeedModifierComponent component, ref ComponentHandleState args)
+    {
+        if (args.Current is ClothingSpeedModifierComponentState state)
+        {
+            component.WalkModifier = state.WalkModifier;
+            component.SprintModifier = state.SprintModifier;
+            component.Enabled = state.Enabled;
+
+            if (_container.TryGetContainingContainer(uid, out var container))
+            {
+                _movementSpeed.RefreshMovementSpeedModifiers(container.Owner);
+            }
+        }
+    }
+
+    private void OnRefreshMoveSpeed(EntityUid uid, ClothingSpeedModifierComponent component, RefreshMovementSpeedModifiersEvent args)
+    {
+        if (!component.Enabled)
+            return;
+
+        args.ModifySpeed(component.WalkModifier, component.SprintModifier);
+    }
+}