Files
tbd-station-14/Content.Server/DeviceLinking/Systems/DoorSignalControlSystem.cs
Kevin Zheng 4e5adc2b86 Add interlocking airlocks (#14177)
Co-authored-by: metalgearsloth <comedian_vs_clown@hotmail.com>
2023-05-07 16:49:11 +10:00

107 lines
4.0 KiB
C#

using Content.Server.DeviceLinking.Components;
using Content.Server.DeviceNetwork;
using Content.Server.Doors.Systems;
using Content.Server.MachineLinking.Events;
using Content.Server.MachineLinking.System;
using Content.Shared.Doors.Components;
using Content.Shared.Doors;
using JetBrains.Annotations;
using SignalReceivedEvent = Content.Server.DeviceLinking.Events.SignalReceivedEvent;
namespace Content.Server.DeviceLinking.Systems
{
[UsedImplicitly]
public sealed class DoorSignalControlSystem : EntitySystem
{
[Dependency] private readonly AirlockSystem _airlockSystem = default!;
[Dependency] private readonly DoorSystem _doorSystem = default!;
[Dependency] private readonly DeviceLinkSystem _signalSystem = default!;
private const string DoorSignalState = "DoorState";
public override void Initialize()
{
base.Initialize();
SubscribeLocalEvent<DoorSignalControlComponent, ComponentInit>(OnInit);
SubscribeLocalEvent<DoorSignalControlComponent, SignalReceivedEvent>(OnSignalReceived);
SubscribeLocalEvent<DoorSignalControlComponent, DoorStateChangedEvent>(OnStateChanged);
}
private void OnInit(EntityUid uid, DoorSignalControlComponent component, ComponentInit args)
{
_signalSystem.EnsureSinkPorts(uid, component.OpenPort, component.ClosePort, component.TogglePort);
_signalSystem.EnsureSourcePorts(uid, component.OutOpen);
}
private void OnSignalReceived(EntityUid uid, DoorSignalControlComponent component, ref SignalReceivedEvent args)
{
if (!TryComp(uid, out DoorComponent? door))
return;
var state = SignalState.Momentary;
args.Data?.TryGetValue(DoorSignalState, out state);
if (args.Port == component.OpenPort)
{
if (state == SignalState.High || state == SignalState.Momentary)
{
if (door.State != DoorState.Open)
_doorSystem.TryOpen(uid, door);
}
}
else if (args.Port == component.ClosePort)
{
if (state == SignalState.High || state == SignalState.Momentary)
{
if (door.State != DoorState.Closed)
_doorSystem.TryClose(uid, door);
}
}
else if (args.Port == component.TogglePort)
{
if (state == SignalState.High || state == SignalState.Momentary)
{
_doorSystem.TryToggleDoor(uid, door);
}
}
else if (args.Port == component.InBolt)
{
if (state == SignalState.High)
{
if(TryComp<AirlockComponent>(uid, out var airlockComponent))
_airlockSystem.SetBoltsWithAudio(uid, airlockComponent, true);
}
else
{
if(TryComp<AirlockComponent>(uid, out var airlockComponent))
_airlockSystem.SetBoltsWithAudio(uid, airlockComponent, false);
}
}
}
private void OnStateChanged(EntityUid uid, DoorSignalControlComponent door, DoorStateChangedEvent args)
{
var data = new NetworkPayload()
{
{ DoorSignalState, SignalState.Momentary }
};
if (args.State == DoorState.Closed)
{
data[DoorSignalState] = SignalState.Low;
_signalSystem.InvokePort(uid, door.OutOpen, data);
}
else if (args.State == DoorState.Open
|| args.State == DoorState.Opening
|| args.State == DoorState.Closing
|| args.State == DoorState.Emagging)
{
data[DoorSignalState] = SignalState.High;
_signalSystem.InvokePort(uid, door.OutOpen, data);
}
}
}
}