tbd-station-14/Content.Server/Atmos/Monitor/Systems/AirAlarmModes.cs

using Content.Server.Atmos.Monitor.Components;
using Content.Server.Atmos.Monitor.Systems;
using Content.Server.DeviceNetwork.Systems;
using Content.Shared.Atmos;
using Content.Shared.Atmos.Monitor.Components;
using Content.Shared.Atmos.Piping.Unary.Components;

namespace Content.Server.Atmos.Monitor;

/// <summary>
///     This is an interface that air alarm modes use
///     in order to execute the defined modes.
/// </summary>
public interface IAirAlarmMode
{
    // This is executed the moment the mode
    // is set. This is to ensure that 'dumb'
    // modes such as Filter/Panic are immediately
    // set.
    /// <summary>
    ///     Executed the mode is set on an air alarm.
    ///     This is to ensure that modes like Filter/Panic
    ///     are immediately set.
    /// </summary>
    public void Execute(EntityUid uid);
}

// IAirAlarmModeUpdate
//
// This is an interface that AirAlarmSystem uses
// in order to 'update' air alarm modes so that
// modes like Replace can be implemented.
/// <summary>
///     An interface that AirAlarmSystem uses
///     in order to update air alarm modes that
///     need updating (e.g., Replace)
/// </summary>
public interface IAirAlarmModeUpdate
{
    /// <summary>
    ///     This is checked by AirAlarmSystem when
    ///     a mode is updated. This should be set
    ///     to a DeviceNetwork address, or some
    ///     unique identifier that ID's the
    ///     owner of the mode's executor.
    /// </summary>
    public string NetOwner { get; set; }
    /// <summary>
    ///     This is executed every time the air alarm
    ///     update loop is fully executed. This should
    ///     be where all the logic goes.
    /// </summary>
    public void Update(EntityUid uid);
}

public sealed class AirAlarmModeFactory
{
    private static IAirAlarmMode _filterMode = new AirAlarmFilterMode();
    private static IAirAlarmMode _wideFilterMode = new AirAlarmWideFilterMode();
    private static IAirAlarmMode _fillMode = new AirAlarmFillMode();
    private static IAirAlarmMode _panicMode = new AirAlarmPanicMode();
    private static IAirAlarmMode _noneMode = new AirAlarmNoneMode();

    // still not a fan since ReplaceMode must have an allocation
    // but it's whatever
    public static IAirAlarmMode? ModeToExecutor(AirAlarmMode mode)
    {
        return mode switch
        {
            AirAlarmMode.Filtering => _filterMode,
            AirAlarmMode.WideFiltering => _wideFilterMode,
            AirAlarmMode.Fill => _fillMode,
            AirAlarmMode.Panic => _panicMode,
            AirAlarmMode.None => _noneMode,
            _ => null
        };
    }
}

// like a tiny little EntitySystem
public abstract class AirAlarmModeExecutor : IAirAlarmMode
{
    [Dependency] public readonly IEntityManager EntityManager = default!;
    public readonly DeviceNetworkSystem DeviceNetworkSystem;
    public readonly AirAlarmSystem AirAlarmSystem;

    public abstract void Execute(EntityUid uid);

    public AirAlarmModeExecutor()
    {
        IoCManager.InjectDependencies(this);

        DeviceNetworkSystem = EntityManager.System<DeviceNetworkSystem>();
        AirAlarmSystem = EntityManager.System<AirAlarmSystem>();
    }
}

public sealed class AirAlarmNoneMode : AirAlarmModeExecutor
{
    public override void Execute(EntityUid uid)
    {
        if (!EntityManager.TryGetComponent(uid, out AirAlarmComponent? alarm))
            return;

        foreach (var (addr, device) in alarm.VentData)
        {
            device.Enabled = false;
            AirAlarmSystem.SetData(uid, addr, device);
        }

        foreach (var (addr, device) in alarm.ScrubberData)
        {
            device.Enabled = false;
            AirAlarmSystem.SetData(uid, addr, device);
        }
    }
}

public sealed class AirAlarmFilterMode : AirAlarmModeExecutor
{
    public override void Execute(EntityUid uid)
    {
        if (!EntityManager.TryGetComponent(uid, out AirAlarmComponent? alarm))
            return;

        foreach (var (addr, device) in alarm.VentData)
        {
            AirAlarmSystem.SetData(uid, addr, GasVentPumpData.FilterModePreset);
        }

        foreach (var (addr, device) in alarm.ScrubberData)
        {
            AirAlarmSystem.SetData(uid, addr, GasVentScrubberData.FilterModePreset);
        }
    }
}

public sealed class AirAlarmWideFilterMode : AirAlarmModeExecutor
{
    public override void Execute(EntityUid uid)
    {
        if (!EntityManager.TryGetComponent(uid, out AirAlarmComponent? alarm))
            return;

        foreach (var (addr, device) in alarm.VentData)
        {
            AirAlarmSystem.SetData(uid, addr, GasVentPumpData.FilterModePreset);
        }

        foreach (var (addr, device) in alarm.ScrubberData)
        {
            AirAlarmSystem.SetData(uid, addr, GasVentScrubberData.WideFilterModePreset);
        }
    }
}

public sealed class AirAlarmPanicMode : AirAlarmModeExecutor
{
    public override void Execute(EntityUid uid)
    {
        if (!EntityManager.TryGetComponent(uid, out AirAlarmComponent? alarm))
            return;

        foreach (var (addr, device) in alarm.VentData)
        {
            AirAlarmSystem.SetData(uid, addr, GasVentPumpData.PanicModePreset);
        }

        foreach (var (addr, device) in alarm.ScrubberData)
        {
            AirAlarmSystem.SetData(uid, addr, GasVentScrubberData.PanicModePreset);
        }
    }
}

public sealed class AirAlarmFillMode : AirAlarmModeExecutor
{
    public override void Execute(EntityUid uid)
    {
        if (!EntityManager.TryGetComponent(uid, out AirAlarmComponent? alarm))
            return;

        foreach (var (addr, device) in alarm.VentData)
        {
            AirAlarmSystem.SetData(uid, addr, GasVentPumpData.FillModePreset);
        }

        foreach (var (addr, device) in alarm.ScrubberData)
        {
            AirAlarmSystem.SetData(uid, addr, GasVentScrubberData.FillModePreset);
        }
    }
}