47 lines
1.8 KiB
C#
47 lines
1.8 KiB
C#
using Content.Server.Power.Components;
|
|
using Content.Server.Station.Systems;
|
|
using Content.Shared.AlertLevel;
|
|
|
|
namespace Content.Server.AlertLevel;
|
|
|
|
public sealed class AlertLevelDisplaySystem : EntitySystem
|
|
{
|
|
[Dependency] private readonly StationSystem _stationSystem = default!;
|
|
[Dependency] private readonly SharedAppearanceSystem _appearance = default!;
|
|
|
|
public override void Initialize()
|
|
{
|
|
SubscribeLocalEvent<AlertLevelChangedEvent>(OnAlertChanged);
|
|
SubscribeLocalEvent<AlertLevelDisplayComponent, ComponentInit>(OnDisplayInit);
|
|
SubscribeLocalEvent<AlertLevelDisplayComponent, PowerChangedEvent>(OnPowerChanged);
|
|
}
|
|
|
|
private void OnAlertChanged(AlertLevelChangedEvent args)
|
|
{
|
|
var query = EntityQueryEnumerator<AlertLevelDisplayComponent, AppearanceComponent>();
|
|
while (query.MoveNext(out var uid, out _, out var appearance))
|
|
{
|
|
_appearance.SetData(uid, AlertLevelDisplay.CurrentLevel, args.AlertLevel, appearance);
|
|
}
|
|
}
|
|
|
|
private void OnDisplayInit(EntityUid uid, AlertLevelDisplayComponent alertLevelDisplay, ComponentInit args)
|
|
{
|
|
if (TryComp(uid, out AppearanceComponent? appearance))
|
|
{
|
|
var stationUid = _stationSystem.GetOwningStation(uid);
|
|
if (stationUid != null && TryComp(stationUid, out AlertLevelComponent? alert))
|
|
{
|
|
_appearance.SetData(uid, AlertLevelDisplay.CurrentLevel, alert.CurrentLevel, appearance);
|
|
}
|
|
}
|
|
}
|
|
private void OnPowerChanged(EntityUid uid, AlertLevelDisplayComponent alertLevelDisplay, ref PowerChangedEvent args)
|
|
{
|
|
if (!TryComp(uid, out AppearanceComponent? appearance))
|
|
return;
|
|
|
|
_appearance.SetData(uid, AlertLevelDisplay.Powered, args.Powered, appearance);
|
|
}
|
|
}
|