Update radio prefix parsing (#13777)
This commit is contained in:
Leon Friedrich
@@ -1,74 +1,71 @@
|
||||
using Content.Server.Chat.Systems;
|
||||
using Content.Server.Popups;
|
||||
using Content.Server.Tools;
|
||||
using Content.Shared.Tools.Components;
|
||||
using Content.Shared.Examine;
|
||||
using Content.Shared.Interaction;
|
||||
using Content.Server.Radio.Components;
|
||||
using Content.Shared.Inventory.Events;
|
||||
using Content.Shared.Radio;
|
||||
using Content.Server.Radio.Components;
|
||||
using Content.Shared.Radio.Components;
|
||||
using Content.Shared.Radio.EntitySystems;
|
||||
using Robust.Server.GameObjects;
|
||||
using Robust.Shared.Containers;
|
||||
using Robust.Shared.Network;
|
||||
using Robust.Shared.Prototypes;
|
||||
using System.Linq;
|
||||
|
||||
namespace Content.Server.Radio.EntitySystems;
|
||||
|
||||
public sealed class HeadsetSystem : EntitySystem
|
||||
public sealed class HeadsetSystem : SharedHeadsetSystem
|
||||
{
|
||||
[Dependency] private readonly IPrototypeManager _protoManager = default!;
|
||||
[Dependency] private readonly INetManager _netMan = default!;
|
||||
[Dependency] private readonly RadioSystem _radio = default!;
|
||||
[Dependency] private readonly ToolSystem _toolSystem = default!;
|
||||
[Dependency] private readonly PopupSystem _popupSystem = default!;
|
||||
[Dependency] private readonly SharedContainerSystem _container = default!;
|
||||
[Dependency] private readonly SharedAudioSystem _audio = default!;
|
||||
|
||||
public override void Initialize()
|
||||
{
|
||||
base.Initialize();
|
||||
SubscribeLocalEvent<HeadsetComponent, ExaminedEvent>(OnExamined);
|
||||
SubscribeLocalEvent<HeadsetComponent, RadioReceiveEvent>(OnHeadsetReceive);
|
||||
SubscribeLocalEvent<HeadsetComponent, GotEquippedEvent>(OnGotEquipped);
|
||||
SubscribeLocalEvent<HeadsetComponent, GotUnequippedEvent>(OnGotUnequipped);
|
||||
SubscribeLocalEvent<WearingHeadsetComponent, EntitySpokeEvent>(OnSpeak);
|
||||
SubscribeLocalEvent<HeadsetComponent, EncryptionChannelsChangedEvent>(OnKeysChanged);
|
||||
|
||||
SubscribeLocalEvent<HeadsetComponent, ComponentStartup>(OnStartup);
|
||||
SubscribeLocalEvent<HeadsetComponent, InteractUsingEvent>(OnInteractUsing);
|
||||
SubscribeLocalEvent<HeadsetComponent, EntInsertedIntoContainerMessage>(OnContainerInserted);
|
||||
SubscribeLocalEvent<WearingHeadsetComponent, EntitySpokeEvent>(OnSpeak);
|
||||
}
|
||||
|
||||
private void OnKeysChanged(EntityUid uid, HeadsetComponent component, EncryptionChannelsChangedEvent args)
|
||||
{
|
||||
UpdateRadioChannels(uid, component, args.Component);
|
||||
}
|
||||
|
||||
private void UpdateRadioChannels(EntityUid uid, HeadsetComponent headset, EncryptionKeyHolderComponent? keyHolder = null)
|
||||
{
|
||||
if (!headset.Enabled)
|
||||
return;
|
||||
|
||||
if (!Resolve(uid, ref keyHolder))
|
||||
return;
|
||||
|
||||
if (keyHolder.Channels.Count == 0)
|
||||
RemComp<ActiveRadioComponent>(uid);
|
||||
else
|
||||
EnsureComp<ActiveRadioComponent>(uid).Channels = new(keyHolder.Channels);
|
||||
}
|
||||
|
||||
private void OnSpeak(EntityUid uid, WearingHeadsetComponent component, EntitySpokeEvent args)
|
||||
{
|
||||
if (args.Channel != null
|
||||
&& TryComp(component.Headset, out HeadsetComponent? headset)
|
||||
&& headset.Channels.Contains(args.Channel.ID))
|
||||
&& TryComp(component.Headset, out EncryptionKeyHolderComponent? keys)
|
||||
&& keys.Channels.Contains(args.Channel.ID))
|
||||
{
|
||||
_radio.SendRadioMessage(uid, args.Message, args.Channel, component.Headset);
|
||||
args.Channel = null; // prevent duplicate messages from other listeners.
|
||||
}
|
||||
}
|
||||
|
||||
private void OnGotEquipped(EntityUid uid, HeadsetComponent component, GotEquippedEvent args)
|
||||
protected override void OnGotEquipped(EntityUid uid, HeadsetComponent component, GotEquippedEvent args)
|
||||
{
|
||||
component.IsEquipped = args.SlotFlags.HasFlag(component.RequiredSlot);
|
||||
|
||||
base.OnGotEquipped(uid, component, args);
|
||||
if (component.IsEquipped && component.Enabled)
|
||||
{
|
||||
EnsureComp<WearingHeadsetComponent>(args.Equipee).Headset = uid;
|
||||
UpdateRadioChannelsInActiveRadio(uid, component, EnsureComp<ActiveRadioComponent>(uid));
|
||||
UpdateRadioChannels(uid, component);
|
||||
}
|
||||
}
|
||||
|
||||
private void UpdateRadioChannelsInActiveRadio(EntityUid uid, HeadsetComponent component, ActiveRadioComponent activeRadio)
|
||||
{
|
||||
activeRadio.Channels.Clear();
|
||||
activeRadio.Channels.UnionWith(component.Channels);
|
||||
}
|
||||
|
||||
private void OnGotUnequipped(EntityUid uid, HeadsetComponent component, GotUnequippedEvent args)
|
||||
protected override void OnGotUnequipped(EntityUid uid, HeadsetComponent component, GotUnequippedEvent args)
|
||||
{
|
||||
base.OnGotUnequipped(uid, component, args);
|
||||
component.IsEquipped = false;
|
||||
RemComp<ActiveRadioComponent>(uid);
|
||||
RemComp<WearingHeadsetComponent>(args.Equipee);
|
||||
@@ -92,7 +89,7 @@ public sealed class HeadsetSystem : EntitySystem
|
||||
else if (component.IsEquipped)
|
||||
{
|
||||
EnsureComp<WearingHeadsetComponent>(Transform(uid).ParentUid).Headset = uid;
|
||||
EnsureComp<ActiveRadioComponent>(uid).Channels.UnionWith(component.Channels);
|
||||
UpdateRadioChannels(uid, component);
|
||||
}
|
||||
}
|
||||
|
||||
@@ -101,127 +98,4 @@ public sealed class HeadsetSystem : EntitySystem
|
||||
if (TryComp(Transform(uid).ParentUid, out ActorComponent? actor))
|
||||
_netMan.ServerSendMessage(args.ChatMsg, actor.PlayerSession.ConnectedClient);
|
||||
}
|
||||
|
||||
private void OnExamined(EntityUid uid, HeadsetComponent component, ExaminedEvent args)
|
||||
{
|
||||
if (!args.IsInDetailsRange)
|
||||
return;
|
||||
if (component.KeyContainer.ContainedEntities.Count == 0)
|
||||
{
|
||||
args.PushMarkup(Loc.GetString("examine-headset-no-keys"));
|
||||
return;
|
||||
}
|
||||
else if (component.Channels.Count > 0)
|
||||
{
|
||||
args.PushMarkup(Loc.GetString("examine-headset-channels-prefix"));
|
||||
EncryptionKeySystem.GetChannelsExamine(component.Channels, args, _protoManager, "examine-headset-channel");
|
||||
args.PushMarkup(Loc.GetString("examine-headset-chat-prefix", ("prefix", ":h")));
|
||||
if (component.DefaultChannel != null)
|
||||
{
|
||||
var proto = _protoManager.Index<RadioChannelPrototype>(component.DefaultChannel);
|
||||
args.PushMarkup(Loc.GetString("examine-headset-default-channel", ("channel", component.DefaultChannel), ("color", proto.Color)));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private void OnStartup(EntityUid uid, HeadsetComponent component, ComponentStartup args)
|
||||
{
|
||||
component.KeyContainer = _container.EnsureContainer<Container>(uid, HeadsetComponent.KeyContainerName);
|
||||
}
|
||||
|
||||
private bool InstallKey(HeadsetComponent component, EntityUid key, EncryptionKeyComponent keyComponent)
|
||||
{
|
||||
return component.KeyContainer.Insert(key);
|
||||
}
|
||||
|
||||
private void UploadChannelsFromKey(HeadsetComponent component, EncryptionKeyComponent key)
|
||||
{
|
||||
foreach (var j in key.Channels)
|
||||
component.Channels.Add(j);
|
||||
}
|
||||
|
||||
public void RecalculateChannels(HeadsetComponent component)
|
||||
{
|
||||
component.Channels.Clear();
|
||||
foreach (EntityUid i in component.KeyContainer.ContainedEntities)
|
||||
{
|
||||
if (TryComp<EncryptionKeyComponent>(i, out var key))
|
||||
{
|
||||
UploadChannelsFromKey(component, key);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private void OnInteractUsing(EntityUid uid, HeadsetComponent component, InteractUsingEvent args)
|
||||
{
|
||||
if (!TryComp<ContainerManagerComponent>(uid, out var storage))
|
||||
return;
|
||||
if(!component.IsKeysUnlocked)
|
||||
{
|
||||
_popupSystem.PopupEntity(Loc.GetString("headset-encryption-keys-are-locked"), uid, args.User);
|
||||
return;
|
||||
}
|
||||
if (TryComp<EncryptionKeyComponent>(args.Used, out var key))
|
||||
{
|
||||
if (component.KeySlots > component.KeyContainer.ContainedEntities.Count)
|
||||
{
|
||||
if (InstallKey(component, args.Used, key))
|
||||
{
|
||||
_popupSystem.PopupEntity(Loc.GetString("headset-encryption-key-successfully-installed"), uid, args.User);
|
||||
_audio.PlayPvs(component.KeyInsertionSound, args.Target);
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
_popupSystem.PopupEntity(Loc.GetString("headset-encryption-key-slots-already-full"), uid, args.User);
|
||||
}
|
||||
}
|
||||
if (TryComp<ToolComponent>(args.Used, out var tool))
|
||||
{
|
||||
if (component.KeyContainer.ContainedEntities.Count > 0)
|
||||
{
|
||||
if (_toolSystem.UseTool(
|
||||
args.Used, args.User, uid,
|
||||
0f, 0f, new String[] { component.KeysExtractionMethod },
|
||||
doAfterCompleteEvent: null, toolComponent: tool)
|
||||
)
|
||||
{
|
||||
var contained = component.KeyContainer.ContainedEntities.ToArray<EntityUid>();
|
||||
foreach (var i in contained)
|
||||
component.KeyContainer.Remove(i);
|
||||
component.Channels.Clear();
|
||||
UpdateDefaultChannel(component);
|
||||
_popupSystem.PopupEntity(Loc.GetString("headset-encryption-keys-all-extracted"), uid, args.User);
|
||||
_audio.PlayPvs(component.KeyExtractionSound, args.Target);
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
_popupSystem.PopupEntity(Loc.GetString("headset-encryption-keys-no-keys"), uid, args.User);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private void UpdateDefaultChannel(HeadsetComponent component)
|
||||
{
|
||||
if (component.KeyContainer.ContainedEntities.Count >= 1)
|
||||
component.DefaultChannel = EnsureComp<EncryptionKeyComponent>(component.KeyContainer.ContainedEntities[0])?.DefaultChannel;
|
||||
else
|
||||
component.DefaultChannel = null;
|
||||
}
|
||||
|
||||
private void OnContainerInserted(EntityUid uid, HeadsetComponent component, EntInsertedIntoContainerMessage args)
|
||||
{
|
||||
if (args.Container.ID != HeadsetComponent.KeyContainerName)
|
||||
{
|
||||
return;
|
||||
}
|
||||
if (TryComp<EncryptionKeyComponent>(args.Entity, out var added))
|
||||
{
|
||||
UpdateDefaultChannel(component);
|
||||
UploadChannelsFromKey(component, added);
|
||||
UpdateRadioChannelsInActiveRadio(uid, component, EnsureComp<ActiveRadioComponent>(uid));
|
||||
}
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user