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

using Content.Server.Disposal.Unit.Components;

namespace Content.Server.Disposal.Tube.Components
{
    // TODO: Different types of tubes eject in random direction with no exit point
    [RegisterComponent]
    [ComponentReference(typeof(IDisposalTubeComponent))]
    [Virtual]
    [ComponentReference(typeof(DisposalTubeComponent))]
    public class DisposalTransitComponent : DisposalTubeComponent
    {
        public override string ContainerId => "DisposalTransit";

        protected override Direction[] ConnectableDirections()
        {
            var rotation = IoCManager.Resolve<IEntityManager>().GetComponent<TransformComponent>(Owner).LocalRotation;
            var opposite = new Angle(rotation.Theta + Math.PI);

            return new[] {rotation.GetDir(), opposite.GetDir()};
        }

        public override Direction NextDirection(DisposalHolderComponent holder)
        {
            var directions = ConnectableDirections();
            var previousDF = holder.PreviousDirectionFrom;
            var forward = directions[0];

            if (previousDF == Direction.Invalid)
            {
                return forward;
            }

            var backward = directions[1];
            return previousDF == forward ? backward : forward;
        }
    }
}