CatStation/Content.Server/GameObjects/Components/Observer/GhostRoles/MakeGhostRoleEui.cs

using Content.Server.Eui;
using Content.Server.GameObjects.EntitySystems;
using Content.Shared.Eui;
using Content.Shared.GameObjects.Components.Observer.GhostRoles;
using Robust.Shared.GameObjects;

namespace Content.Server.GameObjects.Components.Observer.GhostRoles
{
    public class MakeGhostRoleEui : BaseEui
    {
        public MakeGhostRoleEui(EntityUid entityUid)
        {
            EntityUid = entityUid;
        }

        public EntityUid EntityUid { get; }

        public override EuiStateBase GetNewState()
        {
            return new MakeGhostRoleEuiState(EntityUid);
        }

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

            switch (msg)
            {
                case MakeGhostRoleWindowClosedMessage _:
                    Closed();
                    break;
            }
        }

        public override void Closed()
        {
            base.Closed();

            EntitySystem.Get<GhostRoleSystem>().CloseMakeGhostRoleEui(Player);
        }
    }
}