ECS pilots (#8321)

This commit is contained in:
metalgearsloth
2022-05-21 14:51:49 +10:00
committed by GitHub
parent 61dc7d8748
commit d0a7d8af51
2 changed files with 41 additions and 43 deletions

View File

@@ -1,7 +1,5 @@
using Content.Shared.ActionBlocker;
using Robust.Shared.GameStates;
using Robust.Shared.Map;
using Robust.Shared.Serialization;
namespace Content.Shared.Shuttles.Components
{
@@ -12,9 +10,6 @@ namespace Content.Shared.Shuttles.Components
[NetworkedComponent]
public sealed class PilotComponent : Component
{
[Dependency] private readonly IEntitySystemManager _sysMan = default!;
[Dependency] private readonly IEntityManager _entMan = default!;
[ViewVariables] public SharedShuttleConsoleComponent? Console { get; set; }
/// <summary>
@@ -23,43 +18,5 @@ namespace Content.Shared.Shuttles.Components
[ViewVariables] public EntityCoordinates? Position { get; set; }
public const float BreakDistance = 0.25f;
public override void HandleComponentState(ComponentState? curState, ComponentState? nextState)
{
base.HandleComponentState(curState, nextState);
if (curState is not PilotComponentState state) return;
var console = state.Console.GetValueOrDefault();
if (!console.IsValid())
{
Console = null;
return;
}
if (!_entMan.TryGetComponent(console, out SharedShuttleConsoleComponent? shuttleConsoleComponent))
{
Logger.Warning($"Unable to set Helmsman console to {console}");
return;
}
Console = shuttleConsoleComponent;
_sysMan.GetEntitySystem<ActionBlockerSystem>().UpdateCanMove(Owner);
}
public override ComponentState GetComponentState()
{
return Console == null ? new PilotComponentState(null) : new PilotComponentState(Console.Owner);
}
[Serializable, NetSerializable]
private sealed class PilotComponentState : ComponentState
{
public EntityUid? Console { get; }
public PilotComponentState(EntityUid? uid)
{
Console = uid;
}
}
}
}