108 lines
4.0 KiB
C#
108 lines
4.0 KiB
C#
using System.Linq;
|
|
using Content.Client.Administration.Managers;
|
|
using Content.Client.Administration.Systems;
|
|
using Content.Shared.Administration;
|
|
using Content.Shared.IdentityManagement;
|
|
using Robust.Client.GameObjects;
|
|
using Robust.Client.Player;
|
|
using Robust.Client.UserInterface.Controls;
|
|
using Robust.Shared.GameObjects;
|
|
using Robust.Shared.IoC;
|
|
using Robust.Shared.Maths;
|
|
using Robust.Shared.Utility;
|
|
|
|
namespace Content.Client.ContextMenu.UI
|
|
{
|
|
public sealed partial class EntityMenuElement : ContextMenuElement
|
|
{
|
|
public const string StyleClassEntityMenuCountText = "contextMenuCount";
|
|
|
|
[Dependency] private readonly IClientAdminManager _adminManager = default!;
|
|
[Dependency] private readonly IEntityManager _entityManager = default!;
|
|
[Dependency] private readonly IPlayerManager _playerManager = default!;
|
|
|
|
private AdminSystem _adminSystem;
|
|
|
|
/// <summary>
|
|
/// The entity that can be accessed by interacting with this element.
|
|
/// </summary>
|
|
public EntityUid? Entity;
|
|
|
|
/// <summary>
|
|
/// How many entities are accessible through this element's sub-menus.
|
|
/// </summary>
|
|
/// <remarks>
|
|
/// This is used for <see cref="CountLabel"/>
|
|
/// </remarks>
|
|
public int Count;
|
|
|
|
public readonly Label CountLabel;
|
|
public readonly SpriteView EntityIcon = new() { OverrideDirection = Direction.South};
|
|
|
|
public EntityMenuElement(EntityUid? entity = null)
|
|
{
|
|
IoCManager.InjectDependencies(this);
|
|
|
|
_adminSystem = _entityManager.System<AdminSystem>();
|
|
|
|
CountLabel = new Label { StyleClasses = { StyleClassEntityMenuCountText } };
|
|
Icon.AddChild(new LayoutContainer() { Children = { EntityIcon, CountLabel } });
|
|
|
|
LayoutContainer.SetAnchorPreset(CountLabel, LayoutContainer.LayoutPreset.BottomRight);
|
|
LayoutContainer.SetGrowHorizontal(CountLabel, LayoutContainer.GrowDirection.Begin);
|
|
LayoutContainer.SetGrowVertical(CountLabel, LayoutContainer.GrowDirection.Begin);
|
|
|
|
Entity = entity;
|
|
if (Entity == null)
|
|
return;
|
|
|
|
Count = 1;
|
|
CountLabel.Visible = false;
|
|
UpdateEntity();
|
|
}
|
|
|
|
protected override void Dispose(bool disposing)
|
|
{
|
|
base.Dispose(disposing);
|
|
Entity = null;
|
|
Count = 0;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Update the icon and text of this element based on the given entity or this element's own entity if none
|
|
/// is provided.
|
|
/// </summary>
|
|
public void UpdateEntity(EntityUid? entity = null)
|
|
{
|
|
entity ??= Entity;
|
|
|
|
// check whether entity is null, invalid, or has been deleted.
|
|
// _entityManager.Deleted() implicitly checks all of these.
|
|
if (_entityManager.Deleted(entity))
|
|
{
|
|
Text = string.Empty;
|
|
EntityIcon.Sprite = null;
|
|
return;
|
|
}
|
|
|
|
EntityIcon.Sprite = _entityManager.GetComponentOrNull<SpriteComponent>(entity);
|
|
|
|
if (_adminManager.HasFlag(AdminFlags.Admin | AdminFlags.Debug))
|
|
{
|
|
var representation = _entityManager.ToPrettyString(entity.Value);
|
|
var name = representation.Name;
|
|
var id = representation.Uid;
|
|
var prototype = representation.Prototype;
|
|
var playerName =
|
|
representation.Session?.Name ??
|
|
_adminSystem.PlayerList.FirstOrDefault(player => player.EntityUid == entity)?.Username;
|
|
var deleted = representation.Deleted;
|
|
|
|
Text = $"{name} ({id}{(representation.Prototype != null ? $", {prototype}" : "")}{(playerName != null ? $", {playerName}" : "")}){(deleted ? "D" : "")}";
|
|
}
|
|
else
|
|
Text = Identity.Name(entity.Value, _entityManager, _playerManager.LocalPlayer!.ControlledEntity!);
|
|
}
|
|
}
|
|
}
|