58 lines
1.8 KiB
C#
58 lines
1.8 KiB
C#
using System.Diagnostics.CodeAnalysis;
|
|
using Content.Shared.Access.Components;
|
|
using Content.Shared.Hands.Components;
|
|
using Content.Shared.Inventory;
|
|
using Content.Shared.PDA;
|
|
|
|
namespace Content.Shared.Access.Systems;
|
|
|
|
public abstract class SharedIdCardSystem : EntitySystem
|
|
{
|
|
[Dependency] private readonly InventorySystem _inventorySystem = default!;
|
|
|
|
/// <summary>
|
|
/// Attempt to find an ID card on an entity. This will look in the entity itself, in the entity's hands, and
|
|
/// in the entity's inventory.
|
|
/// </summary>
|
|
public bool TryFindIdCard(EntityUid uid, [NotNullWhen(true)] out IdCardComponent? idCard)
|
|
{
|
|
// check held item?
|
|
if (EntityManager.TryGetComponent(uid, out HandsComponent? hands) &&
|
|
hands.ActiveHandEntity is EntityUid heldItem &&
|
|
TryGetIdCard(heldItem, out idCard))
|
|
{
|
|
return true;
|
|
}
|
|
|
|
// check entity itself
|
|
if (TryGetIdCard(uid, out idCard))
|
|
return true;
|
|
|
|
// check inventory slot?
|
|
if (_inventorySystem.TryGetSlotEntity(uid, "id", out var idUid) && TryGetIdCard(idUid.Value, out idCard))
|
|
{
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Attempt to get an id card component from an entity, either by getting it directly from the entity, or by
|
|
/// getting the contained id from a <see cref="PdaComponent"/>.
|
|
/// </summary>
|
|
public bool TryGetIdCard(EntityUid uid, [NotNullWhen(true)] out IdCardComponent? idCard)
|
|
{
|
|
if (EntityManager.TryGetComponent(uid, out idCard))
|
|
return true;
|
|
|
|
if (EntityManager.TryGetComponent(uid, out PdaComponent? pda) && pda.ContainedId != null)
|
|
{
|
|
idCard = pda.ContainedId;
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
}
|