104 lines
3.0 KiB
C#
104 lines
3.0 KiB
C#
using System.Linq;
|
|
using Content.Shared.Alert;
|
|
using JetBrains.Annotations;
|
|
using Robust.Client.Player;
|
|
using Robust.Shared.GameStates;
|
|
using Robust.Shared.Player;
|
|
using Robust.Shared.Prototypes;
|
|
|
|
namespace Content.Client.Alerts;
|
|
|
|
[UsedImplicitly]
|
|
public sealed class ClientAlertsSystem : AlertsSystem
|
|
{
|
|
public AlertOrderPrototype? AlertOrder { get; set; }
|
|
|
|
[Dependency] private readonly IPlayerManager _playerManager = default!;
|
|
[Dependency] private readonly IPrototypeManager _prototypeManager = default!;
|
|
|
|
public event EventHandler? ClearAlerts;
|
|
public event EventHandler<IReadOnlyDictionary<AlertKey, AlertState>>? SyncAlerts;
|
|
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
|
|
SubscribeLocalEvent<AlertsComponent, LocalPlayerAttachedEvent>(OnPlayerAttached);
|
|
SubscribeLocalEvent<AlertsComponent, LocalPlayerDetachedEvent>(OnPlayerDetached);
|
|
SubscribeLocalEvent<AlertsComponent, ComponentHandleState>(OnHandleState);
|
|
}
|
|
protected override void LoadPrototypes()
|
|
{
|
|
base.LoadPrototypes();
|
|
|
|
AlertOrder = _prototypeManager.EnumeratePrototypes<AlertOrderPrototype>().FirstOrDefault();
|
|
if (AlertOrder == null)
|
|
Log.Error("No alertOrder prototype found, alerts will be in random order");
|
|
}
|
|
|
|
public IReadOnlyDictionary<AlertKey, AlertState>? ActiveAlerts
|
|
{
|
|
get
|
|
{
|
|
var ent = _playerManager.LocalEntity;
|
|
return ent is not null
|
|
? GetActiveAlerts(ent.Value)
|
|
: null;
|
|
}
|
|
}
|
|
|
|
private void OnHandleState(Entity<AlertsComponent> alerts, ref ComponentHandleState args)
|
|
{
|
|
if (args.Current is not AlertComponentState cast)
|
|
return;
|
|
|
|
alerts.Comp.Alerts = new(cast.Alerts);
|
|
|
|
UpdateHud(alerts);
|
|
}
|
|
|
|
protected override void AfterShowAlert(Entity<AlertsComponent> alerts)
|
|
{
|
|
UpdateHud(alerts);
|
|
}
|
|
|
|
protected override void AfterClearAlert(Entity<AlertsComponent> alerts)
|
|
{
|
|
UpdateHud(alerts);
|
|
}
|
|
|
|
private void UpdateHud(Entity<AlertsComponent> entity)
|
|
{
|
|
if (_playerManager.LocalEntity == entity.Owner)
|
|
SyncAlerts?.Invoke(this, entity.Comp.Alerts);
|
|
}
|
|
|
|
private void OnPlayerAttached(EntityUid uid, AlertsComponent component, LocalPlayerAttachedEvent args)
|
|
{
|
|
if (_playerManager.LocalEntity != uid)
|
|
return;
|
|
|
|
SyncAlerts?.Invoke(this, component.Alerts);
|
|
}
|
|
|
|
protected override void HandleComponentShutdown(EntityUid uid, AlertsComponent component, ComponentShutdown args)
|
|
{
|
|
base.HandleComponentShutdown(uid, component, args);
|
|
|
|
if (_playerManager.LocalEntity != uid)
|
|
return;
|
|
|
|
ClearAlerts?.Invoke(this, EventArgs.Empty);
|
|
}
|
|
|
|
private void OnPlayerDetached(EntityUid uid, AlertsComponent component, LocalPlayerDetachedEvent args)
|
|
{
|
|
ClearAlerts?.Invoke(this, EventArgs.Empty);
|
|
}
|
|
|
|
public void AlertClicked(ProtoId<AlertPrototype> alertType)
|
|
{
|
|
RaisePredictiveEvent(new ClickAlertEvent(alertType));
|
|
}
|
|
}
|