crystall-punk-14/Content.Server/Ghost/ReturnToBodyEui.cs

using Content.Server.EUI;
using Content.Server.Mind;
using Content.Server.Players;
using Content.Shared.Eui;
using Content.Shared.Ghost;

namespace Content.Server.Ghost;

public sealed class ReturnToBodyEui : BaseEui
{
    [Dependency] private readonly MindSystem _mindSystem = default!;

    private readonly Mind.Mind _mind;

    public ReturnToBodyEui(Mind.Mind mind)
    {
        _mind = mind;
    }

    public override void HandleMessage(EuiMessageBase msg)
    {
        base.HandleMessage(msg);

        if (msg is not ReturnToBodyMessage choice ||
            !choice.Accepted)
        {
            Close();
            return;
        }

        if (_mindSystem.TryGetSession(_mind, out var session))
            _mindSystem.UnVisit(session.ContentData()!.Mind);
        Close();
    }
}