RobustToolbox/Robust.Shared/Containers/Container.cs

using System.Collections.Generic;
using JetBrains.Annotations;
using Robust.Shared.GameObjects;
using Robust.Shared.IoC;
using Robust.Shared.Serialization;
using Robust.Shared.Serialization.Manager.Attributes;

namespace Robust.Shared.Containers
{
    /// <summary>
    /// Default implementation for containers,
    /// cannot be inherited. If additional logic is needed,
    /// this logic should go on the systems that are holding this container.
    /// For example, inventory containers should be modified only through an inventory component.
    /// </summary>
    [UsedImplicitly]
    [SerializedType(ClassName)]
    public sealed class Container : BaseContainer
    {
        private const string ClassName = "Container";

        /// <summary>
        /// The generic container class uses a list of entities
        /// </summary>
        [DataField("ents")]
        private readonly List<EntityUid> _containerList = new();

        private readonly List<EntityUid> _expectedEntities = new();

        /// <inheritdoc />
        public override IReadOnlyList<EntityUid> ContainedEntities => _containerList;

        public override List<EntityUid> ExpectedEntities => _expectedEntities;

        /// <inheritdoc />
        public override string ContainerType => ClassName;

        /// <inheritdoc />
        protected override void InternalInsert(EntityUid toinsert, IEntityManager entMan, MetaDataComponent? meta = null)
        {
            _containerList.Add(toinsert);
            base.InternalInsert(toinsert, entMan, meta);
        }

        /// <inheritdoc />
        protected override void InternalRemove(EntityUid toremove, IEntityManager entMan, MetaDataComponent? meta = null)
        {
            _containerList.Remove(toremove);
            base.InternalRemove(toremove, entMan, meta);
        }

        /// <inheritdoc />
        public override bool Contains(EntityUid contained)
        {
            return _containerList.Contains(contained);
        }

        /// <inheritdoc />
        public override void Shutdown()
        {
            base.Shutdown();

            var entMan = IoCManager.Resolve<IEntityManager>();
            foreach (var entity in _containerList)
            {
                entMan.DeleteEntity(entity);
            }
        }
    }
}