93 lines
3.6 KiB
C#
93 lines
3.6 KiB
C#
using Content.Server.Labels.Components;
|
|
using Content.Server.UserInterface;
|
|
using Content.Shared.Interaction;
|
|
using Content.Shared.Labels;
|
|
using Content.Shared.Popups;
|
|
using JetBrains.Annotations;
|
|
using Robust.Server.GameObjects;
|
|
|
|
namespace Content.Server.Labels
|
|
{
|
|
/// <summary>
|
|
/// A hand labeler system that lets an object apply labels to objects with the <see cref="LabelComponent"/> .
|
|
/// </summary>
|
|
[UsedImplicitly]
|
|
public sealed class HandLabelerSystem : EntitySystem
|
|
{
|
|
[Dependency] private readonly UserInterfaceSystem _userInterfaceSystem = default!;
|
|
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
|
|
SubscribeLocalEvent<HandLabelerComponent, AfterInteractEvent>(AfterInteractOn);
|
|
SubscribeLocalEvent<HandLabelerComponent, ActivateInWorldEvent>(OnActivate);
|
|
// Bound UI subscriptions
|
|
SubscribeLocalEvent<HandLabelerComponent, HandLabelerLabelChangedMessage>(OnHandLabelerLabelChanged);
|
|
}
|
|
|
|
private void AfterInteractOn(EntityUid uid, HandLabelerComponent handLabeler, AfterInteractEvent args)
|
|
{
|
|
if (args.Target is not {Valid: true} target || !handLabeler.Whitelist.IsValid(target) || !args.CanReach)
|
|
return;
|
|
|
|
AddLabelTo(uid, handLabeler, target, out string? result);
|
|
if (result != null)
|
|
handLabeler.Owner.PopupMessage(args.User, result);
|
|
}
|
|
|
|
private void AddLabelTo(EntityUid uid, HandLabelerComponent? handLabeler, EntityUid target, out string? result)
|
|
{
|
|
if (!Resolve(uid, ref handLabeler))
|
|
{
|
|
result = null;
|
|
return;
|
|
}
|
|
|
|
LabelComponent label = target.EnsureComponent<LabelComponent>();
|
|
|
|
if (label.OriginalName != null)
|
|
EntityManager.GetComponent<MetaDataComponent>(target).EntityName = label.OriginalName;
|
|
label.OriginalName = null;
|
|
|
|
if (handLabeler.AssignedLabel == string.Empty)
|
|
{
|
|
label.CurrentLabel = null;
|
|
result = Loc.GetString("hand-labeler-successfully-removed");
|
|
return;
|
|
}
|
|
|
|
label.OriginalName = EntityManager.GetComponent<MetaDataComponent>(target).EntityName;
|
|
string val = EntityManager.GetComponent<MetaDataComponent>(target).EntityName + $" ({handLabeler.AssignedLabel})";
|
|
EntityManager.GetComponent<MetaDataComponent>(target).EntityName = val;
|
|
label.CurrentLabel = handLabeler.AssignedLabel;
|
|
result = Loc.GetString("hand-labeler-successfully-applied");
|
|
}
|
|
|
|
private void OnActivate(EntityUid uid, HandLabelerComponent handLabeler, ActivateInWorldEvent args)
|
|
{
|
|
if (!EntityManager.TryGetComponent(args.User, out ActorComponent? actor))
|
|
return;
|
|
|
|
handLabeler.Owner.GetUIOrNull(HandLabelerUiKey.Key)?.Open(actor.PlayerSession);
|
|
args.Handled = true;
|
|
}
|
|
|
|
private void OnHandLabelerLabelChanged(EntityUid uid, HandLabelerComponent handLabeler, HandLabelerLabelChangedMessage args)
|
|
{
|
|
handLabeler.AssignedLabel = args.Label.Trim().Substring(0, Math.Min(handLabeler.MaxLabelChars, args.Label.Length));
|
|
DirtyUI(uid, handLabeler);
|
|
}
|
|
|
|
private void DirtyUI(EntityUid uid,
|
|
HandLabelerComponent? handLabeler = null)
|
|
{
|
|
if (!Resolve(uid, ref handLabeler))
|
|
return;
|
|
|
|
_userInterfaceSystem.TrySetUiState(uid, HandLabelerUiKey.Key,
|
|
new HandLabelerBoundUserInterfaceState(handLabeler.AssignedLabel));
|
|
}
|
|
}
|
|
}
|