CatStation/Content.Shared/Disposal/SignalRouter/DisposalSignalRouterSystem.cs

using Content.Shared.DeviceLinking;
using Content.Shared.DeviceLinking.Events;
using Content.Shared.Disposal.Tube;
using Content.Shared.Disposal.Unit;
using System.Linq;

namespace Content.Shared.Disposal.SignalRouter;

/// <summary>
/// Handles signal-based routing for entities in the disposal system.
/// </summary>
public sealed partial class DisposalSignalRouterSystem : EntitySystem
{
    [Dependency] private SharedDeviceLinkSystem _deviceLink = default!;
    [Dependency] private DisposalTubeSystem _disposalTube = default!;

    public override void Initialize()
    {
        base.Initialize();

        SubscribeLocalEvent<DisposalSignalRouterComponent, ComponentInit>(OnInit);
        SubscribeLocalEvent<DisposalSignalRouterComponent, SignalReceivedEvent>(OnSignalReceived);
        SubscribeLocalEvent<DisposalSignalRouterComponent, GetDisposalsNextDirectionEvent>(OnGetNextDirection, before: new[] { typeof(DisposalTubeSystem) });
    }

    private void OnInit(Entity<DisposalSignalRouterComponent> ent, ref ComponentInit args)
    {
        _deviceLink.EnsureSinkPorts(ent, ent.Comp.OnPort, ent.Comp.OffPort, ent.Comp.TogglePort);
    }

    private void OnSignalReceived(Entity<DisposalSignalRouterComponent> ent, ref SignalReceivedEvent args)
    {
        // TogglePort flips it
        // OnPort sets it to true
        // OffPort sets it to false
        ent.Comp.Routing = args.Port == ent.Comp.TogglePort
            ? !ent.Comp.Routing
            : args.Port == ent.Comp.OnPort;

        Dirty(ent);
    }

    private void OnGetNextDirection(Entity<DisposalSignalRouterComponent> ent, ref GetDisposalsNextDirectionEvent args)
    {
        if (args.Handled)
            return;

        if (!TryComp<DisposalTubeComponent>(ent, out var disposalTube))
            return;

        var exits = _disposalTube.GetTubeConnectableDirections((ent, disposalTube));

        if (exits.Length < 3 || !ent.Comp.Routing)
        {
            _disposalTube.SelectNextDirection((ent, disposalTube), exits, ref args);
            return;
        }

        _disposalTube.SelectNextDirection((ent, disposalTube), exits.Skip(1).ToArray(), ref args);
        args.Handled = true;
    }
}