ECS MindComponent (#5917)

This commit is contained in:
Vera Aguilera Puerto
2022-01-04 13:37:50 +01:00
committed by GitHub
parent b0ee574c70
commit 675a29ed33
4 changed files with 157 additions and 118 deletions

View File

@@ -3,6 +3,7 @@ using Content.Server.Ghost.Components;
using Content.Shared.Examine;
using Content.Shared.Ghost;
using Content.Shared.MobState.Components;
using Robust.Shared.Analyzers;
using Robust.Shared.GameObjects;
using Robust.Shared.IoC;
using Robust.Shared.Localization;
@@ -17,21 +18,14 @@ namespace Content.Server.Mind.Components
/// <summary>
/// Stores a <see cref="Server.Mind.Mind"/> on a mob.
/// </summary>
[RegisterComponent]
#pragma warning disable 618
public class MindComponent : Component, IExamine
#pragma warning restore 618
[RegisterComponent, ComponentProtoName("Mind"), Friend(typeof(MindSystem))]
public class MindComponent : Component
{
[Dependency] private readonly IEntityManager _entMan = default!;
/// <inheritdoc />
public override string Name => "Mind";
/// <summary>
/// The mind controlling this mob. Can be null.
/// </summary>
[ViewVariables]
public Mind? Mind { get; private set; }
public Mind? Mind { get; set; }
/// <summary>
/// True if we have a mind, false otherwise.
@@ -52,109 +46,6 @@ namespace Content.Server.Mind.Components
[ViewVariables(VVAccess.ReadWrite)]
[DataField("ghostOnShutdown")]
public bool GhostOnShutdown { get; set; } = true;
/// <summary>
/// Don't call this unless you know what the hell you're doing.
/// Use <see cref="Mind.TransferTo(Robust.Shared.GameObjects.EntityUid)"/> instead.
/// If that doesn't cover it, make something to cover it.
/// </summary>
public void InternalEjectMind()
{
if (!Deleted)
_entMan.EventBus.RaiseLocalEvent(Owner, new MindRemovedMessage());
Mind = null;
}
/// <summary>
/// Don't call this unless you know what the hell you're doing.
/// Use <see cref="Mind.TransferTo(Robust.Shared.GameObjects.EntityUid)"/> instead.
/// If that doesn't cover it, make something to cover it.
/// </summary>
public void InternalAssignMind(Mind value)
{
Mind = value;
_entMan.EventBus.RaiseLocalEvent(Owner, new MindAddedMessage());
}
protected override void Shutdown()
{
base.Shutdown();
// Let's not create ghosts if not in the middle of the round.
if (EntitySystem.Get<GameTicker>().RunLevel != GameRunLevel.InRound)
return;
if (HasMind)
{
if (Mind?.VisitingEntity is {Valid: true} visiting)
{
if (_entMan.TryGetComponent(visiting, out GhostComponent? ghost))
{
EntitySystem.Get<SharedGhostSystem>().SetCanReturnToBody(ghost, false);
}
Mind!.TransferTo(visiting);
}
else if (GhostOnShutdown)
{
var spawnPosition = _entMan.GetComponent<TransformComponent>(Owner).Coordinates;
// Use a regular timer here because the entity has probably been deleted.
Timer.Spawn(0, () =>
{
// Async this so that we don't throw if the grid we're on is being deleted.
var mapMan = IoCManager.Resolve<IMapManager>();
var gridId = spawnPosition.GetGridId(_entMan);
if (gridId == GridId.Invalid || !mapMan.GridExists(gridId))
{
spawnPosition = EntitySystem.Get<GameTicker>().GetObserverSpawnPoint();
}
var ghost = _entMan.SpawnEntity("MobObserver", spawnPosition);
var ghostComponent = _entMan.GetComponent<GhostComponent>(ghost);
EntitySystem.Get<SharedGhostSystem>().SetCanReturnToBody(ghostComponent, false);
if (Mind != null)
{
string? val = Mind.CharacterName ?? string.Empty;
_entMan.GetComponent<MetaDataComponent>(ghost).EntityName = val;
Mind.TransferTo(ghost);
}
});
}
}
}
public void Examine(FormattedMessage message, bool inDetailsRange)
{
if (!ShowExamineInfo || !inDetailsRange)
{
return;
}
var dead =
_entMan.TryGetComponent<MobStateComponent?>(Owner, out var state) &&
state.IsDead();
if (dead)
{
if (Mind?.Session == null) {
// Player has no session attached and dead
message.AddMarkup($"[color=yellow]{Loc.GetString("mind-component-no-mind-and-dead-text", ("ent", Owner))}[/color]");
} else {
// Player is dead with session
message.AddMarkup($"[color=red]{Loc.GetString("comp-mind-examined-dead", ("ent", Owner))}[/color]");
}
}
else if (!HasMind)
{
message.AddMarkup($"[color=mediumpurple]{Loc.GetString("comp-mind-examined-catatonic", ("ent", Owner))}[/color]");
}
else if (Mind?.Session == null)
{
message.AddMarkup($"[color=yellow]{Loc.GetString("comp-mind-examined-ssd", ("ent", Owner))}[/color]");
}
}
}
public class MindRemovedMessage : EntityEventArgs