Implement lattice cutting (#10920)

* Add lattice cutting * Add delay when spacing the floor. * Add LatticeCutting to jaws of life * Add warning * Minor fixes * Add atmos decompression delay * Add CanWirecutter and lattice item drops * Remove atmos check * Prevent tile replacement edge case
2022-08-31 01:24:51 -07:00
parent c26074e978
commit 61d975644a
9 changed files with 187 additions and 13 deletions
--- a/Content.Server/Tools/ToolSystem.LatticeCutting.cs
+++ b/Content.Server/Tools/ToolSystem.LatticeCutting.cs
@@ -0,0 +1,101 @@
+using System.Threading;
+using Content.Server.Tools.Components;
+using Content.Shared.Interaction;
+using Content.Shared.Maps;
+using Content.Shared.Tools.Components;
+using Robust.Shared.Map;
+
+namespace Content.Server.Tools;
+
+public sealed partial class ToolSystem
+{
+    private void InitializeLatticeCutting()
+    {
+        SubscribeLocalEvent<LatticeCuttingComponent, AfterInteractEvent>(OnLatticeCuttingAfterInteract);
+        SubscribeLocalEvent<LatticeCuttingComponent, LatticeCuttingCompleteEvent>(OnLatticeCutComplete);
+        SubscribeLocalEvent<LatticeCuttingComponent, LatticeCuttingCancelledEvent>(OnLatticeCutCancelled);
+    }
+
+    private void OnLatticeCutCancelled(EntityUid uid, LatticeCuttingComponent component, LatticeCuttingCancelledEvent args)
+    {
+        component.CancelTokenSource = null;
+    }
+
+    private void OnLatticeCutComplete(EntityUid uid, LatticeCuttingComponent component, LatticeCuttingCompleteEvent args)
+    {
+        component.CancelTokenSource = null;
+        var gridUid = args.Coordinates.GetGridUid(EntityManager);
+        if (gridUid == null)
+            return;
+        var grid = _mapManager.GetGrid(gridUid.Value);
+        var tile = grid.GetTileRef(args.Coordinates);
+
+        if (_tileDefinitionManager[tile.Tile.TypeId] is not ContentTileDefinition tileDef
+            || !tileDef.CanWirecutter
+            || tileDef.BaseTurfs.Count == 0
+            || tile.IsBlockedTurf(true))
+            return;
+
+        tile.CutTile(_mapManager, _tileDefinitionManager, EntityManager);
+    }
+
+    private void OnLatticeCuttingAfterInteract(EntityUid uid, LatticeCuttingComponent component,
+        AfterInteractEvent args)
+    {
+        if (args.Handled || !args.CanReach || args.Target != null)
+            return;
+
+        if (TryCut(args.User, component, args.ClickLocation))
+            args.Handled = true;
+    }
+
+    private bool TryCut(EntityUid user, LatticeCuttingComponent component, EntityCoordinates clickLocation)
+    {
+        if (component.CancelTokenSource != null)
+            return true;
+
+        ToolComponent? tool = null;
+        if (component.ToolComponentNeeded && !TryComp<ToolComponent?>(component.Owner, out tool))
+            return false;
+
+        if (!_mapManager.TryGetGrid(clickLocation.GetGridUid(EntityManager), out var mapGrid))
+            return false;
+
+        var tile = mapGrid.GetTileRef(clickLocation);
+
+        var coordinates = mapGrid.GridTileToLocal(tile.GridIndices);
+
+        if (!_interactionSystem.InRangeUnobstructed(user, coordinates, popup: false))
+            return false;
+
+        if (_tileDefinitionManager[tile.Tile.TypeId] is not ContentTileDefinition tileDef
+            || !tileDef.CanWirecutter
+            || tileDef.BaseTurfs.Count == 0
+            || _tileDefinitionManager[tileDef.BaseTurfs[^1]] is not ContentTileDefinition newDef
+            || tile.IsBlockedTurf(true))
+            return false;
+
+        var tokenSource = new CancellationTokenSource();
+        component.CancelTokenSource = tokenSource;
+
+        if (!UseTool(component.Owner, user, null, 0f, component.Delay, new[] {component.QualityNeeded},
+                new LatticeCuttingCompleteEvent
+                {
+                    Coordinates = clickLocation
+                }, new LatticeCuttingCancelledEvent(), toolComponent: tool, doAfterEventTarget: component.Owner,
+                cancelToken: tokenSource.Token))
+            component.CancelTokenSource = null;
+
+        return true;
+    }
+
+    private sealed class LatticeCuttingCompleteEvent : EntityEventArgs
+    {
+        public EntityCoordinates Coordinates { get; set; }
+    }
+
+    private sealed class LatticeCuttingCancelledEvent : EntityEventArgs
+    {
+    }
+}
+