ECS CablePlacer (#6371)

This commit is contained in:
metalgearsloth
2022-01-29 22:45:57 +11:00
committed by GitHub
parent 803d6056d8
commit b5f816c815
3 changed files with 63 additions and 57 deletions

View File

@@ -1,69 +1,20 @@
using System.Threading.Tasks;
using Content.Server.Stack;
using Content.Shared.ActionBlocker;
using Content.Shared.Interaction;
using Content.Shared.Interaction.Helpers;
using Content.Shared.Maps;
using Robust.Shared.GameObjects;
using Robust.Shared.IoC;
using Robust.Shared.Map;
using Robust.Shared.Prototypes;
using Robust.Shared.Serialization.Manager.Attributes;
using Robust.Shared.Serialization.TypeSerializers.Implementations.Custom.Prototype;
using Robust.Shared.ViewVariables;
namespace Content.Server.Power.Components
{
[RegisterComponent]
internal class CablePlacerComponent : Component, IAfterInteract
[RegisterComponent, ComponentProtoName("CablePlacer")]
public sealed class CablePlacerComponent : Component
{
[Dependency] private readonly IEntityManager _entMan = default!;
[Dependency] private readonly IMapManager _mapManager = default!;
/// <inheritdoc />
public override string Name => "CablePlacer";
[ViewVariables]
[DataField("cablePrototypeID")]
private string? _cablePrototypeID = "CableHV";
[DataField("cablePrototypeID", customTypeSerializer:typeof(PrototypeIdSerializer<EntityPrototype>))]
public string? CablePrototypeId = "CableHV";
[ViewVariables]
[DataField("blockingWireType")]
private CableType _blockingCableType = CableType.HighVoltage;
/// <inheritdoc />
async Task<bool> IAfterInteract.AfterInteract(AfterInteractEventArgs eventArgs)
{
if (!EntitySystem.Get<ActionBlockerSystem>().CanInteract(eventArgs.User))
return false;
if (_cablePrototypeID == null)
return false;
if (!eventArgs.InRangeUnobstructed(ignoreInsideBlocker: true, popup: true))
return false;
if(!_mapManager.TryGetGrid(eventArgs.ClickLocation.GetGridId(_entMan), out var grid))
return false;
var snapPos = grid.TileIndicesFor(eventArgs.ClickLocation);
var tileDef = grid.GetTileRef(snapPos).Tile.GetContentTileDefinition();
if(!tileDef.IsSubFloor || !tileDef.Sturdy)
return false;
foreach (var anchored in grid.GetAnchoredEntities(snapPos))
{
if (_entMan.TryGetComponent<CableComponent>(anchored, out var wire) && wire.CableType == _blockingCableType)
{
return false;
}
}
if (_entMan.TryGetComponent<StackComponent?>(Owner, out var stack)
&& !EntitySystem.Get<StackSystem>().Use(Owner, 1, stack))
return false;
_entMan.SpawnEntity(_cablePrototypeID, grid.GridTileToLocal(snapPos));
return true;
}
public CableType BlockingCableType = CableType.HighVoltage;
}
}