BrainSystem.cs 1.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647
  1. using Content.Server.Body.Components;
  2. using Content.Server.Ghost.Components;
  3. using Content.Shared.Body.Components;
  4. using Content.Shared.Body.Events;
  5. using Content.Shared.Mind;
  6. using Content.Shared.Mind.Components;
  7. using Content.Shared.Pointing;
  8. namespace Content.Server.Body.Systems
  9. {
  10. public sealed class BrainSystem : EntitySystem
  11. {
  12. [Dependency] private readonly SharedMindSystem _mindSystem = default!;
  13. public override void Initialize()
  14. {
  15. base.Initialize();
  16. SubscribeLocalEvent<BrainComponent, OrganAddedToBodyEvent>((uid, _, args) => HandleMind(args.Body, uid));
  17. SubscribeLocalEvent<BrainComponent, OrganRemovedFromBodyEvent>((uid, _, args) => HandleMind(uid, args.OldBody));
  18. SubscribeLocalEvent<BrainComponent, PointAttemptEvent>(OnPointAttempt);
  19. }
  20. private void HandleMind(EntityUid newEntity, EntityUid oldEntity)
  21. {
  22. if (TerminatingOrDeleted(newEntity) || TerminatingOrDeleted(oldEntity))
  23. return;
  24. EnsureComp<MindContainerComponent>(newEntity);
  25. EnsureComp<MindContainerComponent>(oldEntity);
  26. var ghostOnMove = EnsureComp<GhostOnMoveComponent>(newEntity);
  27. if (HasComp<BodyComponent>(newEntity))
  28. ghostOnMove.MustBeDead = true;
  29. if (!_mindSystem.TryGetMind(oldEntity, out var mindId, out var mind))
  30. return;
  31. _mindSystem.TransferTo(mindId, newEntity, mind: mind);
  32. }
  33. private void OnPointAttempt(Entity<BrainComponent> ent, ref PointAttemptEvent args)
  34. {
  35. args.Cancel();
  36. }
  37. }
  38. }