ECS strap component (#12627)

2022-11-18 22:08:28 +01:00
parent b749d0d275
commit 5aeb2ac4a0
15 changed files with 997 additions and 993 deletions
--- a/Content.Shared/Buckle/SharedBuckleSystem.Buckle.cs
+++ b/Content.Shared/Buckle/SharedBuckleSystem.Buckle.cs
@@ -0,0 +1,107 @@
+using Content.Shared.Buckle.Components;
+using Content.Shared.Interaction.Events;
+using Content.Shared.Movement.Events;
+using Content.Shared.Standing;
+using Content.Shared.Throwing;
+using Content.Shared.Vehicle.Components;
+using Robust.Shared.Map;
+using Robust.Shared.Physics.Events;
+
+namespace Content.Shared.Buckle;
+
+public abstract partial class SharedBuckleSystem
+{
+    private void InitializeBuckle()
+    {
+        SubscribeLocalEvent<SharedBuckleComponent, PreventCollideEvent>(PreventCollision);
+        SubscribeLocalEvent<SharedBuckleComponent, DownAttemptEvent>(HandleDown);
+        SubscribeLocalEvent<SharedBuckleComponent, StandAttemptEvent>(HandleStand);
+        SubscribeLocalEvent<SharedBuckleComponent, ThrowPushbackAttemptEvent>(HandleThrowPushback);
+        SubscribeLocalEvent<SharedBuckleComponent, UpdateCanMoveEvent>(HandleMove);
+        SubscribeLocalEvent<SharedBuckleComponent, ChangeDirectionAttemptEvent>(OnBuckleChangeDirectionAttempt);
+    }
+
+    private void PreventCollision(EntityUid uid, SharedBuckleComponent component, ref PreventCollideEvent args)
+    {
+        if (args.BodyB.Owner != component.LastEntityBuckledTo)
+            return;
+
+        if (component.Buckled || component.DontCollide)
+            args.Cancelled = true;
+    }
+
+    private void HandleDown(EntityUid uid, SharedBuckleComponent component, DownAttemptEvent args)
+    {
+        if (component.Buckled)
+            args.Cancel();
+    }
+
+    private void HandleStand(EntityUid uid, SharedBuckleComponent component, StandAttemptEvent args)
+    {
+        if (component.Buckled)
+            args.Cancel();
+    }
+
+    private void HandleThrowPushback(EntityUid uid, SharedBuckleComponent component, ThrowPushbackAttemptEvent args)
+    {
+        if (component.Buckled)
+            args.Cancel();
+    }
+
+    private void HandleMove(EntityUid uid, SharedBuckleComponent component, UpdateCanMoveEvent args)
+    {
+        if (component.LifeStage > ComponentLifeStage.Running)
+            return;
+
+        if (component.Buckled &&
+            !HasComp<VehicleComponent>(Transform(uid).ParentUid)) // buckle+vehicle shitcode
+            args.Cancel();
+    }
+
+    private void OnBuckleChangeDirectionAttempt(EntityUid uid, SharedBuckleComponent component, ChangeDirectionAttemptEvent args)
+    {
+        if (component.Buckled)
+            args.Cancel();
+    }
+
+    public bool IsBuckled(EntityUid uid, SharedBuckleComponent? component = null)
+    {
+        return Resolve(uid, ref component, false) && component.Buckled;
+    }
+
+    /// <summary>
+    ///     Reattaches this entity to the strap, modifying its position and rotation.
+    /// </summary>
+    /// <param name="buckleId">The entity to reattach.</param>
+    /// <param name="strap">The strap to reattach to.</param>
+    /// <param name="buckle">The buckle component of the entity to reattach.</param>
+    public void ReAttach(EntityUid buckleId, SharedStrapComponent strap, SharedBuckleComponent? buckle = null)
+    {
+        if (!Resolve(buckleId, ref buckle, false))
+            return;
+
+        var ownTransform = Transform(buckleId);
+        var strapTransform = Transform(strap.Owner);
+
+        ownTransform.Coordinates = new EntityCoordinates(strapTransform.Owner, strap.BuckleOffset);
+
+        // Buckle subscribes to move for <reasons> so this might fail.
+        // TODO: Make buckle not do that.
+        if (ownTransform.ParentUid != strapTransform.Owner)
+            return;
+
+        ownTransform.LocalRotation = Angle.Zero;
+
+        switch (strap.Position)
+        {
+            case StrapPosition.None:
+                break;
+            case StrapPosition.Stand:
+                _standing.Stand(buckleId);
+                break;
+            case StrapPosition.Down:
+                _standing.Down(buckleId, false, false);
+                break;
+        }
+    }
+}