tbd-station-14/Content.Server/GameObjects/Components/Access/AccessReaderComponent.cs

using System;
using System.Collections.Generic;
using Content.Server.Interfaces;
using Content.Server.Interfaces.GameObjects;
using Content.Shared.GameObjects.Components.Inventory;
using JetBrains.Annotations;
using Robust.Shared.GameObjects;
using Robust.Shared.Interfaces.GameObjects;
using Robust.Shared.Serialization;
using Robust.Shared.ViewVariables;

namespace Content.Server.GameObjects.Components.Access
{
    [RegisterComponent]
    public class AccessReader : Component
    {
        public override string Name => "AccessReader";
        [ViewVariables]
        private List<string> _necessaryTags;
        [ViewVariables]
        private List<string> _sufficientTags;

        /// <summary>
        /// Searches an <see cref="AccessComponent"/> in the entity itself, in its active hand or in its ID slot.
        /// Returns true if an <see cref="AccessComponent"/> is found and its tags list contains
        /// at least one of <see cref="_sufficientTags"/> or all of <see cref="_necessaryTags"/>.
        /// </summary>
        /// <param name="entity">The entity to be searched for access.</param>
        public bool IsAllowed(IEntity entity)
        {
            var tags = FindAccessTags(entity);
            return tags != null && IsAllowed(tags);
        }

        private bool IsAllowed(List<string> accessTags)
        {
            foreach (var sufficient in _sufficientTags)
            {
                if (accessTags.Contains(sufficient))
                {
                    return true;
                }
            }
            foreach (var necessary in _necessaryTags)
            {
                if (!accessTags.Contains(necessary))
                {
                    return false;
                }
            }
            return true;
        }

        [CanBeNull]
        private static List<string> FindAccessTags(IEntity entity)
        {
            if (entity.TryGetComponent(out IAccess accessComponent))
            {
                return accessComponent.GetTags();
            }

            if (entity.TryGetComponent(out IHandsComponent handsComponent))
            {
                var activeHandEntity = handsComponent.GetActiveHand?.Owner;
                if (activeHandEntity != null &&
                    activeHandEntity.TryGetComponent(out IAccess handAccessComponent))
                {
                    return handAccessComponent.GetTags();
                }
            }
            else
            {
                return null;
            }
            if (entity.TryGetComponent(out InventoryComponent inventoryComponent))
            {
                if (inventoryComponent.HasSlot(EquipmentSlotDefines.Slots.IDCARD) &&
                    inventoryComponent.TryGetSlotItem(EquipmentSlotDefines.Slots.IDCARD, out ItemComponent item) &&
                    item.Owner.TryGetComponent(out IAccess idAccessComponent)
                    )
                {
                    return idAccessComponent.GetTags();
                }
            }
            return null;
        }

        public override void ExposeData(ObjectSerializer serializer)
        {
            base.ExposeData(serializer);

            serializer.DataField(ref _necessaryTags, "necessary" ,new List<string>());
            serializer.DataField(ref _sufficientTags, "sufficient", new List<string>());
        }
    }
}