36 lines
1.3 KiB
C#
36 lines
1.3 KiB
C#
using Content.Shared.Gravity;
|
|
using Content.Shared.Movement.Components;
|
|
using Content.Shared.Movement.Events;
|
|
|
|
namespace Content.Shared.Movement.Systems;
|
|
|
|
public sealed class MovementIgnoreGravitySystem : EntitySystem
|
|
{
|
|
[Dependency] SharedGravitySystem _gravity = default!;
|
|
public override void Initialize()
|
|
{
|
|
SubscribeLocalEvent<MovementAlwaysTouchingComponent, CanWeightlessMoveEvent>(OnWeightless);
|
|
SubscribeLocalEvent<MovementIgnoreGravityComponent, IsWeightlessEvent>(OnIsWeightless);
|
|
SubscribeLocalEvent<MovementIgnoreGravityComponent, ComponentStartup>(OnComponentStartup);
|
|
}
|
|
|
|
private void OnWeightless(Entity<MovementAlwaysTouchingComponent> entity, ref CanWeightlessMoveEvent args)
|
|
{
|
|
args.CanMove = true;
|
|
}
|
|
|
|
private void OnIsWeightless(Entity<MovementIgnoreGravityComponent> entity, ref IsWeightlessEvent args)
|
|
{
|
|
// We don't check if the event has been handled as this component takes precedent over other things.
|
|
|
|
args.IsWeightless = entity.Comp.Weightless;
|
|
args.Handled = true;
|
|
}
|
|
|
|
private void OnComponentStartup(Entity<MovementIgnoreGravityComponent> entity, ref ComponentStartup args)
|
|
{
|
|
EnsureComp<GravityAffectedComponent>(entity);
|
|
_gravity.RefreshWeightless(entity.Owner, entity.Comp.Weightless);
|
|
}
|
|
}
|