42 lines
1.2 KiB
C#
42 lines
1.2 KiB
C#
using Content.Server.Ghost.Components;
|
|
using Content.Server.Mind.Components;
|
|
using JetBrains.Annotations;
|
|
using Robust.Shared.GameObjects;
|
|
|
|
namespace Content.Server.Ghost
|
|
{
|
|
[UsedImplicitly]
|
|
public class GhostSystem : EntitySystem
|
|
{
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
|
|
SubscribeLocalEvent<GhostComponent, MindRemovedMessage>(OnMindRemovedMessage);
|
|
SubscribeLocalEvent<GhostComponent, MindUnvisitedMessage>(OnMindUnvisitedMessage);
|
|
}
|
|
|
|
private void OnMindRemovedMessage(EntityUid uid, GhostComponent component, MindRemovedMessage args)
|
|
{
|
|
DeleteEntity(uid);
|
|
}
|
|
|
|
private void OnMindUnvisitedMessage(EntityUid uid, GhostComponent component, MindUnvisitedMessage args)
|
|
{
|
|
DeleteEntity(uid);
|
|
}
|
|
|
|
private void DeleteEntity(EntityUid uid)
|
|
{
|
|
if (!EntityManager.TryGetEntity(uid, out var entity)
|
|
|| entity.Deleted == true
|
|
|| entity.LifeStage == EntityLifeStage.Terminating)
|
|
return;
|
|
|
|
if (entity.TryGetComponent<MindComponent>(out var mind))
|
|
mind.GhostOnShutdown = false;
|
|
entity.Delete();
|
|
}
|
|
}
|
|
}
|