tbd-station-14/Content.Server/Atmos/Monitor/WireActions/AtmosMonitorAlarmWire.cs

using Content.Server.Atmos.Monitor.Components;
using Content.Server.Atmos.Monitor.Systems;
using Content.Server.Wires;
using Content.Shared.Atmos.Monitor;
using Content.Shared.Wires;

namespace Content.Server.Atmos.Monitor;

[DataDefinition]
public sealed class AtmosMonitorDeviceNetWire : BaseWireAction
{
    // whether or not this wire will send out an alarm upon
    // being pulsed
    [DataField("alarmOnPulse")]
    private bool _alarmOnPulse = false;

    private string _text = "NETW";
    private Color _color = Color.Orange;

    private AtmosMonitorSystem _atmosMonitorSystem = default!;

    public override object StatusKey { get; } = AtmosMonitorAlarmWireActionKeys.Network;

    public override StatusLightData? GetStatusLightData(Wire wire)
    {
        var lightState = StatusLightState.Off;

        if (IsPowered(wire.Owner) && EntityManager.TryGetComponent<AtmosMonitorComponent>(wire.Owner, out var monitor))
        {
            lightState = monitor.HighestAlarmInNetwork == AtmosMonitorAlarmType.Danger
                ? StatusLightState.BlinkingFast
                : StatusLightState.On;
        }

        return new StatusLightData(
            _color,
            lightState,
            _text);
    }

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

        _atmosMonitorSystem = EntitySystem.Get<AtmosMonitorSystem>();
    }

    public override bool Cut(EntityUid user, Wire wire)
    {
        if (EntityManager.TryGetComponent<AtmosMonitorComponent>(wire.Owner, out var monitor))
        {
            monitor.NetEnabled = false;
        }

        return true;
    }

    public override bool Mend(EntityUid user, Wire wire)
    {
        if (EntityManager.TryGetComponent<AtmosMonitorComponent>(wire.Owner, out var monitor))
        {
            monitor.NetEnabled = true;
        }

        return true;
    }

    public override bool Pulse(EntityUid user, Wire wire)
    {
        if (_alarmOnPulse)
        {
            _atmosMonitorSystem.Alert(wire.Owner, AtmosMonitorAlarmType.Danger);
        }

        return true;
    }
}