39 lines
1.2 KiB
C#
39 lines
1.2 KiB
C#
using Content.Server.Shuttles.Components;
|
|
using Content.Shared.NodeContainer;
|
|
using Robust.Shared.Map.Components;
|
|
|
|
namespace Content.Server.NodeContainer.Nodes;
|
|
|
|
|
|
[DataDefinition, Virtual]
|
|
public partial class DockablePipeNode : PipeNode
|
|
{
|
|
|
|
public override IEnumerable<Node> GetReachableNodes(TransformComponent xform,
|
|
EntityQuery<NodeContainerComponent> nodeQuery,
|
|
EntityQuery<TransformComponent> xformQuery,
|
|
MapGridComponent? grid,
|
|
IEntityManager entMan)
|
|
{
|
|
foreach (var pipe in base.GetReachableNodes(xform, nodeQuery, xformQuery, grid, entMan))
|
|
{
|
|
yield return pipe;
|
|
}
|
|
|
|
if (!xform.Anchored || grid == null)
|
|
yield break;
|
|
|
|
if (entMan.TryGetComponent(Owner, out DockingComponent? docking)
|
|
&& docking.DockedWith != null
|
|
&& nodeQuery.TryComp(docking.DockedWith, out var otherNode))
|
|
{
|
|
// Hack: this doesn't take into account the direction of the dockable port.
|
|
foreach (var node in otherNode.Nodes.Values)
|
|
{
|
|
if (node is DockablePipeNode pipe)
|
|
yield return pipe;
|
|
}
|
|
}
|
|
}
|
|
}
|