Fix barriers not working (#19774)

2023-09-04 15:33:40 -07:00
parent e4ca6f4fb9
commit 0527d2e719
5 changed files with 89 additions and 58 deletions
--- a/Content.Shared/Security/Systems/DeployableBarrierSystem.cs
+++ b/Content.Shared/Security/Systems/DeployableBarrierSystem.cs
@@ -0,0 +1,67 @@
+using Content.Shared.Lock;
+using Content.Shared.Pulling;
+using Content.Shared.Pulling.Components;
+using Content.Shared.Security.Components;
+using Robust.Shared.Physics.Systems;
+
+namespace Content.Shared.Security.Systems;
+
+public sealed class DeployableBarrierSystem : EntitySystem
+{
+    [Dependency] private readonly SharedAppearanceSystem _appearance = default!;
+    [Dependency] private readonly FixtureSystem _fixtures = default!;
+    [Dependency] private readonly SharedPointLightSystem _pointLight = default!;
+    [Dependency] private readonly SharedPhysicsSystem _physics = default!;
+    [Dependency] private readonly SharedPullingSystem _pulling = default!;
+    [Dependency] private readonly SharedTransformSystem _transform = default!;
+
+    public override void Initialize()
+    {
+        SubscribeLocalEvent<DeployableBarrierComponent, MapInitEvent>(OnMapInit);
+        SubscribeLocalEvent<DeployableBarrierComponent, LockToggledEvent>(OnLockToggled);
+    }
+
+    private void OnMapInit(EntityUid uid, DeployableBarrierComponent component, MapInitEvent args)
+    {
+        if (!TryComp(uid, out LockComponent? lockComponent))
+            return;
+
+        ToggleBarrierDeploy(uid, lockComponent.Locked, component);
+    }
+
+    private void OnLockToggled(EntityUid uid, DeployableBarrierComponent component, ref LockToggledEvent args)
+    {
+        ToggleBarrierDeploy(uid, args.Locked, component);
+    }
+
+    private void ToggleBarrierDeploy(EntityUid uid, bool isDeployed, DeployableBarrierComponent? component)
+    {
+        if (!Resolve(uid, ref component))
+            return;
+
+        var transform = Transform(uid);
+        var fixture = _fixtures.GetFixtureOrNull(uid, component.FixtureId);
+
+        if (isDeployed && transform.GridUid != null)
+        {
+            _transform.AnchorEntity(uid, transform);
+            if (fixture != null)
+                _physics.SetHard(uid, fixture, true);
+        }
+        else
+        {
+            _transform.Unanchor(uid, transform);
+            if (fixture != null)
+                _physics.SetHard(uid, fixture, false);
+        }
+
+        var state = isDeployed ? DeployableBarrierState.Deployed : DeployableBarrierState.Idle;
+        _appearance.SetData(uid, DeployableBarrierVisuals.State, state);
+
+        if (TryComp(uid, out SharedPullableComponent? pullable))
+            _pulling.TryStopPull(pullable);
+
+        if (TryComp(uid, out SharedPointLightComponent? light))
+            _pointLight.SetEnabled(uid, isDeployed, light);
+    }
+}