50 lines
2.1 KiB
C#
50 lines
2.1 KiB
C#
using Content.Server.Body.Components;
|
|
using Content.Server.Ghost.Components;
|
|
using Content.Server.Mind.Components;
|
|
using Content.Shared.Body.Components;
|
|
using Content.Shared.Body.Events;
|
|
using Content.Shared.Movement.Components;
|
|
|
|
namespace Content.Server.Body.Systems
|
|
{
|
|
public sealed class BrainSystem : EntitySystem
|
|
{
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
|
|
SubscribeLocalEvent<BrainComponent, AddedToBodyEvent>((uid, _, args) => HandleMind((args.Body).Owner, uid));
|
|
SubscribeLocalEvent<BrainComponent, AddedToPartEvent>((uid, _, args) => HandleMind((args.Part).Owner, uid));
|
|
SubscribeLocalEvent<BrainComponent, AddedToPartInBodyEvent>((uid, _, args) => HandleMind((args.Body).Owner, uid));
|
|
SubscribeLocalEvent<BrainComponent, RemovedFromBodyEvent>(OnRemovedFromBody);
|
|
SubscribeLocalEvent<BrainComponent, RemovedFromPartEvent>((uid, _, args) => HandleMind(uid, (args.Old).Owner));
|
|
SubscribeLocalEvent<BrainComponent, RemovedFromPartInBodyEvent>((uid, _, args) => HandleMind((args.OldBody).Owner, uid));
|
|
}
|
|
|
|
private void OnRemovedFromBody(EntityUid uid, BrainComponent component, RemovedFromBodyEvent args)
|
|
{
|
|
// This one needs to be special, okay?
|
|
if (!EntityManager.TryGetComponent(uid, out MechanismComponent? mech))
|
|
return;
|
|
|
|
HandleMind((mech.Part!).Owner, (args.Old).Owner);
|
|
}
|
|
|
|
private void HandleMind(EntityUid newEntity, EntityUid oldEntity)
|
|
{
|
|
EntityManager.EnsureComponent<MindComponent>(newEntity);
|
|
var oldMind = EntityManager.EnsureComponent<MindComponent>(oldEntity);
|
|
|
|
EnsureComp<GhostOnMoveComponent>(newEntity);
|
|
if (HasComp<BodyComponent>(newEntity))
|
|
Comp<GhostOnMoveComponent>(newEntity).MustBeDead = true;
|
|
|
|
// TODO: This is an awful solution.
|
|
if (!EntityManager.HasComponent<IMoverComponent>(newEntity))
|
|
EntityManager.AddComponent<SharedDummyInputMoverComponent>(newEntity);
|
|
|
|
oldMind.Mind?.TransferTo(newEntity);
|
|
}
|
|
}
|
|
}
|