36 lines
1.2 KiB
C#
36 lines
1.2 KiB
C#
using Content.Server.Station.Systems;
|
|
using Content.Shared.AlertLevel;
|
|
|
|
namespace Content.Server.AlertLevel;
|
|
|
|
public sealed class AlertLevelDisplaySystem : EntitySystem
|
|
{
|
|
[Dependency] private readonly StationSystem _stationSystem = default!;
|
|
|
|
public override void Initialize()
|
|
{
|
|
SubscribeLocalEvent<AlertLevelChangedEvent>(OnAlertChanged);
|
|
SubscribeLocalEvent<AlertLevelDisplayComponent, ComponentInit>(OnDisplayInit);
|
|
}
|
|
|
|
private void OnAlertChanged(AlertLevelChangedEvent args)
|
|
{
|
|
foreach (var (_, appearance) in EntityManager.EntityQuery<AlertLevelDisplayComponent, AppearanceComponent>())
|
|
{
|
|
appearance.SetData(AlertLevelDisplay.CurrentLevel, args.AlertLevel);
|
|
}
|
|
}
|
|
|
|
private void OnDisplayInit(EntityUid uid, AlertLevelDisplayComponent component, ComponentInit args)
|
|
{
|
|
if (TryComp(uid, out AppearanceComponent? appearance))
|
|
{
|
|
var stationUid = _stationSystem.GetOwningStation(uid);
|
|
if (stationUid != null && TryComp(stationUid, out AlertLevelComponent? alert))
|
|
{
|
|
appearance.SetData(AlertLevelDisplay.CurrentLevel, alert.CurrentLevel);
|
|
}
|
|
}
|
|
}
|
|
}
|