Files
tbd-station-14/Content.Shared/Shuttles/Components/PilotComponent.cs
Vera Aguilera Puerto 2bfec7ec62 Some more fixes
2021-12-04 14:14:22 +01:00

70 lines
2.2 KiB
C#

using System;
using Robust.Shared.GameObjects;
using Robust.Shared.GameStates;
using Robust.Shared.IoC;
using Robust.Shared.Log;
using Robust.Shared.Map;
using Robust.Shared.Players;
using Robust.Shared.Serialization;
using Robust.Shared.ViewVariables;
namespace Content.Shared.Shuttles.Components
{
/// <summary>
/// Stores what shuttle this entity is currently piloting.
/// </summary>
[RegisterComponent]
[NetworkedComponent]
public sealed class PilotComponent : Component
{
public override string Name => "Pilot";
[ViewVariables] public SharedShuttleConsoleComponent? Console { get; set; }
/// <summary>
/// Where we started piloting from to check if we should break from moving too far.
/// </summary>
[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;
if (state.Console == null)
{
Console = null;
return;
}
var entityManager = IoCManager.Resolve<IEntityManager>();
if (!entityManager.TryGetEntity(state.Console.Value, out var consoleEnt) ||
!entityManager.TryGetComponent(consoleEnt.Value, out SharedShuttleConsoleComponent? shuttleConsoleComponent))
{
Logger.Warning($"Unable to set Helmsman console to {state.Console.Value}");
return;
}
Console = shuttleConsoleComponent;
}
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;
}
}
}
}