Files
tbd-station-14/Content.Server/GameObjects/Components/Power/WirePlacerComponent.cs
2020-01-24 02:52:44 +01:00

63 lines
2.1 KiB
C#

using Content.Server.GameObjects.Components.Stack;
using Content.Server.GameObjects.EntitySystems;
using Robust.Server.GameObjects;
using Robust.Server.Interfaces.GameObjects;
using Robust.Shared.GameObjects;
using Robust.Shared.GameObjects.Components.Transform;
using Robust.Shared.Interfaces.Map;
using Robust.Shared.IoC;
namespace Content.Server.GameObjects.Components.Power
{
[RegisterComponent]
internal class WirePlacerComponent : Component, IAfterAttack
{
#pragma warning disable 649
[Dependency] private readonly IServerEntityManager _entityManager;
[Dependency] private readonly IMapManager _mapManager;
#pragma warning restore 649
/// <inheritdoc />
public override string Name => "WirePlacer";
/// <inheritdoc />
public void AfterAttack(AfterAttackEventArgs eventArgs)
{
if(!_mapManager.TryGetGrid(eventArgs.ClickLocation.GridID, out var grid))
return;
var snapPos = grid.SnapGridCellFor(eventArgs.ClickLocation.ToWorld(_mapManager), SnapGridOffset.Center);
var snapCell = grid.GetSnapGridCell(snapPos, SnapGridOffset.Center);
if(grid.GetTileRef(snapPos).Tile.IsEmpty)
return;
var found = false;
foreach (var snapComp in snapCell)
{
if (!snapComp.Owner.HasComponent<PowerTransferComponent>())
continue;
found = true;
break;
}
if (found)
return;
bool hasItemSpriteComp = Owner.TryGetComponent(out SpriteComponent itemSpriteComp);
if (Owner.TryGetComponent(out StackComponent stack) && !stack.Use(1))
return;
var newWire = _entityManager.SpawnEntityAt("Wire", grid.GridTileToLocal(snapPos));
if (newWire.TryGetComponent(out SpriteComponent wireSpriteComp) && hasItemSpriteComp)
{
wireSpriteComp.Color = itemSpriteComp.Color;
}
//TODO: There is no way to set this wire as above or below the floor
}
}
}