65bffbebf1
* Initial commit * Fixing merge conflict * Updated for deployment * Whitespace fixes * Linter fixes * Test fail fix * Fixed test failure * Add separate command circuitboards * Addressed review comments
44 lines
1.5 KiB
C#
44 lines
1.5 KiB
C#
using Content.Shared.Construction.Components;
|
|
using JetBrains.Annotations;
|
|
using Robust.Shared.Containers;
|
|
|
|
namespace Content.Shared.Construction.NodeEntities;
|
|
|
|
/// <summary>
|
|
/// Works for both <see cref="ComputerBoardComponent"/> and <see cref="MachineBoardComponent"/>
|
|
/// because duplicating code just for this is really stinky.
|
|
/// </summary>
|
|
[UsedImplicitly]
|
|
[DataDefinition]
|
|
public sealed partial class BoardNodeEntity : IGraphNodeEntity
|
|
{
|
|
[DataField]
|
|
public string Container { get; private set; } = string.Empty;
|
|
|
|
public string? GetId(EntityUid? uid, EntityUid? userUid, GraphNodeEntityArgs args)
|
|
{
|
|
if (uid == null)
|
|
return null;
|
|
|
|
var containerSystem = args.EntityManager.EntitySysManager.GetEntitySystem<SharedContainerSystem>();
|
|
|
|
if (!containerSystem.TryGetContainer(uid.Value, Container, out var container)
|
|
|| container.ContainedEntities.Count == 0)
|
|
return null;
|
|
|
|
var board = container.ContainedEntities[0];
|
|
|
|
// There should not be a case where more than one of these components exist on the same entity
|
|
if (args.EntityManager.TryGetComponent(board, out MachineBoardComponent? machine))
|
|
return machine.Prototype;
|
|
|
|
if (args.EntityManager.TryGetComponent(board, out ComputerBoardComponent? computer))
|
|
return computer.Prototype;
|
|
|
|
if (args.EntityManager.TryGetComponent(board, out ElectronicsBoardComponent? electronics))
|
|
return electronics.Prototype;
|
|
|
|
return null;
|
|
}
|
|
}
|