111 lines
3.4 KiB
C#
111 lines
3.4 KiB
C#
using System.Linq;
|
|
using Content.Shared.Alert;
|
|
using JetBrains.Annotations;
|
|
using Robust.Client.GameObjects;
|
|
using Robust.Client.Player;
|
|
using Robust.Shared.GameStates;
|
|
using Robust.Shared.Prototypes;
|
|
|
|
namespace Content.Client.Alerts;
|
|
|
|
[UsedImplicitly]
|
|
internal 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, PlayerAttachedEvent>(OnPlayerAttached);
|
|
SubscribeLocalEvent<AlertsComponent, PlayerDetachedEvent>(OnPlayerDetached);
|
|
|
|
SubscribeLocalEvent<AlertsComponent, ComponentHandleState>(ClientAlertsHandleState);
|
|
}
|
|
protected override void LoadPrototypes()
|
|
{
|
|
base.LoadPrototypes();
|
|
|
|
AlertOrder = _prototypeManager.EnumeratePrototypes<AlertOrderPrototype>().FirstOrDefault();
|
|
if (AlertOrder == null)
|
|
Logger.ErrorS("alert", "no alertOrder prototype found, alerts will be in random order");
|
|
}
|
|
|
|
public IReadOnlyDictionary<AlertKey, AlertState>? ActiveAlerts
|
|
{
|
|
get
|
|
{
|
|
var ent = _playerManager.LocalPlayer?.ControlledEntity;
|
|
return ent is not null
|
|
? GetActiveAlerts(ent.Value)
|
|
: null;
|
|
}
|
|
}
|
|
|
|
protected override void AfterShowAlert(AlertsComponent alertsComponent)
|
|
{
|
|
if (_playerManager.LocalPlayer?.ControlledEntity != alertsComponent.Owner)
|
|
return;
|
|
|
|
SyncAlerts?.Invoke(this, alertsComponent.Alerts);
|
|
}
|
|
|
|
protected override void AfterClearAlert(AlertsComponent alertsComponent)
|
|
{
|
|
if (_playerManager.LocalPlayer?.ControlledEntity != alertsComponent.Owner)
|
|
return;
|
|
|
|
SyncAlerts?.Invoke(this, alertsComponent.Alerts);
|
|
}
|
|
|
|
private void ClientAlertsHandleState(EntityUid uid, AlertsComponent component, ref ComponentHandleState args)
|
|
{
|
|
if (_playerManager.LocalPlayer?.ControlledEntity != uid)
|
|
return;
|
|
|
|
var componentAlerts = (args.Current as AlertsComponentState)?.Alerts;
|
|
if (componentAlerts == null) return;
|
|
|
|
component.Alerts = new(componentAlerts);
|
|
|
|
SyncAlerts?.Invoke(this, componentAlerts);
|
|
}
|
|
|
|
private void OnPlayerAttached(EntityUid uid, AlertsComponent component, PlayerAttachedEvent args)
|
|
{
|
|
if (_playerManager.LocalPlayer?.ControlledEntity != uid)
|
|
return;
|
|
|
|
SyncAlerts?.Invoke(this, component.Alerts);
|
|
}
|
|
|
|
protected override void HandleComponentShutdown(EntityUid uid, AlertsComponent component, ComponentShutdown args)
|
|
{
|
|
base.HandleComponentShutdown(uid, component, args);
|
|
|
|
if (_playerManager.LocalPlayer?.ControlledEntity != uid)
|
|
return;
|
|
|
|
ClearAlerts?.Invoke(this, EventArgs.Empty);
|
|
}
|
|
|
|
private void OnPlayerDetached(EntityUid uid, AlertsComponent component, PlayerDetachedEvent args)
|
|
{
|
|
if (_playerManager.LocalPlayer?.ControlledEntity != uid)
|
|
return;
|
|
|
|
ClearAlerts?.Invoke(this, EventArgs.Empty);
|
|
}
|
|
|
|
public void AlertClicked(AlertType alertType)
|
|
{
|
|
RaiseNetworkEvent(new ClickAlertEvent(alertType));
|
|
}
|
|
}
|