108 lines
3.5 KiB
C#
108 lines
3.5 KiB
C#
using System.Linq;
|
|
using Content.Server.DeviceNetwork.Components;
|
|
using Content.Shared.Interaction;
|
|
using JetBrains.Annotations;
|
|
|
|
namespace Content.Server.DeviceNetwork.Systems;
|
|
|
|
[UsedImplicitly]
|
|
public sealed class DeviceListSystem : EntitySystem
|
|
{
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
SubscribeLocalEvent<DeviceListComponent, BeforeBroadcastAttemptEvent>(OnBeforeBroadcast);
|
|
SubscribeLocalEvent<DeviceListComponent, BeforePacketSentEvent>(OnBeforePacketSent);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Replaces or merges the current device list with the given one
|
|
/// </summary>
|
|
public void UpdateDeviceList(EntityUid uid, IEnumerable<EntityUid> devices, bool merge = false, DeviceListComponent? deviceList = null)
|
|
{
|
|
if (!Resolve(uid, ref deviceList))
|
|
return;
|
|
|
|
if (!merge)
|
|
deviceList.Devices.Clear();
|
|
|
|
var devicesList = devices.ToList();
|
|
deviceList.Devices.UnionWith(devicesList);
|
|
|
|
RaiseLocalEvent(uid, new DeviceListUpdateEvent(devicesList));
|
|
}
|
|
|
|
/// <summary>
|
|
/// Gets the given device list as a dictionary
|
|
/// </summary>
|
|
public Dictionary<string, EntityUid> GetDeviceList(EntityUid uid, DeviceListComponent? deviceList = null)
|
|
{
|
|
if (!Resolve(uid, ref deviceList))
|
|
return new Dictionary<string, EntityUid>();
|
|
|
|
var devices = new Dictionary<string, EntityUid>(deviceList.Devices.Count);
|
|
|
|
foreach (var deviceUid in deviceList.Devices)
|
|
{
|
|
if (!TryComp(deviceUid, out DeviceNetworkComponent? deviceNet))
|
|
continue;
|
|
|
|
devices.Add(deviceNet.Address, deviceUid);
|
|
|
|
}
|
|
|
|
return devices;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Toggles the given device lists connection visualisation on and off.
|
|
/// TODO: Implement an overlay that draws a line between the given entity and the entities in the device list
|
|
/// </summary>
|
|
public void ToggleVisualization(EntityUid uid, bool ensureOff = false, DeviceListComponent? deviceList = null)
|
|
{
|
|
if (!Resolve(uid, ref deviceList))
|
|
return;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Filters the broadcasts recipient list against the device list as either an allow or deny list depending on the components IsAllowList field
|
|
/// </summary>
|
|
private void OnBeforeBroadcast(EntityUid uid, DeviceListComponent component, BeforeBroadcastAttemptEvent args)
|
|
{
|
|
//Don't filter anything if the device list is empty
|
|
if (component.Devices.Count == 0)
|
|
{
|
|
if (component.IsAllowList) args.Cancel();
|
|
return;
|
|
}
|
|
|
|
HashSet<DeviceNetworkComponent> filteredRecipients = new(args.Recipients.Count);
|
|
|
|
foreach (var recipient in args.Recipients)
|
|
{
|
|
if (component.Devices.Contains(recipient.Owner) == component.IsAllowList) filteredRecipients.Add(recipient);
|
|
}
|
|
|
|
args.ModifiedRecipients = filteredRecipients;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Filters incoming packets if that is enabled <see cref="OnBeforeBroadcast"/>
|
|
/// </summary>
|
|
private void OnBeforePacketSent(EntityUid uid, DeviceListComponent component, BeforePacketSentEvent args)
|
|
{
|
|
if (component.HandleIncomingPackets && component.Devices.Contains(args.Sender) != component.IsAllowList)
|
|
args.Cancel();
|
|
}
|
|
}
|
|
|
|
public sealed class DeviceListUpdateEvent : EntityEventArgs
|
|
{
|
|
public DeviceListUpdateEvent(List<EntityUid> devices)
|
|
{
|
|
Devices = devices;
|
|
}
|
|
|
|
public List<EntityUid> Devices { get; }
|
|
}
|