ECS buckle (#12586)

Content.Client/Buckle/BuckleSystem.cs

@@ -1,18 +1,67 @@
 using Content.Client.Buckle.Strap;
+using Content.Shared.ActionBlocker;
 using Content.Shared.Buckle;
 using Content.Shared.Buckle.Components;
+using Content.Shared.Vehicle.Components;
+using Robust.Client.GameObjects;
 using Robust.Shared.GameStates;
 
 namespace Content.Client.Buckle
 {
     internal sealed class BuckleSystem : SharedBuckleSystem
     {
+        [Dependency] private readonly ActionBlockerSystem _actionBlocker = default!;
+
         public override void Initialize()
         {
             base.Initialize();
+
+            SubscribeLocalEvent<BuckleComponent, ComponentHandleState>(OnBuckleHandleState);
+
             SubscribeLocalEvent<StrapComponent, ComponentHandleState>(OnStrapHandleState);
         }
 
+        private void OnBuckleHandleState(EntityUid uid, BuckleComponent buckle, ref ComponentHandleState args)
+        {
+            if (args.Current is not BuckleComponentState state)
+                return;
+
+            buckle.Buckled = state.Buckled;
+            buckle.LastEntityBuckledTo = state.LastEntityBuckledTo;
+            buckle.DontCollide = state.DontCollide;
+
+            _actionBlocker.UpdateCanMove(uid);
+
+            if (!TryComp(uid, out SpriteComponent? ownerSprite))
+                return;
+
+            if (HasComp<VehicleComponent>(buckle.LastEntityBuckledTo))
+                return;
+
+            // Adjust draw depth when the chair faces north so that the seat back is drawn over the player.
+            // Reset the draw depth when rotated in any other direction.
+            // TODO when ECSing, make this a visualizer
+            // This code was written before rotatable viewports were introduced, so hard-coding Direction.North
+            // and comparing it against LocalRotation now breaks this in other rotations. This is a FIXME, but
+            // better to get it working for most people before we look at a more permanent solution.
+            if (buckle.Buckled &&
+                buckle.LastEntityBuckledTo != null &&
+                Transform(buckle.LastEntityBuckledTo.Value).LocalRotation.GetCardinalDir() == Direction.North &&
+                TryComp<SpriteComponent>(buckle.LastEntityBuckledTo, out var buckledSprite))
+            {
+                buckle.OriginalDrawDepth ??= ownerSprite.DrawDepth;
+                ownerSprite.DrawDepth = buckledSprite.DrawDepth - 1;
+                return;
+            }
+
+            // If here, we're not turning north and should restore the saved draw depth.
+            if (buckle.OriginalDrawDepth.HasValue)
+            {
+                ownerSprite.DrawDepth = buckle.OriginalDrawDepth.Value;
+                buckle.OriginalDrawDepth = null;
+            }
+        }
+
         private void OnStrapHandleState(EntityUid uid, StrapComponent component, ref ComponentHandleState args)
         {
             if (args.Current is not StrapComponentState state) return;