Predict tile-prying (#21167)

Content.Shared/Tools/Systems/SharedToolSystem.TilePrying.cs

@@ -0,0 +1,82 @@
+using Content.Shared.Database;
+using Content.Shared.Fluids.Components;
+using Content.Shared.Interaction;
+using Content.Shared.Maps;
+using Content.Shared.Tools.Components;
+using Robust.Shared.Map;
+using Robust.Shared.Map.Components;
+
+namespace Content.Shared.Tools.Systems;
+
+public abstract partial class SharedToolSystem
+{
+    private void InitializeTilePrying()
+    {
+        SubscribeLocalEvent<TilePryingComponent, AfterInteractEvent>(OnTilePryingAfterInteract);
+        SubscribeLocalEvent<TilePryingComponent, TilePryingDoAfterEvent>(OnTilePryComplete);
+    }
+
+    private void OnTilePryingAfterInteract(EntityUid uid, TilePryingComponent component, AfterInteractEvent args)
+    {
+        if (args.Handled || !args.CanReach || args.Target != null && !HasComp<PuddleComponent>(args.Target))
+            return;
+
+        if (TryPryTile(uid, args.User, component, args.ClickLocation))
+            args.Handled = true;
+    }
+
+    private void OnTilePryComplete(EntityUid uid, TilePryingComponent component, TilePryingDoAfterEvent args)
+    {
+        if (args.Cancelled)
+            return;
+
+        var coords = GetCoordinates(args.Coordinates);
+        var gridUid = coords.GetGridUid(EntityManager);
+        if (!TryComp(gridUid, out MapGridComponent? grid))
+        {
+            Log.Error("Attempted to pry from a non-existent grid?");
+            return;
+        }
+
+        var tile = _maps.GetTileRef(gridUid.Value, grid, coords);
+        var center = _turfs.GetTileCenter(tile);
+
+        if (args.Used != null)
+        {
+            _adminLogger.Add(LogType.Tile, LogImpact.Low,
+                $"{ToPrettyString(args.User):actor} used {ToPrettyString(args.Used.Value):tool} to pry {_tileDefManager[tile.Tile.TypeId].Name} at {center}");
+        }
+        else
+        {
+            _adminLogger.Add(LogType.Tile, LogImpact.Low,
+                $"{ToPrettyString(args.User):actor} pried {_tileDefManager[tile.Tile.TypeId].Name} at {center}");
+        }
+
+        if (_netManager.IsServer)
+            _tiles.PryTile(tile, component.Advanced);
+    }
+
+    private bool TryPryTile(EntityUid toolEntity, EntityUid user, TilePryingComponent component, EntityCoordinates clickLocation)
+    {
+        if (!TryComp<ToolComponent>(toolEntity, out var tool) && component.ToolComponentNeeded)
+            return false;
+
+        if (!_mapManager.TryFindGridAt(clickLocation.ToMap(EntityManager, _transformSystem), out var gridUid, out var mapGrid))
+            return false;
+
+        var tile = _maps.GetTileRef(gridUid, mapGrid, clickLocation);
+        var coordinates = _maps.GridTileToLocal(gridUid, mapGrid, tile.GridIndices);
+
+        if (!InteractionSystem.InRangeUnobstructed(user, coordinates, popup: false))
+            return false;
+
+        var tileDef = (ContentTileDefinition) _tileDefManager[tile.Tile.TypeId];
+
+        if (!tileDef.CanCrowbar && !(tileDef.CanAxe && component.Advanced))
+            return false;
+
+        var ev = new TilePryingDoAfterEvent(GetNetCoordinates(coordinates));
+
+        return UseTool(toolEntity, user, toolEntity, component.Delay, component.QualityNeeded, ev, toolComponent: tool);
+    }
+}