crystall-punk-14/Content.Server/Disposal/Tube/Components/DisposalEntryComponent.cs

using System.Linq;
using Content.Server.Atmos.EntitySystems;
using Content.Server.Disposal.Unit.Components;
using Content.Server.Disposal.Unit.EntitySystems;

namespace Content.Server.Disposal.Tube.Components
{
    [RegisterComponent]
    [ComponentReference(typeof(IDisposalTubeComponent))]
    [ComponentReference(typeof(DisposalTubeComponent))]
    public sealed class DisposalEntryComponent : DisposalTubeComponent
    {
        [Dependency] private readonly IEntityManager _entMan = default!;

        private const string HolderPrototypeId = "DisposalHolder";
        public override string ContainerId => "DisposalEntry";

        public bool TryInsert(DisposalUnitComponent from, IEnumerable<string>? tags = default)
        {
            var holder = _entMan.SpawnEntity(HolderPrototypeId, _entMan.GetComponent<TransformComponent>(Owner).MapPosition);
            var holderComponent = _entMan.GetComponent<DisposalHolderComponent>(holder);

            foreach (var entity in from.Container.ContainedEntities.ToArray())
            {
                holderComponent.TryInsert(entity);
            }

            EntitySystem.Get<AtmosphereSystem>().Merge(holderComponent.Air, from.Air);
            from.Air.Clear();

            if (tags != default)
                holderComponent.Tags.UnionWith(tags);

            return EntitySystem.Get<DisposableSystem>().EnterTube((holderComponent).Owner, Owner, holderComponent, null, this);
        }

        protected override Direction[] ConnectableDirections()
        {
            return new[] {_entMan.GetComponent<TransformComponent>(Owner).LocalRotation.GetDir()};
        }

        /// <summary>
        ///     Ejects contents when they come from the same direction the entry is facing.
        /// </summary>
        public override Direction NextDirection(DisposalHolderComponent holder)
        {
            if (holder.PreviousDirectionFrom != Direction.Invalid)
            {
                return Direction.Invalid;
            }

            return ConnectableDirections()[0];
        }
    }
}