RobustToolbox/Robust.Shared/Player/ActorSystem.cs

using JetBrains.Annotations;
using Robust.Shared.GameObjects;
using Robust.Shared.IoC;

namespace Robust.Shared.Player;

/// <summary>
///     System that handles <see cref="ActorComponent"/>.
/// </summary>
public sealed partial class ActorSystem : EntitySystem
{
    [Dependency] private ISharedPlayerManager _playerManager = default!;

    public override void Initialize()
    {
        base.Initialize();
        SubscribeLocalEvent<ActorComponent, ComponentShutdown>(OnActorShutdown);
    }

    private void OnActorShutdown(EntityUid entity, ActorComponent component, ComponentShutdown args)
    {
        _playerManager.SetAttachedEntity(component.PlayerSession, null);
    }

    /// <summary>
    ///     Retrieves the session on a given entity, if one exists.
    /// </summary>
    [PublicAPI]
    public bool TryGetSession(EntityUid? uid, out ICommonSession? session)
    {
        if (TryComp(uid, out ActorComponent? actorComp))
        {
            session = actorComp.PlayerSession;
            return true;
        }

        session = null;
        return false;
    }

    /// <summary>
    ///     Retrieves the session on a given entity, if one exists.
    /// </summary>
    [PublicAPI]
    [Pure]
    public ICommonSession? GetSession(EntityUid? uid)
    {
        if (TryComp(uid, out ActorComponent? actorComp))
        {
            return actorComp.PlayerSession;
        }

        return null;
    }
}