72 lines
2.5 KiB
C#
72 lines
2.5 KiB
C#
using System.Diagnostics.CodeAnalysis;
|
|
using Content.Server.Atmos.Piping.Binary.Components;
|
|
using Content.Server.Atmos.Piping.Unary.Components;
|
|
using Content.Server.NodeContainer;
|
|
using Content.Server.NodeContainer.Nodes;
|
|
using Content.Shared.Atmos.Piping.Unary.Components;
|
|
using Content.Shared.Construction.Components;
|
|
using JetBrains.Annotations;
|
|
using Robust.Shared.Map;
|
|
|
|
namespace Content.Server.Atmos.Piping.Unary.EntitySystems
|
|
{
|
|
[UsedImplicitly]
|
|
public sealed class GasPortableSystem : EntitySystem
|
|
{
|
|
[Dependency] private readonly IMapManager _mapManager = default!;
|
|
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
|
|
SubscribeLocalEvent<GasPortableComponent, AnchorAttemptEvent>(OnPortableAnchorAttempt);
|
|
// Shouldn't need re-anchored event.
|
|
SubscribeLocalEvent<GasPortableComponent, AnchorStateChangedEvent>(OnAnchorChanged);
|
|
}
|
|
|
|
private void OnPortableAnchorAttempt(EntityUid uid, GasPortableComponent component, AnchorAttemptEvent args)
|
|
{
|
|
if (!EntityManager.TryGetComponent(uid, out TransformComponent? transform))
|
|
return;
|
|
|
|
// If we can't find any ports, cancel the anchoring.
|
|
if(!FindGasPortIn(transform.GridUid, transform.Coordinates, out _))
|
|
args.Cancel();
|
|
}
|
|
|
|
private void OnAnchorChanged(EntityUid uid, GasPortableComponent portable, ref AnchorStateChangedEvent args)
|
|
{
|
|
if (!EntityManager.TryGetComponent(uid, out NodeContainerComponent? nodeContainer))
|
|
return;
|
|
|
|
if (!nodeContainer.TryGetNode(portable.PortName, out PipeNode? portableNode))
|
|
return;
|
|
|
|
portableNode.ConnectionsEnabled = args.Anchored;
|
|
|
|
if (EntityManager.TryGetComponent(uid, out AppearanceComponent? appearance))
|
|
{
|
|
appearance.SetData(GasPortableVisuals.ConnectedState, args.Anchored);
|
|
}
|
|
}
|
|
|
|
public bool FindGasPortIn(EntityUid? gridId, EntityCoordinates coordinates, [NotNullWhen(true)] out GasPortComponent? port)
|
|
{
|
|
port = null;
|
|
|
|
if (!_mapManager.TryGetGrid(gridId, out var grid))
|
|
return false;
|
|
|
|
foreach (var entityUid in grid.GetLocal(coordinates))
|
|
{
|
|
if (EntityManager.TryGetComponent(entityUid, out port))
|
|
{
|
|
return true;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
}
|
|
}
|