Ensnaring Component and Bola Update (#9968)

2022-08-24 10:50:31 -04:00
parent 16be5184a4
commit cd78c5451d
29 changed files with 681 additions and 2 deletions
--- a/Content.Shared/Ensnaring/SharedEnsnareableSystem.cs
+++ b/Content.Shared/Ensnaring/SharedEnsnareableSystem.cs
@@ -0,0 +1,59 @@
+using Content.Shared.Ensnaring.Components;
+using Content.Shared.Movement.Systems;
+
+namespace Content.Shared.Ensnaring;
+
+public abstract class SharedEnsnareableSystem : EntitySystem
+{
+    [Dependency] private readonly MovementSpeedModifierSystem _speedModifier = default!;
+
+    public override void Initialize()
+    {
+        base.Initialize();
+
+        SubscribeLocalEvent<SharedEnsnareableComponent, RefreshMovementSpeedModifiersEvent>(MovementSpeedModify);
+        SubscribeLocalEvent<SharedEnsnareableComponent, EnsnareEvent>(OnEnsnare);
+        SubscribeLocalEvent<SharedEnsnareableComponent, EnsnareRemoveEvent>(OnEnsnareRemove);
+        SubscribeLocalEvent<SharedEnsnareableComponent, EnsnaredChangedEvent>(OnEnsnareChange);
+    }
+
+    private void OnEnsnare(EntityUid uid, SharedEnsnareableComponent component, EnsnareEvent args)
+    {
+        component.WalkSpeed = args.WalkSpeed;
+        component.SprintSpeed = args.SprintSpeed;
+
+        _speedModifier.RefreshMovementSpeedModifiers(uid);
+
+        var ev = new EnsnaredChangedEvent(component.IsEnsnared);
+        RaiseLocalEvent(uid, ev, true);
+    }
+
+    private void OnEnsnareRemove(EntityUid uid, SharedEnsnareableComponent component, EnsnareRemoveEvent args)
+    {
+        _speedModifier.RefreshMovementSpeedModifiers(uid);
+
+        var ev = new EnsnaredChangedEvent(component.IsEnsnared);
+        RaiseLocalEvent(uid, ev, true);
+    }
+
+    private void OnEnsnareChange(EntityUid uid, SharedEnsnareableComponent component, EnsnaredChangedEvent args)
+    {
+        UpdateAppearance(uid, component);
+    }
+
+    private void UpdateAppearance(EntityUid uid, SharedEnsnareableComponent? component, AppearanceComponent? appearance = null)
+    {
+        if (!Resolve(uid, ref component, ref appearance, false))
+            return;
+
+        appearance.SetData(EnsnareableVisuals.IsEnsnared, component.IsEnsnared);
+    }
+
+    private void MovementSpeedModify(EntityUid uid, SharedEnsnareableComponent component, RefreshMovementSpeedModifiersEvent args)
+    {
+        if (!component.IsEnsnared)
+            return;
+
+        args.ModifySpeed(component.WalkSpeed, component.SprintSpeed);
+    }
+}